mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-28 17:51:59 +08:00
37 lines
943 B
Go
37 lines
943 B
Go
package implorganization
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/SigNoz/signoz/pkg/modules/organization"
|
|
"github.com/SigNoz/signoz/pkg/sharder"
|
|
"github.com/SigNoz/signoz/pkg/types"
|
|
"github.com/SigNoz/signoz/pkg/valuer"
|
|
)
|
|
|
|
type getter struct {
|
|
store types.OrganizationStore
|
|
sharder sharder.Sharder
|
|
}
|
|
|
|
func NewGetter(store types.OrganizationStore, sharder sharder.Sharder) organization.Getter {
|
|
return &getter{store: store, sharder: sharder}
|
|
}
|
|
|
|
func (module *getter) Get(ctx context.Context, id valuer.UUID) (*types.Organization, error) {
|
|
return module.store.Get(ctx, id)
|
|
}
|
|
|
|
func (module *getter) List(ctx context.Context) ([]*types.Organization, error) {
|
|
return module.store.GetAll(ctx)
|
|
}
|
|
|
|
func (module *getter) ListByOwnedKeyRange(ctx context.Context) ([]*types.Organization, error) {
|
|
start, end, err := module.sharder.GetMyOwnedKeyRange(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return module.store.ListByKeyRange(ctx, start, end)
|
|
}
|