feat: add AI workflow service foundation

internal/repositories/ai_workflow_node_run_repository.go

@@ -0,0 +1,37 @@
+package repositories
+
+import (
+	"agent-desk/internal/models"
+
+	"github.com/mlogclub/simple/sqls"
+	"gorm.io/gorm"
+)
+
+var AIWorkflowNodeRunRepository = newAIWorkflowNodeRunRepository()
+
+func newAIWorkflowNodeRunRepository() *aiWorkflowNodeRunRepository {
+	return &aiWorkflowNodeRunRepository{}
+}
+
+type aiWorkflowNodeRunRepository struct{}
+
+func (r *aiWorkflowNodeRunRepository) Get(db *gorm.DB, id int64) *models.AIWorkflowNodeRun {
+	ret := &models.AIWorkflowNodeRun{}
+	if err := db.First(ret, "id = ?", id).Error; err != nil {
+		return nil
+	}
+	return ret
+}
+
+func (r *aiWorkflowNodeRunRepository) Find(db *gorm.DB, cnd *sqls.Cnd) (list []models.AIWorkflowNodeRun) {
+	cnd.Find(db, &list)
+	return
+}
+
+func (r *aiWorkflowNodeRunRepository) Create(db *gorm.DB, t *models.AIWorkflowNodeRun) error {
+	return db.Create(t).Error
+}
+
+func (r *aiWorkflowNodeRunRepository) Updates(db *gorm.DB, id int64, columns map[string]interface{}) error {
+	return db.Model(&models.AIWorkflowNodeRun{}).Where("id = ?", id).Updates(columns).Error
+}

internal/repositories/ai_workflow_repository.go

@@ -0,0 +1,61 @@
+package repositories
+
+import (
+	"agent-desk/internal/models"
+	"agent-desk/internal/pkg/httpx/params"
+
+	"github.com/mlogclub/simple/sqls"
+	"gorm.io/gorm"
+)
+
+var AIWorkflowRepository = newAIWorkflowRepository()
+
+func newAIWorkflowRepository() *aiWorkflowRepository {
+	return &aiWorkflowRepository{}
+}
+
+type aiWorkflowRepository struct{}
+
+func (r *aiWorkflowRepository) Get(db *gorm.DB, id int64) *models.AIWorkflow {
+	ret := &models.AIWorkflow{}
+	if err := db.First(ret, "id = ?", id).Error; err != nil {
+		return nil
+	}
+	return ret
+}
+
+func (r *aiWorkflowRepository) Take(db *gorm.DB, where ...interface{}) *models.AIWorkflow {
+	ret := &models.AIWorkflow{}
+	if err := db.Take(ret, where...).Error; err != nil {
+		return nil
+	}
+	return ret
+}
+
+func (r *aiWorkflowRepository) Find(db *gorm.DB, cnd *sqls.Cnd) (list []models.AIWorkflow) {
+	cnd.Find(db, &list)
+	return
+}
+
+func (r *aiWorkflowRepository) FindPageByParams(db *gorm.DB, params *params.QueryParams) (list []models.AIWorkflow, paging *sqls.Paging) {
+	return r.FindPageByCnd(db, &params.Cnd)
+}
+
+func (r *aiWorkflowRepository) FindPageByCnd(db *gorm.DB, cnd *sqls.Cnd) (list []models.AIWorkflow, paging *sqls.Paging) {
+	cnd.Find(db, &list)
+	count := cnd.Count(db, &models.AIWorkflow{})
+	paging = &sqls.Paging{
+		Page:  cnd.Paging.Page,
+		Limit: cnd.Paging.Limit,
+		Total: count,
+	}
+	return
+}
+
+func (r *aiWorkflowRepository) Create(db *gorm.DB, t *models.AIWorkflow) error {
+	return db.Create(t).Error
+}
+
+func (r *aiWorkflowRepository) Updates(db *gorm.DB, id int64, columns map[string]interface{}) error {
+	return db.Model(&models.AIWorkflow{}).Where("id = ?", id).Updates(columns).Error
+}

internal/repositories/ai_workflow_run_repository.go

@@ -0,0 +1,37 @@
+package repositories
+
+import (
+	"agent-desk/internal/models"
+
+	"github.com/mlogclub/simple/sqls"
+	"gorm.io/gorm"
+)
+
+var AIWorkflowRunRepository = newAIWorkflowRunRepository()
+
+func newAIWorkflowRunRepository() *aiWorkflowRunRepository {
+	return &aiWorkflowRunRepository{}
+}
+
+type aiWorkflowRunRepository struct{}
+
+func (r *aiWorkflowRunRepository) Get(db *gorm.DB, id int64) *models.AIWorkflowRun {
+	ret := &models.AIWorkflowRun{}
+	if err := db.First(ret, "id = ?", id).Error; err != nil {
+		return nil
+	}
+	return ret
+}
+
+func (r *aiWorkflowRunRepository) Find(db *gorm.DB, cnd *sqls.Cnd) (list []models.AIWorkflowRun) {
+	cnd.Find(db, &list)
+	return
+}
+
+func (r *aiWorkflowRunRepository) Create(db *gorm.DB, t *models.AIWorkflowRun) error {
+	return db.Create(t).Error
+}
+
+func (r *aiWorkflowRunRepository) Updates(db *gorm.DB, id int64, columns map[string]interface{}) error {
+	return db.Model(&models.AIWorkflowRun{}).Where("id = ?", id).Updates(columns).Error
+}

internal/repositories/ai_workflow_version_repository.go

@@ -0,0 +1,58 @@
+package repositories
+
+import (
+	"agent-desk/internal/models"
+	"agent-desk/internal/pkg/httpx/params"
+
+	"github.com/mlogclub/simple/sqls"
+	"gorm.io/gorm"
+)
+
+var AIWorkflowVersionRepository = newAIWorkflowVersionRepository()
+
+func newAIWorkflowVersionRepository() *aiWorkflowVersionRepository {
+	return &aiWorkflowVersionRepository{}
+}
+
+type aiWorkflowVersionRepository struct{}
+
+func (r *aiWorkflowVersionRepository) Get(db *gorm.DB, id int64) *models.AIWorkflowVersion {
+	ret := &models.AIWorkflowVersion{}
+	if err := db.First(ret, "id = ?", id).Error; err != nil {
+		return nil
+	}
+	return ret
+}
+
+func (r *aiWorkflowVersionRepository) Find(db *gorm.DB, cnd *sqls.Cnd) (list []models.AIWorkflowVersion) {
+	cnd.Find(db, &list)
+	return
+}
+
+func (r *aiWorkflowVersionRepository) FindPageByParams(db *gorm.DB, params *params.QueryParams) (list []models.AIWorkflowVersion, paging *sqls.Paging) {
+	return r.FindPageByCnd(db, &params.Cnd)
+}
+
+func (r *aiWorkflowVersionRepository) FindPageByCnd(db *gorm.DB, cnd *sqls.Cnd) (list []models.AIWorkflowVersion, paging *sqls.Paging) {
+	cnd.Find(db, &list)
+	count := cnd.Count(db, &models.AIWorkflowVersion{})
+	paging = &sqls.Paging{
+		Page:  cnd.Paging.Page,
+		Limit: cnd.Paging.Limit,
+		Total: count,
+	}
+	return
+}
+
+func (r *aiWorkflowVersionRepository) Create(db *gorm.DB, t *models.AIWorkflowVersion) error {
+	return db.Create(t).Error
+}
+
+func (r *aiWorkflowVersionRepository) MaxVersionByWorkflowID(db *gorm.DB, workflowID int64) int {
+	var maxVersion int
+	db.Model(&models.AIWorkflowVersion{}).
+		Where("workflow_id = ?", workflowID).
+		Select("COALESCE(MAX(version), 0)").
+		Scan(&maxVersion)
+	return maxVersion
+}