neofs: Write object payload if exists

Signed-off-by: Evgenii Baidakov <[email protected]>
nspcc-dev · Sep 12, 2023 · 27779de · 27779de
1 parent f9c431c
commit 27779de
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/neofs/neofs.go b/internal/neofs/neofs.go
@@ -277,6 +277,15 @@ func (x *NeoFS) CreateObject(ctx context.Context, prm layer.PrmObjectCreate) (oi
 		return oid.ID{}, fmt.Errorf("save object via connection pool: %w", err)
 	}
 
+	// Lock payload has been written to header directly, thus we shouldn't miss this payload too.
+	if data := obj.Payload(); len(data) > 0 {
+		if prm.Payload != nil {
+			prm.Payload = io.MultiReader(bytes.NewReader(data), prm.Payload)
+		} else {
+			prm.Payload = bytes.NewReader(data)
+		}
+	}
+
 	chunk := make([]byte, x.maxObjectSize)
 	_, err = io.CopyBuffer(writer, prm.Payload, chunk)
 	if err != nil {