totole/services
1
0
Fork 0
mirror of https://github.com/kevin-DL/services.git synced 2026-01-13 11:35:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update threads to use model/store (#97)

Browse Source
This commit is contained in:
Asim Aslam
2021-05-11 12:41:53 +01:00
committed by GitHub
parent f74cfdf674
commit 7e13403d9b
31 changed files with 1309 additions and 1143 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
threads/handler/create_message.go
View File

@@ -2,17 +2,16 @@ package handler
import (
"context"
"strings"
"github.com/google/uuid"
"github.com/micro/micro/v3/service/auth"
"github.com/micro/micro/v3/service/errors"
"github.com/micro/micro/v3/service/logger"
"github.com/micro/services/pkg/model"
pb "github.com/micro/services/threads/proto"
"gorm.io/gorm"
)
// Create a message within a conversation
// Create a message within a thread
func (s *Threads) CreateMessage(ctx context.Context, req *pb.CreateMessageRequest, rsp *pb.CreateMessageResponse) error {
_, ok := auth.AccountFromContext(ctx)
if !ok {
@@ -22,52 +21,54 @@ func (s *Threads) CreateMessage(ctx context.Context, req *pb.CreateMessageReques
if len(req.AuthorId) == 0 {
return ErrMissingAuthorID
}
if len(req.ConversationId) == 0 {
return ErrMissingConversationID
if len(req.ThreadId) == 0 {
return ErrMissingThreadID
}
if len(req.Text) == 0 {
return ErrMissingText
}
db, err := s.GetDBConn(ctx)
if err != nil {
logger.Errorf("Error connecting to DB: %v", err)
return errors.InternalServerError("DB_ERROR", "Error connecting to DB")
}
// lookup the conversation
var conv Conversation
if err := db.Where(&Conversation{ID: req.ConversationId}).First(&conv).Error; err == gorm.ErrRecordNotFound {
// lookup the thread
conv := Thread{ID: req.ThreadId}
if err := model.Read(ctx, &conv); err == model.ErrNotFound {
return ErrNotFound
} else if err != nil {
logger.Errorf("Error reading conversation: %v", err)
logger.Errorf("Error reading thread: %v", err)
return errors.InternalServerError("DATABASE_ERROR", "Error connecting to database")
}
// create the message
msg := &Message{
ID: req.Id,
SentAt: s.Time(),
Text: req.Text,
AuthorID: req.AuthorId,
ConversationID: req.ConversationId,
ID: req.Id,
SentAt: s.Time(),
Text: req.Text,
AuthorID: req.AuthorId,
ThreadID: req.ThreadId,
}
if len(msg.ID) == 0 {
msg.ID = uuid.New().String()
}
if err := db.Create(msg).Error; err == nil {
if err := model.Create(ctx, msg); err == nil {
rsp.Message = msg.Serialize()
return nil
} else if !strings.Contains(err.Error(), "messages_pkey") {
} else if err != model.ErrAlreadyExists {
logger.Errorf("Error creating message: %v", err)
return errors.InternalServerError("DATABASE_ERROR", "Error connecting to database")
}
// a message already exists with this id
var existing Message
if err := db.Where(&Message{ID: msg.ID}).First(&existing).Error; err != nil {
existing := &Message{ID: msg.ID, ThreadID: req.ThreadId}
if err := model.Read(ctx, existing); err == model.ErrNotFound {
return ErrNotFound
} else if err != nil {
logger.Errorf("Error creating message: %v", err)
return errors.InternalServerError("DATABASE_ERROR", "Error connecting to database")
}
// return the message
rsp.Message = existing.Serialize()
return nil
}