support for "<timestamp>-*"

Author: alicebob

cmd_stream_test.go

@@ -56,9 +56,13 @@ func TestStream(t *testing.T) {
 		ok(t, err)
 		equals(t, "978321845004-0", id)
 
+		id, err = s.XAdd("s1", "978321845004-*", []string{"name", "bazz"})
+		ok(t, err)
+		equals(t, "978321845004-1", id)
+
 		stream, err := s.Stream("s1")
 		ok(t, err)
-		equals(t, 2, len(stream))
+		equals(t, 3, len(stream))
 		equals(t, StreamEntry{
 			ID:     "12345-67",
 			Values: []string{"name", "bar"},
@@ -67,6 +71,10 @@ func TestStream(t *testing.T) {
 			ID:     "978321845004-0",
 			Values: []string{"name", "baz"},
 		}, stream[1])
+		equals(t, StreamEntry{
+			ID:     "978321845004-1",
+			Values: []string{"name", "bazz"},
+		}, stream[2])
 	})
 
 	useRESP3(t, c)

stream.go

@@ -230,14 +230,23 @@ func (s *streamKey) createGroup(group, id string) error {
 }
 
 // streamAdd adds an entry to a stream. Returns the new entry ID.
-// If id is empty or "*" the ID will be generated automatically.
+// If id is empty, "*", or "123-*", the ID will be generated automatically.
 // `values` should have an even length.
 func (s *streamKey) add(entryID string, values []string, now time.Time) (string, error) {
 	s.mu.Lock()
 	defer s.mu.Unlock()
 
-	if entryID == "" || entryID == "*" {
+	switch {
+	case entryID == "" || entryID == "*":
 		entryID = s.generateID(now)
+	default:
+		// "<timestamp>-*"
+		parts := strings.Split(entryID, "-")
+		if len(parts) == 2 && parts[1] == "*" {
+			if ts, err := strconv.Atoi(parts[0]); err == nil {
+				entryID = s.generateID(time.Unix(int64(ts), 0))
+			}
+		}
 	}
 
 	entryID, err := formatStreamID(entryID)