mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-02 12:22:18 +00:00
ankitaryan10andGitHub
fe1eec1962
## What changed? Adding interface methods to Set Idempotency and SpeculativeRetryPolicy on gocql queries ## Why? We want to experiment with setting speculative retry policy in queries from cds to allow for faster detection of faulty nodes and return result before timeout by trying a different node. ## How did you test it? - [x] built - [ ] run locally and tested manually - [ ] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s)
63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
package gocql
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gocql/gocql"
|
|
)
|
|
|
|
// Note: this file defines the minimal interface that is needed by Temporal's cassandra
|
|
// persistence implementation and should be implemented for all gocql libraries if
|
|
// they need to be used.
|
|
// Please add more methods to the interface if needed by the cassandra implementation.
|
|
|
|
type (
|
|
// Session is the interface for interacting with the database.
|
|
Session interface {
|
|
Query(string, ...interface{}) Query
|
|
NewBatch(BatchType) *Batch
|
|
ExecuteBatch(*Batch) error
|
|
MapExecuteBatchCAS(*Batch, map[string]interface{}) (bool, Iter, error)
|
|
AwaitSchemaAgreement(ctx context.Context) error
|
|
Close()
|
|
}
|
|
|
|
// Query is the interface for query object.
|
|
Query interface {
|
|
Exec() error
|
|
Scan(...interface{}) error
|
|
ScanCAS(...interface{}) (bool, error)
|
|
MapScan(map[string]interface{}) error
|
|
MapScanCAS(map[string]interface{}) (bool, error)
|
|
Iter() Iter
|
|
PageSize(int) Query
|
|
PageState([]byte) Query
|
|
WithContext(context.Context) Query
|
|
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.
|
|
Iter interface {
|
|
Scan(...interface{}) bool
|
|
MapScan(map[string]interface{}) bool
|
|
PageState() []byte
|
|
Close() error
|
|
}
|
|
|
|
// BatchType is the type of the Batch operation
|
|
BatchType byte
|
|
|
|
// Consistency is the consistency level used by a Query
|
|
Consistency uint16
|
|
|
|
// SerialConsistency is the serial consistency level used by a Query
|
|
SerialConsistency uint16
|
|
|
|
// SpeculativeExecutionPolicy is a gocql SpeculativeExecutionPolicy
|
|
SpeculativeExecutionPolicy gocql.SpeculativeExecutionPolicy
|
|
)
|