288 lines
9.0 KiB
Go
288 lines
9.0 KiB
Go
package dingtalk
|
||
|
||
import (
|
||
"ai_scheduler/internal/data/constants"
|
||
"ai_scheduler/internal/pkg"
|
||
"context"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
|
||
dingtalkim_1_0 "github.com/alibabacloud-go/dingtalk/im_1_0"
|
||
util "github.com/alibabacloud-go/tea-utils/v2/service"
|
||
"github.com/alibabacloud-go/tea/tea"
|
||
"github.com/gofiber/fiber/v2/log"
|
||
"github.com/google/uuid"
|
||
)
|
||
|
||
const DefaultInterval = 100 * time.Millisecond
|
||
const HeardBeatX = 50
|
||
|
||
type SendCardClient struct {
|
||
Auth *Auth
|
||
CardClient *sync.Map
|
||
mu sync.RWMutex // 保护 CardClient 的并发访问
|
||
logger log.AllLogger // 日志记录
|
||
botOption *Bot
|
||
}
|
||
|
||
func NewSendCardClient(auth *Auth, logger log.AllLogger) *SendCardClient {
|
||
return &SendCardClient{
|
||
Auth: auth,
|
||
CardClient: &sync.Map{},
|
||
logger: logger,
|
||
botOption: &Bot{},
|
||
}
|
||
}
|
||
|
||
// initClient 初始化或复用 DingTalk 客户端
|
||
func (s *SendCardClient) initClient(robotCode string) (*dingtalkim_1_0.Client, error) {
|
||
if client, ok := s.CardClient.Load(robotCode); ok {
|
||
return client.(*dingtalkim_1_0.Client), nil
|
||
}
|
||
s.botOption.BotCode = robotCode
|
||
config := &openapi.Config{
|
||
Protocol: tea.String("https"),
|
||
RegionId: tea.String("central"),
|
||
}
|
||
client, err := dingtalkim_1_0.NewClient(config)
|
||
if err != nil {
|
||
s.logger.Error("failed to init DingTalk client")
|
||
return nil, fmt.Errorf("init client failed: %w", err)
|
||
}
|
||
|
||
s.CardClient.Store(robotCode, client)
|
||
return client, nil
|
||
}
|
||
|
||
func (s *SendCardClient) NewCard(ctx context.Context, cardSend *CardSend) error {
|
||
// 参数校验
|
||
if (len(cardSend.ContentSlice) == 0 || cardSend.ContentSlice == nil) && cardSend.ContentChannel == nil {
|
||
return errors.New("卡片内容不能为空")
|
||
}
|
||
if cardSend.UpdateInterval == 0 {
|
||
cardSend.UpdateInterval = DefaultInterval // 默认更新间隔
|
||
}
|
||
if cardSend.Title == "" {
|
||
cardSend.Title = "钉钉卡片"
|
||
}
|
||
//替换标题
|
||
replace, err := pkg.SafeReplace(string(cardSend.Template), "${title}", cardSend.Title)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
cardSend.Template = constants.CardTemp(replace)
|
||
// 初始化客户端
|
||
client, err := s.initClient(cardSend.RobotCode)
|
||
if err != nil {
|
||
return fmt.Errorf("初始化client失败: %w", err)
|
||
}
|
||
|
||
// 生成卡片实例ID
|
||
cardInstanceId, err := uuid.NewUUID()
|
||
if err != nil {
|
||
return fmt.Errorf("创建uuid失败: %w", err)
|
||
}
|
||
|
||
// 构建初始请求
|
||
request, err := s.buildBaseRequest(cardSend, cardInstanceId.String())
|
||
if err != nil {
|
||
return fmt.Errorf("请求失败: %w", err)
|
||
}
|
||
|
||
// 发送初始卡片
|
||
if _, err := s.SendInteractiveCard(ctx, request, cardSend.RobotCode, client); err != nil {
|
||
return fmt.Errorf("发送初始卡片失败: %w", err)
|
||
}
|
||
|
||
// 处理切片内容（同步）
|
||
if len(cardSend.ContentSlice) > 0 {
|
||
if err := s.processContentSlice(ctx, cardSend, cardInstanceId.String(), client); err != nil {
|
||
return fmt.Errorf("内容同步失败: %w", err)
|
||
}
|
||
}
|
||
|
||
// 处理通道内容（异步）
|
||
if cardSend.ContentChannel != nil {
|
||
var wg sync.WaitGroup
|
||
wg.Add(1)
|
||
go func() {
|
||
defer wg.Done()
|
||
s.processContentChannel(ctx, cardSend, cardInstanceId.String(), client)
|
||
}()
|
||
wg.Wait()
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// buildBaseRequest 构建基础请求
|
||
func (s *SendCardClient) buildBaseRequest(cardSend *CardSend, cardInstanceId string) (*dingtalkim_1_0.SendRobotInteractiveCardRequest, error) {
|
||
cardData := fmt.Sprintf(string(cardSend.Template), "") // 初始空内容
|
||
request := &dingtalkim_1_0.SendRobotInteractiveCardRequest{
|
||
CardTemplateId: tea.String("StandardCard"),
|
||
CardBizId: tea.String(cardInstanceId),
|
||
CardData: tea.String(cardData),
|
||
RobotCode: tea.String(cardSend.RobotCode),
|
||
SendOptions: &dingtalkim_1_0.SendRobotInteractiveCardRequestSendOptions{},
|
||
PullStrategy: tea.Bool(false),
|
||
}
|
||
|
||
switch cardSend.ConversationType {
|
||
case constants.ConversationTypeGroup:
|
||
request.SetOpenConversationId(cardSend.ConversationId)
|
||
case constants.ConversationTypeSingle:
|
||
receiver, err := json.Marshal(map[string]string{"userId": cardSend.SenderStaffId})
|
||
if err != nil {
|
||
return nil, fmt.Errorf("数据整理失败: %w", err)
|
||
}
|
||
request.SetSingleChatReceiver(string(receiver))
|
||
default:
|
||
return nil, errors.New("未知的聊天场景")
|
||
}
|
||
|
||
return request, nil
|
||
}
|
||
|
||
// processContentChannel 处理通道内容（异步更新）
|
||
func (s *SendCardClient) processContentChannel(ctx context.Context, cardSend *CardSend, cardInstanceId string, client *dingtalkim_1_0.Client) {
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
s.logger.Error("panic in processContentChannel")
|
||
}
|
||
}()
|
||
|
||
ticker := time.NewTicker(cardSend.UpdateInterval)
|
||
defer ticker.Stop()
|
||
heartbeatTicker := time.NewTicker(time.Duration(HeardBeatX) * DefaultInterval)
|
||
defer heartbeatTicker.Stop()
|
||
|
||
var (
|
||
contentBuilder strings.Builder
|
||
lastUpdate time.Time
|
||
)
|
||
for {
|
||
|
||
select {
|
||
case content, ok := <-cardSend.ContentChannel:
|
||
if !ok {
|
||
// 通道关闭，发送最终内容
|
||
if contentBuilder.Len() > 0 {
|
||
if err := s.updateCardContent(ctx, cardSend, cardInstanceId, contentBuilder.String(), client); err != nil {
|
||
s.logger.Errorf("更新卡片失败1:%s", err.Error())
|
||
}
|
||
}
|
||
return
|
||
}
|
||
contentBuilder.WriteString(content)
|
||
if contentBuilder.Len() > 0 {
|
||
if err := s.updateCardContent(ctx, cardSend, cardInstanceId, contentBuilder.String(), client); err != nil {
|
||
s.logger.Errorf("更新卡片失败2：%s", err.Error())
|
||
}
|
||
}
|
||
lastUpdate = time.Now()
|
||
|
||
case <-heartbeatTicker.C:
|
||
if time.Now().Unix()-lastUpdate.Unix() >= HeardBeatX {
|
||
return
|
||
}
|
||
|
||
case <-ctx.Done():
|
||
s.logger.Info("context canceled, stop channel processing")
|
||
return
|
||
}
|
||
}
|
||
|
||
}
|
||
|
||
// processContentSlice 处理切片内容（同步更新）
|
||
func (s *SendCardClient) processContentSlice(ctx context.Context, cardSend *CardSend, cardInstanceId string, client *dingtalkim_1_0.Client) error {
|
||
var contentBuilder strings.Builder
|
||
for _, content := range cardSend.ContentSlice {
|
||
|
||
contentBuilder.WriteString(content)
|
||
err := s.updateCardRequest(ctx, &UpdateCardRequest{
|
||
Template: string(cardSend.Template),
|
||
Content: contentBuilder.String(),
|
||
Client: client,
|
||
RobotCode: cardSend.RobotCode,
|
||
CardInstanceId: cardInstanceId,
|
||
})
|
||
if err != nil {
|
||
return fmt.Errorf("更新卡片失败: %w", err)
|
||
}
|
||
time.Sleep(cardSend.UpdateInterval) // 控制更新频率
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// updateCardContent 封装卡片更新逻辑
|
||
func (s *SendCardClient) updateCardContent(ctx context.Context, cardSend *CardSend, cardInstanceId, content string, client *dingtalkim_1_0.Client) error {
|
||
err := s.updateCardRequest(ctx, &UpdateCardRequest{
|
||
Template: string(cardSend.Template),
|
||
Content: content,
|
||
Client: client,
|
||
RobotCode: cardSend.RobotCode,
|
||
CardInstanceId: cardInstanceId,
|
||
})
|
||
|
||
return err
|
||
}
|
||
|
||
func (s *SendCardClient) updateCardRequest(ctx context.Context, updateCardRequest *UpdateCardRequest) error {
|
||
content, err := pkg.SafeReplace(updateCardRequest.Template, "%s", updateCardRequest.Content)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
updateRequest := &dingtalkim_1_0.UpdateRobotInteractiveCardRequest{
|
||
CardBizId: tea.String(updateCardRequest.CardInstanceId),
|
||
CardData: tea.String(content),
|
||
}
|
||
_, err = s.UpdateInteractiveCard(ctx, updateRequest, updateCardRequest.RobotCode, updateCardRequest.Client)
|
||
return err
|
||
}
|
||
|
||
// UpdateInteractiveCard 更新交互卡片（封装错误处理）
|
||
func (s *SendCardClient) UpdateInteractiveCard(ctx context.Context, request *dingtalkim_1_0.UpdateRobotInteractiveCardRequest, robotCode string, client *dingtalkim_1_0.Client) (*dingtalkim_1_0.UpdateRobotInteractiveCardResponse, error) {
|
||
authInfo, err := s.Auth.GetTokenFromBotOption(ctx, WithBot(s.botOption))
|
||
if err != nil {
|
||
return nil, fmt.Errorf("get token failed: %w", err)
|
||
}
|
||
|
||
headers := &dingtalkim_1_0.UpdateRobotInteractiveCardHeaders{
|
||
XAcsDingtalkAccessToken: tea.String(authInfo.AccessToken),
|
||
}
|
||
|
||
response, err := client.UpdateRobotInteractiveCardWithOptions(request, headers, &util.RuntimeOptions{})
|
||
if err != nil {
|
||
return nil, fmt.Errorf("API call failed: %w,request:%v", err, request.String())
|
||
}
|
||
return response, nil
|
||
}
|
||
|
||
// SendInteractiveCard 发送交互卡片（封装错误处理）
|
||
func (s *SendCardClient) SendInteractiveCard(ctx context.Context, request *dingtalkim_1_0.SendRobotInteractiveCardRequest, robotCode string, client *dingtalkim_1_0.Client) (res *dingtalkim_1_0.SendRobotInteractiveCardResponse, err error) {
|
||
err = s.Auth.GetBotConfigFromModel(s.botOption)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("初始化bot失败: %w", err)
|
||
}
|
||
authInfo, err := s.Auth.GetTokenFromBotOption(ctx, WithBot(s.botOption))
|
||
if err != nil {
|
||
return nil, fmt.Errorf("get token failed: %w", err)
|
||
}
|
||
|
||
headers := &dingtalkim_1_0.SendRobotInteractiveCardHeaders{
|
||
XAcsDingtalkAccessToken: tea.String(authInfo.AccessToken),
|
||
}
|
||
|
||
response, err := client.SendRobotInteractiveCardWithOptions(request, headers, &util.RuntimeOptions{})
|
||
if err != nil {
|
||
return nil, fmt.Errorf("API call failed: %w", err)
|
||
}
|
||
return response, nil
|
||
}
|