Merge branch 'master' into master

Author: artursouza

state/cassandra/cassandra.go

@@ -33,6 +33,7 @@ const (
 	defaultTable             = "items"
 	defaultKeyspace          = "dapr"
 	defaultPort              = 9042
+	metadataTTLKey           = "ttlInSeconds"
 )
 
 // Cassandra is a state store implementation for Apache Cassandra
@@ -286,6 +287,15 @@ func (c *Cassandra) Set(req *state.SetRequest) error {
 		session = sess
 	}
 
+	ttl, err := parseTTL(req.Metadata)
+	if err != nil {
+		return fmt.Errorf("error parsing TTL from Metadata: %s", err)
+	}
+
+	if ttl != nil {
+		return session.Query("INSERT INTO ? (key, value) VALUES (?, ?) USING TTL ?", c.table, req.Key, bt, *ttl).Exec()
+	}
+
 	return session.Query("INSERT INTO ? (key, value) VALUES (?, ?)", c.table, req.Key, bt).Exec()
 }
 
@@ -303,3 +313,17 @@ func (c *Cassandra) createSession(consistency gocql.Consistency) (*gocql.Session
 
 	return session, nil
 }
+
+func parseTTL(requestMetadata map[string]string) (*int, error) {
+	if val, found := requestMetadata[metadataTTLKey]; found && val != "" {
+		parsedVal, err := strconv.ParseInt(val, 10, 0)
+		if err != nil {
+			return nil, err
+		}
+		parsedInt := int(parsedVal)
+
+		return &parsedInt, nil
+	}
+
+	return nil, nil
+}

state/cassandra/cassandra_test.go

@@ -6,6 +6,7 @@
 package cassandra
 
 import (
+	"strconv"
 	"testing"
 
 	"github.com/dapr/components-contrib/state"
@@ -98,3 +99,35 @@ func TestGetCassandraMetadata(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 }
+
+func TestParseTTL(t *testing.T) {
+	t.Run("TTL Not an integer", func(t *testing.T) {
+		ttlInSeconds := "not an integer"
+		ttl, err := parseTTL(map[string]string{
+			"ttlInSeconds": ttlInSeconds,
+		})
+		assert.Error(t, err)
+		assert.Nil(t, ttl)
+	})
+	t.Run("TTL specified with wrong key", func(t *testing.T) {
+		ttlInSeconds := 12345
+		ttl, err := parseTTL(map[string]string{
+			"expirationTime": strconv.Itoa(ttlInSeconds),
+		})
+		assert.NoError(t, err)
+		assert.Nil(t, ttl)
+	})
+	t.Run("TTL is a number", func(t *testing.T) {
+		ttlInSeconds := 12345
+		ttl, err := parseTTL(map[string]string{
+			"ttlInSeconds": strconv.Itoa(ttlInSeconds),
+		})
+		assert.NoError(t, err)
+		assert.Equal(t, *ttl, ttlInSeconds)
+	})
+	t.Run("TTL not set", func(t *testing.T) {
+		ttl, err := parseTTL(map[string]string{})
+		assert.NoError(t, err)
+		assert.Nil(t, ttl)
+	})
+}