testbench/tcp/tc/mgr.go

package tc

import (
    "sync"
    "wb/lg"
    "wb/st"
)

type ITConn interface {
    GetTermId() string
    GetType()     string
    SendCmd(cmd string)
    InitDtu(startId ...string)bool
    LogInfo(v ...interface{})
    LogError(v... interface{})
    Close()
}

var lockMapConn = sync.Mutex{}
var mapConn = make(map[string]ITConn)

func AddMConn(mConn ITConn) {
    lockMapConn.Lock()
    defer lockMapConn.Unlock()
    termId := mConn.GetTermId()
    oldConn, ok := mapConn[termId];
    if ok {
        if oldConn != mConn{
            //delete(mapConn, termId)
            oldConn.Close()
            mConn.LogInfo("TC AddMConn new: ", mConn.GetType(), "Id: ", termId)
            lg.Info("TC AddMConn  new", termId)
        }
        lg.Info("TC AddMConn delete old = new", oldConn==mConn)
        oldConn.LogInfo("TC AddMConn delete old", termId, " is exist, close it use new connect")
    }
    mapConn[termId] = mConn
    lg.Info("TC AddMConn:", mConn.GetType(), "Id: ", termId)
    mConn.LogInfo("TC AddMConn:", termId)
}
func DeleteConn(mConn ITConn) {
    lockMapConn.Lock()
    defer lockMapConn.Unlock()
    termId := mConn.GetTermId()
    oldConn, ok := mapConn[termId];
    if ok {
        if oldConn == mConn{
            //delete(mapConn, termId)
            mConn.LogInfo("TC DeleteConn: ", mConn.GetType(), "Id: ", termId)
            lg.Info("TC DeleteConn ", termId)
        }
    }
}
func SendCmd(sn string, cmd string) {
    lockMapConn.Lock()
    defer lockMapConn.Unlock()
    if bConn, ok := mapConn[sn]; ok {
        lg.Info("TC SEND CMD[", cmd, "] TO [", sn, "]")
        bConn.SendCmd(cmd)
    } else {
        lg.Error("SendCmd sn not online:", sn)
    }
}
func InitDtu(sn string, startId string)string {
    lockMapConn.Lock()
    defer lockMapConn.Unlock()
    if bConn, ok := mapConn[sn]; ok {
        lg.Info("TC InitDtu startId [", startId, "] TO [", sn, "]")
        bConn.InitDtu(startId)
    } else {
        lg.Error("SendCmd no such sn:", sn)
    }
    return st.Success
}