Merge pull request #244 from stgraber/main

incusd/auth/openfga: Use chunking
lxc · Nov 28, 2023 · 4bf57b1 · 4bf57b1
2 parents 9e9ff11 + 28123a2
commit 4bf57b1
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/internal/server/auth/driver_openfga.go b/internal/server/auth/driver_openfga.go
@@ -779,14 +779,27 @@ func (f *fga) updateTuples(ctx context.Context, writes []client.ClientTupleKey,
 	ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
 	defer cancel()
 
-	opts := client.ClientWriteOptions{AuthorizationModelId: openfga.PtrString(f.authModelID)}
+	opts := client.ClientWriteOptions{
+		AuthorizationModelId: openfga.PtrString(f.authModelID),
+		Transaction: &client.TransactionOptions{
+			Disable:             true,
+			MaxParallelRequests: 5,
+			MaxPerChunk:         50,
+		},
+	}
+
 	body := client.ClientWriteRequest{}
+
 	if writes != nil {
 		body.Writes = &writes
+	} else {
+		body.Writes = &[]client.ClientTupleKey{}
 	}
 
 	if deletions != nil {
 		body.Deletes = &deletions
+	} else {
+		body.Deletes = &[]client.ClientTupleKey{}
 	}
 
 	clientWriteResponse, err := f.client.Write(ctx).Options(opts).Body(body).Execute()