490 lines
15 KiB
Go
490 lines
15 KiB
Go
package event
|
||
|
||
import (
|
||
"encoding/hex"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"git.rosy.net.cn/baseapi/utils"
|
||
"git.rosy.net.cn/jx-callback/business/model"
|
||
"git.rosy.net.cn/jx-callback/business/model/dao"
|
||
"git.rosy.net.cn/jx-callback/globals"
|
||
"io"
|
||
"net"
|
||
"strings"
|
||
"time"
|
||
"unicode/utf8"
|
||
)
|
||
|
||
var Poll *Pool
|
||
|
||
func init() {
|
||
Poll = NewPool(100)
|
||
Poll.Start()
|
||
|
||
}
|
||
|
||
//入口
|
||
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 {
|
||
fmt.Println("accept error:", err)
|
||
break
|
||
}
|
||
|
||
fn := func() {
|
||
defer func() {
|
||
// 捕获异常 防止waitGroup阻塞
|
||
if err := recover(); err != nil {
|
||
fmt.Println("recover err = ", err)
|
||
return
|
||
}
|
||
}()
|
||
|
||
t := NewTcpClient()
|
||
if err := handleConn(c, t); err != nil {
|
||
t = nil
|
||
return
|
||
}
|
||
}
|
||
Poll.AddJob(fn)
|
||
//go func() {
|
||
// defer func() {
|
||
// // 捕获异常 防止waitGroup阻塞
|
||
// if err := recover(); err != nil {
|
||
// fmt.Println("recover err = ", err)
|
||
// return
|
||
// }
|
||
// }()
|
||
//
|
||
// t := NewTcpClient()
|
||
// if err := handleConn(c, t); err != nil {
|
||
// t = nil
|
||
// return
|
||
// }
|
||
//}()
|
||
}
|
||
}
|
||
|
||
func handleConn(c net.Conn, t *TcpClient) error {
|
||
if c == nil {
|
||
globals.SugarLogger.Debugf("conn is nil")
|
||
return errors.New("conn is nil")
|
||
}
|
||
defer c.Close()
|
||
for {
|
||
var (
|
||
printNo string //打印机编号
|
||
buffer = make([]byte, 1024)
|
||
)
|
||
n, err := c.Read(buffer)
|
||
if err != nil {
|
||
if err == io.EOF {
|
||
fmt.Println("connection close")
|
||
} else {
|
||
fmt.Println("ReadString err", err)
|
||
}
|
||
return err
|
||
}
|
||
|
||
//看是心跳还是打印回调
|
||
data := hex.EncodeToString(buffer[:n])
|
||
if strings.Contains(string(buffer[0:n]), "print_no_clear") { // 清理缓存
|
||
param := struct {
|
||
PrintNoClear json.Number `json:"print_no_clear"`
|
||
}{}
|
||
if err := json.Unmarshal(buffer[0:n], ¶m); err != nil {
|
||
c.Write([]byte(err.Error()))
|
||
return err
|
||
}
|
||
t.buildMsgMap(param.PrintNoClear.String())
|
||
c.Write([]byte("ok"))
|
||
return err
|
||
}
|
||
//证明是心跳
|
||
if strings.Contains(data, heartText) || strings.Contains(data, heartTextNew) {
|
||
printNoData, _ := hex.DecodeString(data[len(heartText) : len(data)-8])
|
||
printNo = string(printNoData)
|
||
status := printStatus2JxStatus(data[len(data)-8 : len(data)-6])
|
||
//如果没在连接池里
|
||
//1、加到连接池中，不同的打印机no开不同的goroutine
|
||
//2、初始化channel，每个打印机一个，放打印消息和打印回调消息
|
||
//3、读数据库里的待打印信息，放到打印channel中
|
||
//4、读打印channel并打印，并切等待回调channel中的消息
|
||
//5、修改数据库中打印机状态（没在连接池中说明是重新连接的）
|
||
//6、监听心跳时间，超过1分多钟就clear掉
|
||
if t.getClients(printNo) == nil {
|
||
t.addConn(c, printNo, status)
|
||
t.buildAllMap(printNo)
|
||
//t.TimeoutMap[printNo] <- true
|
||
t.HandleTcpMessages(printNo)
|
||
t.doPrint(printNo)
|
||
if status == printerStatusOnline {
|
||
//t.printFail()
|
||
}
|
||
changePrinterStatus(printNo, status)
|
||
t.HandleCheckTcpHeart(printNo)
|
||
// todo 证明打印机已经被激活,将激活打印机存入数据库,保证用户不能无限制绑定打印机
|
||
if err := dao.NotExistsCreate(printNo); err != nil {
|
||
globals.SugarLogger.Debugf("监听打印机心跳,不存在则创建 :[%v]", err)
|
||
}
|
||
} else {
|
||
//在加到连接池中已经更新了时间，所以放在else里
|
||
t.setPrintStatusTime(printNo)
|
||
}
|
||
//状态不一致再更新状态（可能缺纸了，过热了等）
|
||
if t.getPrintStatus(printNo) != status {
|
||
t.setPrintStatus(printNo, status)
|
||
changePrinterStatus(printNo, status)
|
||
}
|
||
} else if strings.Contains(data, printText) || strings.Contains(data, printTextNew) { //打印回调
|
||
//打印消息发送后，打印机会回调该条打印消息的状态（打印成功or失败，失败原因..）
|
||
//将回调的信息放到回调channel中，打印成功后再打印下一条消息
|
||
_, printNo = getCallbackMsgInfo(data)
|
||
//更新打印机心跳时间（打印机本身不会在打印的同时，或回调的同时发心跳消息，会导致心跳判断超时，这里更新一下）
|
||
t.setPrintStatusTime(printNo)
|
||
t.addCallbackChan(printNo, data)
|
||
}
|
||
}
|
||
}
|
||
|
||
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) changePrintMsg(data string, orderNo int64, printNo string) (err error) {
|
||
var (
|
||
db = dao.GetDB()
|
||
comment string
|
||
status int
|
||
)
|
||
//1、先找出打印机编号和订单序列号，这两个确定唯一一条消息?
|
||
//通过参数传进来
|
||
//2、打印成功改变打印表的状态
|
||
if strings.Contains(data, printSuccessText) || strings.Contains(data, printSuccessTextNew) {
|
||
status = printMsgSuccess
|
||
comment = "回调成功,修改打印状态"
|
||
} else {
|
||
//打印失败也改变状态并更新失败原因
|
||
status = printMsgFail
|
||
comment = printErrMap[data[12:14]]
|
||
}
|
||
//这里序号重复会有问题
|
||
if printMsgs, err := dao.GetPrintMsgNoPage(db, printNo, orderNo); err != nil {
|
||
globals.SugarLogger.Debugf("changePrintMsg err :[%v]", err)
|
||
return err
|
||
} else if len(printMsgs) == 0 {
|
||
globals.SugarLogger.Debugf("changePrintMsg err ,not found printMsg printNo:[%v], orderNo :[%v]", printNo, orderNo)
|
||
} else if len(printMsgs) > 0 {
|
||
for _, v := range printMsgs {
|
||
v.Comment = comment
|
||
v.Status = status
|
||
dao.UpdateEntity(db, v, "Comment", "Status")
|
||
}
|
||
}
|
||
return err
|
||
}
|
||
|
||
func (t *TcpClient) HandleTcpMessages(printNo string) {
|
||
var (
|
||
db = dao.GetDB()
|
||
offset, pageSize = 0, 10
|
||
)
|
||
if !t.isExistMsg(printNo) {
|
||
return
|
||
}
|
||
|
||
fn := func() {
|
||
for {
|
||
if t.TimeoutMap[printNo] == true {
|
||
if time.Now().Unix()%3 == 0 {
|
||
timeNow := time.Now()
|
||
timeStart := time.Date(timeNow.Year(), timeNow.Month(), timeNow.Day(), 0, 0, 0, 0, timeNow.Location())
|
||
timeEnd := time.Date(timeNow.Year(), timeNow.Month(), timeNow.Day(), 23, 59, 59, 0, timeNow.Location())
|
||
prints, _ := dao.GetPrintMsgs(db, printNo, []int{printMsgWait}, timeStart.AddDate(0, 0, -1), timeEnd, offset, pageSize)
|
||
for _, printMsg := range prints {
|
||
printMsg.Status = printMsgAlreadyLoad
|
||
//先避免重复读再插到channel？
|
||
if _, err := dao.UpdateEntity(db, printMsg, "Status"); err == nil {
|
||
if err = t.addMsgChan(printMsg); err != nil {
|
||
globals.SugarLogger.Debugf("HandleTcpMessages addMsgChan Err: %v", err)
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
continue
|
||
}
|
||
} else {
|
||
globals.SugarLogger.Debugf("HandleTcpMessages timeout")
|
||
return
|
||
}
|
||
}
|
||
}
|
||
Poll.AddJob(fn)
|
||
|
||
//go func() {
|
||
// for {
|
||
// if t.TimeoutMap[printNo] == true {
|
||
// timeNow := time.Now()
|
||
// timeStart := time.Date(timeNow.Year(), timeNow.Month(), timeNow.Day(), 0, 0, 0, 0, timeNow.Location())
|
||
// timeEnd := time.Date(timeNow.Year(), timeNow.Month(), timeNow.Day(), 23, 59, 59, 0, timeNow.Location())
|
||
// prints, _ := dao.GetPrintMsgs(db, printNo, []int{printMsgWait}, timeStart.AddDate(0, 0, -1), timeEnd, offset, pageSize)
|
||
// for _, printMsg := range prints {
|
||
// printMsg.Status = printMsgAlreadyLoad
|
||
// //先避免重复读再插到channel？
|
||
// if _, err := dao.UpdateEntity(db, printMsg, "Status"); err == nil {
|
||
// if err = t.addMsgChan(printMsg); err != nil {
|
||
// globals.SugarLogger.Debugf("HandleTcpMessages addMsgChan Err: %v", err)
|
||
// }
|
||
// }
|
||
// }
|
||
// } else {
|
||
// globals.SugarLogger.Debugf("HandleTcpMessages timeout")
|
||
// return
|
||
// }
|
||
// }
|
||
//}()
|
||
}
|
||
|
||
func (t *TcpClient) readTimeoutMap(key string) bool {
|
||
t.Lock()
|
||
defer t.Unlock()
|
||
return t.TimeoutMap[key]
|
||
}
|
||
|
||
func (t *TcpClient) doPrint(key string) (err error) {
|
||
var (
|
||
db = dao.GetDB()
|
||
)
|
||
|
||
if !t.isExistMsg(key) {
|
||
return err
|
||
}
|
||
|
||
fn := func() {
|
||
if t.TimeoutMap[key] == true {
|
||
select {
|
||
case printMsg, ok := <-t.MsgMap[key]:
|
||
if !ok {
|
||
globals.SugarLogger.Debugf("doPrint err !ok ...")
|
||
return
|
||
}
|
||
var (
|
||
data []byte
|
||
c net.Conn
|
||
)
|
||
if printMsg != nil {
|
||
if err = checkPrintMsg(db, printMsg); err == nil {
|
||
status := t.getPrintStatus(printMsg.PrintNo)
|
||
switch status {
|
||
//只有在线才打印内容
|
||
case printerStatusOnline:
|
||
if c = t.getPrintConn(printMsg.PrintNo); c != nil {
|
||
data, err = buildMsg(printMsg)
|
||
}
|
||
case printerStatusOffline:
|
||
err = fmt.Errorf("打印机离线！")
|
||
case printerStatusOnlineWithoutPaper:
|
||
err = fmt.Errorf("打印机缺纸！")
|
||
default:
|
||
err = fmt.Errorf("打印机状态未知！")
|
||
}
|
||
}
|
||
if err != nil {
|
||
printMsg.Status = printMsgErr
|
||
printMsg.Comment = err.Error()
|
||
dao.UpdateEntity(db, printMsg, "Status", "Comment")
|
||
delete(t.TimeoutMap, key)
|
||
} else {
|
||
if c != nil {
|
||
if _, err = c.Write(data); err != nil {
|
||
globals.SugarLogger.Debugf("handleTcpMessages err [%v]", err)
|
||
//close(t.TimeoutMap[key])
|
||
delete(t.TimeoutMap, key)
|
||
} else {
|
||
//等待回调
|
||
dataStr := <-t.CallBackMap[key]
|
||
if dataStr != "" {
|
||
a, b := getCallbackMsgInfo(dataStr)
|
||
t.changePrintMsg(dataStr, a, b)
|
||
// 查询打印机是否扣费,未扣费就扣费,已经扣费不做处理
|
||
have, err := dao.QueryOrderDeductionRecord(db, b, utils.Int64ToStr(a))
|
||
if err != nil && !have {
|
||
// 扣除打印机账号金额
|
||
if err = dao.DeductionPrintBalance(db, b); err != nil {
|
||
globals.SugarLogger.Debugf("扣除用户打印机金额错误 %s", err)
|
||
} else {
|
||
// 添加打印记录(支出记录)
|
||
if err = dao.AddPrintRecord(db, &model.PrintBillRecord{
|
||
CreatedAt: time.Now(),
|
||
UpdatedAt: time.Now(),
|
||
PrintNo: b,
|
||
PayType: 2,
|
||
PayMoney: 1, // 固定支出一分钱
|
||
OrderId: utils.Int64ToStr(a),
|
||
UserId: "",
|
||
}); err != nil {
|
||
globals.SugarLogger.Debugf("添加打印机订单支付记录错误 %s", err)
|
||
}
|
||
}
|
||
} else {
|
||
globals.SugarLogger.Debugf("查询打印机扣费记录错误 %s", err)
|
||
}
|
||
//判断音频暂停？
|
||
//收到打印成功回调后，如果消息中有音频，需要等待一下,等上一个音频播完
|
||
//暂停时间就暂时取的sound标签内内容长度/2
|
||
if sounds := regexpSoundSpan.FindStringSubmatch(printMsg.Content); len(sounds) > 0 {
|
||
sound := sounds[1]
|
||
lenTime := time.Duration(utf8.RuneCountInString(sound)) * time.Second
|
||
time.Sleep(lenTime / 2)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
globals.SugarLogger.Debugf("msgMap is nil")
|
||
}
|
||
}
|
||
} else {
|
||
globals.SugarLogger.Debugf("doPrint timeout")
|
||
return
|
||
}
|
||
}
|
||
|
||
Poll.AddJob(fn)
|
||
//go func() {
|
||
// for {
|
||
// if t.TimeoutMap[key] == true {
|
||
// select {
|
||
// case printMsg, ok := <-t.MsgMap[key]:
|
||
// if !ok {
|
||
// globals.SugarLogger.Debugf("doPrint err !ok ...")
|
||
// return
|
||
// }
|
||
// var (
|
||
// data []byte
|
||
// c net.Conn
|
||
// )
|
||
// if printMsg != nil {
|
||
// if err = checkPrintMsg(db, printMsg); err == nil {
|
||
// status := t.getPrintStatus(printMsg.PrintNo)
|
||
// switch status {
|
||
// //只有在线才打印内容
|
||
// case printerStatusOnline:
|
||
// if c = t.getPrintConn(printMsg.PrintNo); c != nil {
|
||
// data, err = buildMsg(printMsg)
|
||
// }
|
||
// case printerStatusOffline:
|
||
// err = fmt.Errorf("打印机离线！")
|
||
// case printerStatusOnlineWithoutPaper:
|
||
// err = fmt.Errorf("打印机缺纸！")
|
||
// default:
|
||
// err = fmt.Errorf("打印机状态未知！")
|
||
// }
|
||
// }
|
||
// if err != nil {
|
||
// printMsg.Status = printMsgErr
|
||
// printMsg.Comment = err.Error()
|
||
// dao.UpdateEntity(db, printMsg, "Status", "Comment")
|
||
// delete(t.TimeoutMap, key)
|
||
// } else {
|
||
// if c != nil {
|
||
// if _, err = c.Write(data); err != nil {
|
||
// globals.SugarLogger.Debugf("handleTcpMessages err [%v]", err)
|
||
// //close(t.TimeoutMap[key])
|
||
// delete(t.TimeoutMap, key)
|
||
// } else {
|
||
// //等待回调
|
||
// dataStr := <-t.CallBackMap[key]
|
||
// if dataStr != "" {
|
||
// a, b := getCallbackMsgInfo(dataStr)
|
||
// t.changePrintMsg(dataStr, a, b)
|
||
// // 查询打印机是否扣费,未扣费就扣费,已经扣费不做处理
|
||
// have, err := dao.QueryOrderDeductionRecord(db, b, utils.Int64ToStr(a))
|
||
// if err != nil && !have {
|
||
// // 扣除打印机账号金额
|
||
// if err = dao.DeductionPrintBalance(db, b); err != nil {
|
||
// globals.SugarLogger.Debugf("扣除用户打印机金额错误 %s", err)
|
||
// } else {
|
||
// // 添加打印记录(支出记录)
|
||
// if err = dao.AddPrintRecord(db, &model.PrintBillRecord{
|
||
// CreatedAt: time.Now(),
|
||
// UpdatedAt: time.Now(),
|
||
// PrintNo: b,
|
||
// PayType: 2,
|
||
// PayMoney: 1, // 固定支出一分钱
|
||
// OrderId: utils.Int64ToStr(a),
|
||
// UserId: "",
|
||
// }); err != nil {
|
||
// globals.SugarLogger.Debugf("添加打印机订单支付记录错误 %s", err)
|
||
// }
|
||
// }
|
||
// } else {
|
||
// globals.SugarLogger.Debugf("查询打印机扣费记录错误 %s", err)
|
||
// }
|
||
// //判断音频暂停？
|
||
// //收到打印成功回调后，如果消息中有音频，需要等待一下,等上一个音频播完
|
||
// //暂停时间就暂时取的sound标签内内容长度/2
|
||
// if sounds := regexpSoundSpan.FindStringSubmatch(printMsg.Content); len(sounds) > 0 {
|
||
// sound := sounds[1]
|
||
// lenTime := time.Duration(utf8.RuneCountInString(sound)) * time.Second
|
||
// time.Sleep(lenTime / 2)
|
||
// }
|
||
// }
|
||
// }
|
||
// }
|
||
// }
|
||
// } else {
|
||
// globals.SugarLogger.Debugf("msgMap is nil")
|
||
// }
|
||
// }
|
||
// } else {
|
||
// globals.SugarLogger.Debugf("doPrint timeout")
|
||
// return
|
||
// }
|
||
//
|
||
// }
|
||
//}()
|
||
return err
|
||
}
|
||
|
||
//检测心跳
|
||
func (t *TcpClient) HandleCheckTcpHeart(key string) {
|
||
go func() {
|
||
for {
|
||
if t.TimeoutMap[key] == true {
|
||
statusTime := t.getPrintStatusTime(key)
|
||
if !utils.IsTimeZero(statusTime) {
|
||
//1分钟内没心跳判断打印机掉线了
|
||
if time.Now().Sub(statusTime) > time.Minute+time.Second*10 {
|
||
changePrinterStatus(key, printerStatusOffline)
|
||
delete(t.TimeoutMap, key)
|
||
}
|
||
}
|
||
} else {
|
||
t.getClients(key).C.Close()
|
||
close(t.MsgMap[key])
|
||
close(t.CallBackMap[key])
|
||
t.delConn(key)
|
||
return
|
||
}
|
||
}
|
||
}()
|
||
}
|