package rest_websocket import ( "log" "net/http" "git.ali33.ru/fcg-xvii/rest" "github.com/gorilla/websocket" ) func New(app rest.IApplicationStream, core any) *Rest { return &Rest{ upgrader: &websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, CheckOrigin: func(r *http.Request) bool { // Проверьте ориджин здесь, и верните true, если он разрешен. //return r.Header.Get("Origin") == "http://example.com" return true }, }, appConf: &appConfig{ app: app, core: core, }, } } type Rest struct { upgrader *websocket.Upgrader appConf *appConfig server rest.IServer } func (s *Rest) Prepare(server rest.IServer, httpPrefix string) { s.appConf.ctx = server.Context() s.appConf.tokenGenerator = server.TokenGenerate server.HandleFunc(httpPrefix, s.handle) s.server = server } func (s *Rest) handle(w http.ResponseWriter, r *http.Request) { // Преобразование HTTP соединения в WebSocket соединение ws, err := s.upgrader.Upgrade(w, r, nil) if err != nil { log.Println(err) } socket := NewSocketServer( NewSocket(ws, true), s.appConf, ) s.appConf.app.Connect() <- socket }