87 lines
1.7 KiB
Go
87 lines
1.7 KiB
Go
package fns
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/chaitin/panda-wiki/domain"
|
|
"github.com/chaitin/panda-wiki/log"
|
|
)
|
|
|
|
type MigrationCreateFirstNavs struct {
|
|
Name string
|
|
logger *log.Logger
|
|
}
|
|
|
|
func NewMigrationCreateFirstNavs(logger *log.Logger) *MigrationCreateFirstNavs {
|
|
return &MigrationCreateFirstNavs{
|
|
Name: "0005_create_first_navs",
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
func (m *MigrationCreateFirstNavs) Execute(tx *gorm.DB) error {
|
|
|
|
var kbs []*domain.KnowledgeBaseListItem
|
|
if err := tx.Model(&domain.KnowledgeBase{}).
|
|
Order("created_at ASC").
|
|
Find(&kbs).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, kb := range kbs {
|
|
|
|
nav := &domain.Nav{
|
|
ID: uuid.New().String(),
|
|
Name: kb.Name,
|
|
KbID: kb.ID,
|
|
}
|
|
|
|
if err := tx.Model(&domain.Nav{}).Create(nav).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := tx.Model(&domain.Node{}).
|
|
Where("kb_id = ?", kb.ID).
|
|
Update("nav_id", nav.ID).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
var release domain.KBRelease
|
|
err := tx.Model(&domain.KBRelease{}).
|
|
Where("kb_id = ?", kb.ID).
|
|
Order("created_at DESC").
|
|
First(&release).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
|
|
navRelease := &domain.NavRelease{
|
|
ID: uuid.New().String(),
|
|
NavID: nav.ID,
|
|
ReleaseID: release.ID,
|
|
KbID: release.KBID,
|
|
Name: nav.Name,
|
|
Position: nav.Position,
|
|
CreatedAt: time.Now(),
|
|
}
|
|
if err := tx.Model(&domain.NavRelease{}).Create(navRelease).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := tx.Model(&domain.KBReleaseNodeRelease{}).
|
|
Where("kb_id = ? AND release_id = ?", kb.ID, release.ID).
|
|
Update("nav_id", nav.ID).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|