EdgeAPI/internal/installers/deploy_manager.go

package installers

import (
	"github.com/iwind/TeaGo/Tea"
	"github.com/iwind/TeaGo/files"
	stringutil "github.com/iwind/TeaGo/utils/string"
	"regexp"
	"sync"
)

var SharedDeployManager = NewDeployManager()

// DeployManager 节点部署文件管理器
// 如果节点部署文件有变化，需要重启API节点以便于生效
type DeployManager struct {
	dir string

	nodeFiles   []*DeployFile
	nsNodeFiles []*DeployFile

	locker sync.Mutex
}

// NewDeployManager 获取新节点部署文件管理器
func NewDeployManager() *DeployManager {
	var manager = &DeployManager{
		dir: Tea.Root + "/deploy",
	}
	manager.LoadNodeFiles()
	manager.LoadNSNodeFiles()
	return manager
}

// LoadNodeFiles 加载所有边缘节点文件
func (this *DeployManager) LoadNodeFiles() []*DeployFile {
	this.locker.Lock()
	defer this.locker.Unlock()

	if len(this.nodeFiles) > 0 {
		return this.nodeFiles
	}

	var keyMap = map[string]*DeployFile{} // key => File

	var reg = regexp.MustCompile(`^edge-node-(\w+)-(\w+)-v([0-9.]+)\.zip$`)
	for _, file := range files.NewFile(this.dir).List() {
		var name = file.Name()
		if !reg.MatchString(name) {
			continue
		}
		var matches = reg.FindStringSubmatch(name)
		var osName = matches[1]
		var arch = matches[2]
		var version = matches[3]

		var key = osName + "_" + arch
		oldFile, ok := keyMap[key]
		if ok && stringutil.VersionCompare(oldFile.Version, version) > 0 {
			continue
		}
		keyMap[key] = &DeployFile{
			OS:      osName,
			Arch:    arch,
			Version: version,
			Path:    file.Path(),
		}
	}

	var result = []*DeployFile{}
	for _, v := range keyMap {
		result = append(result, v)
	}

	this.nodeFiles = result

	return result
}

// FindNodeFile 查找特别平台的节点文件
func (this *DeployManager) FindNodeFile(os string, arch string) *DeployFile {
	for _, file := range this.LoadNodeFiles() {
		if file.OS == os && file.Arch == arch {
			return file
		}
	}
	return nil
}

// LoadNSNodeFiles 加载所有NS节点安装文件
func (this *DeployManager) LoadNSNodeFiles() []*DeployFile {
	this.locker.Lock()
	defer this.locker.Unlock()

	if len(this.nsNodeFiles) > 0 {
		return this.nsNodeFiles
	}

	var keyMap = map[string]*DeployFile{} // key => File

	var reg = regexp.MustCompile(`^edge-dns-(\w+)-(\w+)-v([0-9.]+)\.zip$`)
	for _, file := range files.NewFile(this.dir).List() {
		var name = file.Name()
		if !reg.MatchString(name) {
			continue
		}
		var matches = reg.FindStringSubmatch(name)
		var osName = matches[1]
		var arch = matches[2]
		var version = matches[3]

		var key = osName + "_" + arch
		oldFile, ok := keyMap[key]
		if ok && stringutil.VersionCompare(oldFile.Version, version) > 0 {
			continue
		}
		keyMap[key] = &DeployFile{
			OS:      osName,
			Arch:    arch,
			Version: version,
			Path:    file.Path(),
		}
	}

	var result = []*DeployFile{}
	for _, v := range keyMap {
		result = append(result, v)
	}

	this.nsNodeFiles = result

	return result
}

// FindNSNodeFile 查找特别平台的NS节点安装文件
func (this *DeployManager) FindNSNodeFile(os string, arch string) *DeployFile {
	for _, file := range this.LoadNSNodeFiles() {
		if file.OS == os && file.Arch == arch {
			return file
		}
	}
	return nil
}

// Reload 重置缓存
func (this *DeployManager) Reload() {
	this.locker.Lock()
	defer this.locker.Unlock()

	this.nodeFiles = nil
	this.nsNodeFiles = nil
}