126 lines
4.3 KiB
Go
126 lines
4.3 KiB
Go
|
package script
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"os"
|
|||
|
|
"os/exec"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ExecuteScriptRequest 定义了执行脚本所需的参数和配置
|
|||
|
|
type ExecuteScriptRequest struct {
|
|||
|
|
ScriptPath string
|
|||
|
|
Args []string
|
|||
|
|
Assets []string // 资产目录列表
|
|||
|
|
|
|||
|
|
// 脚本工作目录（默认当前目录）
|
|||
|
|
workDir string
|
|||
|
|
// 脚本执行环境变量（默认空）
|
|||
|
|
envVars map[string]string
|
|||
|
|
// 脚本执行日志文件路径（默认空，表示不记录日志）
|
|||
|
|
logFile string
|
|||
|
|
// 脚本执行超时时间（默认 0，表示不超时）
|
|||
|
|
timeout time.Duration
|
|||
|
|
// 脚本执行的命令元数据（可选），用于日志记录和监控
|
|||
|
|
metadata *CommandMetadata
|
|||
|
|
// 脚本执行结果文件路径（默认空，表示不保存结果）
|
|||
|
|
resultFile string
|
|||
|
|
// 需要收集内容的文件列表
|
|||
|
|
collectFiles []string
|
|||
|
|
|
|||
|
|
// 日志回调函数, 用于实时输出日志（默认 nil，表示不使用回调）
|
|||
|
|
logCallback func(string)
|
|||
|
|
|
|||
|
|
// 进程组管理控制参数, 用于避免脚本执行过程中，产生的子进程无法被正确杀死的问题
|
|||
|
|
// 是否创建新的进程组（默认 true，用于杀死子进程树）
|
|||
|
|
// 设为 false 时，子进程不会被放入新的进程组，不能被组杀
|
|||
|
|
createProcessGroup bool
|
|||
|
|
// 是否自定义 Cancel 函数以杀死进程组（默认 true）
|
|||
|
|
// 设为 false 时，使用默认的进程杀死方式
|
|||
|
|
useProcessGroupKill bool
|
|||
|
|
|
|||
|
|
// 命令执行信息
|
|||
|
|
cmd *exec.Cmd
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SetEnv 设置环境变量, key会被强制转换为大写
|
|||
|
|
func (s *ExecuteScriptRequest) SetEnv(key, value string) {
|
|||
|
|
key = strings.ToUpper(strings.TrimSpace(key))
|
|||
|
|
s.envVars[key] = value
|
|||
|
|
if s.metadata != nil {
|
|||
|
|
s.metadata.EnvVars[key] = value
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// buildEnv 构建环境变量, 将请求中的环境变量与系统环境变量合并，返回一个新的环境变量列表
|
|||
|
|
func (s *ExecuteScriptRequest) buildEnv() []string {
|
|||
|
|
env := os.Environ() // 获取系统环境变量
|
|||
|
|
|
|||
|
|
// 补充自定义环境变量
|
|||
|
|
for key, value := range s.envVars {
|
|||
|
|
env = append(env, fmt.Sprintf("%s=%s", key, value))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return env
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// isDebugScriptEnabled 检查是否启用调试脚本
|
|||
|
|
// 通过环境变量 DEBUG_SCRIPT 控制 (值为 true 或 1 时启用)
|
|||
|
|
func (s *ExecuteScriptRequest) isDebugScriptEnabled() bool {
|
|||
|
|
if val, ok := s.envVars["DEBUG_SCRIPT"]; ok {
|
|||
|
|
return strings.EqualFold(val, "true") || val == "1"
|
|||
|
|
}
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (r *ExecuteScriptRequest) WithWorkspacePrefix(workDirPrefix string) *ExecuteScriptRequest {
|
|||
|
|
if strings.HasPrefix(r.workDir, workDirPrefix) {
|
|||
|
|
return r
|
|||
|
|
}
|
|||
|
|
r.workDir = workDirPrefix + "/" + r.workDir
|
|||
|
|
return r
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CommandMetadata 命令元数据
|
|||
|
|
type CommandMetadata struct {
|
|||
|
|
ID string `json:"id"` // 命令唯一ID
|
|||
|
|
Name string `json:"name"` // 命令名称
|
|||
|
|
Description string `json:"description,omitempty"` // 命令描述
|
|||
|
|
Tags []string `json:"tags,omitempty"` // 标签
|
|||
|
|
CreatedBy string `json:"created_by"` // 创建者
|
|||
|
|
CreatedAt time.Time `json:"created_at"` // 创建时间
|
|||
|
|
Timeout time.Duration `json:"timeout"` // 超时时间
|
|||
|
|
EnvVars map[string]string `json:"env_vars"` // 环境变量
|
|||
|
|
WorkDir string `json:"work_dir"` // 工作目录
|
|||
|
|
RefTask string `json:"ref_task,omitempty"` // 关联的任务
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ExecutionResult 命令执行结果
|
|||
|
|
type ExecutionResult struct {
|
|||
|
|
// 命令
|
|||
|
|
Command string `json:"command"`
|
|||
|
|
// 错误原因
|
|||
|
|
Error string `json:"error,omitempty"`
|
|||
|
|
// 命令退出码
|
|||
|
|
ExitCode int `json:"exit_code"`
|
|||
|
|
// 命令开始执行时间
|
|||
|
|
StartTime time.Time `json:"start_time"`
|
|||
|
|
// 命令结束执行时间
|
|||
|
|
EndTime *time.Time `json:"end_time"`
|
|||
|
|
// 命令执行时长
|
|||
|
|
Duration time.Duration `json:"duration"`
|
|||
|
|
// 命令执行是否成功
|
|||
|
|
Success bool `json:"success"`
|
|||
|
|
// 是否跳过执行（跳过视为成功，但标记为 Skip 以便管道状态同步）
|
|||
|
|
Skipped bool `json:"skipped,omitempty"`
|
|||
|
|
// 非错误的说明信息（比如跳过原因等）
|
|||
|
|
Message string `json:"message,omitempty"`
|
|||
|
|
// 元数据
|
|||
|
|
Metadata *CommandMetadata `json:"metadata"`
|
|||
|
|
// 文件内容集合
|
|||
|
|
FileContents map[string]string `json:"file_contents,omitempty"`
|
|||
|
|
// 脚本输出参数（供下一个任务使用）
|
|||
|
|
OutputParams map[string]string `json:"output_params,omitempty"`
|
|||
|
|
}
|