1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
package datastore
import (
"context"
v1 "github.com/authzed/authzed-go/proto/authzed/api/v1"
)
// DefinitionsOf returns just the schema definitions found in the list of revisioned
// definitions.
func DefinitionsOf[T SchemaDefinition](revisionedDefinitions []RevisionedDefinition[T]) []T {
definitions := make([]T, 0, len(revisionedDefinitions))
for _, revDef := range revisionedDefinitions {
definitions = append(definitions, revDef.Definition)
}
return definitions
}
// DeleteAllData deletes all data from the datastore. Should only be used when explicitly requested.
// The data is transactionally deleted, which means it may time out.
func DeleteAllData(ctx context.Context, ds Datastore) error {
_, err := ds.ReadWriteTx(ctx, func(ctx context.Context, rwt ReadWriteTransaction) error {
nsDefs, err := rwt.ListAllNamespaces(ctx)
if err != nil {
return err
}
// Delete all relationships.
namespaceNames := make([]string, 0, len(nsDefs))
for _, nsDef := range nsDefs {
_, _, err = rwt.DeleteRelationships(ctx, &v1.RelationshipFilter{
ResourceType: nsDef.Definition.Name,
})
if err != nil {
return err
}
namespaceNames = append(namespaceNames, nsDef.Definition.Name)
}
// Delete all caveats.
caveatDefs, err := rwt.ListAllCaveats(ctx)
if err != nil {
return err
}
caveatNames := make([]string, 0, len(caveatDefs))
for _, caveatDef := range caveatDefs {
caveatNames = append(caveatNames, caveatDef.Definition.Name)
}
if err := rwt.DeleteCaveats(ctx, caveatNames); err != nil {
return err
}
// Delete all namespaces.
if err := rwt.DeleteNamespaces(ctx, namespaceNames...); err != nil {
return err
}
return nil
})
return err
}
|
