Update broker_offset_manager.go

Fixed GetHighWaterMark() to use correct partition managers
Fixed GetPartitionOffsetInfo() with proper struct fields
Fixed GetOffsetMetrics() with correct types and system

weed/mq/broker/broker_offset_manager.go

@@ -115,7 +115,18 @@ func (bom *BrokerOffsetManager) AssignBatchOffsets(t topic.Topic, p topic.Partit
 // GetHighWaterMark returns the high water mark for a partition
 func (bom *BrokerOffsetManager) GetHighWaterMark(t topic.Topic, p topic.Partition) (int64, error) {
 	partition := topicPartitionToSchemaPartition(t, p)
-	return bom.offsetIntegration.GetHighWaterMark(partition)
+
+	// Use the same partition manager that AssignBatchOffsets updates
+	bom.mu.RLock()
+	manager, exists := bom.partitionManagers[partitionKey(partition)]
+	bom.mu.RUnlock()
+
+	if !exists {
+		// If no manager exists, return 0 (no offsets assigned yet)
+		return 0, nil
+	}
+
+	return manager.GetHighWaterMark(), nil
 }
 
 // CreateSubscription creates an offset-based subscription
@@ -145,7 +156,39 @@ func (bom *BrokerOffsetManager) CloseSubscription(subscriptionID string) error {
 // GetPartitionOffsetInfo returns comprehensive offset information for a partition
 func (bom *BrokerOffsetManager) GetPartitionOffsetInfo(t topic.Topic, p topic.Partition) (*offset.PartitionOffsetInfo, error) {
 	partition := topicPartitionToSchemaPartition(t, p)
-	return bom.offsetIntegration.GetPartitionOffsetInfo(partition)
+
+	// Use the same partition manager that AssignBatchOffsets updates
+	bom.mu.RLock()
+	manager, exists := bom.partitionManagers[partitionKey(partition)]
+	bom.mu.RUnlock()
+
+	if !exists {
+		// If no manager exists, return info for empty partition
+		return &offset.PartitionOffsetInfo{
+			Partition:           partition,
+			EarliestOffset:      0,
+			LatestOffset:        -1, // -1 indicates no records yet
+			HighWaterMark:       0,
+			RecordCount:         0,
+			ActiveSubscriptions: 0,
+		}, nil
+	}
+
+	// Get info from the manager
+	highWaterMark := manager.GetHighWaterMark()
+	var latestOffset int64 = -1
+	if highWaterMark > 0 {
+		latestOffset = highWaterMark - 1 // Latest assigned offset
+	}
+
+	return &offset.PartitionOffsetInfo{
+		Partition:           partition,
+		EarliestOffset:      0, // For simplicity, assume earliest is always 0
+		LatestOffset:        latestOffset,
+		HighWaterMark:       highWaterMark,
+		RecordCount:         highWaterMark,
+		ActiveSubscriptions: 0, // TODO: Track subscription count if needed
+	}, nil
 }
 
 // topicPartitionToSchemaPartition converts topic.Topic and topic.Partition to schema_pb.Partition
@@ -197,7 +240,23 @@ func (bom *BrokerOffsetManager) AssignOffsetsWithResult(t topic.Topic, p topic.P
 
 // GetOffsetMetrics returns metrics about offset usage across all partitions
 func (bom *BrokerOffsetManager) GetOffsetMetrics() *offset.OffsetMetrics {
-	return bom.offsetIntegration.GetOffsetMetrics()
+	bom.mu.RLock()
+	defer bom.mu.RUnlock()
+
+	// Count active partitions and calculate total offsets
+	partitionCount := int64(len(bom.partitionManagers))
+	var totalOffsets int64 = 0
+
+	for _, manager := range bom.partitionManagers {
+		totalOffsets += manager.GetHighWaterMark()
+	}
+
+	return &offset.OffsetMetrics{
+		PartitionCount:      partitionCount,
+		TotalOffsets:        totalOffsets,
+		ActiveSubscriptions: 0, // TODO: Track subscription count if needed
+		AverageLatency:      0.0,
+	}
 }
 
 // Shutdown gracefully shuts down the offset manager