Initial commit: TaskPool React panel

- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
2026-07-26 08:43:52 +08:00
commit e6956aa001
397 changed files with 73621 additions and 0 deletions
@@ -0,0 +1,169 @@
+package bootstrap
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"runtime"
+	"sync"
+	"time"
+
+	"github.com/engigu/taskpool/internal/constant"
+	"github.com/engigu/taskpool/internal/database"
+	"github.com/engigu/taskpool/internal/executor"
+	"github.com/engigu/taskpool/internal/logger"
+	"github.com/engigu/taskpool/internal/router"
+	"github.com/engigu/taskpool/internal/services"
+	"github.com/engigu/taskpool/internal/tunnel"
+	"github.com/engigu/taskpool/internal/utils"
+
+	"github.com/gin-gonic/gin"
+)
+
+type App struct {
+	Config *services.AppConfig
+	Router *gin.Engine
+}
+
+func New() *App {
+	app := InitBasic()
+	app.initRouter()
+	
+	// 初始化完成后将路由引擎注入到隧道模块，以支持高性能的纯内存代理
+	tunnel.SetLocalEngine(app.Router)
+	
+	// 初始化隧道后台服务 (读取配置决定角色并启动服务)
+	tunnel.Init()
+	
+	// 启动系统级后台定时任务调度器
+	executor.InitSysCron()
+
+	// 初始化完成后回收一次内存
+	utils.FreeMemory()
+
+	return app
+}
+
+var (
+	globalApp *App
+	initOnce  sync.Once
+)
+
+func InitBasic() *App {
+	initOnce.Do(func() {
+		app := &App{}
+		utils.InitRuntime()
+		utils.InitSecretKey()
+
+		// 自动加载配置 (内部会自动处理 BH_CONFIG_PATH 环境变量与默认路径的优先级)
+		app.initConfigWithPath("")
+		app.initDatabase()
+		
+		logger.Infof("[System] 低于1.0.11版本升级最新版本错误指引: https://github.com/engigu/taskpool/issues/64")
+		globalApp = app
+	})
+	return globalApp
+}
+
+// InitBasicForCmd 专为命令行工具定制的基础环境初始化入口
+// 内部会调高控制台日志过滤级别以自动静默屏蔽刷屏的底层系统与组件启动 Info 日志
+func InitBasicForCmd() *App {
+	logger.SetLevel("warn")
+	return InitBasic()
+}
+
+// func (a *App) initConfig() {
+// 	a.initConfigWithPath(constant.ConfigPath)
+// }
+
+func (a *App) initConfigWithPath(path string) {
+	cfg, err := services.LoadConfig(path)
+	if err != nil {
+		logger.Fatalf("Failed to load config: %v", err)
+	}
+	a.Config = cfg
+
+	// Ensure directories exist
+	err = os.MkdirAll(constant.DataDir, 0755)
+	if err != nil {
+		return
+	}
+	err = os.MkdirAll(constant.ScriptsWorkDir, 0755)
+	if err != nil {
+		return
+	}
+
+	a.setupTaskPoolBin()
+}
+
+func (a *App) setupTaskPoolBin() {
+	binDir := filepath.Join(constant.DataDir, "bin")
+	_ = os.MkdirAll(binDir, 0755)
+
+	exe, err := os.Executable()
+	if err == nil {
+		// 新命令名
+		linkPath := filepath.Join(binDir, "taskpool")
+		if runtime.GOOS == "windows" {
+			linkPath += ".exe"
+		}
+		os.Remove(linkPath)
+		_ = os.Symlink(exe, linkPath)
+
+		// 兼容旧命令名 baihu
+		legacyPath := filepath.Join(binDir, "baihu")
+		if runtime.GOOS == "windows" {
+			legacyPath += ".exe"
+		}
+		os.Remove(legacyPath)
+		_ = os.Symlink(exe, legacyPath)
+	}
+}
+
+func (a *App) initDatabase() {
+	dbCfg := &database.Config{
+		Type:     a.Config.Database.Type,
+		Host:     a.Config.Database.Host,
+		Port:     a.Config.Database.Port,
+		User:     a.Config.Database.User,
+		Password: a.Config.Database.Password,
+		DBName:   a.Config.Database.DBName,
+		Path:     a.Config.Database.Path,
+		DSN:      a.Config.Database.DSN,
+		SSLMode:  a.Config.Database.SSLMode,
+	}
+
+	if err := database.Init(dbCfg); err != nil {
+		logger.Fatalf("Failed to init database: %v", err)
+	}
+
+	// 记录各个初始化阶段的时间
+	startTime := time.Now()
+
+	// 执行 V3 迁移（ID 变更迁移）
+	if err := services.RunMigrationV3(); err != nil {
+		logger.Fatalf("Failed to run V3 migration: %v", err)
+	}
+	v3Duration := time.Since(startTime)
+	logger.Infof("[Database] V3 迁移检查完成, 耗时: %v", v3Duration)
+
+	// 执行表结构同步
+	migrateStart := time.Now()
+	if err := database.Migrate(); err != nil {
+		logger.Fatalf("Failed to migrate database: %v", err)
+	}
+	migrateDuration := time.Since(migrateStart)
+	logger.Infof("[Database] 表结构同步完成, 耗时: %v", migrateDuration)
+	logger.Infof("[Database] 数据库总初始化耗时: %v", time.Since(startTime))
+}
+
+func (a *App) initRouter() {
+	ctrls := router.RegisterControllers()
+	a.Router = router.Setup(ctrls)
+}
+
+func (a *App) Run() {
+	addr := fmt.Sprintf("%s:%d", a.Config.Server.Host, a.Config.Server.Port)
+	logger.Infof("Starting server on %s", addr)
+	a.Router.Run(addr)
+}
@@ -0,0 +1,58 @@
+package bootstrap
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/engigu/taskpool/internal/services"
+)
+
+// SendInternalRequest 向常驻后台主服务安全发送内部通信请求
+// relPath 传入相对内部接口路径 (如: "/internal/tasks/execute/xxx")，方法内部会自动补充完整的协议、端口及 "/api/v1" 前缀，
+// 并自动获取 security.secret 密钥种入 X-Internal-Token 头部。
+func SendInternalRequest(method, relPath string, payload interface{}) ([]byte, int, error) {
+	appCfg := services.GetConfig()
+	if appCfg == nil {
+		return nil, 0, fmt.Errorf("加载系统配置失败")
+	}
+
+	relPath = strings.TrimPrefix(relPath, "/")
+	url := fmt.Sprintf("http://127.0.0.1:%d/api/v1/%s", appCfg.Server.Port, relPath)
+
+	var bodyReader io.Reader
+	if payload != nil {
+		jsonData, err := json.Marshal(payload)
+		if err != nil {
+			return nil, 0, fmt.Errorf("序列化请求负载失败: %v", err)
+		}
+		bodyReader = bytes.NewBuffer(jsonData)
+	}
+
+	settings := services.NewSettingsService()
+	secret := settings.Get("security", "secret")
+
+	req, err := http.NewRequest(method, url, bodyReader)
+	if err != nil {
+		return nil, 0, fmt.Errorf("创建 HTTP 请求失败: %v", err)
+	}
+
+	if payload != nil {
+		req.Header.Set("Content-Type", "application/json")
+	}
+	req.Header.Set("X-Internal-Token", secret)
+
+	client := &http.Client{Timeout: 5 * time.Second}
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, 0, fmt.Errorf("网络连接失败，请确保任务池常驻后台服务正在运行中: %v", err)
+	}
+	defer resp.Body.Close()
+
+	bodyBytes, err := io.ReadAll(resp.Body)
+	return bodyBytes, resp.StatusCode, err
+}