35 lines
721 B
Go
35 lines
721 B
Go
package pwd
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/googollee/go-socket.io"
|
|
)
|
|
|
|
type BroadcastApi interface {
|
|
BroadcastTo(sessionId, eventName string, args ...interface{})
|
|
GetHandler() http.Handler
|
|
}
|
|
|
|
type broadcast struct {
|
|
sio *socketio.Server
|
|
}
|
|
|
|
func (b *broadcast) BroadcastTo(sessionId, eventName string, args ...interface{}) {
|
|
b.sio.BroadcastTo(sessionId, eventName, args...)
|
|
}
|
|
|
|
func (b *broadcast) GetHandler() http.Handler {
|
|
return b.sio
|
|
}
|
|
|
|
func NewBroadcast(connectionEvent, errorEvent interface{}) (*broadcast, error) {
|
|
server, err := socketio.NewServer(nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
server.On("connection", connectionEvent)
|
|
server.On("error", errorEvent)
|
|
return &broadcast{sio: server}, nil
|
|
}
|