49 lines
1.6 KiB
Go
49 lines
1.6 KiB
Go
package repository
|
|
|
|
import (
|
|
"infinite-canvas/backend/internal/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func (r *Repository) AgentProfile(userID, scope, projectID, canvasID string) (*model.AgentProfile, error) {
|
|
var profile model.AgentProfile
|
|
if err := r.db.Where("user_id = ? AND scope = ? AND project_id = ? AND canvas_id = ?", userID, scope, projectID, canvasID).First(&profile).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &profile, nil
|
|
}
|
|
|
|
func (r *Repository) SaveAgentProfile(profile *model.AgentProfile, expectedRevision int64) error {
|
|
if profile == nil {
|
|
return gorm.ErrInvalidData
|
|
}
|
|
return r.db.Transaction(func(tx *gorm.DB) error {
|
|
var current model.AgentProfile
|
|
err := tx.Where("user_id = ? AND scope = ? AND project_id = ? AND canvas_id = ?", profile.UserID, profile.Scope, profile.ProjectID, profile.CanvasID).First(¤t).Error
|
|
if err != nil && err != gorm.ErrRecordNotFound {
|
|
return err
|
|
}
|
|
if err == gorm.ErrRecordNotFound {
|
|
if expectedRevision != 0 {
|
|
return ErrTaskStateConflict
|
|
}
|
|
return tx.Create(profile).Error
|
|
}
|
|
if expectedRevision != current.Revision {
|
|
return ErrTaskStateConflict
|
|
}
|
|
profile.ID, profile.Revision, profile.CreatedAt = current.ID, current.Revision+1, current.CreatedAt
|
|
result := tx.Model(&model.AgentProfile{}).Where("id = ? AND revision = ?", current.ID, current.Revision).Updates(map[string]any{
|
|
"content": profile.Content, "revision": profile.Revision, "hash": profile.Hash, "updated_at": profile.UpdatedAt,
|
|
})
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
if result.RowsAffected != 1 {
|
|
return ErrTaskStateConflict
|
|
}
|
|
return nil
|
|
})
|
|
}
|