event

2020-10-13 12:02:07 +08:00
parent f2f7e511a3
commit 078327e8d5
2 changed files with 186 additions and 0 deletions
--- a/controllers/event.go
+++ b/controllers/event.go
@@ -0,0 +1,181 @@
+package controllers
+
+import (
+	"log"
+	"net/http"
+
+	"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
+
+	"git.rosy.net.cn/jx-callback/globals"
+
+	"git.rosy.net.cn/jx-callback/business/model/dao"
+
+	"git.rosy.net.cn/jx-callback/business/model"
+
+	"github.com/gorilla/websocket"
+
+	"git.rosy.net.cn/baseapi/utils"
+
+	"github.com/astaxie/beego"
+)
+
+// 操作事件明细相关API
+type EventController struct {
+	beego.Controller
+}
+
+//连接的客户端,吧每个客户端都放进来
+var clients = make(map[string][]*websocket.Conn)
+
+//广播频道(通道)
+var broadcast = make(chan map[string]*model.ImMessageRecord)
+
+// 配置升级程序(升级为websocket)
+var upgrader = websocket.Upgrader{}
+
+// 定义我们的消息对象
+type Message struct {
+	Data interface{} `json:"data"`
+}
+
+// @Title 得到微信小程序码
+// @Description 得到微信小程序码
+// @Param	token		header		string	true	"认证token"
+// @Param	scene		formData	string	true	"最大32个可见字符，只支持数字，大小写英文以及部分特殊字符：!#$&'()*+,/:;=?@-._~，其它字符请自行编码为合法字符（因不支持%，中文无法使用 urlencode 处理，请使用其他编码方式）"
+// @Param	page		formData	string	false	"必须是已经发布的小程序存在的页面（否则报错），例如 pages/index/index, 根路径前不要填加 /,不能携带参数（参数请放在scene字段里），如果不填写这个字段，默认跳主页面"
+// @Success 200 {object} controllers.CallResult
+// @Failure 200 {object} controllers.CallResult
+// @router /GetWeixinUnlimited [post]
+func (c *EventController) GetWeixinUnlimited() {
+	// c.callGetWeixinUnlimited(func(params *tEventGetWeixinUnlimitedParams) (retVal interface{}, errCode string, err error) {
+	// 	// api.WeixinMiniAPI2.CBRetrieveToken()
+	// 	result, err := api.WeixinMiniAPI2.GetUnlimited(params.Scene, params.Page)
+	// 	if err != nil || result == nil {
+	// 		return retVal, "", err
+	// 	}
+	// 	fileName := "wxcode" + utils.Int64ToStr(time.Now().Unix()) + ".jpg"
+	// 	retVal, err = jxutils.UploadExportContent([]byte(result["fakeData"].(string)), fileName)
+	// 	// tasks.RefreshWeixinToken()
+	// 	return retVal, "", err
+	// })
+}
+
+// @Title 测试websocket
+// @Description 测试websocket
+// @Success 200 {object} controllers.CallResult
+// @Failure 200 {object} controllers.CallResult
+// @router /TestWebsocket [get]
+func (c *EventController) TestWebsocket() {
+	// 解决跨域问题(微信小程序)
+	upgrader.CheckOrigin = func(r *http.Request) bool {
+		return true
+	}
+	//升级将HTTP服务器连接升级到WebSocket协议。
+	//responseHeader包含在对客户端升级的响应中
+	//请求。使用responseHeader指定Cookie（设置Cookie）和
+	//应用程序协商的子目录（Sec WebSocket协议）。
+	//如果升级失败，则升级将向客户端答复一个HTTP错误
+	ws, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer ws.Close()
+	var (
+		vendorOrderID = c.GetString("vendorOrderID")
+		msgMap        = make(map[string]*model.ImMessageRecord)
+	)
+	//将当前客户端放入map中
+	clients[vendorOrderID] = append(clients[vendorOrderID], ws)
+
+	db := dao.GetDB()
+	if globals.IsProductEnv() {
+		_, _, err = jxcontext.New(nil, c.GetString("token"), c.Ctx.ResponseWriter, c.Ctx.Request)
+		if err != nil {
+			msg := &CallResult{
+				Code: model.ErrCodeGeneralFailed,
+				Desc: err.Error(),
+			}
+			ws.WriteJSON(&msg)
+		}
+	}
+
+	c.EnableRender = false //Beego不启用渲染
+
+	var s *model.ImMessageRecord
+	for {
+		//接收客户端的消息
+		err := ws.ReadJSON(&s)
+		if err != nil {
+			globals.SugarLogger.Debugf("页面可能断开啦 ws.ReadJSON error: %v", err.Error())
+			index := 0
+			for k, v := range clients[vendorOrderID] {
+				if v == ws {
+					index = k
+				}
+			}
+			wsList := clients[vendorOrderID]
+			clients[vendorOrderID] = clients[vendorOrderID][0:0]
+			clients[vendorOrderID] = append(clients[vendorOrderID], wsList[0:index]...)
+			clients[vendorOrderID] = append(clients[vendorOrderID], wsList[index+1:len(wsList)]...)
+			// delete(clients, ws) //删除map中的客户端
+			break //结束循环
+		} else {
+			//接受消息 业务逻辑
+			msgMap[vendorOrderID] = s
+			broadcast <- msgMap
+			dao.WrapAddIDCULDEntity(s, "")
+			dao.CreateEntity(db, s)
+			// fmt.Println("接受到从页面上反馈回来的信息 ", utils.Format4Output(s, false))
+		}
+	}
+}
+
+func init() {
+	go handleMessages()
+}
+
+//广播推送消息
+func handleMessages() {
+	for {
+		//读取通道中的消息
+		msg := <-broadcast
+		globals.SugarLogger.Debugf("clients len %v", len(clients))
+		//循环map客户端
+		for vendorOrderID, mmsg := range msg {
+			for _, client := range clients[vendorOrderID] {
+				//把通道中的消息发送给客户端
+				globals.SugarLogger.Debugf("msg %v", utils.Format4Output(mmsg, false))
+				err := client.WriteJSON(mmsg)
+				if err != nil {
+					globals.SugarLogger.Debugf("client.WriteJSON error: %v", err)
+					client.Close() //关闭
+					index := 0
+					for k, v := range clients[vendorOrderID] {
+						if v == client {
+							index = k
+						}
+					}
+					wsList := clients[vendorOrderID]
+					clients[vendorOrderID] = clients[vendorOrderID][0:0]
+					clients[vendorOrderID] = append(clients[vendorOrderID], wsList[0:index]...)
+					clients[vendorOrderID] = append(clients[vendorOrderID], wsList[index+1:len(wsList)]...)
+					// delete(clients, client) //删除map中的客户端
+				}
+			}
+		}
+	}
+}
+
+// @Title 查询聊天记录
+// @Description 查询聊天记录
+// @Param	token			header		string	true	"认证token"
+// @Param	vendorOrderID	query		string	true	"订单号"
+// @Success 200 {object} controllers.CallResult
+// @Failure 200 {object} controllers.CallResult
+// @router /GetImMessageRecord [get]
+func (c *EventController) GetImMessageRecord() {
+	// c.callGetImMessageRecord(func(params *tEventGetImMessageRecordParams) (retVal interface{}, errCode string, err error) {
+	// 	retVal, err = dao.GetImMessageRecord(dao.GetDB(), params.VendorOrderID, "", 0, -1)
+	// 	return retVal, "", err
+	// })
+}
--- a/routers/router.go
+++ b/routers/router.go
@@ -46,6 +46,11 @@ func init() {
 				&controllers.JobController{},
 			),
 		),
+		beego.NSNamespace("/event",
+			beego.NSInclude(
+				&controllers.EventController{},
+			),
+		),
 	)
 	beego.AddNamespace(ns)
 	beego.AutoRouter(&controllers.TongLianController{})