Adding interface methods for setting Idempotency and SpeculativeRetryPolicy on the underlying gocql query

Author: ankitaryan10

common/persistence/nosql/nosqlplugin/cassandra/gocql/interfaces.go

@@ -2,6 +2,8 @@ package gocql
 
 import (
 	"context"
+
+	"github.com/gocql/gocql"
 )
 
 // Note: this file defines the minimal interface that is needed by Temporal's cassandra
@@ -34,6 +36,8 @@ type (
 		WithTimestamp(int64) Query
 		Consistency(Consistency) Query
 		Bind(...interface{}) Query
+		Idempotent(bool) Query
+		SetSpeculativeExecutionPolicy(SpeculativeExecutionPolicy) Query
 	}
 
 	// Iter is the interface for executing and iterating over all resulting rows.
@@ -52,4 +56,7 @@ type (
 
 	// SerialConsistency is the serial consistency level used by a Query
 	SerialConsistency uint16
+
+	// SpeculativeExecutionPolicy is a gocql SpeculativeExecutionPolicy
+	SpeculativeExecutionPolicy gocql.SpeculativeExecutionPolicy
 )

common/persistence/nosql/nosqlplugin/cassandra/gocql/query.go

@@ -100,3 +100,11 @@ func (q *query) Bind(v ...interface{}) Query {
 	q.gocqlQuery.Bind(v...)
 	return newQuery(q.session, q.gocqlQuery)
 }
+
+func (q *query) Idempotent(value bool) Query {
+	return newQuery(q.session, q.gocqlQuery.Idempotent(value))
+}
+
+func (q *query) SetSpeculativeExecutionPolicy(policy SpeculativeExecutionPolicy) Query {
+	return newQuery(q.session, q.gocqlQuery.SetSpeculativeExecutionPolicy(policy))
+}