fix document check when node specific

sourcenetwork · Jan 16, 2025 · da75557 · da75557
1 parent 10e0f38
commit da75557
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/internal/db/collection_acp.go b/internal/db/collection_acp.go
@@ -55,6 +55,10 @@ func (c *collection) checkAccessOfDocWithACP(
 	if !c.db.acp.HasValue() {
 		return true, nil
 	}
+	ident := identity.FromContext(ctx)
+	if ident.HasValue() && c.db.nodeIdentity.HasValue() && ident.Value().DID == c.db.nodeIdentity.Value().DID {
+		return true, nil
+	}
 	return permission.CheckAccessOfDocOnCollectionWithACP(
 		ctx,
 		identity.FromContext(ctx),

diff --git a/internal/db/p2p_replicator.go b/internal/db/p2p_replicator.go
@@ -21,6 +21,7 @@ import (
 
 	"github.com/sourcenetwork/corelog"
 
+	"github.com/sourcenetwork/defradb/acp/identity"
 	"github.com/sourcenetwork/defradb/client"
 	"github.com/sourcenetwork/defradb/datastore"
 	dbErrors "github.com/sourcenetwork/defradb/errors"
@@ -156,6 +157,8 @@ func (db *db) getDocsHeads(
 		}
 		defer txn.Discard(ctx)
 		ctx = SetContextTxn(ctx, txn)
+		// This is a node specific action which means the actor is the node itself.
+		ctx = identity.WithContext(ctx, db.nodeIdentity)
 		for _, col := range cols {
 			keysCh, err := col.GetAllDocIDs(ctx)
 			if err != nil {

diff --git a/internal/db/p2p_schema_root.go b/internal/db/p2p_schema_root.go
@@ -17,6 +17,7 @@ import (
 	"github.com/libp2p/go-libp2p/core/peer"
 	"github.com/sourcenetwork/immutable"
 
+	"github.com/sourcenetwork/defradb/acp/identity"
 	"github.com/sourcenetwork/defradb/client"
 	"github.com/sourcenetwork/defradb/event"
 	"github.com/sourcenetwork/defradb/internal/keys"
@@ -72,6 +73,8 @@ func (db *db) AddP2PCollections(ctx context.Context, collectionIDs []string) err
 		evt.ToAdd = append(evt.ToAdd, col.SchemaRoot())
 	}
 
+	// This is a node specific action which means the actor is the node itself.
+	ctx = identity.WithContext(ctx, db.nodeIdentity)
 	for _, col := range storeCollections {
 		keyChan, err := col.GetAllDocIDs(ctx)
 		if err != nil {
@@ -129,6 +132,8 @@ func (db *db) RemoveP2PCollections(ctx context.Context, collectionIDs []string)
 		evt.ToRemove = append(evt.ToRemove, col.SchemaRoot())
 	}
 
+	// This is a node specific action which means the actor is the node itself.
+	ctx = identity.WithContext(ctx, db.nodeIdentity)
 	for _, col := range storeCollections {
 		keyChan, err := col.GetAllDocIDs(ctx)
 		if err != nil {
@@ -202,6 +207,8 @@ func (db *db) loadAndPublishP2PCollections(ctx context.Context) error {
 		colMap[schemaRoot] = struct{}{}
 	}
 
+	// This is a node specific action which means the actor is the node itself.
+	ctx = identity.WithContext(ctx, db.nodeIdentity)
 	for _, col := range cols {
 		// If we subscribed to the collection, we skip subscribing to the collection's docIDs.
 		if _, ok := colMap[col.SchemaRoot()]; ok {