48 lines
1020 B
Go
48 lines
1020 B
Go
package network
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/kercylan98/minotaur/server/internal/v2"
|
|
"github.com/kercylan98/minotaur/utils/collection"
|
|
"github.com/panjf2000/gnet/v2"
|
|
"time"
|
|
)
|
|
|
|
func WebSocket(addr string, pattern ...string) server.server {
|
|
ws := &websocketCore{
|
|
addr: addr,
|
|
pattern: collection.FindFirstOrDefaultInSlice(pattern, "/"),
|
|
}
|
|
return ws
|
|
}
|
|
|
|
type websocketCore struct {
|
|
ctx context.Context
|
|
core server.NetworkCore
|
|
handler *websocketHandler
|
|
addr string
|
|
pattern string
|
|
}
|
|
|
|
func (w *websocketCore) OnSetup(ctx context.Context, core server.NetworkCore) (err error) {
|
|
w.ctx = ctx
|
|
w.handler = newWebsocketHandler(w)
|
|
w.core = core
|
|
return
|
|
}
|
|
|
|
func (w *websocketCore) OnRun() (err error) {
|
|
err = gnet.Run(w.handler, fmt.Sprintf("tcp://%s", w.addr))
|
|
return
|
|
}
|
|
|
|
func (w *websocketCore) OnShutdown() error {
|
|
if w.handler.engine != nil {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
return w.handler.engine.Stop(ctx)
|
|
}
|
|
return nil
|
|
}
|