44 lines
910 B
Go
44 lines
910 B
Go
package communicator
|
|
|
|
import (
|
|
"flag"
|
|
"github.com/gorilla/mux"
|
|
_ "net/http/pprof"
|
|
"sync"
|
|
)
|
|
|
|
const (
|
|
CAPTURE_PACKET_RATE = "capture_packet_rate"
|
|
QPS = "qps"
|
|
)
|
|
|
|
var (
|
|
communicatePort int
|
|
router = mux.NewRouter()
|
|
)
|
|
|
|
var (
|
|
configMapLock sync.RWMutex
|
|
configMap map[string]configItem
|
|
catpurePacketRate *capturePacketRateConfig
|
|
catpurePacketRateVal float64
|
|
)
|
|
|
|
func init() {
|
|
catpurePacketRate = newCapturePacketRateConfig()
|
|
|
|
flag.IntVar(&communicatePort, "communicate_port", 8088, "http server port. Default is 8088")
|
|
flag.Float64Var(&catpurePacketRateVal, CAPTURE_PACKET_RATE, 1.0, "capture packet rate. Default is 1.0")
|
|
|
|
if err := catpurePacketRate.setVal(catpurePacketRateVal); err != nil {
|
|
panic(err.Error())
|
|
}
|
|
configMap = make(map[string]configItem)
|
|
regsiterConfig()
|
|
}
|
|
|
|
func regsiterConfig() {
|
|
configMap[CAPTURE_PACKET_RATE] = catpurePacketRate
|
|
configMap[QPS] = &qpsConfig{}
|
|
}
|