azure-container-networking/ipam/manager.go

336 строки
6.8 KiB
Go
Исходник Обычный вид История

2016-09-16 06:27:36 +03:00
// Copyright Microsoft Corp.
// All rights reserved.
package ipam
import (
"sync"
"github.com/Azure/azure-container-networking/common"
"github.com/Azure/azure-container-networking/log"
"github.com/Azure/azure-container-networking/network"
"github.com/Azure/azure-container-networking/store"
2016-09-16 06:27:36 +03:00
)
const (
// IPAM store key.
storeKey = "IPAM"
)
// AddressManager manages the set of address spaces and pools allocated to containers.
type addressManager struct {
AddrSpaces map[string]*addressSpace `json:"AddressSpaces"`
store store.KeyValueStore
source addressConfigSource
netApi network.NetworkManager
2016-09-16 06:27:36 +03:00
sync.Mutex
}
// AddressManager API.
type AddressManager interface {
Initialize(config *common.PluginConfig, options map[string]interface{}) error
Uninitialize()
GetDefaultAddressSpaces() (string, string)
RequestPool(asId, poolId, subPoolId string, options map[string]string, v6 bool) (string, string, error)
ReleasePool(asId, poolId string) error
GetPoolInfo(asId, poolId string) (*AddressPoolInfo, error)
RequestAddress(asId, poolId, address string, options map[string]string) (string, error)
ReleaseAddress(asId, poolId, address string) error
}
2016-09-16 06:27:36 +03:00
// AddressConfigSource configures the address pools managed by AddressManager.
type addressConfigSource interface {
start(sink addressConfigSink) error
stop()
refresh() error
}
// AddressConfigSink interface is used by AddressConfigSources to configure address pools.
type addressConfigSink interface {
newAddressSpace(id string, scope string) (*addressSpace, error)
setAddressSpace(*addressSpace) error
}
// Creates a new address manager.
func NewAddressManager() (AddressManager, error) {
2016-09-16 06:27:36 +03:00
am := &addressManager{
AddrSpaces: make(map[string]*addressSpace),
}
return am, nil
}
// Initialize configures address manager.
func (am *addressManager) Initialize(config *common.PluginConfig, options map[string]interface{}) error {
2016-09-16 06:27:36 +03:00
am.store = config.Store
am.netApi, _ = config.NetApi.(network.NetworkManager)
2016-09-16 06:27:36 +03:00
// Restore persisted state.
err := am.restore()
if err != nil {
return err
}
// Start source.
err = am.startSource(options)
2016-09-16 06:27:36 +03:00
return err
}
// Uninitialize cleans up address manager.
func (am *addressManager) Uninitialize() {
am.stopSource()
}
// Restore reads address manager state from persistent store.
func (am *addressManager) restore() error {
// Skip if a store is not provided.
if am.store == nil {
return nil
}
2016-09-16 06:27:36 +03:00
// Read any persisted state.
err := am.store.Read(storeKey, am)
if err != nil {
if err == store.ErrKeyNotFound {
return nil
} else {
log.Printf("[ipam] Failed to restore state, err:%v\n", err)
return err
}
}
// Populate pointers.
for _, as := range am.AddrSpaces {
for _, ap := range as.Pools {
ap.as = as
}
}
log.Printf("[ipam] Restored state, %+v\n", am)
return nil
}
// Save writes address manager state to persistent store.
func (am *addressManager) save() error {
// Skip if a store is not provided.
if am.store == nil {
return nil
}
2016-09-16 06:27:36 +03:00
err := am.store.Write(storeKey, am)
if err == nil {
log.Printf("[ipam] Save succeeded.\n")
} else {
log.Printf("[ipam] Save failed, err:%v\n", err)
}
return err
}
// Starts configuration source.
func (am *addressManager) startSource(options map[string]interface{}) error {
2016-09-16 06:27:36 +03:00
var err error
environment, _ := options[common.OptEnvironment].(string)
switch environment {
case common.OptEnvironmentAzure:
am.source, err = newAzureSource(options)
2016-09-16 06:27:36 +03:00
case common.OptEnvironmentMAS:
am.source, err = newMasSource(options)
2016-09-16 06:27:36 +03:00
case "null":
am.source, err = newNullSource()
case "":
am.source = nil
2016-09-16 06:27:36 +03:00
default:
return errInvalidConfiguration
}
if am.source != nil {
err = am.source.start(am)
}
return err
}
// Stops the configuration source.
func (am *addressManager) stopSource() {
if am.source != nil {
am.source.stop()
am.source = nil
}
}
// Signals configuration source to refresh.
func (am *addressManager) refreshSource() {
if am.source != nil {
err := am.source.refresh()
if err != nil {
log.Printf("[ipam] Source refresh failed, err:%v.\n", err)
}
}
}
2016-09-22 01:37:35 +03:00
//
// AddressManager API
//
// Provides atomic stateful wrappers around core IPAM functionality.
//
2016-09-16 06:27:36 +03:00
// GetDefaultAddressSpaces returns the default local and global address space IDs.
func (am *addressManager) GetDefaultAddressSpaces() (string, string) {
var localId, globalId string
am.Lock()
defer am.Unlock()
am.refreshSource()
local := am.AddrSpaces[localDefaultAddressSpaceId]
if local != nil {
localId = local.Id
}
global := am.AddrSpaces[globalDefaultAddressSpaceId]
if global != nil {
globalId = global.Id
}
return localId, globalId
}
// RequestPool reserves an address pool.
func (am *addressManager) RequestPool(asId, poolId, subPoolId string, options map[string]string, v6 bool) (string, string, error) {
am.Lock()
defer am.Unlock()
am.refreshSource()
as, err := am.getAddressSpace(asId)
if err != nil {
return "", "", err
}
pool, err := as.requestPool(poolId, subPoolId, options, v6)
if err != nil {
return "", "", err
}
err = am.save()
if err != nil {
return "", "", err
}
return pool.Id, pool.Subnet.String(), nil
}
// ReleasePool releases a previously reserved address pool.
func (am *addressManager) ReleasePool(asId string, poolId string) error {
am.Lock()
defer am.Unlock()
am.refreshSource()
as, err := am.getAddressSpace(asId)
if err != nil {
return err
}
err = as.releasePool(poolId)
if err != nil {
return err
}
err = am.save()
if err != nil {
return err
}
return nil
}
// GetPoolInfo returns information about the given address pool.
func (am *addressManager) GetPoolInfo(asId string, poolId string) (*AddressPoolInfo, error) {
am.Lock()
defer am.Unlock()
as, err := am.getAddressSpace(asId)
if err != nil {
return nil, err
}
ap, err := as.getAddressPool(poolId)
if err != nil {
return nil, err
}
return ap.getInfo(), nil
}
2016-09-16 06:27:36 +03:00
// RequestAddress reserves a new address from the address pool.
func (am *addressManager) RequestAddress(asId, poolId, address string, options map[string]string) (string, error) {
am.Lock()
defer am.Unlock()
am.refreshSource()
as, err := am.getAddressSpace(asId)
if err != nil {
return "", err
}
ap, err := as.getAddressPool(poolId)
if err != nil {
return "", err
}
addr, err := ap.requestAddress(address, options)
if err != nil {
return "", err
}
err = am.save()
if err != nil {
return "", err
}
return addr, nil
}
// ReleaseAddress releases a previously reserved address.
func (am *addressManager) ReleaseAddress(asId string, poolId string, address string) error {
am.Lock()
defer am.Unlock()
am.refreshSource()
as, err := am.getAddressSpace(asId)
if err != nil {
return err
}
ap, err := as.getAddressPool(poolId)
if err != nil {
return err
}
err = ap.releaseAddress(address)
if err != nil {
return err
}
err = am.save()
if err != nil {
return err
}
return nil
}