DESKTOP-72TV0V4\caoxiaozhu
2a9326ef5f
- model/skill.go: Status字段从 int 改为 string，支持 "active"/"inactive" - handler/skill_handler.go: 适配Status字段的类型变化，处理"1"/"0"和"active"/"inactive"两种格式 - repository/skill_repo.go: 更新Status字段的空值判断逻辑 - service/skill_service.go: 同步更新 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"x-agents/server/internal/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type SkillRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewSkillRepository(db *gorm.DB) *SkillRepository {
|
|
return &SkillRepository{db: db}
|
|
}
|
|
|
|
// DB 获取数据库连接
|
|
func (r *SkillRepository) DB() *gorm.DB {
|
|
return r.db
|
|
}
|
|
|
|
func (r *SkillRepository) Create(skill *model.Skill) error {
|
|
return r.db.Create(skill).Error
|
|
}
|
|
|
|
func (r *SkillRepository) FindAll() ([]model.Skill, error) {
|
|
var skills []model.Skill
|
|
err := r.db.Order("skill_type, skill_name").Find(&skills).Error
|
|
return skills, err
|
|
}
|
|
|
|
func (r *SkillRepository) FindByID(id string) (*model.Skill, error) {
|
|
var skill model.Skill
|
|
err := r.db.First(&skill, "id = ?", id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &skill, nil
|
|
}
|
|
|
|
func (r *SkillRepository) FindByType(skillType string) ([]model.Skill, error) {
|
|
var skills []model.Skill
|
|
err := r.db.Where("skill_type = ?", skillType).Order("skill_name").Find(&skills).Error
|
|
return skills, err
|
|
}
|
|
|
|
func (r *SkillRepository) FindByName(skillName string) (*model.Skill, error) {
|
|
var skill model.Skill
|
|
err := r.db.First(&skill, "skill_name = ?", skillName).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &skill, nil
|
|
}
|
|
|
|
// Update 更新技能（只更新传入的字段）
|
|
func (r *SkillRepository) Update(skill *model.Skill) error {
|
|
updates := make(map[string]interface{})
|
|
if skill.SkillName != "" {
|
|
updates["skill_name"] = skill.SkillName
|
|
}
|
|
if skill.SkillDesc != "" {
|
|
updates["skill_desc"] = skill.SkillDesc
|
|
}
|
|
if skill.SkillType != "" {
|
|
updates["skill_type"] = skill.SkillType
|
|
}
|
|
if skill.Status != "" {
|
|
updates["status"] = skill.Status
|
|
}
|
|
if skill.Path != "" {
|
|
updates["path"] = skill.Path
|
|
}
|
|
return r.db.Model(&model.Skill{}).Where("id = ?", skill.ID).Updates(updates).Error
|
|
}
|
|
|
|
func (r *SkillRepository) Delete(id string) error {
|
|
return r.db.Delete(&model.Skill{}, "id = ?", id).Error
|
|
}
|
|
|
|
// UpsertBatch 批量upsert skills
|
|
func (r *SkillRepository) UpsertBatch(skills []model.Skill) error {
|
|
for _, skill := range skills {
|
|
var existing model.Skill
|
|
err := r.db.First(&existing, "skill_name = ? AND skill_type = ?", skill.SkillName, skill.SkillType).Error
|
|
if err == gorm.ErrRecordNotFound {
|
|
if err := r.db.Create(&skill).Error; err != nil {
|
|
return err
|
|
}
|
|
} else if err != nil {
|
|
return err
|
|
} else {
|
|
existing.SkillDesc = skill.SkillDesc
|
|
existing.Path = skill.Path
|
|
existing.Status = skill.Status
|
|
if err := r.db.Save(&existing).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DeleteByType 根据类型删除
|
|
func (r *SkillRepository) DeleteByType(skillType string) error {
|
|
return r.db.Where("skill_type = ?", skillType).Delete(&model.Skill{}).Error
|
|
}
|