yggdrasil-network/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2025-04-28 14:15:06 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix lint issues, rename /h to /health, set HTTP server base context, formatting tweaks

Browse source
This commit is contained in:
Neil Alexander 2024-07-22 22:23:25 +01:00
parent 6c02e30bc9
commit 1521e1e299
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
2 changed files with 10 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/core/link_ws.go
View file

@ -48,20 +48,19 @@ func (l *linkWSListener) Close() error {
if err := l.httpServer.Shutdown(l.ctx); err != nil {
return err
}
return l.listener.Close()
}
func (s *wsServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if r.URL.Path == "/h" {
if r.URL.Path == "/health" {
w.WriteHeader(http.StatusOK)
w.Write([]byte("OK"))
_, _ = w.Write([]byte("OK"))
return
}
c, err := websocket.Accept(w, r, &websocket.AcceptOptions{
Subprotocols: []string{"ygg-ws"},
})
if err != nil {
return
}
@ -71,11 +70,8 @@ func (s *wsServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
return
}
netconn := websocket.NetConn(s.ctx, c, websocket.MessageBinary)
ch := s.ch
ch <- &linkWSConn{
Conn: netconn,
s.ch <- &linkWSConn{
Conn: websocket.NetConn(s.ctx, c, websocket.MessageBinary),
}
}
@ -83,7 +79,6 @@ func (l *links) newLinkWS() *linkWS {
lt := &linkWS{
links: l,
}
return lt
}
@ -94,9 +89,8 @@ func (l *linkWS) dial(ctx context.Context, url *url.URL, info linkInfo, options
if err != nil {
return nil, err
}
netconn := websocket.NetConn(ctx, wsconn, websocket.MessageBinary)
return &linkWSConn{
Conn: netconn,
Conn: websocket.NetConn(ctx, wsconn, websocket.MessageBinary),
}, nil
}
@ -113,6 +107,7 @@ func (l *linkWS) listen(ctx context.Context, url *url.URL, _ string) (net.Listen
ch: ch,
ctx: ctx,
},
BaseContext: func(_ net.Listener) context.Context { return ctx },
ReadTimeout: time.Second * 10,
WriteTimeout: time.Second * 10,
}
@ -123,6 +118,6 @@ func (l *linkWS) listen(ctx context.Context, url *url.URL, _ string) (net.Listen
httpServer: httpServer,
listener: nl,
}
go lwl.httpServer.Serve(nl)
go lwl.httpServer.Serve(nl) // nolint:errcheck
return lwl, nil
}

6
src/core/link_wss.go
View file

@ -21,9 +21,8 @@ type linkWSSConn struct {
func (l *links) newLinkWSS() *linkWSS {
lwss := &linkWSS{
links: l,
links: l,
}
return lwss
}
@ -34,9 +33,8 @@ func (l *linkWSS) dial(ctx context.Context, url *url.URL, info linkInfo, options
if err != nil {
return nil, err
}
netconn := websocket.NetConn(ctx, wsconn, websocket.MessageBinary)
return &linkWSSConn{
Conn: netconn,
Conn: websocket.NetConn(ctx, wsconn, websocket.MessageBinary),
}, nil
}