all: Make PdataStorage SetPdata return the stored pdata size

3 files changed, 7 insertions, 7 deletions

diff --git a/pkg/api/api0/accounts.go b/pkg/api/api0/accounts.go
index ed95ad1..c87530e 100644
--- a/pkg/api/api0/accounts.go
+++ b/pkg/api/api0/accounts.go
@@ -147,7 +147,7 @@ func (h *Handler) handleAccountsWritePersistence(w http.ResponseWriter, r *http.
 		}
 	}
 
-	if err := h.PdataStorage.SetPdata(uid, buf); err != nil {
+	if _, err := h.PdataStorage.SetPdata(uid, buf); err != nil {
 		hlog.FromRequest(r).Error().
 			Err(err).
 			Uint64("uid", uid).
diff --git a/pkg/api/api0/api0testutil/storage.go b/pkg/api/api0/api0testutil/storage.go
index 80020fd..f6972a5 100644
--- a/pkg/api/api0/api0testutil/storage.go
+++ b/pkg/api/api0/api0testutil/storage.go
@@ -364,7 +364,7 @@ func TestPdataStorage(t *testing.T, s api0.PdataStorage) {
 			}
 		})
 		t.Run("PutUser1Pdata1", func(t *testing.T) {
-			if err := s.SetPdata(user1, pdata1); err != nil {
+			if _, err := s.SetPdata(user1, pdata1); err != nil {
 				t.Fatalf("unexpected error: %v", err)
 			}
 		})
@@ -416,7 +416,7 @@ func TestPdataStorage(t *testing.T, s api0.PdataStorage) {
 			}
 		})
 		t.Run("PutUser1Pdata2", func(t *testing.T) {
-			if err := s.SetPdata(user1, pdata2); err != nil {
+			if _, err := s.SetPdata(user1, pdata2); err != nil {
 				t.Fatalf("unexpected error: %v", err)
 			}
 		})
@@ -504,7 +504,7 @@ func TestPdataStorage(t *testing.T, s api0.PdataStorage) {
 				}
 				randSched()
 
-				if err := s.SetPdata(uid, data1); err != nil {
+				if _, err := s.SetPdata(uid, data1); err != nil {
 					fail.Store(3)
 					return
 				}
@@ -528,7 +528,7 @@ func TestPdataStorage(t *testing.T, s api0.PdataStorage) {
 				}
 				randSched()
 
-				if err := s.SetPdata(uid, data2); err != nil {
+				if _, err := s.SetPdata(uid, data2); err != nil {
 					fail.Store(7)
 					return
 				}
diff --git a/pkg/api/api0/storage.go b/pkg/api/api0/storage.go
index 3fe7378..a6de4cd 100644
--- a/pkg/api/api0/storage.go
+++ b/pkg/api/api0/storage.go
@@ -61,6 +61,6 @@ type PdataStorage interface {
 	// matches, buf is nil. If another error occurs, err is non-nil.
 	GetPdataCached(uid uint64, sha [sha256.Size]byte) (buf []byte, exists bool, err error)
 
-	// SetPdata sets the raw pdata for uid.
-	SetPdata(uid uint64, buf []byte) (err error)
+	// SetPdata sets the raw pdata for uid, returning the actual size stored.
+	SetPdata(uid uint64, buf []byte) (n int, err error)
 }