Make read/write debug messages a bit easier to read

command.go

@@ -90,7 +90,7 @@ func (srv *Server) consumeCommands(ctx context.Context, conn net.Conn, reader *b
 		// NOTE: we increase the wait group by one in order to make sure that idle
 		// connections are not blocking a close.
 		srv.wg.Add(1)
-		srv.logger.Debug("incoming command", slog.Int("length", length), slog.String("type", string(t)))
+		srv.logger.Debug("<- incoming command", slog.Int("length", length), slog.String("type", t.String()))
 		err = srv.handleCommand(ctx, conn, t, reader, writer)
 		srv.wg.Done()
 		if errors.Is(err, io.EOF) {

pkg/buffer/writer.go

@@ -136,7 +136,7 @@ func (writer *Writer) End() error {
 	binary.BigEndian.PutUint32(bytes[1:5], length)
 	_, err := writer.Writer.Write(bytes)
 
-	writer.logger.Debug("writing message", slog.String("type", string(bytes[0])))
+	writer.logger.Debug("-> writing message", slog.String("type", types.ServerMessage(bytes[0]).String()))
 	return err
 }
 

pkg/types/types.go

@@ -39,3 +39,75 @@ const (
 	ServerReady                ServerMessage = 'Z'
 	ServerRowDescription       ServerMessage = 'T'
 )
+
+func (m ClientMessage) String() string {
+	switch m {
+	case ClientBind:
+		return "Bind"
+	case ClientClose:
+		return "Close"
+	case ClientCopyData:
+		return "CopyData"
+	case ClientCopyDone:
+		return "CopyDone"
+	case ClientCopyFail:
+		return "CopyFail"
+	case ClientDescribe:
+		return "Describe"
+	case ClientExecute:
+		return "Execute"
+	case ClientFlush:
+		return "Flush"
+	case ClientParse:
+		return "Parse"
+	case ClientPassword:
+		return "Password"
+	case ClientSimpleQuery:
+		return "SimpleQuery"
+	case ClientSync:
+		return "Sync"
+	case ClientTerminate:
+		return "Terminate"
+	default:
+		return "Unknown"
+	}
+}
+
+func (m ServerMessage) String() string {
+	switch m {
+	case ServerAuth:
+		return "Auth"
+	case ServerBindComplete:
+		return "BindComplete"
+	case ServerCommandComplete:
+		return "CommandComplete"
+	case ServerCloseComplete:
+		return "CloseComplete"
+	case ServerCopyInResponse:
+		return "CopyInResponse"
+	case ServerDataRow:
+		return "DataRow"
+	case ServerEmptyQuery:
+		return "EmptyQuery"
+	case ServerErrorResponse:
+		return "ErrorResponse"
+	case ServerNoticeResponse:
+		return "NoticeResponse"
+	case ServerNoData:
+		return "NoData"
+	case ServerParameterDescription:
+		return "ParameterDescription"
+	case ServerParameterStatus:
+		return "ParameterStatus"
+	case ServerParseComplete:
+		return "ParseComplete"
+	case ServerPortalSuspended:
+		return "PortalSuspended"
+	case ServerReady:
+		return "Ready"
+	case ServerRowDescription:
+		return "RowDescription"
+	default:
+		return "Unknown"
+	}
+}