IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-25 18:55:37 -06:00
Code Issues Packages Projects Releases Wiki Activity

Zanzana: Refactor stores listing (#99098)

Browse Source 
Zanzana: Refactor store loading
This commit is contained in:
Alexander Zobnin 2025-01-17 11:10:22 +01:00 committed by GitHub
parent 5e23b2c07f
commit c5f14407cc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
pkg/services/authz/zanzana/server/server_store.go
View File

@ -40,40 +40,25 @@ func (s *Server) getStoreInfo(ctx context.Context, namespace string) (*storeInfo
}
func (s *Server) getOrCreateStore(ctx context.Context, namespace string) (*openfgav1.Store, error) {
var continuationToken string
for {
res, err := s.openfga.ListStores(ctx, &openfgav1.ListStoresRequest{
PageSize: &wrapperspb.Int32Value{Value: 100},
ContinuationToken: continuationToken,
})
if err != nil {
return nil, fmt.Errorf("failed to load zanzana stores: %w", err)
}
for _, s := range res.GetStores() {
if s.GetName() == namespace {
return s, nil
}
}
// we have no more stores to check
if res.GetContinuationToken() == "" {
break
}
continuationToken = res.GetContinuationToken()
res, err := s.openfga.ListStores(ctx, &openfgav1.ListStoresRequest{Name: namespace})
if err != nil {
return nil, fmt.Errorf("failed to load zanzana stores: %w", err)
}
res, err := s.openfga.CreateStore(ctx, &openfgav1.CreateStoreRequest{Name: namespace})
for _, s := range res.GetStores() {
if s.GetName() == namespace {
return s, nil
}
}
createStoreRes, err := s.openfga.CreateStore(ctx, &openfgav1.CreateStoreRequest{Name: namespace})
if err != nil {
return nil, err
}
return &openfgav1.Store{
Id: res.GetId(),
Name: res.GetName(),
Id: createStoreRes.GetId(),
Name: createStoreRes.GetName(),
}, nil
}