31 lines
764 B
Go
31 lines
764 B
Go
|
package cassandra
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/scylladb/gocqlx/v3"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// queryHelper 封裝查詢相關的輔助方法
|
|||
|
|
type queryHelper struct{}
|
|||
|
|
|
|||
|
|
// withTimestamp 為查詢添加時間戳
|
|||
|
|
func (h *queryHelper) withTimestamp(q *gocqlx.Queryx) *gocqlx.Queryx {
|
|||
|
|
return q.WithTimestamp(time.Now().UnixNano() / 1e3)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// withContextAndTimestamp 為查詢添加 context 和時間戳
|
|||
|
|
func (h *queryHelper) withContextAndTimestamp(ctx context.Context, q *gocqlx.Queryx) *gocqlx.Queryx {
|
|||
|
|
return q.WithContext(ctx).WithTimestamp(time.Now().UnixNano() / 1e3)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getKeyspace 獲取 keyspace，如果為空則使用預設值
|
|||
|
|
func getKeyspace(db *CassandraDB, keyspace string) string {
|
|||
|
|
if keyspace == "" {
|
|||
|
|
return db.defaultKeyspace
|
|||
|
|
}
|
|||
|
|
return keyspace
|
|||
|
|
}
|
|||
|
|
|