Clean up
This commit is contained in:
parent
43ec360d08
commit
95452e4e02
4 changed files with 4 additions and 14 deletions
|
@ -107,19 +107,10 @@ func main() {
|
|||
}
|
||||
}()
|
||||
|
||||
ticker := time.NewTicker(time.Second)
|
||||
defer ticker.Stop()
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-done:
|
||||
return
|
||||
case t := <-ticker.C:
|
||||
err := c.WriteMessage(websocket.TextMessage, []byte(t.String()))
|
||||
if err != nil {
|
||||
log.Println("write:", err)
|
||||
return
|
||||
}
|
||||
case <-interrupt:
|
||||
log.Println("interrupt")
|
||||
|
||||
|
|
|
@ -2,7 +2,6 @@ package auth
|
|||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
|
@ -103,7 +102,7 @@ func (a *authenticator) routeAuth(w http.ResponseWriter, r *http.Request) {
|
|||
}
|
||||
username, password := r.PostFormValue("username"), r.PostFormValue("password")
|
||||
if username == "" || password == "" {
|
||||
http.Error(w, "blank credentials"+fmt.Sprint(r.Form), http.StatusBadRequest)
|
||||
http.Error(w, "blank credentials", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
|
|
|
@ -87,7 +87,7 @@ func (conn *wsConn) readPump(reg Registry, c Client) {
|
|||
_, message, err := conn.ReadMessage()
|
||||
if err != nil {
|
||||
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
|
||||
log.Error().Err(err).Msg("read failed")
|
||||
return
|
||||
}
|
||||
|
||||
break
|
||||
|
@ -153,6 +153,6 @@ func (conn *wsConn) writeMessage(w io.WriteCloser, msg *pb.Message) {
|
|||
}
|
||||
}
|
||||
|
||||
func (n *wsManager) InstallPrivateRoutes(r chi.Router) {
|
||||
func (n *wsManager) PrivateRoutes(r chi.Router) {
|
||||
r.HandleFunc("/ws", n.serveWS)
|
||||
}
|
||||
|
|
|
@ -18,7 +18,7 @@ func (s *Server) setupRoutes() {
|
|||
r.Group(func(r chi.Router) {
|
||||
// authenticated routes
|
||||
r.Use(s.auth.VerifyMiddleware(), s.auth.AuthMiddleware())
|
||||
s.nex.InstallPrivateRoutes(r)
|
||||
s.nex.PrivateRoutes(r)
|
||||
})
|
||||
|
||||
r.Group(func(r chi.Router) {
|
||||
|
|
Loading…
Reference in a new issue