services/export/Task.go - develop - wax_boy/semaphore_custom

services/export/Task.go
101 строка2 KB
Michael Gukov
fix(be): migration tool
05 мар 2026, 19:24
05 мар 2026, 19:24cea5bc8
package export

import (
	"slices"
	"strconv"

	"github.com/semaphoreui/semaphore/db"
)

type TaskExporter struct {
	ValueMap[db.Task]
}

func (e *TaskExporter) load(store db.Store, exporter DataExporter, progress Progress) error {
	projs, err := exporter.getLoadedKeysInt(Project, GlobalScope)
	if err != nil {
		return err
	}

	for _, proj := range projs {
		tasksTmpl, err := store.GetProjectTasks(proj, db.RetrieveQueryParams{})
		if err != nil {
			return err
		}

		tasks := make([]db.Task, len(tasksTmpl))
		for i, task := range tasksTmpl {
			tasks[i] = task.Task
		}

		slices.Reverse(tasks)

		err = e.appendValues(tasks, strconv.Itoa(proj))
		if err != nil {
			return err
		}
	}

	return nil
}
func (e *TaskExporter) restore(store db.Store, exporter DataExporter, progress Progress) (err error) {
	return e.restoreValues(store, exporter, progress, e)
}

func (e *TaskExporter) restoreValue(val EntityObject[db.Task], store db.Store, exporter DataExporter) (err error) {
	old := val.value

	old.ProjectID, err = exporter.getNewKeyInt(Project, GlobalScope, old.ProjectID)
	if err != nil {
		return err
	}

	old.TemplateID, err = exporter.getNewKeyInt(Template, val.scope, old.TemplateID)
	if err != nil {
		return err
	}

	old.InventoryID, err = exporter.getNewKeyIntRef(Inventory, val.scope, old.InventoryID, e)
	if err != nil {
		return err
	}

	old.ScheduleID, err = exporter.getNewKeyIntRef(Schedule, val.scope, old.ScheduleID, e)
	if err != nil {
		return err
	}

	old.UserID, err = exporter.getNewKeyIntRef(User, GlobalScope, old.UserID, e)
	if err != nil {
		return err
	}

	old.IntegrationID, err = exporter.getNewKeyIntRef(Integration, val.scope, old.IntegrationID, e)
	if err != nil {
		return err
	}

	old.BuildTaskID, err = exporter.getNewKeyIntRef(Task, val.scope, old.BuildTaskID, e)
	if err != nil {
		return err
	}

	newObj, err := store.CreateTask(old, 0)
	if err != nil {
		return err
	}

	return exporter.mapKeys(e.getName(), val.scope, old.GetDbKey(), newObj.GetDbKey())
}

func (e *TaskExporter) getName() string {
	return Task
}

func (e *TaskExporter) exportDependsOn() []string {
	return []string{Project}
}

func (e *TaskExporter) importDependsOn() []string {
	return []string{Project, Template, Inventory, Integration, Schedule, User}
}