Files
tasks/common/persistence/nosql/nosqlplugin/cassandra/gocql/query.go
Stephan BehnkeandGitHub c9e360a3fb go fix (#9337)
## What changed?

Integrate Go 1.26's new `go fix` into workflow.

NOTE that the changes caused our linter to fire; a [separate
commit](https://github.com/temporalio/temporal/pull/9337/commits/1b23f787aef370098b35829c4d17dcc06230b4c1)
addresses those.

## Why?

Ensure Go code is standardized/modernized.

## How did you test it?
- [ ] built
- [ ] run locally and tested manually
- [x] covered by existing tests
- [ ] added new unit test(s)
- [ ] added new functional test(s)
2026-02-18 09:12:19 -08:00

111 lines
2.2 KiB
Go

package gocql
import (
"context"
"github.com/gocql/gocql"
)
var _ Query = (*query)(nil)
type (
query struct {
session *session
gocqlQuery *gocql.Query
}
)
func newQuery(
session *session,
gocqlQuery *gocql.Query,
) *query {
return &query{
session: session,
gocqlQuery: gocqlQuery,
}
}
func (q *query) Exec() (retError error) {
defer func() { q.session.handleError(retError) }()
return q.gocqlQuery.Exec()
}
func (q *query) Scan(
dest ...any,
) (retError error) {
defer func() { q.session.handleError(retError) }()
return q.gocqlQuery.Scan(dest...)
}
func (q *query) ScanCAS(
dest ...any,
) (_ bool, retError error) {
defer func() { q.session.handleError(retError) }()
return q.gocqlQuery.ScanCAS(dest...)
}
func (q *query) MapScan(
m map[string]any,
) (retError error) {
defer func() { q.session.handleError(retError) }()
return q.gocqlQuery.MapScan(m)
}
func (q *query) MapScanCAS(
dest map[string]any,
) (_ bool, retError error) {
defer func() { q.session.handleError(retError) }()
return q.gocqlQuery.MapScanCAS(dest)
}
func (q *query) Iter() Iter {
iter := q.gocqlQuery.Iter()
return newIter(q.session, iter)
}
func (q *query) PageSize(n int) Query {
q.gocqlQuery.PageSize(n)
return newQuery(q.session, q.gocqlQuery)
}
func (q *query) PageState(state []byte) Query {
q.gocqlQuery.PageState(state)
return newQuery(q.session, q.gocqlQuery)
}
func (q *query) Consistency(c Consistency) Query {
q.gocqlQuery.Consistency(mustConvertConsistency(c))
return newQuery(q.session, q.gocqlQuery)
}
func (q *query) WithTimestamp(timestamp int64) Query {
q.gocqlQuery.WithTimestamp(timestamp)
return newQuery(q.session, q.gocqlQuery)
}
func (q *query) WithContext(ctx context.Context) Query {
q2 := q.gocqlQuery.WithContext(ctx)
if q2 == nil {
return nil
}
return newQuery(q.session, q2)
}
func (q *query) Bind(v ...any) 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))
}