复制项目
This commit is contained in:
kim.dev.6789
288
internal/rpc/chat/scheduled_task.go
Normal file
288
internal/rpc/chat/scheduled_task.go
Normal file
@@ -0,0 +1,288 @@
|
||||
// Copyright © 2023 OpenIM open source community. All rights reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package chat
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
chatdb "git.imall.cloud/openim/chat/pkg/common/db/table/chat"
|
||||
"git.imall.cloud/openim/chat/pkg/common/mctx"
|
||||
"git.imall.cloud/openim/chat/pkg/protocol/chat"
|
||||
"github.com/openimsdk/tools/errs"
|
||||
)
|
||||
|
||||
// ==================== 定时任务相关 RPC ====================
|
||||
|
||||
// CreateScheduledTask 创建定时任务
|
||||
func (o *chatSvr) CreateScheduledTask(ctx context.Context, req *chat.CreateScheduledTaskReq) (*chat.CreateScheduledTaskResp, error) {
|
||||
// 获取当前用户ID
|
||||
userID, _, err := mctx.Check(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 验证必填字段
|
||||
if req.Name == "" {
|
||||
return nil, errs.ErrArgs.WrapMsg("task name is required")
|
||||
}
|
||||
if req.CronExpression == "" {
|
||||
return nil, errs.ErrArgs.WrapMsg("cron expression is required")
|
||||
}
|
||||
if len(req.Messages) == 0 {
|
||||
return nil, errs.ErrArgs.WrapMsg("messages is required")
|
||||
}
|
||||
if len(req.RecvIDs) == 0 && len(req.GroupIDs) == 0 {
|
||||
return nil, errs.ErrArgs.WrapMsg("recvIDs or groupIDs is required")
|
||||
}
|
||||
|
||||
// 验证消息类型
|
||||
for _, msg := range req.Messages {
|
||||
if msg.Type < 1 || msg.Type > 3 {
|
||||
return nil, errs.ErrArgs.WrapMsg("invalid message type")
|
||||
}
|
||||
}
|
||||
|
||||
// 转换消息列表
|
||||
messages := make([]chatdb.Message, 0, len(req.Messages))
|
||||
for _, msg := range req.Messages {
|
||||
messages = append(messages, chatdb.Message{
|
||||
Type: msg.Type,
|
||||
Content: msg.Content,
|
||||
Thumbnail: msg.Thumbnail,
|
||||
Duration: msg.Duration,
|
||||
FileSize: msg.FileSize,
|
||||
Width: msg.Width,
|
||||
Height: msg.Height,
|
||||
})
|
||||
}
|
||||
|
||||
// 创建定时任务对象
|
||||
task := &chatdb.ScheduledTask{
|
||||
UserID: userID,
|
||||
Name: req.Name,
|
||||
CronExpression: req.CronExpression,
|
||||
Messages: messages,
|
||||
RecvIDs: req.RecvIDs,
|
||||
GroupIDs: req.GroupIDs,
|
||||
Status: req.Status,
|
||||
CreateTime: time.Now(),
|
||||
UpdateTime: time.Now(),
|
||||
}
|
||||
|
||||
// 如果状态未设置，默认为启用
|
||||
if task.Status == 0 {
|
||||
task.Status = 1
|
||||
}
|
||||
|
||||
// 保存到数据库
|
||||
if err := o.Database.CreateScheduledTask(ctx, task); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &chat.CreateScheduledTaskResp{
|
||||
TaskID: task.ID,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetScheduledTask 获取定时任务详情
|
||||
func (o *chatSvr) GetScheduledTask(ctx context.Context, req *chat.GetScheduledTaskReq) (*chat.GetScheduledTaskResp, error) {
|
||||
// 获取当前用户ID
|
||||
userID, _, err := mctx.Check(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 获取任务
|
||||
task, err := o.Database.GetScheduledTask(ctx, req.TaskID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 验证是否为当前用户的任务
|
||||
if task.UserID != userID {
|
||||
return nil, errs.ErrNoPermission.WrapMsg("not your task")
|
||||
}
|
||||
|
||||
return &chat.GetScheduledTaskResp{
|
||||
Task: convertScheduledTaskToProto(task),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetScheduledTasks 获取定时任务列表
|
||||
func (o *chatSvr) GetScheduledTasks(ctx context.Context, req *chat.GetScheduledTasksReq) (*chat.GetScheduledTasksResp, error) {
|
||||
// 获取当前用户ID
|
||||
userID, _, err := mctx.Check(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 获取任务列表
|
||||
total, tasks, err := o.Database.GetScheduledTasksByUserID(ctx, userID, req.Pagination)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 转换为响应格式
|
||||
taskInfos := make([]*chat.ScheduledTaskInfo, 0, len(tasks))
|
||||
for _, task := range tasks {
|
||||
taskInfos = append(taskInfos, convertScheduledTaskToProto(task))
|
||||
}
|
||||
|
||||
return &chat.GetScheduledTasksResp{
|
||||
Total: uint32(total),
|
||||
Tasks: taskInfos,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// UpdateScheduledTask 更新定时任务
|
||||
func (o *chatSvr) UpdateScheduledTask(ctx context.Context, req *chat.UpdateScheduledTaskReq) (*chat.UpdateScheduledTaskResp, error) {
|
||||
// 获取当前用户ID
|
||||
userID, _, err := mctx.Check(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 获取任务，验证所有权
|
||||
task, err := o.Database.GetScheduledTask(ctx, req.TaskID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if task.UserID != userID {
|
||||
return nil, errs.ErrNoPermission.WrapMsg("not your task")
|
||||
}
|
||||
|
||||
// 构建更新数据
|
||||
updateData := make(map[string]any)
|
||||
if req.Name != "" {
|
||||
updateData["name"] = req.Name
|
||||
}
|
||||
if req.CronExpression != "" {
|
||||
updateData["cron_expression"] = req.CronExpression
|
||||
}
|
||||
if len(req.Messages) > 0 {
|
||||
// 验证消息类型
|
||||
for _, msg := range req.Messages {
|
||||
if msg.Type < 1 || msg.Type > 3 {
|
||||
return nil, errs.ErrArgs.WrapMsg("invalid message type")
|
||||
}
|
||||
}
|
||||
// 转换消息列表
|
||||
messages := make([]chatdb.Message, 0, len(req.Messages))
|
||||
for _, msg := range req.Messages {
|
||||
messages = append(messages, chatdb.Message{
|
||||
Type: msg.Type,
|
||||
Content: msg.Content,
|
||||
Thumbnail: msg.Thumbnail,
|
||||
Duration: msg.Duration,
|
||||
FileSize: msg.FileSize,
|
||||
Width: msg.Width,
|
||||
Height: msg.Height,
|
||||
})
|
||||
}
|
||||
updateData["messages"] = messages
|
||||
}
|
||||
if req.RecvIDs != nil {
|
||||
updateData["recv_ids"] = req.RecvIDs
|
||||
}
|
||||
if req.GroupIDs != nil {
|
||||
updateData["group_ids"] = req.GroupIDs
|
||||
}
|
||||
// status字段：0-已禁用，1-已启用，允许设置为0
|
||||
if req.Status == 0 || req.Status == 1 {
|
||||
updateData["status"] = req.Status
|
||||
}
|
||||
|
||||
// 验证：如果更新后没有接收者，返回错误
|
||||
if req.RecvIDs != nil && req.GroupIDs != nil {
|
||||
if len(req.RecvIDs) == 0 && len(req.GroupIDs) == 0 {
|
||||
return nil, errs.ErrArgs.WrapMsg("recvIDs or groupIDs is required")
|
||||
}
|
||||
} else if req.RecvIDs != nil && len(req.RecvIDs) == 0 {
|
||||
// 如果只更新了RecvIDs且为空，检查GroupIDs是否也为空
|
||||
if len(task.GroupIDs) == 0 {
|
||||
return nil, errs.ErrArgs.WrapMsg("recvIDs or groupIDs is required")
|
||||
}
|
||||
} else if req.GroupIDs != nil && len(req.GroupIDs) == 0 {
|
||||
// 如果只更新了GroupIDs且为空，检查RecvIDs是否也为空
|
||||
if len(task.RecvIDs) == 0 {
|
||||
return nil, errs.ErrArgs.WrapMsg("recvIDs or groupIDs is required")
|
||||
}
|
||||
}
|
||||
|
||||
// 更新任务
|
||||
if err := o.Database.UpdateScheduledTask(ctx, req.TaskID, updateData); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &chat.UpdateScheduledTaskResp{}, nil
|
||||
}
|
||||
|
||||
// DeleteScheduledTask 删除定时任务
|
||||
func (o *chatSvr) DeleteScheduledTask(ctx context.Context, req *chat.DeleteScheduledTaskReq) (*chat.DeleteScheduledTaskResp, error) {
|
||||
// 获取当前用户ID
|
||||
userID, _, err := mctx.Check(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 验证所有权（批量验证）
|
||||
for _, taskID := range req.TaskIDs {
|
||||
task, err := o.Database.GetScheduledTask(ctx, taskID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if task.UserID != userID {
|
||||
return nil, errs.ErrNoPermission.WrapMsg("not your task")
|
||||
}
|
||||
}
|
||||
|
||||
// 删除任务
|
||||
if err := o.Database.DeleteScheduledTask(ctx, req.TaskIDs); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &chat.DeleteScheduledTaskResp{}, nil
|
||||
}
|
||||
|
||||
// convertScheduledTaskToProto 将数据库模型转换为 protobuf 消息
|
||||
func convertScheduledTaskToProto(task *chatdb.ScheduledTask) *chat.ScheduledTaskInfo {
|
||||
messages := make([]*chat.ScheduledTaskMessage, 0, len(task.Messages))
|
||||
for _, msg := range task.Messages {
|
||||
messages = append(messages, &chat.ScheduledTaskMessage{
|
||||
Type: msg.Type,
|
||||
Content: msg.Content,
|
||||
Thumbnail: msg.Thumbnail,
|
||||
Duration: msg.Duration,
|
||||
FileSize: msg.FileSize,
|
||||
Width: msg.Width,
|
||||
Height: msg.Height,
|
||||
})
|
||||
}
|
||||
|
||||
return &chat.ScheduledTaskInfo{
|
||||
Id: task.ID,
|
||||
UserID: task.UserID,
|
||||
Name: task.Name,
|
||||
CronExpression: task.CronExpression,
|
||||
Messages: messages,
|
||||
RecvIDs: task.RecvIDs,
|
||||
GroupIDs: task.GroupIDs,
|
||||
Status: task.Status,
|
||||
CreateTime: task.CreateTime.UnixMilli(),
|
||||
UpdateTime: task.UpdateTime.UnixMilli(),
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user