refactor: always request higher consistency on export

openfga · Sep 3, 2024 · eb16366 · eb16366
1 parent 12ba9f5
commit eb16366
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/cmd/store/export.go b/cmd/store/export.go
@@ -22,6 +22,7 @@ import (
 	"math"
 	"os"
 
+	openfga "github.com/openfga/go-sdk"
 	"github.com/openfga/go-sdk/client"
 	"github.com/spf13/cobra"
 	"gopkg.in/yaml.v3"
@@ -68,7 +69,12 @@ func buildStoreData(config fga.ClientConfig, fgaClient client.SdkClient, maxTupl
 	// get the tuples
 	maxPages := int(math.Ceil(float64(maxTupleCount) / float64(tuple.DefaultReadPageSize)))
 
-	rawTuples, err := tuple.Read(fgaClient, &client.ClientReadRequest{}, maxPages, nil)
+	rawTuples, err := tuple.Read(
+		fgaClient,
+		&client.ClientReadRequest{},
+		maxPages,
+		openfga.CONSISTENCYPREFERENCE_HIGHER_CONSISTENCY.Ptr(),
+	)
 	if err != nil {
 		return nil, fmt.Errorf("unable to read tuples: %w", err)
 	}

diff --git a/cmd/store/export_test.go b/cmd/store/export_test.go
@@ -136,6 +136,7 @@ func TestExportSuccess(t *testing.T) {
 	readOptions := client.ClientReadOptions{
 		PageSize:          openfga.PtrInt32(tuple.DefaultReadPageSize),
 		ContinuationToken: openfga.PtrString(""),
+		Consistency:       openfga.CONSISTENCYPREFERENCE_HIGHER_CONSISTENCY.Ptr(),
 	}
 
 	mockReadRequest := mock_client.NewMockSdkClientReadRequestInterface(mockCtrl)