forked from TrueCloudLab/frostfs-node
Dmitrii Stepanov
d62c6e4ce6
Add tracing spans for PUT requests. Add tracing spans for DELETE requests. Add tracing spans for SELECT requests. Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
package engine
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/pkg/tracing"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/shard"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
// FlushWriteCachePrm groups the parameters of FlushWriteCache operation.
|
|
type FlushWriteCachePrm struct {
|
|
shardID *shard.ID
|
|
ignoreErrors bool
|
|
}
|
|
|
|
// SetShardID is an option to set shard ID.
|
|
//
|
|
// Option is required.
|
|
func (p *FlushWriteCachePrm) SetShardID(id *shard.ID) {
|
|
p.shardID = id
|
|
}
|
|
|
|
// SetIgnoreErrors sets errors ignore flag..
|
|
func (p *FlushWriteCachePrm) SetIgnoreErrors(ignore bool) {
|
|
p.ignoreErrors = ignore
|
|
}
|
|
|
|
// FlushWriteCacheRes groups the resulting values of FlushWriteCache operation.
|
|
type FlushWriteCacheRes struct{}
|
|
|
|
// FlushWriteCache flushes write-cache on a single shard.
|
|
func (e *StorageEngine) FlushWriteCache(ctx context.Context, p FlushWriteCachePrm) (FlushWriteCacheRes, error) {
|
|
ctx, span := tracing.StartSpanFromContext(ctx, "StorageEngine.FlushWriteCache",
|
|
trace.WithAttributes(
|
|
attribute.String("shard)id", p.shardID.String()),
|
|
attribute.Bool("ignore_errors", p.ignoreErrors),
|
|
))
|
|
defer span.End()
|
|
|
|
e.mtx.RLock()
|
|
sh, ok := e.shards[p.shardID.String()]
|
|
e.mtx.RUnlock()
|
|
|
|
if !ok {
|
|
return FlushWriteCacheRes{}, errShardNotFound
|
|
}
|
|
|
|
var prm shard.FlushWriteCachePrm
|
|
prm.SetIgnoreErrors(p.ignoreErrors)
|
|
|
|
return FlushWriteCacheRes{}, sh.FlushWriteCache(ctx, prm)
|
|
}
|