126 lines
2.9 KiB
Go
126 lines
2.9 KiB
Go
package ipam
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/docker/docker/libnetwork/datastore"
|
|
"github.com/docker/docker/libnetwork/types"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// Key provides the Key to be used in KV Store
|
|
func (aSpace *addrSpace) Key() []string {
|
|
aSpace.Lock()
|
|
defer aSpace.Unlock()
|
|
return []string{aSpace.id}
|
|
}
|
|
|
|
// KeyPrefix returns the immediate parent key that can be used for tree walk
|
|
func (aSpace *addrSpace) KeyPrefix() []string {
|
|
aSpace.Lock()
|
|
defer aSpace.Unlock()
|
|
return []string{dsConfigKey}
|
|
}
|
|
|
|
// Value marshals the data to be stored in the KV store
|
|
func (aSpace *addrSpace) Value() []byte {
|
|
b, err := json.Marshal(aSpace)
|
|
if err != nil {
|
|
logrus.Warnf("Failed to marshal ipam configured pools: %v", err)
|
|
return nil
|
|
}
|
|
return b
|
|
}
|
|
|
|
// SetValue unmarshalls the data from the KV store.
|
|
func (aSpace *addrSpace) SetValue(value []byte) error {
|
|
rc := &addrSpace{subnets: make(map[SubnetKey]*PoolData)}
|
|
if err := json.Unmarshal(value, rc); err != nil {
|
|
return err
|
|
}
|
|
aSpace.subnets = rc.subnets
|
|
return nil
|
|
}
|
|
|
|
// Index returns the latest DB Index as seen by this object
|
|
func (aSpace *addrSpace) Index() uint64 {
|
|
aSpace.Lock()
|
|
defer aSpace.Unlock()
|
|
return aSpace.dbIndex
|
|
}
|
|
|
|
// SetIndex method allows the datastore to store the latest DB Index into this object
|
|
func (aSpace *addrSpace) SetIndex(index uint64) {
|
|
aSpace.Lock()
|
|
aSpace.dbIndex = index
|
|
aSpace.dbExists = true
|
|
aSpace.Unlock()
|
|
}
|
|
|
|
// Exists method is true if this object has been stored in the DB.
|
|
func (aSpace *addrSpace) Exists() bool {
|
|
aSpace.Lock()
|
|
defer aSpace.Unlock()
|
|
return aSpace.dbExists
|
|
}
|
|
|
|
// Skip provides a way for a KV Object to avoid persisting it in the KV Store
|
|
func (aSpace *addrSpace) Skip() bool {
|
|
return false
|
|
}
|
|
|
|
func (a *Allocator) getStore(as string) datastore.DataStore {
|
|
a.Lock()
|
|
defer a.Unlock()
|
|
|
|
if aSpace, ok := a.addrSpaces[as]; ok {
|
|
return aSpace.ds
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *Allocator) getAddressSpaceFromStore(as string) (*addrSpace, error) {
|
|
store := a.getStore(as)
|
|
|
|
// IPAM may not have a valid store. In such cases it is just in-memory state.
|
|
if store == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
pc := &addrSpace{id: dsConfigKey + "/" + as, ds: store, alloc: a}
|
|
if err := store.GetObject(datastore.Key(pc.Key()...), pc); err != nil {
|
|
if err == datastore.ErrKeyNotFound {
|
|
return nil, nil
|
|
}
|
|
|
|
return nil, types.InternalErrorf("could not get pools config from store: %v", err)
|
|
}
|
|
|
|
return pc, nil
|
|
}
|
|
|
|
func (a *Allocator) writeToStore(aSpace *addrSpace) error {
|
|
store := aSpace.store()
|
|
|
|
// IPAM may not have a valid store. In such cases it is just in-memory state.
|
|
if store == nil {
|
|
return nil
|
|
}
|
|
|
|
err := store.PutObjectAtomic(aSpace)
|
|
if err == datastore.ErrKeyModified {
|
|
return types.RetryErrorf("failed to perform atomic write (%v). retry might fix the error", err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// DataScope method returns the storage scope of the datastore
|
|
func (aSpace *addrSpace) DataScope() string {
|
|
aSpace.Lock()
|
|
defer aSpace.Unlock()
|
|
|
|
return aSpace.scope
|
|
}
|