mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
a52bcf48f2
Previously the bitseq alloc was allocating the first available bit from the begining of the sequence. With this commit the bitseq alloc will proceed from the current allocation. This change will affect the way ipam and vni allocation is done currently. The ip allocation will be done sequentially from the previous allocation as opposed to the first available IP. Signed-off-by: Abhinandan Prativadi <abhi@docker.com>
142 lines
2.9 KiB
Go
142 lines
2.9 KiB
Go
package bitseq
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/docker/libnetwork/datastore"
|
|
"github.com/docker/libnetwork/types"
|
|
)
|
|
|
|
// Key provides the Key to be used in KV Store
|
|
func (h *Handle) Key() []string {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
return []string{h.app, h.id}
|
|
}
|
|
|
|
// KeyPrefix returns the immediate parent key that can be used for tree walk
|
|
func (h *Handle) KeyPrefix() []string {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
return []string{h.app}
|
|
}
|
|
|
|
// Value marshals the data to be stored in the KV store
|
|
func (h *Handle) Value() []byte {
|
|
b, err := json.Marshal(h)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
return b
|
|
}
|
|
|
|
// SetValue unmarshals the data from the KV store
|
|
func (h *Handle) SetValue(value []byte) error {
|
|
return json.Unmarshal(value, h)
|
|
}
|
|
|
|
// Index returns the latest DB Index as seen by this object
|
|
func (h *Handle) Index() uint64 {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
return h.dbIndex
|
|
}
|
|
|
|
// SetIndex method allows the datastore to store the latest DB Index into this object
|
|
func (h *Handle) SetIndex(index uint64) {
|
|
h.Lock()
|
|
h.dbIndex = index
|
|
h.dbExists = true
|
|
h.Unlock()
|
|
}
|
|
|
|
// Exists method is true if this object has been stored in the DB.
|
|
func (h *Handle) Exists() bool {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
return h.dbExists
|
|
}
|
|
|
|
// New method returns a handle based on the receiver handle
|
|
func (h *Handle) New() datastore.KVObject {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
return &Handle{
|
|
app: h.app,
|
|
store: h.store,
|
|
}
|
|
}
|
|
|
|
// CopyTo deep copies the handle into the passed destination object
|
|
func (h *Handle) CopyTo(o datastore.KVObject) error {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
dstH := o.(*Handle)
|
|
if h == dstH {
|
|
return nil
|
|
}
|
|
dstH.Lock()
|
|
dstH.bits = h.bits
|
|
dstH.unselected = h.unselected
|
|
dstH.head = h.head.getCopy()
|
|
dstH.app = h.app
|
|
dstH.id = h.id
|
|
dstH.dbIndex = h.dbIndex
|
|
dstH.dbExists = h.dbExists
|
|
dstH.store = h.store
|
|
dstH.curr = h.curr
|
|
dstH.Unlock()
|
|
|
|
return nil
|
|
}
|
|
|
|
// Skip provides a way for a KV Object to avoid persisting it in the KV Store
|
|
func (h *Handle) Skip() bool {
|
|
return false
|
|
}
|
|
|
|
// DataScope method returns the storage scope of the datastore
|
|
func (h *Handle) DataScope() string {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
return h.store.Scope()
|
|
}
|
|
|
|
func (h *Handle) fromDsValue(value []byte) error {
|
|
var ba []byte
|
|
if err := json.Unmarshal(value, &ba); err != nil {
|
|
return fmt.Errorf("failed to decode json: %s", err.Error())
|
|
}
|
|
if err := h.FromByteArray(ba); err != nil {
|
|
return fmt.Errorf("failed to decode handle: %s", err.Error())
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (h *Handle) writeToStore() error {
|
|
h.Lock()
|
|
store := h.store
|
|
h.Unlock()
|
|
if store == nil {
|
|
return nil
|
|
}
|
|
err := store.PutObjectAtomic(h)
|
|
if err == datastore.ErrKeyModified {
|
|
return types.RetryErrorf("failed to perform atomic write (%v). Retry might fix the error", err)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (h *Handle) deleteFromStore() error {
|
|
h.Lock()
|
|
store := h.store
|
|
h.Unlock()
|
|
if store == nil {
|
|
return nil
|
|
}
|
|
return store.DeleteObjectAtomic(h)
|
|
}
|