fix 标识初始化

Author: liuguangw

common/billing_packet.go

@@ -119,3 +119,12 @@ func (packet *BillingPacket) String() string {
 		"MsgID: [%x, %x]\n"+
 		"Data: \n%s", packet.OpType, packet.MsgID[0], packet.MsgID[1], hex.Dump(packet.OpData))
 }
+
+// InitBillingPacketHead 初始化头部标识
+func InitBillingPacketHead(billType byte) {
+	if billType == BillTypeCommon {
+		BillingPacketHead = [2]byte{0xAA, 0x55}
+	} else if billType == BillTypeHuaiJiu {
+		BillingPacketHead = [2]byte{0x55, 0xAA}
+	}
+}

services/billing/run.go

@@ -6,8 +6,6 @@ import (
 	"os"
 	"os/signal"
 	"syscall"
-
-	"github.com/liuguangw/billing_go/common"
 )
 
 // Run 运行billing
@@ -29,12 +27,6 @@ func (s *Server) Run(logFilePath string) error {
 	if err := s.initDatabase(); err != nil {
 		s.logger.Fatal("init database connection failed: " + err.Error())
 	}
-	//初始化头部标识
-	if s.config.BillType == common.BillTypeCommon {
-		common.BillingPacketHead = [2]byte{0xAA, 0x55}
-	} else if s.config.BillType == common.BillTypeHuaiJiu {
-		common.BillingPacketHead = [2]byte{0x55, 0xAA}
-	}
 	//标记为已启动
 	s.running = true
 	s.logger.Info("billing server run at " + s.listener.Addr().String())

services/billing/server.go

@@ -2,11 +2,12 @@ package billing
 
 import (
 	"database/sql"
+	"net"
+	"os"
+
 	"github.com/liuguangw/billing_go/common"
 	"github.com/liuguangw/billing_go/services"
 	"go.uber.org/zap"
-	"net"
-	"os"
 )
 
 // Server billing 服务
@@ -27,6 +28,8 @@ func NewServer() (*Server, error) {
 	if err != nil {
 		return nil, err
 	}
+	//初始化头部标识
+	common.InitBillingPacketHead(byte(serverConfig.BillType))
 	return &Server{config: serverConfig}, nil
 }