mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-06 10:00:24 +08:00
133 lines
2.9 KiB
Go
133 lines
2.9 KiB
Go
package installers
|
||
|
||
import (
|
||
"github.com/iwind/TeaGo/Tea"
|
||
"github.com/iwind/TeaGo/files"
|
||
stringutil "github.com/iwind/TeaGo/utils/string"
|
||
"regexp"
|
||
)
|
||
|
||
var SharedDeployManager = NewDeployManager()
|
||
|
||
// DeployManager 节点部署文件管理器
|
||
// 如果节点部署文件有变化，需要重启API节点以便于生效
|
||
type DeployManager struct {
|
||
dir string
|
||
nodeFiles []*DeployFile
|
||
nsNodeFiles []*DeployFile
|
||
}
|
||
|
||
// NewDeployManager 获取新节点部署文件管理器
|
||
func NewDeployManager() *DeployManager {
|
||
var manager = &DeployManager{
|
||
dir: Tea.Root + "/deploy",
|
||
}
|
||
manager.LoadNodeFiles()
|
||
manager.LoadNSNodeFiles()
|
||
return manager
|
||
}
|
||
|
||
// LoadNodeFiles 加载所有边缘节点文件
|
||
func (this *DeployManager) LoadNodeFiles() []*DeployFile {
|
||
if len(this.nodeFiles) > 0 {
|
||
return this.nodeFiles
|
||
}
|
||
|
||
keyMap := map[string]*DeployFile{} // key => File
|
||
|
||
reg := regexp.MustCompile(`^edge-node-(\w+)-(\w+)-v([0-9.]+)\.zip$`)
|
||
for _, file := range files.NewFile(this.dir).List() {
|
||
name := file.Name()
|
||
if !reg.MatchString(name) {
|
||
continue
|
||
}
|
||
matches := reg.FindStringSubmatch(name)
|
||
osName := matches[1]
|
||
arch := matches[2]
|
||
version := matches[3]
|
||
|
||
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(),
|
||
}
|
||
}
|
||
|
||
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 {
|
||
if len(this.nsNodeFiles) > 0 {
|
||
return this.nsNodeFiles
|
||
}
|
||
|
||
keyMap := map[string]*DeployFile{} // key => File
|
||
|
||
reg := regexp.MustCompile(`^edge-dns-(\w+)-(\w+)-v([0-9.]+)\.zip$`)
|
||
for _, file := range files.NewFile(this.dir).List() {
|
||
name := file.Name()
|
||
if !reg.MatchString(name) {
|
||
continue
|
||
}
|
||
matches := reg.FindStringSubmatch(name)
|
||
osName := matches[1]
|
||
arch := matches[2]
|
||
version := matches[3]
|
||
|
||
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(),
|
||
}
|
||
}
|
||
|
||
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
|
||
}
|