init push
This commit is contained in:
290
backend/usecase/conversation.go
Normal file
290
backend/usecase/conversation.go
Normal file
@@ -0,0 +1,290 @@
|
||||
package usecase
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"regexp"
|
||||
|
||||
"github.com/samber/lo"
|
||||
|
||||
"github.com/chaitin/panda-wiki/domain"
|
||||
"github.com/chaitin/panda-wiki/log"
|
||||
"github.com/chaitin/panda-wiki/repo/cache"
|
||||
"github.com/chaitin/panda-wiki/repo/ipdb"
|
||||
"github.com/chaitin/panda-wiki/repo/pg"
|
||||
)
|
||||
|
||||
type ConversationUsecase struct {
|
||||
repo *pg.ConversationRepository
|
||||
nodeRepo *pg.NodeRepository
|
||||
geoCacheRepo *cache.GeoRepo
|
||||
logger *log.Logger
|
||||
ipRepo *ipdb.IPAddressRepo
|
||||
authRepo *pg.AuthRepo
|
||||
}
|
||||
|
||||
func NewConversationUsecase(
|
||||
repo *pg.ConversationRepository,
|
||||
nodeRepo *pg.NodeRepository,
|
||||
geoCacheRepo *cache.GeoRepo,
|
||||
logger *log.Logger,
|
||||
ipRepo *ipdb.IPAddressRepo,
|
||||
authRepo *pg.AuthRepo,
|
||||
) *ConversationUsecase {
|
||||
return &ConversationUsecase{
|
||||
repo: repo,
|
||||
nodeRepo: nodeRepo,
|
||||
geoCacheRepo: geoCacheRepo,
|
||||
ipRepo: ipRepo,
|
||||
authRepo: authRepo,
|
||||
logger: logger.WithModule("usecase.conversation"),
|
||||
}
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) CreateChatConversationMessage(ctx context.Context, kbID string, conversation *domain.ConversationMessage) error {
|
||||
references := extractReferencesBlock(conversation.ID, conversation.AppID, conversation.Content)
|
||||
return u.repo.CreateConversationMessage(ctx, conversation, references)
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) GetConversationList(ctx context.Context, request *domain.ConversationListReq) (*domain.PaginatedResult[[]*domain.ConversationListItem], error) {
|
||||
conversations, total, err := u.repo.GetConversationList(ctx, request)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// get feedback info
|
||||
conversationIDs := make([]string, 0, len(conversations))
|
||||
// get all conversation authID
|
||||
authIDs := make([]uint, 0, len(conversations))
|
||||
|
||||
for _, c := range conversations {
|
||||
conversationIDs = append(conversationIDs, c.ID)
|
||||
// 检查 s_id 是否有效，避免查询无效数据
|
||||
if c.Info.UserInfo.AuthUserID != 0 {
|
||||
authIDs = append(authIDs, c.Info.UserInfo.AuthUserID)
|
||||
}
|
||||
}
|
||||
|
||||
// 遍历拿到的c，去数据库里面搜索最新的用户回复
|
||||
feedbackMap, err := u.repo.GetConversationFeedBackInfoByIDs(ctx, conversationIDs)
|
||||
if err != nil {
|
||||
u.logger.Error("get latest feedback by conversation id failed", log.Error(err))
|
||||
}
|
||||
// get user info according authIDs
|
||||
authMap, err := u.authRepo.GetAuthUserinfoByIDs(ctx, authIDs)
|
||||
if err != nil {
|
||||
u.logger.Error("get user info failed", log.Error(err))
|
||||
}
|
||||
|
||||
// get ip address
|
||||
ipAddressMap := make(map[string]*domain.IPAddress)
|
||||
lo.Map(conversations, func(conversation *domain.ConversationListItem, _ int) *domain.ConversationListItem {
|
||||
if _, ok := ipAddressMap[conversation.RemoteIP]; !ok {
|
||||
ipAddress, err := u.ipRepo.GetIPAddress(ctx, conversation.RemoteIP)
|
||||
if err != nil {
|
||||
u.logger.Error("get ip address failed", log.Error(err), log.String("ip", conversation.RemoteIP))
|
||||
return conversation
|
||||
}
|
||||
ipAddressMap[conversation.RemoteIP] = ipAddress
|
||||
conversation.IPAddress = ipAddress
|
||||
} else {
|
||||
conversation.IPAddress = ipAddressMap[conversation.RemoteIP]
|
||||
}
|
||||
if _, ok := feedbackMap[conversation.ID]; ok {
|
||||
conversation.FeedBackInfo = feedbackMap[conversation.ID]
|
||||
}
|
||||
if _, ok := authMap[conversation.Info.UserInfo.AuthUserID]; ok {
|
||||
conversation.Info.UserInfo = domain.UserInfo{
|
||||
NickName: authMap[conversation.Info.UserInfo.AuthUserID].AuthUserInfo.Username,
|
||||
Avatar: authMap[conversation.Info.UserInfo.AuthUserID].AuthUserInfo.AvatarUrl,
|
||||
Email: authMap[conversation.Info.UserInfo.AuthUserID].AuthUserInfo.Email,
|
||||
}
|
||||
}
|
||||
return conversation
|
||||
})
|
||||
return domain.NewPaginatedResult(conversations, total), nil
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) GetConversationDetail(ctx context.Context, kbID, conversationID string) (*domain.ConversationDetailResp, error) {
|
||||
conversation, err := u.repo.GetConversationDetail(ctx, kbID, conversationID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// get ip address
|
||||
ipAddress, err := u.ipRepo.GetIPAddress(ctx, conversation.RemoteIP)
|
||||
if err != nil {
|
||||
u.logger.Error("get ip address failed", log.Error(err), log.String("ip", conversation.RemoteIP))
|
||||
} else {
|
||||
conversation.IPAddress = ipAddress
|
||||
}
|
||||
// get messages
|
||||
messages, err := u.repo.GetConversationMessagesByID(ctx, conversationID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
conversation.Messages = messages
|
||||
// get references
|
||||
references, err := u.repo.GetConversationReferences(ctx, conversationID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
conversation.References = references
|
||||
return conversation, nil
|
||||
}
|
||||
|
||||
func extractReferencesBlock(conversationID, appID, text string) []*domain.ConversationReference {
|
||||
// match whole reference block
|
||||
reBlock := regexp.MustCompile(`(?ms)((?:>|\\u003e)\s*\[\d+\]\.\s*\[.*?\]\(.*?\)\s*\n?)+$`)
|
||||
// find the last match index
|
||||
lastIndex := -1
|
||||
allMatches := reBlock.FindAllStringIndex(text, -1)
|
||||
if len(allMatches) > 0 {
|
||||
lastIndex = allMatches[len(allMatches)-1][0]
|
||||
}
|
||||
|
||||
if lastIndex == -1 {
|
||||
return nil
|
||||
}
|
||||
|
||||
// extract all references in the last reference block
|
||||
block := text[lastIndex:]
|
||||
reLine := regexp.MustCompile(`(?m)^(?:>|\\u003e)\s*\[(\d+)\]\.\s*\[(.*?)\]\((.*?)\)`)
|
||||
matches := reLine.FindAllStringSubmatch(block, -1)
|
||||
|
||||
refs := make([]*domain.ConversationReference, 0)
|
||||
for _, match := range matches {
|
||||
if len(match) == 4 {
|
||||
refs = append(refs, &domain.ConversationReference{
|
||||
Name: match[2],
|
||||
URL: match[3],
|
||||
|
||||
ConversationID: conversationID,
|
||||
AppID: appID,
|
||||
})
|
||||
}
|
||||
}
|
||||
return refs
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) ValidateConversationNonce(ctx context.Context, conversationID, nonce string) error {
|
||||
return u.repo.ValidateConversationNonce(ctx, conversationID, nonce)
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) CreateConversation(ctx context.Context, conversation *domain.Conversation) error {
|
||||
if err := u.repo.CreateConversation(ctx, conversation); err != nil {
|
||||
return err
|
||||
}
|
||||
remoteIP := conversation.RemoteIP
|
||||
ipAddress, err := u.ipRepo.GetIPAddress(ctx, remoteIP)
|
||||
if err != nil {
|
||||
u.logger.Warn("get ip address failed", log.Error(err), log.String("ip", remoteIP), log.String("conversation_id", conversation.ID))
|
||||
} else {
|
||||
location := fmt.Sprintf("%s|%s|%s", ipAddress.Country, ipAddress.Province, ipAddress.City)
|
||||
if err := u.geoCacheRepo.SetGeo(ctx, conversation.KBID, location); err != nil {
|
||||
u.logger.Warn("set geo cache failed", log.Error(err), log.String("conversation_id", conversation.ID), log.String("ip", remoteIP))
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) FeedBack(ctx context.Context, feedback *domain.FeedbackRequest) error {
|
||||
// 先查询数据库，看看目前message的信息
|
||||
messages, err := u.repo.GetConversationMessagesDetailByID(ctx, feedback.MessageId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
u.logger.Debug("feedback info", log.Any("feedback_info", messages.Info))
|
||||
|
||||
// 后端校验一下，只是允许用户进行一次投票
|
||||
if messages.Info.Score == 0 {
|
||||
// 用户可以提供建议
|
||||
if err := u.repo.UpdateMessageFeedback(ctx, feedback); err != nil {
|
||||
return err
|
||||
}
|
||||
} else {
|
||||
return fmt.Errorf("already voted for this message, please do not vote again")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) GetMessageList(ctx context.Context, req *domain.MessageListReq) (*domain.PaginatedResult[[]*domain.ConversationMessageListItem], error) {
|
||||
total, messageList, err := u.repo.GetMessageFeedBackList(ctx, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// get auth userinfo --> auth_user_id is not 0
|
||||
authIDs := make([]uint, 0, len(messageList))
|
||||
for _, message := range messageList {
|
||||
if message.ConversationInfo.UserInfo.AuthUserID != 0 {
|
||||
authIDs = append(authIDs, message.ConversationInfo.UserInfo.AuthUserID)
|
||||
}
|
||||
}
|
||||
// get user info according authIDs
|
||||
authMap, err := u.authRepo.GetAuthUserinfoByIDs(ctx, authIDs)
|
||||
if err != nil {
|
||||
u.logger.Error("get user info failed", log.Error(err))
|
||||
}
|
||||
|
||||
// get ip address
|
||||
ipAddressMap := make(map[string]*domain.IPAddress)
|
||||
lo.Map(messageList, func(message *domain.ConversationMessageListItem, _ int) *domain.ConversationMessageListItem {
|
||||
if _, ok := ipAddressMap[message.RemoteIP]; !ok {
|
||||
ipAddress, err := u.ipRepo.GetIPAddress(ctx, message.RemoteIP)
|
||||
if err != nil {
|
||||
u.logger.Error("get ip address failed", log.Error(err), log.String("ip", message.RemoteIP))
|
||||
return message
|
||||
}
|
||||
ipAddressMap[message.RemoteIP] = ipAddress
|
||||
message.IPAddress = ipAddress
|
||||
} else {
|
||||
message.IPAddress = ipAddressMap[message.RemoteIP]
|
||||
}
|
||||
if _, ok := authMap[message.ConversationInfo.UserInfo.AuthUserID]; ok {
|
||||
message.ConversationInfo.UserInfo = domain.UserInfo{
|
||||
NickName: authMap[message.ConversationInfo.UserInfo.AuthUserID].AuthUserInfo.Username,
|
||||
Avatar: authMap[message.ConversationInfo.UserInfo.AuthUserID].AuthUserInfo.AvatarUrl,
|
||||
Email: authMap[message.ConversationInfo.UserInfo.AuthUserID].AuthUserInfo.Email,
|
||||
}
|
||||
}
|
||||
return message
|
||||
})
|
||||
|
||||
return domain.NewPaginatedResult(messageList, uint64(total)), nil
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) GetMessageDetail(ctx context.Context, kbId, messageId string) (*domain.ConversationMessage, error) {
|
||||
message, err := u.repo.GetConversationMessagesDetailByKbID(ctx, kbId, messageId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return message, nil
|
||||
}
|
||||
|
||||
func (u *ConversationUsecase) GetShareConversationDetail(ctx context.Context, kbID, conversationID string) (*domain.ShareConversationDetailResp, error) {
|
||||
conversation, err := u.repo.GetConversationDetail(ctx, kbID, conversationID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// get messages
|
||||
messages, err := u.repo.GetConversationMessagesByID(ctx, conversationID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var shareMessages []*domain.ShareConversationMessage
|
||||
for _, message := range messages {
|
||||
shareMessages = append(shareMessages, &domain.ShareConversationMessage{
|
||||
Role: message.Role,
|
||||
Content: message.Content,
|
||||
ImagePaths: message.ImagePaths,
|
||||
CreatedAt: message.CreatedAt,
|
||||
})
|
||||
}
|
||||
shareConversationDetail := domain.ShareConversationDetailResp{
|
||||
ID: conversation.ID,
|
||||
Subject: conversation.Subject,
|
||||
CreatedAt: conversation.CreatedAt,
|
||||
|
||||
Messages: shareMessages,
|
||||
}
|
||||
conversation.Messages = messages
|
||||
return &shareConversationDetail, nil
|
||||
}
|
||||
Reference in New Issue
Block a user