meter.go

package goovn

import (
	"math"
	"strings"

	"github.com/ebay/libovsdb"
)

type Meter struct {
	UUID        string
	Name        string                      `json:"name"`
	Unit        string                      `json:"unit"`
	Bands       []string                    `json:"bands"`
	ExternalIds map[interface{}]interface{} `json:"external_ids"`
}

type MeterBand struct {
	UUID        string
	Action      string                      `json:"action"`
	Rate        int                         `json:"rate"`
	BurstSize   int                         `json:"burst_size"`
	ExternalIds map[interface{}]interface{} `json:"external_ids"`
}

func (odbi *ovndb) rowToMeter(uuid string) *Meter {
	cacheMeter, ok := odbi.cache[TableMeter][uuid]
	if !ok {
		return nil
	}
	meter := &Meter{
		UUID:        uuid,
		Name:        cacheMeter.Fields["name"].(string),
		Unit:        cacheMeter.Fields["unit"].(string),
		Bands:       []string{cacheMeter.Fields["bands"].(libovsdb.UUID).GoUUID},
		ExternalIds: cacheMeter.Fields["external_ids"].(libovsdb.OvsMap).GoMap,
	}
	return meter
}

func (odbi *ovndb) rowToMeterBand(uuid string) (*MeterBand, error) {
	cacheMeterBand, ok := odbi.cache[TableMeterBand][uuid]
	if !ok {
		return nil, ErrorNotFound
	}
	meterBand := &MeterBand{
		UUID:        uuid,
		Action:      cacheMeterBand.Fields["action"].(string),
		Rate:        cacheMeterBand.Fields["rate"].(int),
		BurstSize:   cacheMeterBand.Fields["burst_size"].(int),
		ExternalIds: cacheMeterBand.Fields["external_ids"].(libovsdb.OvsMap).GoMap,
	}
	return meterBand, nil
}

func (odbi *ovndb) meterAddImp(name, action string, rate int, unit string, external_ids map[string]string, burst int) (*OvnCommand, error) {

	//Names  that  start  with "__" (two underscores) are reserved for
	//internal use by OVN.
	if strings.HasPrefix(name, "__") {
		return nil, ErrorOption
	}

	// The only supported action is drop.
	if action != "drop" {
		return nil, ErrorOption
	}

	MeterUUID, err := newRowUUID()
	if err != nil {
		return nil, err
	}
	MeterBandUUID, err := newRowUUID()
	if err != nil {
		return nil, err
	}
	//meter row
	mRow := make(OVNRow)

	mRow["name"] = name
	if uuid := odbi.getRowUUID(TableMeter, mRow); len(uuid) > 0 {
		return nil, ErrorExist
	}

	mRow["bands"] = libovsdb.UUID{GoUUID: MeterBandUUID}

	switch unit {
	case "kbps", "pktps":
		mRow["unit"] = unit
	default:
		return nil, ErrorOption
	}

	//Meter Band row
	mbRow := make(OVNRow)

	mbRow["action"] = action

	//rate must be in the range 1...4294967295
	if rate < 1 || rate > math.MaxInt32 {
		return nil, ErrorOption
	}
	mbRow["rate"] = rate

	//burst must be in the range 0...4294967295
	if burst >= 0 && burst <= math.MaxInt32 {
		mbRow["burst_size"] = burst
	}

	if external_ids != nil {
		oMap, err := libovsdb.NewOvsMap(external_ids)
		if err != nil {
			return nil, err
		}
		mRow["external_ids"] = oMap
		//mbRow["external_ids"] = oMap
	}

	mbInsterOp := libovsdb.Operation{
		Op:       opInsert,
		Table:    TableMeterBand,
		Row:      mbRow,
		UUIDName: MeterBandUUID,
	}

	mInsertOp := libovsdb.Operation{
		Op:       opInsert,
		Table:    TableMeter,
		Row:      mRow,
		UUIDName: MeterUUID,
	}
	operations := []libovsdb.Operation{mbInsterOp, mInsertOp}
	return &OvnCommand{operations, odbi, make([][]map[string]interface{}, len(operations))}, nil
}

/*
meter-del [name]
Deletes meters. By default, all meters are deleted. If  name  is
supplied, only the meter with that name will be deleted.
*/
func (odbi *ovndb) meterDelImp(name ...string) (*OvnCommand, error) {
	var operations []libovsdb.Operation
	var err error

	switch len(name) {
	case 0:
		for uuid := range odbi.cache[TableMeter] {
			name := odbi.cache[TableMeter][uuid].Fields["name"].(string)
			operations, err = odbi.singleMeterDel(name, operations)
			if err != nil {
				return nil, err
			}
		}
	default:
		for i := 0; i < len(name); i++ {
			operations, err = odbi.singleMeterDel(name[i], operations)
			if err != nil {
				return nil, err
			}
		}
	}
	return &OvnCommand{operations, odbi, make([][]map[string]interface{}, len(operations))}, nil

}

//meter-list
//Lists all meters.
//but not like ovn-nbctl , it can't show meter bands information
func (odbi *ovndb) meterListImp() ([]*Meter, error) {
	odbi.cachemutex.RLock()
	defer odbi.cachemutex.RUnlock()
	cacheMeter, ok := odbi.cache[TableMeter]
	if !ok {
		return nil, ErrorNotFound
	}
	ListMeter := make([]*Meter, 0, len(cacheMeter))
	for uuid := range cacheMeter {
		ListMeter = append(ListMeter, odbi.rowToMeter(uuid))
	}
	return ListMeter, nil
}

//Because meterList can't show meter bands , add this method as a solution
func (odbi *ovndb) meterBandsListImp() ([]*MeterBand, error) {
	odbi.cachemutex.RLock()
	defer odbi.cachemutex.RUnlock()
	cacheMeterBands, ok := odbi.cache[TableMeterBand]
	if !ok {
		return nil, ErrorNotFound
	}
	ListMeterBands := make([]*MeterBand, 0, len(cacheMeterBands))
	for uuid := range cacheMeterBands {
		meterBand, err := odbi.rowToMeterBand(uuid)
		if err != nil {
			return nil, ErrorNotFound
		}
		ListMeterBands = append(ListMeterBands, meterBand)
	}
	return ListMeterBands, nil
}

func (odbi *ovndb) meterFind(name string) bool {
	odbi.cachemutex.RLock()
	defer odbi.cachemutex.RUnlock()
	row := make(OVNRow)
	row["name"] = name
	meterUUID := odbi.getRowUUID(TableMeter, row)
	if len(meterUUID) == 0 {
		return false
	}
	return true
}

func (odbi *ovndb) singleMeterDel(name string, operations []libovsdb.Operation) ([]libovsdb.Operation, error) {
	meterName := name
	row := make(OVNRow)
	row["name"] = meterName
	meterUUID := odbi.getRowUUID(TableMeter, row)
	if len(meterUUID) == 0 {
		return nil, ErrorNotFound
	}
	bands := odbi.cache[TableMeter][meterUUID].Fields["bands"].(libovsdb.UUID)
	mCondition := libovsdb.NewCondition("name", "==", meterName)
	mDeleteOp := libovsdb.Operation{
		Op:    opDelete,
		Table: TableMeter,
		Where: []interface{}{mCondition},
	}

	bCondition := libovsdb.NewCondition("_uuid", "==", bands)
	bDeleteOp := libovsdb.Operation{
		Op:    opDelete,
		Table: TableMeterBand,
		Where: []interface{}{bCondition},
	}
	operations = append(operations, bDeleteOp)
	operations = append(operations, mDeleteOp)
	return operations, nil
}