From ff479565301fad09c43a1397a52fc6aa7cd50207 Mon Sep 17 00:00:00 2001
From: chrislu <chris.lu@gmail.com>
Date: Thu, 3 Oct 2024 00:13:46 -0700
Subject: [PATCH] add helper functions

---
 weed/mq/schema/schema.go   | 27 +++++++++++++++++++++++++++
 weed/mq/topic/partition.go | 13 ++++++++++++-
 2 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/weed/mq/schema/schema.go b/weed/mq/schema/schema.go
index 5fadf2cc2..ca31ce534 100644
--- a/weed/mq/schema/schema.go
+++ b/weed/mq/schema/schema.go
@@ -24,3 +24,30 @@ func (s *Schema) GetField(name string) (*schema_pb.Field, bool) {
 	field, ok := s.fieldMap[name]
 	return field, ok
 }
+
+func TypeToString(t *schema_pb.Type) string {
+	switch t.Kind.(type) {
+	case *schema_pb.Type_ScalarType:
+		switch t.GetScalarType() {
+		case schema_pb.ScalarType_BOOL:
+			return "bool"
+		case schema_pb.ScalarType_INT32:
+			return "int32"
+		case schema_pb.ScalarType_INT64:
+			return "int64"
+		case schema_pb.ScalarType_FLOAT:
+			return "float"
+		case schema_pb.ScalarType_DOUBLE:
+			return "double"
+		case schema_pb.ScalarType_BYTES:
+			return "bytes"
+		case schema_pb.ScalarType_STRING:
+			return "string"
+		}
+	case *schema_pb.Type_ListType:
+		return "list"
+	case *schema_pb.Type_RecordType:
+		return "record"
+	}
+	return "unknown"
+}
diff --git a/weed/mq/topic/partition.go b/weed/mq/topic/partition.go
index ba1accce1..192af6c98 100644
--- a/weed/mq/topic/partition.go
+++ b/weed/mq/topic/partition.go
@@ -1,6 +1,9 @@
 package topic
 
-import "github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
+import (
+	"fmt"
+	"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
+)
 
 const PartitionCount = 4096
 
@@ -81,3 +84,11 @@ func (partition Partition) Overlaps(partition2 Partition) bool {
 	}
 	return true
 }
+
+func (partition Partition) String() string {
+	return fmt.Sprintf("%04d-%04d", partition.RangeStart, partition.RangeStop)
+}
+
+func ToString(partition *mq_pb.Partition) string {
+	return fmt.Sprintf("%04d-%04d", partition.RangeStart, partition.RangeStop)
+}