how get active sessions

[imanborumand]

hi how i get active sessions for use in private chat?

thanks

[hah]

fork this repo and merge this change https://github.com/olahol/melody/pull/52

[lesismal]

fork this repo and merge this change #52

That's not a good way to manage all sessions if you copy all sessions to another map every time in a high concurrency service, it is very performance unfriendly.

[lesismal]

hi how i get active sessions for use in private chat?

thanks

if you just need to do some broadcast, use:

m.Broadcast
m.BroadcastFilter
m.BroadcastOthers
m.BroadcastMultiple
m.BroadcastBinary
m.BroadcastBinaryFilter
m.BroadcastBinaryOthers 

else if you need to do some more, you can manage all sessions at application layer, for example:

package main

import (
    "sync"

    "github.com/gin-gonic/gin"
    "github.com/olahol/melody"
)

type SessionMgr struct {
    mux      sync.RWMutex
    sessions map[interface{}]*melody.Session
}

func (mgr *SessionMgr) Get(id interface{}) (*melody.Session, bool) {
    mgr.mux.RLock()
    defer mgr.mux.RUnlock()
    sess, ok := mgr.sessions[id]
    return sess, ok
}

func (mgr *SessionMgr) Set(id interface{}, sess *melody.Session) {
    mgr.mux.Lock()
    defer mgr.mux.Unlock()
    mgr.sessions[id] = sess
}

func (mgr *SessionMgr) Delete(id interface{}) {
    mgr.mux.Lock()
    defer mgr.mux.Unlock()
    delete(mgr.sessions, id)
}

func (mgr *SessionMgr) ForEach(filter func(sess *melody.Session) bool, handler func(sess *melody.Session)) {
    mgr.mux.RLock()
    defer mgr.mux.RUnlock()
    for _, sess := range mgr.sessions {
        if filter(sess) {
            handler(sess)
        }
    }
}

var (
    sessionMgr = SessionMgr{
        sessions: map[interface{}]*melody.Session{},
    }
)

func main() {
    r := gin.Default()
    m := melody.New()

    m.HandleConnect(func(sess *melody.Session) {
        id := getIDFromSession(sess)
        sessionMgr.Set(id, sess)
    })

    m.HandleDisconnect(func(sess *melody.Session) {
        id := getIDFromSession(sess)
        sessionMgr.Delete(id)
    })

    r.GET("/ws", func(c *gin.Context) {
        m.HandleRequest(c.Writer, c.Request)
    })

    m.HandleMessage(func(s *melody.Session, msg []byte) {
        // ...
    })

    r.Run(":5000")
}

[olahol]

I merged #52, but as @lesismal says it's not very performant, although the package probably needs a performance rewrite anyway.