feat: add creation time for stream and subscription

hstreamdb · Jan 18, 2023 · 43b741f · 43b741f
1 parent 6ff4a75
commit 43b741f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/hstream/stream.go b/hstream/stream.go
@@ -2,6 +2,9 @@ package hstream
 
 import (
 	"fmt"
+	"time"
+
+	"google.golang.org/protobuf/types/known/timestamppb"
 
 	"github.com/hstreamdb/hstreamdb-go/internal/hstreamrpc"
 	hstreampb "github.com/hstreamdb/hstreamdb-go/proto/gen-proto/hstreamdb/hstream/server"
@@ -14,6 +17,7 @@ type Stream struct {
 	// backlog duration == 0 means forbidden backlog
 	BacklogDuration uint32
 	ShardCount      uint32
+	CreationTime    time.Time
 }
 
 func (s *Stream) StreamToPb() *hstreampb.Stream {
@@ -22,6 +26,7 @@ func (s *Stream) StreamToPb() *hstreampb.Stream {
 		ReplicationFactor: s.ReplicationFactor,
 		BacklogDuration:   s.BacklogDuration,
 		ShardCount:        s.ShardCount,
+		CreationTime:      timestamppb.New(s.CreationTime),
 	}
 }
 
@@ -31,6 +36,7 @@ func StreamFromPb(pb *hstreampb.Stream) Stream {
 		ReplicationFactor: pb.ReplicationFactor,
 		BacklogDuration:   pb.BacklogDuration,
 		ShardCount:        pb.ShardCount,
+		CreationTime:      pb.CreationTime.AsTime(),
 	}
 }
 

diff --git a/hstream/subscription.go b/hstream/subscription.go
@@ -2,6 +2,9 @@ package hstream
 
 import (
 	"fmt"
+	"time"
+
+	"google.golang.org/protobuf/types/known/timestamppb"
 
 	"github.com/hstreamdb/hstreamdb-go/internal/hstreamrpc"
 	hstreampb "github.com/hstreamdb/hstreamdb-go/proto/gen-proto/hstreamdb/hstream/server"
@@ -23,6 +26,7 @@ type Subscription struct {
 	AckTimeoutSeconds int32
 	MaxUnackedRecords int32
 	Offset            SubscriptionOffset
+	CreationTime      time.Time
 }
 
 func (s *Subscription) SubscriptionToPb() *hstreampb.Subscription {
@@ -32,6 +36,7 @@ func (s *Subscription) SubscriptionToPb() *hstreampb.Subscription {
 		AckTimeoutSeconds: s.AckTimeoutSeconds,
 		MaxUnackedRecords: s.MaxUnackedRecords,
 		Offset:            SubscriptionOffsetToPb(s.Offset),
+		CreationTime:      timestamppb.New(s.CreationTime),
 	}
 }
 
@@ -42,6 +47,7 @@ func SubscriptionFromPb(pb *hstreampb.Subscription) Subscription {
 		AckTimeoutSeconds: pb.AckTimeoutSeconds,
 		MaxUnackedRecords: pb.MaxUnackedRecords,
 		Offset:            SubscriptionOffsetFromPb(pb.Offset),
+		CreationTime:      pb.CreationTime.AsTime(),
 	}
 }