cubefs/remotecache/flashgroupmanager/config.go

package flashgroupmanager

import (
	"fmt"
	syslog "log"
	"strconv"
	"strings"

	"github.com/cubefs/cubefs/depends/tiglabs/raft/proto"
	"github.com/cubefs/cubefs/raftstore"
)

const (
	commaSplit = ","
	colonSplit = ":"
)
const (
	defaultFlashNodeHandleReadTimeout   = 1000
	defaultFlashNodeReadDataNodeTimeout = 3000
	defaultHttpReversePoolSize          = 1024
	defaultRetainLogs                   = 20000
)

var AddrDatabase = make(map[uint64]string)

type clusterConfig struct {
	flashNodeHandleReadTimeout   int
	flashNodeReadDataNodeTimeout int
	httpProxyPoolSize            uint64
	heartbeatPort                int64
	replicaPort                  int64
	peerAddrs                    []string
	peers                        []raftstore.PeerAddress
}

func newClusterConfig() (cfg *clusterConfig) {
	cfg = new(clusterConfig)

	cfg.flashNodeHandleReadTimeout = defaultFlashNodeHandleReadTimeout
	cfg.flashNodeReadDataNodeTimeout = defaultFlashNodeReadDataNodeTimeout
	return
}

func parsePeerAddr(peerAddr string) (id uint64, ip string, port uint64, err error) {
	peerStr := strings.Split(peerAddr, colonSplit)
	id, err = strconv.ParseUint(peerStr[0], 10, 64)
	if err != nil {
		return
	}
	port, err = strconv.ParseUint(peerStr[2], 10, 64)
	if err != nil {
		return
	}
	ip = peerStr[1]
	return
}

func (cfg *clusterConfig) parsePeers(peerStr string) error {
	peerArr := strings.Split(peerStr, commaSplit)
	cfg.peerAddrs = peerArr
	for _, peerAddr := range peerArr {
		id, ip, port, err := parsePeerAddr(peerAddr)
		if err != nil {
			return err
		}
		cfg.peers = append(cfg.peers, raftstore.PeerAddress{Peer: proto.Peer{ID: id}, Address: ip, HeartbeatPort: int(cfg.heartbeatPort), ReplicaPort: int(cfg.replicaPort)})
		address := fmt.Sprintf("%v:%v", ip, port)
		syslog.Println(address)
		AddrDatabase[id] = address
	}
	return nil
}