modify file name
This commit is contained in:
苏尹岚
268
controllers/event_controller.go
Normal file
268
controllers/event_controller.go
Normal file
@@ -0,0 +1,268 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"log"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"git.rosy.net.cn/jx-callback/business/jxstore/event"
|
||||
|
||||
"git.rosy.net.cn/jx-callback/business/jxutils"
|
||||
"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
|
||||
"git.rosy.net.cn/jx-callback/globals/api"
|
||||
|
||||
"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[int][]*websocket.Conn)
|
||||
|
||||
//广播频道(通道)
|
||||
var broadcast = make(chan map[int]*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 (
|
||||
// userID = c.GetString("userID")
|
||||
msgMap = make(map[int]*model.ImMessageRecord)
|
||||
groupID int
|
||||
)
|
||||
//将当前客户端放入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)
|
||||
clients[s.GroupID] = append(clients[s.GroupID], ws)
|
||||
if err != nil {
|
||||
globals.SugarLogger.Debugf("页面可能断开啦 ws.ReadJSON error: %v", err.Error())
|
||||
index := 0
|
||||
for k, v := range clients[groupID] {
|
||||
if v == ws {
|
||||
index = k
|
||||
}
|
||||
}
|
||||
wsList := clients[groupID]
|
||||
clients[groupID] = clients[groupID][0:0]
|
||||
clients[groupID] = append(clients[groupID], wsList[0:index]...)
|
||||
clients[groupID] = append(clients[groupID], wsList[index+1:len(wsList)]...)
|
||||
// delete(clients, ws) //删除map中的客户端
|
||||
break //结束循环
|
||||
} else {
|
||||
//接受消息 业务逻辑
|
||||
msgMap[s.GroupID] = 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 groupID, mmsg := range msg {
|
||||
for _, client := range clients[groupID] {
|
||||
//把通道中的消息发送给客户端
|
||||
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[groupID] {
|
||||
if v == client {
|
||||
index = k
|
||||
}
|
||||
}
|
||||
wsList := clients[groupID]
|
||||
clients[groupID] = clients[groupID][0:0]
|
||||
clients[groupID] = append(clients[groupID], wsList[0:index]...)
|
||||
clients[groupID] = append(clients[groupID], 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
|
||||
})
|
||||
}
|
||||
|
||||
// @Title 创建聊天组
|
||||
// @Description 创建聊天组
|
||||
// @Param token header string true "认证token"
|
||||
// @Param userID formData string true "创建者id"
|
||||
// @Param userID2 formData string false "被拉的id 如果userID2为空就默认为是创建的群聊"
|
||||
// @Param name formData string false "如果是群聊，则要传入群名"
|
||||
// @Param dividePercentage formData int false "如果是群聊，则要传入分成比例"
|
||||
// @Param quitPrice formData int false "如果是群聊，则要传入退团金额"
|
||||
// @Success 200 {object} controllers.CallResult
|
||||
// @Failure 200 {object} controllers.CallResult
|
||||
// @router /CreateMessageGroup [post]
|
||||
func (c *EventController) CreateMessageGroup() {
|
||||
c.callCreateMessageGroup(func(params *tEventCreateMessageGroupParams) (retVal interface{}, errCode string, err error) {
|
||||
retVal, err = event.CreateMessageGroup(params.Ctx, params.UserID, params.UserID2, params.Name, params.DividePercentage, params.QuitPrice)
|
||||
return retVal, "", err
|
||||
})
|
||||
}
|
||||
|
||||
// @Title 查询某个用户所有聊天组
|
||||
// @Description 查询某个用户所有聊天组
|
||||
// @Param token header string true "认证token"
|
||||
// @Param userID formData string true "userid"
|
||||
// @Success 200 {object} controllers.CallResult
|
||||
// @Failure 200 {object} controllers.CallResult
|
||||
// @router /GetMessageGroupByUser [get]
|
||||
func (c *EventController) GetMessageGroupByUser() {
|
||||
c.callGetMessageGroupByUser(func(params *tEventGetMessageGroupByUserParams) (retVal interface{}, errCode string, err error) {
|
||||
retVal, err = event.GetMessageGroupByUser(params.Ctx, params.UserID)
|
||||
return retVal, "", err
|
||||
})
|
||||
}
|
||||
|
||||
// @Title 加入用户组
|
||||
// @Description 加入用户组
|
||||
// @Param token header string true "认证token"
|
||||
// @Param groupID formData int true "组号"
|
||||
// @Param userID formData string true "被邀请人ID"
|
||||
// @Success 200 {object} controllers.CallResult
|
||||
// @Failure 200 {object} controllers.CallResult
|
||||
// @router /AddMessageGroup [post]
|
||||
func (c *EventController) AddMessageGroup() {
|
||||
c.callAddMessageGroup(func(params *tEventAddMessageGroupParams) (retVal interface{}, errCode string, err error) {
|
||||
err = event.AddMessageGroup(params.Ctx, params.GroupID, params.UserID)
|
||||
return retVal, "", err
|
||||
})
|
||||
}
|
||||
|
||||
// @Title 修改群组信息
|
||||
// @Description 修改群组信息
|
||||
// @Param token header string true "认证token"
|
||||
// @Param groupID formData int true "组号"
|
||||
// @Param payload formData string true "群组 payload"
|
||||
// @Success 200 {object} controllers.CallResult
|
||||
// @Failure 200 {object} controllers.CallResult
|
||||
// @router /UpdateMessageGroup [put]
|
||||
func (c *EventController) UpdateMessageGroup() {
|
||||
c.callUpdateMessageGroup(func(params *tEventUpdateMessageGroupParams) (retVal interface{}, errCode string, err error) {
|
||||
payload := make(map[string]interface{})
|
||||
if err = utils.UnmarshalUseNumber([]byte(params.Payload), &payload); err == nil {
|
||||
retVal, err = event.UpdateMessageGroup(params.Ctx, params.GroupID, payload)
|
||||
}
|
||||
return retVal, "", err
|
||||
})
|
||||
}
|
||||
|
||||
// @Title 退出用户组（踢人）
|
||||
// @Description 退出用户组（踢人）
|
||||
// @Param token header string true "认证token"
|
||||
// @Param groupID formData int true "组号"
|
||||
// @Param userID formData string true "userID"
|
||||
// @Success 200 {object} controllers.CallResult
|
||||
// @Failure 200 {object} controllers.CallResult
|
||||
// @router /DeleteMessageGroup [delete]
|
||||
func (c *EventController) DeleteMessageGroup() {
|
||||
c.callDeleteMessageGroup(func(params *tEventDeleteMessageGroupParams) (retVal interface{}, errCode string, err error) {
|
||||
errCode, err = event.DeleteMessageGroup(params.Ctx, params.GroupID, params.UserID)
|
||||
return retVal, errCode, err
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user