forked from TrueCloudLab/frostfs-node
48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
|
package replicator
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
|
||
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs"
|
||
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/engine"
|
||
|
|
tracingPkg "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/tracing"
|
||
|
|
"git.frostfs.info/TrueCloudLab/frostfs-observability/tracing"
|
||
|
|
"go.opentelemetry.io/otel/attribute"
|
||
|
|
"go.opentelemetry.io/otel/trace"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
var errObjectNotDefined = errors.New("object is not defined")
|
||
|
|
|
||
|
|
func (p *Replicator) HandleLocalPutTask(ctx context.Context, task Task) {
|
||
|
|
p.metrics.IncInFlightRequest()
|
||
|
|
defer p.metrics.DecInFlightRequest()
|
||
|
|
defer func() {
|
||
|
|
p.log.Debug(logs.ReplicatorFinishWork, zap.String("type", "pull"))
|
||
|
|
}()
|
||
|
|
|
||
|
|
ctx, span := tracing.StartSpanFromContext(ctx, "Replicator.HandleLocalPutTask",
|
||
|
|
trace.WithAttributes(
|
||
|
|
attribute.Stringer("address", task.Addr),
|
||
|
|
attribute.Int("nodes_count", len(task.Nodes)),
|
||
|
|
))
|
||
|
|
defer span.End()
|
||
|
|
|
||
|
|
if task.Obj == nil {
|
||
|
|
p.log.Error(logs.ReplicatorCouldNotPutObjectToLocalStorage,
|
||
|
|
zap.Stringer("object", task.Addr),
|
||
|
|
zap.Error(errObjectNotDefined),
|
||
|
|
zap.String("trace_id", tracingPkg.GetTraceID(ctx)))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
err := engine.Put(ctx, p.localStorage, task.Obj)
|
||
|
|
if err != nil {
|
||
|
|
p.log.Error(logs.ReplicatorCouldNotPutObjectToLocalStorage,
|
||
|
|
zap.Stringer("object", task.Addr),
|
||
|
|
zap.Error(err),
|
||
|
|
zap.String("trace_id", tracingPkg.GetTraceID(ctx)))
|
||
|
|
}
|
||
|
|
}
|