Files
tasks/common/persistence/sql/sqlplugin/sqlite/shard.go
T
Alex ShtinandGitHub 8e8f1b0f7c Remove license header from every file (#7689)
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.

## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
2025-05-01 18:50:21 -07:00

98 lines
2.0 KiB
Go

package sqlite
import (
"context"
"database/sql"
"go.temporal.io/server/common/persistence/sql/sqlplugin"
)
const (
createShardQry = `INSERT INTO
shards (shard_id, range_id, data, data_encoding) VALUES (?, ?, ?, ?)`
getShardQry = `SELECT
shard_id, range_id, data, data_encoding
FROM shards WHERE shard_id = ?`
updateShardQry = `UPDATE shards
SET range_id = ?, data = ?, data_encoding = ?
WHERE shard_id = ?`
lockShardQry = `SELECT range_id FROM shards WHERE shard_id = ?`
readLockShardQry = lockShardQry
)
// InsertIntoShards inserts one or more rows into shards table
func (mdb *db) InsertIntoShards(
ctx context.Context,
row *sqlplugin.ShardsRow,
) (sql.Result, error) {
return mdb.conn.ExecContext(ctx,
createShardQry,
row.ShardID,
row.RangeID,
row.Data,
row.DataEncoding,
)
}
// UpdateShards updates one or more rows into shards table
func (mdb *db) UpdateShards(
ctx context.Context,
row *sqlplugin.ShardsRow,
) (sql.Result, error) {
return mdb.conn.ExecContext(ctx,
updateShardQry,
row.RangeID,
row.Data,
row.DataEncoding,
row.ShardID,
)
}
// SelectFromShards reads one or more rows from shards table
func (mdb *db) SelectFromShards(
ctx context.Context,
filter sqlplugin.ShardsFilter,
) (*sqlplugin.ShardsRow, error) {
var row sqlplugin.ShardsRow
err := mdb.conn.GetContext(ctx,
&row,
getShardQry,
filter.ShardID,
)
if err != nil {
return nil, err
}
return &row, err
}
// ReadLockShards acquires a read lock on a single row in shards table
func (mdb *db) ReadLockShards(
ctx context.Context,
filter sqlplugin.ShardsFilter,
) (int64, error) {
var rangeID int64
err := mdb.conn.GetContext(ctx,
&rangeID,
readLockShardQry,
filter.ShardID,
)
return rangeID, err
}
// WriteLockShards acquires a write lock on a single row in shards table
func (mdb *db) WriteLockShards(
ctx context.Context,
filter sqlplugin.ShardsFilter,
) (int64, error) {
var rangeID int64
err := mdb.conn.GetContext(ctx,
&rangeID,
lockShardQry,
filter.ShardID,
)
return rangeID, err
}