diff --git a/watchtower/wtserver/create_session.go b/watchtower/wtserver/create_session.go index 3217d601..3d5c9361 100644 --- a/watchtower/wtserver/create_session.go +++ b/watchtower/wtserver/create_session.go @@ -142,6 +142,6 @@ func (s *Server) replyCreateSession(peer Peer, id *wtdb.SessionID, // disconnect the client. return &connFailure{ ID: *id, - Code: uint16(code), + Code: code, } } diff --git a/watchtower/wtserver/delete_session.go b/watchtower/wtserver/delete_session.go index a5b517c1..00b9b165 100644 --- a/watchtower/wtserver/delete_session.go +++ b/watchtower/wtserver/delete_session.go @@ -52,6 +52,6 @@ func (s *Server) replyDeleteSession(peer Peer, id *wtdb.SessionID, // disconnect the client. return &connFailure{ ID: *id, - Code: uint16(code), + Code: code, } } diff --git a/watchtower/wtserver/server.go b/watchtower/wtserver/server.go index 4a49c3ad..940b41cd 100644 --- a/watchtower/wtserver/server.go +++ b/watchtower/wtserver/server.go @@ -283,12 +283,12 @@ func (s *Server) handleClient(peer Peer) { // error code. type connFailure struct { ID wtdb.SessionID - Code uint16 + Code wtwire.ErrorCode } // Error displays the SessionID and Code that caused the connection failure. func (f *connFailure) Error() string { - return fmt.Sprintf("connection with %s failed with code=%v", + return fmt.Sprintf("connection with %s failed with code=%s", f.ID, f.Code, ) } diff --git a/watchtower/wtserver/state_update.go b/watchtower/wtserver/state_update.go index 7b3e0941..63a0a37d 100644 --- a/watchtower/wtserver/state_update.go +++ b/watchtower/wtserver/state_update.go @@ -117,7 +117,7 @@ func (s *Server) handleStateUpdate(peer Peer, id *wtdb.SessionID, if s.cfg.NoAckUpdates { return &connFailure{ ID: *id, - Code: uint16(failCode), + Code: failCode, } } @@ -152,6 +152,6 @@ func (s *Server) replyStateUpdate(peer Peer, id *wtdb.SessionID, // disconnect the client. return &connFailure{ ID: *id, - Code: uint16(code), + Code: code, } }