【feat】完成http数据上报

iothub/server/tcpserver/hook.go

@@ -0,0 +1,109 @@
+package tcpserver
+
+import (
+	"context"
+	"encoding/hex"
+	"github.com/emicklei/go-restful/v3"
+	"log"
+	"net"
+	"pandax/iothub/hook_message_work"
+	"pandax/iothub/netbase"
+	"pandax/pkg/global"
+	"pandax/pkg/rule_engine/message"
+	"strings"
+	"time"
+)
+
+type HookTcpService struct {
+	HookService *hook_message_work.HookService
+	keepAlive   int64
+	conn        *net.TCPConn
+}
+
+func InitTcpHook(addr string, hs *hook_message_work.HookService) {
+	hhs := &HookTcpService{
+		HookService: hs,
+		keepAlive:   20,
+	}
+	server := NewTcpServer(addr)
+	err := server.Start(context.TODO())
+	if err != nil {
+		global.Log.Error("IOTHUB HTTP服务启动错误", err)
+		return
+	} else {
+		global.Log.Infof("TCP IOTHUB HOOK Start SUCCESS, Server listen: %s", addr)
+	}
+	go func() {
+		for {
+			conn, err := server.listener.AcceptTCP()
+			if err != nil {
+				global.Log.Error("Error accepting connection:", err)
+				continue
+			}
+			conn.SetReadDeadline(time.Now().Add(20 * time.Second))
+			hhs.conn = conn
+			go hhs.hook()
+
+		}
+	}()
+}
+
+// 获取token进行认证
+func basicAuthenticate(req *restful.Request, resp *restful.Response, chain *restful.FilterChain) {
+	path := req.Request.URL.Path
+	log.Println(path)
+	split := strings.Split(path, "/")
+	log.Println(split)
+	chain.ProcessFilter(req, resp)
+}
+
+func (hhs *HookTcpService) hook() {
+	isAuth := false
+	for {
+		buf := make([]byte, 128)
+		n := 0
+		n, err := hhs.conn.Read(buf)
+		if err != nil {
+			// 断开连接 掉线
+			log.Println("断开连接")
+			_ = hhs.conn.Close()
+			isAuth = false
+			return
+		}
+		if !isAuth {
+			token := string(buf[:n])
+			log.Println(token)
+			isAuth = true
+		} else {
+			hexData := hex.EncodeToString(buf[:n])
+			log.Println(hexData)
+		}
+	}
+
+}
+
+func (hhs *HookTcpService) Send(message string) error {
+	return hhs.SendBytes([]byte(message))
+}
+
+func (hhs *HookTcpService) SendHex(msg string) error {
+	b, err := hex.DecodeString(msg)
+	if err != nil {
+		return err
+	}
+	return hhs.SendBytes(b)
+}
+
+func (hhs *HookTcpService) SendBytes(msg []byte) error {
+	_, err := hhs.conn.Write(msg)
+	if err != nil {
+		hhs.conn.Close()
+		data := &netbase.DeviceEventInfo{
+			DeviceId: "",
+			Datas:    "",
+			Type:     message.ConnectMes,
+		}
+		hhs.HookService.MessageCh <- data
+	}
+	return err
+}

iothub/server/tcpserver/tcp_server.go

@@ -0,0 +1,64 @@
+package tcpserver
+
+import (
+	"context"
+	"crypto/tls"
+	"fmt"
+	"net"
+	"pandax/pkg/global"
+)
+
+const DefaultPort = ":9003"
+
+type TcpServer struct {
+	Addr     string
+	listener *net.TCPListener
+}
+
+func NewTcpServer(addr string) *TcpServer {
+	if addr == "" {
+		addr = DefaultPort
+	}
+	return &TcpServer{
+		Addr: addr,
+	}
+}
+
+func (s *TcpServer) GetServe() *net.TCPListener {
+	return s.listener
+}
+
+func (s *TcpServer) Type() string {
+	return "TCP"
+}
+
+func (s *TcpServer) Start(ctx context.Context) error {
+	addr, _ := net.ResolveTCPAddr("tcp", s.Addr)
+	listener, err := net.ListenTCP("tcp", addr)
+	if err != nil {
+		global.Log.Errorf("error http serve: %s", err)
+		return err
+	}
+	s.listener = listener
+	return nil
+}
+
+func (s *TcpServer) Stop(ctx context.Context) error {
+	s.listener.Close()
+	return nil
+}
+
+func (s *TcpServer) TlsConfig() (*tls.Config, error) {
+	var certificates []tls.Certificate
+	cert, err := tls.LoadX509KeyPair(global.Conf.Server.Tls.CertFile, global.Conf.Server.Tls.KeyFile)
+	if err != nil {
+		return nil, fmt.Errorf("generate x509 key pair failed: %s ", err)
+	}
+	certificates = append(certificates, cert)
+
+	if len(certificates) == 0 {
+		return nil, fmt.Errorf("none valid certs and secret")
+	}
+
+	return &tls.Config{Certificates: certificates}, nil
+}