mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-02 10:43:08 +00:00
Chetan GowdaandGitHub
627e29b8f7
## What changed? - Removed the existing unused interface `gocql.Batch`. - Exported `gocql.batch`. This was the only implementation of the above interface. - Fixed few references and types. ## Why? This interface was unused. i.e there were no other implementations other than `gocql.batch`. ## How did you test it? Existing tests ## Potential risks N/A ## Documentation N/A ## Is hotfix candidate? No
84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
// The MIT License
|
|
//
|
|
// Copyright (c) 2020 Temporal Technologies Inc. All rights reserved.
|
|
//
|
|
// Copyright (c) 2020 Uber Technologies, Inc.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
// THE SOFTWARE.
|
|
|
|
package gocql
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/gocql/gocql"
|
|
)
|
|
|
|
type (
|
|
Batch struct {
|
|
session *session
|
|
|
|
gocqlBatch *gocql.Batch
|
|
}
|
|
)
|
|
|
|
// Definition of all BatchTypes
|
|
const (
|
|
LoggedBatch BatchType = iota
|
|
UnloggedBatch
|
|
CounterBatch
|
|
)
|
|
|
|
func newBatch(
|
|
session *session,
|
|
gocqlBatch *gocql.Batch,
|
|
) *Batch {
|
|
return &Batch{
|
|
session: session,
|
|
gocqlBatch: gocqlBatch,
|
|
}
|
|
}
|
|
|
|
func (b *Batch) Query(stmt string, args ...interface{}) {
|
|
b.gocqlBatch.Query(stmt, args...)
|
|
}
|
|
|
|
func (b *Batch) WithContext(ctx context.Context) *Batch {
|
|
return newBatch(b.session, b.gocqlBatch.WithContext(ctx))
|
|
}
|
|
|
|
func (b *Batch) WithTimestamp(timestamp int64) *Batch {
|
|
b.gocqlBatch.WithTimestamp(timestamp)
|
|
return newBatch(b.session, b.gocqlBatch)
|
|
}
|
|
|
|
func mustConvertBatchType(batchType BatchType) gocql.BatchType {
|
|
switch batchType {
|
|
case LoggedBatch:
|
|
return gocql.LoggedBatch
|
|
case UnloggedBatch:
|
|
return gocql.UnloggedBatch
|
|
case CounterBatch:
|
|
return gocql.CounterBatch
|
|
default:
|
|
panic(fmt.Sprintf("Unknown gocql BatchType: %v", batchType))
|
|
}
|
|
}
|