Files
go-gin-api/internal/api/tool/func_sendmessage.go
T
新亮 4c37a7e6b5 feature(1.2.8): swagger 接口文档新增 Security
- 将 middleware 命名为 interceptor
- 将 deploy 命名为 deployments
- 移除 pkg/errno
- 使用 proposal 目录
- 优化代码
2021-11-28 13:25:27 +08:00

91 lines
2.2 KiB
Go

package tool
import (
"encoding/json"
"net/http"
"github.com/xinliangnote/go-gin-api/internal/code"
"github.com/xinliangnote/go-gin-api/internal/pkg/core"
"github.com/xinliangnote/go-gin-api/internal/pkg/validation"
"github.com/xinliangnote/go-gin-api/internal/websocket/sysmessage"
"github.com/xinliangnote/go-gin-api/pkg/timeutil"
)
type sendMessageRequest struct {
Message string `form:"message"` // 消息内容
}
type sendMessageResponse struct {
Status string `json:"status"` // 状态
}
// SendMessage 发送消息
// @Summary 发送消息
// @Description 发送消息
// @Tags API.tool
// @Accept application/x-www-form-urlencoded
// @Produce json
// @Param message formData string true "消息内容"
// @Success 200 {object} sendMessageResponse
// @Failure 400 {object} code.Failure
// @Router /api/tool/send_message [post]
// @Security LoginToken
func (h *handler) SendMessage() core.HandlerFunc {
type messageBody struct {
Username string `json:"username"`
Message string `json:"message"`
Time string `json:"time"`
}
return func(ctx core.Context) {
req := new(sendMessageRequest)
res := new(sendMessageResponse)
if err := ctx.ShouldBindForm(req); err != nil {
ctx.AbortWithError(core.Error(
http.StatusBadRequest,
code.ParamBindError,
validation.Error(err)).WithError(err),
)
return
}
conn, err := sysmessage.GetConn()
if err != nil {
ctx.AbortWithError(core.Error(
http.StatusBadRequest,
code.SocketConnectError,
code.Text(code.SocketConnectError)).WithError(err),
)
return
}
messageData := new(messageBody)
messageData.Username = ctx.SessionUserInfo().UserName
messageData.Message = req.Message
messageData.Time = timeutil.CSTLayoutString()
messageJsonData, err := json.Marshal(messageData)
if err != nil {
ctx.AbortWithError(core.Error(
http.StatusBadRequest,
code.SocketSendError,
code.Text(code.SocketSendError)).WithError(err),
)
return
}
err = conn.OnSend(messageJsonData)
if err != nil {
ctx.AbortWithError(core.Error(
http.StatusBadRequest,
code.SocketSendError,
code.Text(code.SocketSendError)).WithError(err),
)
return
}
res.Status = "OK"
ctx.Payload(res)
}
}