|
@@ -50,8 +50,7 @@ func (c *Controller) getNetworks() ([]*Network, error) {
|
|
|
return nil, nil
|
|
|
}
|
|
|
|
|
|
- kvol, err := store.List(datastore.Key(datastore.NetworkKeyPrefix),
|
|
|
- &Network{ctrlr: c})
|
|
|
+ kvol, err := store.List(&Network{ctrlr: c})
|
|
|
if err != nil && err != datastore.ErrKeyNotFound {
|
|
|
return nil, fmt.Errorf("failed to get networks: %w", err)
|
|
|
}
|
|
@@ -61,7 +60,7 @@ func (c *Controller) getNetworks() ([]*Network, error) {
|
|
|
n.ctrlr = c
|
|
|
|
|
|
ec := &endpointCnt{n: n}
|
|
|
- err = store.GetObject(datastore.Key(ec.Key()...), ec)
|
|
|
+ err = store.GetObject(ec)
|
|
|
if err != nil && !n.inDelete {
|
|
|
log.G(context.TODO()).Warnf("Could not find endpoint count key %s for network %s while listing: %v", datastore.Key(ec.Key()...), n.Name(), err)
|
|
|
continue
|
|
@@ -81,7 +80,7 @@ func (c *Controller) getNetworksFromStore(ctx context.Context) []*Network { // F
|
|
|
var nl []*Network
|
|
|
|
|
|
store := c.getStore()
|
|
|
- kvol, err := store.List(datastore.Key(datastore.NetworkKeyPrefix), &Network{ctrlr: c})
|
|
|
+ kvol, err := store.List(&Network{ctrlr: c})
|
|
|
if err != nil {
|
|
|
if err != datastore.ErrKeyNotFound {
|
|
|
log.G(ctx).Debugf("failed to get networks from store: %v", err)
|
|
@@ -118,7 +117,7 @@ func (c *Controller) getNetworksFromStore(ctx context.Context) []*Network { // F
|
|
|
func (n *Network) getEndpointFromStore(eid string) (*Endpoint, error) {
|
|
|
store := n.ctrlr.getStore()
|
|
|
ep := &Endpoint{id: eid, network: n}
|
|
|
- err := store.GetObject(datastore.Key(ep.Key()...), ep)
|
|
|
+ err := store.GetObject(ep)
|
|
|
if err != nil {
|
|
|
return nil, fmt.Errorf("could not find endpoint %s: %w", eid, err)
|
|
|
}
|
|
@@ -128,9 +127,8 @@ func (n *Network) getEndpointFromStore(eid string) (*Endpoint, error) {
|
|
|
func (n *Network) getEndpointsFromStore() ([]*Endpoint, error) {
|
|
|
var epl []*Endpoint
|
|
|
|
|
|
- tmp := Endpoint{network: n}
|
|
|
store := n.getController().getStore()
|
|
|
- kvol, err := store.List(datastore.Key(tmp.KeyPrefix()...), &Endpoint{network: n})
|
|
|
+ kvol, err := store.List(&Endpoint{network: n})
|
|
|
if err != nil {
|
|
|
if err != datastore.ErrKeyNotFound {
|
|
|
return nil, fmt.Errorf("failed to get endpoints for network %s: %w",
|
|
@@ -172,7 +170,7 @@ func (c *Controller) deleteFromStore(kvObject datastore.KVObject) error {
|
|
|
retry:
|
|
|
if err := cs.DeleteObjectAtomic(kvObject); err != nil {
|
|
|
if err == datastore.ErrKeyModified {
|
|
|
- if err := cs.GetObject(datastore.Key(kvObject.Key()...), kvObject); err != nil {
|
|
|
+ if err := cs.GetObject(kvObject); err != nil {
|
|
|
return fmt.Errorf("could not update the kvobject to latest when trying to delete: %v", err)
|
|
|
}
|
|
|
log.G(context.TODO()).Warnf("Error (%v) deleting object %v, retrying....", err, kvObject.Key())
|