From 84b4ff0755d1d7d0f5c2321b26a696519cf9968b Mon Sep 17 00:00:00 2001 From: Evgenii Stratonikov Date: Tue, 15 Sep 2020 17:10:38 +0300 Subject: [PATCH] [#27] Support JSON format for placement policy Signed-off-by: Evgenii Stratonikov --- pkg/policy/json.go | 221 ++++++++++++++++++++++++++++++++++++++++ pkg/policy/json_test.go | 84 +++++++++++++++ pkg/policy/query.go | 1 + 3 files changed, 306 insertions(+) create mode 100644 pkg/policy/json.go create mode 100644 pkg/policy/json_test.go diff --git a/pkg/policy/json.go b/pkg/policy/json.go new file mode 100644 index 00000000..33eb7986 --- /dev/null +++ b/pkg/policy/json.go @@ -0,0 +1,221 @@ +package policy + +import ( + "encoding/json" + "fmt" + "strings" + + "github.com/nspcc-dev/neofs-api-go/v2/netmap" +) + +type ( + filter struct { + Name string `json:"name,omitempty"` + Key string `json:"key,omitempty"` + Op string `json:"op,omitempty"` + Value string `json:"value,omitempty"` + Filters []filter `json:"filters,omitempty"` + } + replica struct { + Count uint32 `json:"count"` + Selector string `json:"selector,omitempty"` + } + selector struct { + Count uint32 `json:"count"` + Attribute string `json:"attribute"` + Filter string `json:"filter,omitempty"` + Name string `json:"name,omitempty"` + Clause string `json:"clause,omitempty"` + } + placement struct { + Replicas []replica `json:"replicas"` + CBF uint32 `json:"container_backup_factor,omitempty"` + Selectors []selector `json:"selectors,omitempty"` + Filters []filter `json:"filters,omitempty"` + } +) + +// ToJSON converts placement policy to JSON. +func ToJSON(np *netmap.PlacementPolicy) ([]byte, error) { + p := new(placement) + p.CBF = np.GetContainerBackupFactor() + p.Filters = make([]filter, len(np.GetFilters())) + for i, f := range np.GetFilters() { + p.Filters[i].fromNetmap(f) + } + p.Selectors = make([]selector, len(np.GetSelectors())) + for i, s := range np.GetSelectors() { + p.Selectors[i].fromNetmap(s) + } + p.Replicas = make([]replica, len(np.GetReplicas())) + for i, r := range np.GetReplicas() { + p.Replicas[i].fromNetmap(r) + } + return json.Marshal(p) +} + +// ToJSON creates placement policy from JSON. +func FromJSON(data []byte) (*netmap.PlacementPolicy, error) { + p := new(placement) + if err := json.Unmarshal(data, p); err != nil { + return nil, err + } + + rs := make([]*netmap.Replica, len(p.Replicas)) + for i := range p.Replicas { + rs[i] = p.Replicas[i].toNetmap() + } + + var fs []*netmap.Filter + if len(p.Filters) != 0 { + fs = make([]*netmap.Filter, len(p.Filters)) + for i := range p.Filters { + f, err := p.Filters[i].toNetmap() + if err != nil { + return nil, err + } + fs[i] = f + } + } + + var ss []*netmap.Selector + if len(p.Selectors) != 0 { + ss = make([]*netmap.Selector, len(p.Selectors)) + for i := range p.Selectors { + s, err := p.Selectors[i].toNetmap() + if err != nil { + return nil, err + } + ss[i] = s + } + } + + pp := new(netmap.PlacementPolicy) + pp.SetReplicas(rs) + pp.SetContainerBackupFactor(p.CBF) + pp.SetFilters(fs) + pp.SetSelectors(ss) + return pp, nil +} + +func (r *replica) toNetmap() *netmap.Replica { + nr := new(netmap.Replica) + nr.SetCount(r.Count) + nr.SetSelector(r.Selector) + return nr +} + +func (r *replica) fromNetmap(nr *netmap.Replica) { + r.Count = nr.GetCount() + r.Selector = nr.GetSelector() +} + +func (f *filter) toNetmap() (*netmap.Filter, error) { + var op netmap.Operation + switch strings.ToUpper(f.Op) { + case "EQ": + op = netmap.EQ + case "NE": + op = netmap.NE + case "GT": + op = netmap.GT + case "GE": + op = netmap.GE + case "LT": + op = netmap.LT + case "LE": + op = netmap.LE + case "AND": + op = netmap.AND + case "OR": + op = netmap.OR + case "": + op = netmap.UnspecifiedOperation + default: + return nil, fmt.Errorf("%w: '%s'", ErrUnknownOp, f.Op) + } + + var fs []*netmap.Filter + if len(f.Filters) != 0 { + fs = make([]*netmap.Filter, len(f.Filters)) + for i := range f.Filters { + var err error + fs[i], err = f.Filters[i].toNetmap() + if err != nil { + return nil, err + } + } + } + + nf := new(netmap.Filter) + nf.SetFilters(fs) + nf.SetOp(op) + nf.SetName(f.Name) + nf.SetValue(f.Value) + nf.SetKey(f.Key) + return nf, nil +} + +func (f *filter) fromNetmap(nf *netmap.Filter) { + f.Name = nf.GetName() + f.Key = nf.GetKey() + f.Value = nf.GetValue() + switch nf.GetOp() { + case netmap.EQ: + f.Op = "EQ" + case netmap.NE: + f.Op = "NE" + case netmap.GT: + f.Op = "GT" + case netmap.GE: + f.Op = "GE" + case netmap.LT: + f.Op = "LT" + case netmap.LE: + f.Op = "LE" + case netmap.AND: + f.Op = "AND" + case netmap.OR: + f.Op = "OR" + } + if nf.GetFilters() != nil { + f.Filters = make([]filter, len(nf.GetFilters())) + for i, sf := range nf.GetFilters() { + f.Filters[i].fromNetmap(sf) + } + } +} + +func (s *selector) toNetmap() (*netmap.Selector, error) { + var c netmap.Clause + switch strings.ToUpper(s.Clause) { + case "SAME": + c = netmap.Same + case "DISTINCT": + c = netmap.Distinct + case "": + default: + return nil, fmt.Errorf("%w: '%s'", ErrUnknownClause, s.Clause) + } + ns := new(netmap.Selector) + ns.SetName(s.Name) + ns.SetAttribute(s.Attribute) + ns.SetCount(s.Count) + ns.SetClause(c) + ns.SetFilter(s.Filter) + return ns, nil +} + +func (s *selector) fromNetmap(ns *netmap.Selector) { + s.Name = ns.GetName() + s.Filter = ns.GetFilter() + s.Count = ns.GetCount() + s.Attribute = ns.GetAttribute() + switch ns.GetClause() { + case netmap.Same: + s.Clause = "same" + case netmap.Distinct: + s.Clause = "distinct" + } + s.Name = ns.GetName() +} diff --git a/pkg/policy/json_test.go b/pkg/policy/json_test.go new file mode 100644 index 00000000..60b1a15d --- /dev/null +++ b/pkg/policy/json_test.go @@ -0,0 +1,84 @@ +package policy + +import ( + "testing" + + "github.com/nspcc-dev/neofs-api-go/v2/netmap" + "github.com/stretchr/testify/require" +) + +func TestToJSON(t *testing.T) { + check := func(t *testing.T, p *netmap.PlacementPolicy, json string) { + data, err := ToJSON(p) + require.NoError(t, err) + require.JSONEq(t, json, string(data)) + + np, err := FromJSON(data) + require.NoError(t, err) + require.Equal(t, p, np) + } + t.Run("SimpleREP", func(t *testing.T) { + p := new(netmap.PlacementPolicy) + p.SetReplicas([]*netmap.Replica{newReplica("", 3)}) + check(t, p, `{"replicas":[{"count":3}]}`) + }) + t.Run("REPWithCBF", func(t *testing.T) { + p := new(netmap.PlacementPolicy) + p.SetReplicas([]*netmap.Replica{newReplica("", 3)}) + p.SetContainerBackupFactor(3) + check(t, p, `{"replicas":[{"count":3}],"container_backup_factor":3}`) + }) + t.Run("REPFromSelector", func(t *testing.T) { + p := new(netmap.PlacementPolicy) + p.SetReplicas([]*netmap.Replica{newReplica("Nodes", 3)}) + p.SetContainerBackupFactor(3) + p.SetSelectors([]*netmap.Selector{ + newSelector(1, netmap.Distinct, "City", "", "Nodes"), + }) + check(t, p, `{ + "replicas":[{"count":3,"selector":"Nodes"}], + "container_backup_factor":3, + "selectors": [{ + "name":"Nodes", + "attribute":"City", + "clause":"distinct", + "count":1 + }]}`) + }) + t.Run("FilterOps", func(t *testing.T) { + p := new(netmap.PlacementPolicy) + p.SetReplicas([]*netmap.Replica{newReplica("Nodes", 3)}) + p.SetContainerBackupFactor(3) + p.SetSelectors([]*netmap.Selector{ + newSelector(1, netmap.Same, "City", "Good", "Nodes"), + }) + p.SetFilters([]*netmap.Filter{ + newFilter("GoodRating", "Rating", "5", netmap.GE), + newFilter("Good", "", "", netmap.OR, + newFilter("GoodRating", "", "", netmap.UnspecifiedOperation), + newFilter("", "Attr1", "Val1", netmap.EQ), + newFilter("", "Attr2", "Val2", netmap.NE), + newFilter("", "", "", netmap.AND, + newFilter("", "Attr4", "2", netmap.LT), + newFilter("", "Attr5", "3", netmap.LE)), + newFilter("", "Attr3", "1", netmap.GT)), + }) + check(t, p, `{ + "replicas":[{"count":3,"selector":"Nodes"}], + "container_backup_factor":3, + "selectors": [{"name":"Nodes","attribute":"City","clause":"same","count":1,"filter":"Good"}], + "filters": [ + {"name":"GoodRating","key":"Rating","op":"GE","value":"5"}, + {"name":"Good","op":"OR","filters":[ + {"name":"GoodRating"}, + {"key":"Attr1","op":"EQ","value":"Val1"}, + {"key":"Attr2","op":"NE","value":"Val2"}, + {"op":"AND","filters":[ + {"key":"Attr4","op":"LT","value":"2"}, + {"key":"Attr5","op":"LE","value":"3"} + ]}, + {"key":"Attr3","op":"GT","value":"1"} + ]} + ]}`) + }) +} diff --git a/pkg/policy/query.go b/pkg/policy/query.go index b7e055e8..8d5c2dce 100644 --- a/pkg/policy/query.go +++ b/pkg/policy/query.go @@ -11,6 +11,7 @@ import ( var ( ErrInvalidNumber = errors.New("policy: expected positive integer") + ErrUnknownClause = errors.New("policy: unknown clause") ErrUnknownOp = errors.New("policy: unknown operation") ErrUnknownFilter = errors.New("policy: filter not found") ErrUnknownSelector = errors.New("policy: selector not found")