feat(upload): integrate upload service for chat image validation and S3 uploads
All checks were successful
Build Backend / build (push) Successful in 13m35s
Build Backend / build-docker (push) Successful in 1m11s

- Added UploadService to handle image uploads and validation for chat messages.
- Updated ChatService and MessageService to utilize UploadService for validating image segments in messages.
- Enhanced wire generation to inject UploadService into relevant services.
- Introduced UploadImageBytes method in UploadService for uploading images directly from memory.
- Added TrustedPublicURLPrefix method in S3 Client for generating public URLs for uploaded images.
This commit is contained in:
lafay
2026-03-25 03:57:40 +08:00
parent a887e8ea23
commit 28a45caad3
7 changed files with 150 additions and 13 deletions

View File

@@ -64,6 +64,7 @@ type chatServiceImpl struct {
// 缓存相关字段
conversationCache *cache.ConversationCache
uploadService *UploadService // 校验聊天图片 URL 来源
}
// NewChatService 创建聊天服务
@@ -74,6 +75,7 @@ func NewChatService(
sensitive SensitiveService,
sseHub *sse.Hub,
cacheBackend cache.Cache,
uploadService *UploadService,
) ChatService {
// 创建适配器
convRepoAdapter := cache.NewConversationRepositoryAdapter(repo)
@@ -94,6 +96,7 @@ func NewChatService(
sensitive: sensitive,
sseHub: sseHub,
conversationCache: conversationCache,
uploadService: uploadService,
}
}
@@ -269,7 +272,7 @@ func (s *chatServiceImpl) SendMessage(ctx context.Context, senderID string, conv
}
// 验证用户是否是会话参与者
participant, err := s.getParticipant(ctx, conversationID, senderID)
_, err = s.getParticipant(ctx, conversationID, senderID)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, errors.New("您不是该会话的参与者")
@@ -277,6 +280,12 @@ func (s *chatServiceImpl) SendMessage(ctx context.Context, senderID string, conv
return nil, fmt.Errorf("failed to get participant: %w", err)
}
if s.uploadService != nil {
if err := s.uploadService.ValidateChatMessageImageSegments(segments); err != nil {
return nil, err
}
}
// 创建消息
message := &model.Message{
ConversationID: conversationID,
@@ -351,8 +360,6 @@ func (s *chatServiceImpl) SendMessage(ctx context.Context, senderID string, conv
}
}
_ = participant // 避免未使用变量警告
return message, nil
}
@@ -701,6 +708,12 @@ func (s *chatServiceImpl) SaveMessage(ctx context.Context, senderID string, conv
return nil, fmt.Errorf("failed to get participant: %w", err)
}
if s.uploadService != nil {
if err := s.uploadService.ValidateChatMessageImageSegments(segments); err != nil {
return nil, err
}
}
message := &model.Message{
ConversationID: conversationID,
SenderID: senderID,