forked from TrueCloudLab/policy-engine
4d8242584a
* Make LocalOverrideStorage methods to receive Target type instead resource * Refactor unit-tests and dependencies * Make default chain router check local overrides not only for container but also for namespaces Signed-off-by: Airat Arifullin <aarifullin@yadro.com>
52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
package inmemory
|
|
|
|
import (
|
|
"git.frostfs.info/TrueCloudLab/policy-engine/pkg/chain"
|
|
"git.frostfs.info/TrueCloudLab/policy-engine/pkg/engine"
|
|
)
|
|
|
|
type inmemoryMorphRuleChainStorage struct {
|
|
nameToNamespaceChains engine.LocalOverrideStorage
|
|
nameToContainerChains engine.LocalOverrideStorage
|
|
}
|
|
|
|
func NewInmemoryMorphRuleChainStorage() engine.MorphRuleChainStorage {
|
|
return &inmemoryMorphRuleChainStorage{
|
|
nameToNamespaceChains: NewInmemoryLocalStorage(),
|
|
nameToContainerChains: NewInmemoryLocalStorage(),
|
|
}
|
|
}
|
|
|
|
func (s *inmemoryMorphRuleChainStorage) AddMorphRuleChain(name chain.Name, target engine.Target, c *chain.Chain) (err error) {
|
|
switch target.Type {
|
|
case engine.Namespace:
|
|
_, err = s.nameToNamespaceChains.AddOverride(name, target, c)
|
|
case engine.Container:
|
|
_, err = s.nameToContainerChains.AddOverride(name, target, c)
|
|
default:
|
|
err = engine.ErrUnknownTarget
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *inmemoryMorphRuleChainStorage) RemoveMorphRuleChain(name chain.Name, target engine.Target, chainID chain.ID) error {
|
|
switch target.Type {
|
|
case engine.Namespace:
|
|
return s.nameToNamespaceChains.RemoveOverride(name, target, chainID)
|
|
case engine.Container:
|
|
return s.nameToContainerChains.RemoveOverride(name, target, chainID)
|
|
default:
|
|
return engine.ErrUnknownTarget
|
|
}
|
|
}
|
|
|
|
func (s *inmemoryMorphRuleChainStorage) ListMorphRuleChains(name chain.Name, target engine.Target) ([]*chain.Chain, error) {
|
|
switch target.Type {
|
|
case engine.Namespace:
|
|
return s.nameToNamespaceChains.ListOverrides(name, target)
|
|
case engine.Container:
|
|
return s.nameToContainerChains.ListOverrides(name, target)
|
|
default:
|
|
}
|
|
return nil, engine.ErrUnknownTarget
|
|
}
|