diff options
Diffstat (limited to 'server')
| -rw-r--r-- | server/message.go | 6 | ||||
| -rw-r--r-- | server/remote.go | 2 | ||||
| -rw-r--r-- | server/user.go | 2 |
3 files changed, 7 insertions, 3 deletions
diff --git a/server/message.go b/server/message.go index 11644dc..cd5add6 100644 --- a/server/message.go +++ b/server/message.go @@ -30,7 +30,11 @@ func (s *Server) SendBroadcast(msg string) { } } -func (s *Server) handleMessage(sender net.Conn, msg core.Message) error { +func (s *Server) handleMessage(sender net.Conn, connType core.ConnType, msg core.Message) error { + if connType == core.ConnTypeUser { + msg.Timestamp = time.Now() + } + log.Println("message:", msg.Source, "->", msg.Target, msg.Content) addr, err := core.ReadAddr(msg.Target) diff --git a/server/remote.go b/server/remote.go index 5d86da2..6f9bd66 100644 --- a/server/remote.go +++ b/server/remote.go @@ -126,7 +126,7 @@ func (s *Server) readRemoteInput(conn net.Conn) error { func (s *Server) handleRemotePayload(sender net.Conn, payload any) error { switch v := payload.(type) { case core.Message: - return s.handleMessage(sender, v) + return s.handleMessage(sender, core.ConnTypeServer, v) default: return core.ErrUnexpectedPayloadType } diff --git a/server/user.go b/server/user.go index 4f78b6a..002ed89 100644 --- a/server/user.go +++ b/server/user.go @@ -115,7 +115,7 @@ func (s *Server) readUserInput(user *User, conn net.Conn) error { func (s *Server) handleUserPayload(user *User, sender net.Conn, payload any) error { switch v := payload.(type) { case core.Message: - return s.handleMessage(sender, v) + return s.handleMessage(sender, core.ConnTypeUser, v) case core.Usermode: return s.handleUsermode(user, sender, v) default: |