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

services/export/IntegrationMatcher.go

76 строк2 KB

Michael Gukov

fix(be): migration tool

05 мар 2026, 19:24

05 мар 2026, 19:24cea5bc8

package export

import (
	"strconv"

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

type IntegrationMatcherExporter struct {
	ValueMap[db.IntegrationMatcher]
}

func (e *IntegrationMatcherExporter) load(store db.Store, exporter DataExporter, progress Progress) error {

	projs, err := exporter.getLoadedKeysInt(Project, GlobalScope)
	if err != nil {
		return err
	}

	for _, proj := range projs {

		integrations, err := exporter.getLoadedKeysInt(Integration, strconv.Itoa(proj))
		if err != nil {
			return err
		}

		allValues := make([]db.IntegrationMatcher, 0)

		for _, integration := range integrations {
			vals, err := store.GetIntegrationMatchers(proj, db.RetrieveQueryParams{}, integration)
			if err != nil {
				return err
			}
			allValues = append(allValues, vals...)
		}
		err = e.appendValues(allValues, strconv.Itoa(proj))
		if err != nil {
			return err
		}
	}

	return nil
}

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

func (e *IntegrationMatcherExporter) restoreValue(val EntityObject[db.IntegrationMatcher], store db.Store, exporter DataExporter) (err error) {

	old := val.value

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

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

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

func (e *IntegrationMatcherExporter) getName() string {
	return IntegrationMatcher
}

func (e *IntegrationMatcherExporter) exportDependsOn() []string {
	return []string{Project, Integration}
}

func (e *IntegrationMatcherExporter) importDependsOn() []string {
	return []string{Project, Integration}
}