change throw_package_rate to capture_package_rate

communicator/config.go

@@ -8,23 +8,28 @@ import (
 )
 
 const (
-	THROW_PACKET_RATE = "throw_packet_rate"
+	CAPTURE_PACKET_RATE = "capture_packet_rate"
 )
 
 var (
 	communicatePort int
+	// capturePacketRate float64
 	router = mux.NewRouter()
 )
 
 var (
-	configMapLock   sync.RWMutex
-	configMap       map[string]configItem
-	throwPacketRate *throwPacketRateConfig
+	configMapLock     sync.RWMutex
+	configMap         map[string]configItem
+	catpurePacketRate *capturePacketRateConfig
 )
 
 func init() {
+	catpurePacketRate = newCapturePacketRateConfig()
+
 	flag.IntVar(&communicatePort, "communicate_port", 8088, "http server port. Default is 8088")
+	var cpr float64
+	flag.Float64Var(&cpr, CAPTURE_PACKET_RATE, 1, "capture packet rate. Default is 1.0")
+	_ = catpurePacketRate.setVal(cpr)
 
 	configMap = make(map[string]configItem)
-	throwPacketRate = newThrowPacketRateConfig()
 }

communicator/controller.go

@@ -82,10 +82,10 @@ func outletSetConfig(resp http.ResponseWriter, req *http.Request) {
 	mp.Err = SetConfig(ep.ConfigName, ep.Value)
 }
 
-func GetTCPThrowPacketRate() float64 {
-	return throwPacketRate.tcpTPR
+func GetTCPCapturePacketRate() float64 {
+	return catpurePacketRate.tcpTPR
 }
 
-func GetMysqlThrowPacketRate() float64 {
-	return throwPacketRate.mysqlTPR
+func GetMysqlCapturePacketRate() float64 {
+	return catpurePacketRate.mysqlTPR
 }

communicator/major_function.go

@@ -4,8 +4,8 @@ import "fmt"
 
 // SetConfig set config by config key(name) and value
 func SetConfig(key string, val interface{}) (err error) {
-	if key == throwPacketRate.name {
-		err = throwPacketRate.setVal(val)
+	if key == catpurePacketRate.name {
+		err = catpurePacketRate.setVal(val)
 		return
 	}
 

communicator/model.go

@@ -10,33 +10,33 @@ type configItem interface {
 	getVal () interface{}
 }
 
-type throwPacketRateConfig struct {
+type capturePacketRateConfig struct {
 	name string
 	tcpTPR float64
 	mysqlTPR float64
 }
 
-func newThrowPacketRateConfig() (tpr *throwPacketRateConfig) {
-	tpr = &throwPacketRateConfig{
-		name: THROW_PACKET_RATE,
-		tcpTPR: 0.0,
-		mysqlTPR: 0.0,
+func newCapturePacketRateConfig() (cprc *capturePacketRateConfig) {
+	cprc = &capturePacketRateConfig{
+		name:     CAPTURE_PACKET_RATE,
+		tcpTPR:   1.0,
+		mysqlTPR: 1.0,
 	}
 	return
 }
 
-func (tc *throwPacketRateConfig) setVal (val interface{}) (err error){
+func (cprc *capturePacketRateConfig) setVal (val interface{}) (err error){
 	realVal, ok := val.(float64)
 	if !ok {
 		err = fmt.Errorf("cannot reansform val: %v to float64", val)
 		return
 	}
 
-	tc.mysqlTPR = realVal
-	tc.tcpTPR = math.Sqrt(realVal)
+	cprc.mysqlTPR = realVal
+	cprc.tcpTPR = math.Sqrt(realVal)
 	return
 }
 
-func (tc *throwPacketRateConfig) getVal () (val interface{}){
-	return tc.mysqlTPR
+func (cprc *capturePacketRateConfig) getVal () (val interface{}){
+	return cprc.mysqlTPR
 }