cloudreve/models/scripts/invoker/invoker.go

package invoker

import (
	"context"
	"fmt"
	"github.com/cloudreve/Cloudreve/v3/pkg/util"
	"strings"
)

type DBScript interface {
	Run(ctx context.Context)
}

var availableScripts = make(map[string]DBScript)

func RunDBScript(name string, ctx context.Context) error {
	if script, ok := availableScripts[name]; ok {
		util.Log().Info("Start executing database script %q.", name)
		script.Run(ctx)
		return nil
	}

	return fmt.Errorf("Database script %q not exist.", name)
}

func Register(name string, script DBScript) {
	availableScripts[name] = script
}

func ListPrefix(prefix string) []string {
	var scripts []string
	for name := range availableScripts {
		if strings.HasPrefix(name, prefix) {
			scripts = append(scripts, name)
		}
	}
	return scripts
}