124 lines
2.6 KiB
Go
124 lines
2.6 KiB
Go
package event
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
"git.rosy.net.cn/jx-callback/business/model/dao"
|
|
"io"
|
|
"net"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// ConnRead 获取链接数据
|
|
func ConnRead(c net.Conn) ([]byte, int, error) {
|
|
buffer := make([]byte, 1024*2)
|
|
n, err := c.Read(buffer)
|
|
return buffer, n, err
|
|
}
|
|
|
|
// ListenTcp 入口
|
|
func ListenTcp() {
|
|
l, err := net.Listen("tcp", ":8000")
|
|
if err != nil {
|
|
fmt.Println("listen error:", err)
|
|
return
|
|
}
|
|
|
|
for {
|
|
c, err := l.Accept()
|
|
if err != nil || c == nil {
|
|
fmt.Println("accept error:", err)
|
|
break
|
|
}
|
|
|
|
fn := func() {
|
|
// 捕获异常 防止waitGroup阻塞
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
fmt.Println("recover err = ", err)
|
|
return
|
|
}
|
|
}()
|
|
|
|
if err := handleConn(c); err != nil {
|
|
c.Close()
|
|
Pool.Wait()
|
|
Pool.Stop()
|
|
return
|
|
}
|
|
}
|
|
Pool.AddJob(fn)
|
|
}
|
|
}
|
|
|
|
func handleConn(c net.Conn) error {
|
|
if c == nil {
|
|
return errors.New("conn is nil")
|
|
}
|
|
for {
|
|
buffer, n, err := ConnRead(c)
|
|
printRemoteAddr := c.RemoteAddr().String()
|
|
printRemoteAddr = strings.Split(printRemoteAddr, ":")[0]
|
|
if err != nil {
|
|
if err == io.EOF {
|
|
fmt.Println("connection close")
|
|
} else {
|
|
fmt.Println("ReadString err:", err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
//看是心跳还是打印回调
|
|
data := hex.EncodeToString(buffer[:n])
|
|
var (
|
|
printNo string = "" //打印机编号
|
|
heartbeat bool = false
|
|
callback bool = false
|
|
t *TcpClient = nil
|
|
)
|
|
if strings.Contains(data, heartText) || strings.Contains(data, heartTextNew) {
|
|
printNoData, _ := hex.DecodeString(data[len(heartText) : len(data)-8])
|
|
printNo = string(printNoData)
|
|
heartbeat = true
|
|
} else if strings.Contains(data, printText) || strings.Contains(data, printTextNew) { //打印回调
|
|
_, printNo = getCallbackMsgInfo(data)
|
|
callback = true
|
|
}
|
|
|
|
if _, have := TcpClientList[printNo]; !have {
|
|
t = NewTcpClient()
|
|
TcpClientList[printNo] = t
|
|
} else {
|
|
t = TcpClientList[printNo]
|
|
}
|
|
if heartbeat {
|
|
// 证明是心跳
|
|
Heartbeat(c, t, data, printNo, printRemoteAddr)
|
|
} else if callback {
|
|
// 打印回调
|
|
Callback(c, t, data, printNo)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (t *TcpClient) printFail() (err error) {
|
|
//新开机的打印失败和错误的
|
|
var (
|
|
db = dao.GetDB()
|
|
)
|
|
prints, _ := dao.GetPrintMsgs(db, "", []int{printMsgFail, printMsgErr, PrintMsgAlreadyLoad, printMsgAlreadySend}, time.Now().Add(-time.Hour*3), time.Now(), 0, 999)
|
|
for _, printMsg := range prints {
|
|
t.addMsgChan(printMsg)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (t *TcpClient) readTimeoutMap(key string) bool {
|
|
t.Lock()
|
|
defer t.Unlock()
|
|
return t.TimeoutMap[key]
|
|
}
|