Update from https://github.com/cloudimpl/next-coder/commit/fec13351ef969cbad85f82ae513912917396b280

Author: CloudimplBot

polycode/client.go

@@ -94,7 +94,7 @@ type PutRequest struct {
 	Collection string   `json:"collection"`
 	Key        string   `json:"key"`
 	Item       any      `json:"item"`
-	Counter    *Counter `json:"counter.,omitempty"`
+	TTL        int64    `json:"TTL"`
 }
 
 type Counter struct {

polycode/datastore.go

@@ -3,6 +3,7 @@ package polycode
 import (
 	"fmt"
 	"reflect"
+	"time"
 )
 
 type DataStore struct {
@@ -27,6 +28,17 @@ type Collection struct {
 }
 
 func (c Collection) InsertOne(item interface{}) error {
+	return c.InsertOneWithTTL(item, -1)
+}
+
+func (c Collection) InsertOneWithTTL(item interface{}, expireIn time.Duration) error {
+	var ttl int64
+	if expireIn == -1 {
+		ttl = -1
+	} else {
+		ttl = time.Now().Unix() + int64(expireIn.Seconds())
+	}
+
 	id, err := GetId(item)
 	if err != nil {
 		fmt.Printf("failed to get id: %s\n", err.Error())
@@ -38,6 +50,7 @@ func (c Collection) InsertOne(item interface{}) error {
 		Collection: c.name,
 		Key:        id,
 		Item:       item,
+		TTL:        ttl,
 	}
 
 	if c.isGlobal {
@@ -54,6 +67,17 @@ func (c Collection) InsertOne(item interface{}) error {
 }
 
 func (c Collection) UpdateOne(item interface{}) error {
+	return c.UpdateOneWithTTL(item, -1)
+}
+
+func (c Collection) UpdateOneWithTTL(item interface{}, expireIn time.Duration) error {
+	var ttl int64
+	if expireIn == -1 {
+		ttl = -1
+	} else {
+		ttl = time.Now().Unix() + int64(expireIn.Seconds())
+	}
+
 	id, err := GetId(item)
 	if err != nil {
 		fmt.Printf("failed to get id: %s\n", err.Error())
@@ -65,6 +89,7 @@ func (c Collection) UpdateOne(item interface{}) error {
 		Collection: c.name,
 		Key:        id,
 		Item:       item,
+		TTL:        ttl,
 	}
 
 	if c.isGlobal {
@@ -81,6 +106,17 @@ func (c Collection) UpdateOne(item interface{}) error {
 }
 
 func (c Collection) UpsertOne(item interface{}) error {
+	return c.UpsertOneWithTTL(item, -1)
+}
+
+func (c Collection) UpsertOneWithTTL(item interface{}, expireIn time.Duration) error {
+	var ttl int64
+	if expireIn == -1 {
+		ttl = -1
+	} else {
+		ttl = time.Now().Unix() + int64(expireIn.Seconds())
+	}
+
 	id, err := GetId(item)
 	if err != nil {
 		fmt.Printf("failed to get id: %s\n", err.Error())
@@ -92,6 +128,7 @@ func (c Collection) UpsertOne(item interface{}) error {
 		Collection: c.name,
 		Key:        id,
 		Item:       item,
+		TTL:        ttl,
 	}
 
 	if c.isGlobal {