index: GetStatusEnttry method

Author: cheggaaa

nodestorage/indexstorage.go

@@ -17,6 +17,17 @@ import (
 
 type SpaceStatus int
 
+type SpaceStatusEntry struct {
+	SpaceId                 string
+	Status                  SpaceStatus
+	Error                   string
+	NewHash                 string
+	OldHash                 string
+	LastAccess              time.Time
+	ArchiveSizeCompressed   int64
+	ArchiveSizeUncompressed int64
+}
+
 const (
 	SpaceStatusOk SpaceStatus = iota
 	SpaceStatusRemove
@@ -152,6 +163,25 @@ func (d *indexStorage) SpaceStatus(ctx context.Context, spaceId string) (status
 	return SpaceStatus(doc.Value().GetInt(statusKey)), nil
 }
 
+func (d *indexStorage) SpaceStatusEntry(ctx context.Context, spaceId string) (entry SpaceStatusEntry, err error) {
+	doc, err := d.spaceColl.FindId(ctx, spaceId)
+	if err != nil {
+		return entry, err
+	}
+	v := doc.Value()
+	entry = SpaceStatusEntry{
+		SpaceId:                 spaceId,
+		Status:                  SpaceStatus(v.GetInt(statusKey)),
+		Error:                   v.GetString(errorKey),
+		NewHash:                 v.GetString(newHashKey),
+		OldHash:                 v.GetString(oldHashKey),
+		LastAccess:              time.Unix(int64(v.GetInt(lastAccessKey)), 0),
+		ArchiveSizeCompressed:   int64(v.GetInt(archiveSizeCompressedKey)),
+		ArchiveSizeUncompressed: int64(v.GetInt(archiveSizeUncompressedKey)),
+	}
+	return entry, nil
+}
+
 func (d *indexStorage) SetSpaceStatus(ctx context.Context, spaceId string, status SpaceStatus, recId string) (err error) {
 	tx, err := d.db.WriteTx(ctx)
 	if err != nil {

nodestorage/indexstorage_test.go

@@ -114,6 +114,58 @@ func TestIndexStorage_MarkArchived(t *testing.T) {
 	assert.Equal(t, SpaceStatusArchived, status)
 }
 
+func TestIndexStorage_SpaceStatusEntry(t *testing.T) {
+	t.Run("not found", func(t *testing.T) {
+		tempDir := t.TempDir()
+		fx, err := createTestIndexStorage(ctx, tempDir)
+		require.NoError(t, err)
+		defer fx.Close()
+
+		_, err = fx.(*indexStorage).SpaceStatusEntry(ctx, "nonexistent")
+		require.ErrorIs(t, err, anystore.ErrDocNotFound)
+	})
+
+	t.Run("existing entry", func(t *testing.T) {
+		tempDir := t.TempDir()
+		fx, err := createTestIndexStorage(ctx, tempDir)
+		require.NoError(t, err)
+		defer fx.Close()
+
+		require.NoError(t, fx.SetSpaceStatus(ctx, "space1", SpaceStatusOk, ""))
+		require.NoError(t, fx.UpdateHash(ctx, SpaceUpdate{
+			SpaceId: "space1",
+			OldHash: "old",
+			NewHash: "new",
+		}))
+		require.NoError(t, fx.MarkArchived(ctx, "space1", 100, 200))
+
+		entry, err := fx.(*indexStorage).SpaceStatusEntry(ctx, "space1")
+		require.NoError(t, err)
+		assert.Equal(t, "space1", entry.SpaceId)
+		assert.Equal(t, SpaceStatusArchived, entry.Status)
+		assert.Equal(t, "new", entry.NewHash)
+		assert.Equal(t, "old", entry.OldHash)
+		assert.Equal(t, int64(100), entry.ArchiveSizeCompressed)
+		assert.Equal(t, int64(200), entry.ArchiveSizeUncompressed)
+		assert.False(t, entry.LastAccess.IsZero())
+	})
+
+	t.Run("error status", func(t *testing.T) {
+		tempDir := t.TempDir()
+		fx, err := createTestIndexStorage(ctx, tempDir)
+		require.NoError(t, err)
+		defer fx.Close()
+
+		require.NoError(t, fx.SetSpaceStatus(ctx, "space1", SpaceStatusOk, ""))
+		require.NoError(t, fx.MarkError(ctx, "space1", "some error"))
+
+		entry, err := fx.(*indexStorage).SpaceStatusEntry(ctx, "space1")
+		require.NoError(t, err)
+		assert.Equal(t, SpaceStatusError, entry.Status)
+		assert.Equal(t, "some error", entry.Error)
+	})
+}
+
 func TestIndexStorage_MarkError(t *testing.T) {
 	tempDir := t.TempDir()
 	fx, err := createTestIndexStorage(ctx, tempDir)