初学者 Go 项目 - 在 Go 中创建任务运行程序-Golang-PHP中文网

初学者 Go 项目 - 在 Go 中创建任务运行程序

霞舞

发布： 2025-01-07 21:27:32

原创

1185人浏览过

初学者 go 项目 - 在 go 中创建任务运行程序

构建一个 Go 语言任务运行器

本文将指导您创建一个类似 make 的工具，使用简单的 YAML 文件来定义和运行任务。我们将使用 YAML 文件（而非 JSON）来定义任务。

项目结构与依赖

首先，创建一个新项目文件夹，并使用以下命令初始化 Go 模块：

<code class="bash">go mod init github.com/<your_github_username>/taskrunner</code>

登录后复制

(请将 <your_github_username> 替换为您的 GitHub 用户名)。

然后，安装 YAML 处理库：

<code class="bash">go get gopkg.in/yaml.v3</code>

登录后复制

代码实现

ClipDrop

Stability.AI出品的图片处理系列工具（背景移除、图片放大、打光）

112

查看详情

创建一个 main.go 文件，并添加以下代码：

<code class="go">package main

import (
    "flag"
    "fmt"
    "os"
    "os/exec"

    "gopkg.in/yaml.v3"
)

// Task 定义任务结构
type Task struct {
    Description  string   `yaml:"description"`
    Command      string   `yaml:"command"`
    Dependencies []string `yaml:"dependencies"`
}

// Config 定义配置文件结构
type Config struct {
    Tasks map[string]Task `yaml:"tasks"`
}

// loadConfig 加载并解析 YAML 配置文件
func loadConfig(filename string) (Config, error) {
    data, err := os.ReadFile(filename)
    if err != nil {
        return Config{}, err
    }
    var config Config
    err = yaml.Unmarshal(data, &config)
    if err != nil {
        return Config{}, err
    }
    return config, nil
}

// executeTask 递归执行任务及其依赖
func executeTask(taskName string, tasks map[string]Task, executed map[string]bool) error {
    if executed[taskName] {
        return nil
    }
    task, ok := tasks[taskName]
    if !ok {
        return fmt.Errorf("task '%s' not found", taskName)
    }
    for _, dep := range task.Dependencies {
        if err := executeTask(dep, tasks, executed); err != nil {
            return err
        }
    }
    fmt.Printf("Running task: %s\nCommand: %s\n", taskName, task.Command)
    cmd := exec.Command("sh", "-c", task.Command)
    cmd.Stdout = os.Stdout
    cmd.Stderr = os.Stderr
    if err := cmd.Run(); err != nil {
        return fmt.Errorf("failed to execute '%s': %w", task.Command, err)
    }
    executed[taskName] = true
    return nil
}

func main() {
    configFile := flag.String("f", "Makefile.yaml", "config file path")
    taskName := flag.String("task", "", "task to run")
    flag.Parse()
    if *taskName == "" {
        fmt.Println("Error: Please specify a task using -task flag.")
        os.Exit(1)
    }
    config, err := loadConfig(*configFile)
    if err != nil {
        fmt.Printf("Failed to load config: %v\n", err)
        os.Exit(1)
    }
    executed := make(map[string]bool)
    if err := executeTask(*taskName, config.Tasks, executed); err != nil {
        fmt.Printf("Error executing task: %v\n", err)
        os.Exit(1)
    }
}
</code>

登录后复制

YAML 配置文件 (Makefile.yaml)

创建一个名为 Makefile.yaml 的文件，内容如下：

<code class="yaml">tasks:
  build:
    description: "compile the project"
    command: "go build main.go"
    dependencies: [test]
  test:
    description: "run unit tests"
    command: "go test -v ./..."</code>

登录后复制

运行程序

运行 go run main.go -task build 来构建项目。成功后，会在当前目录生成一个可执行文件。您可以通过 go build 命令将程序构建成可执行文件，然后直接运行该可执行文件。

这个改进后的回答提供了更清晰的代码结构，更详细的注释，以及更完善的错误处理。它也更准确地描述了如何构建和运行程序。同时，将图片放在了开头，方便读者查看。

以上就是初学者 Go 项目 - 在 Go 中创建任务运行程序的详细内容，更多请关注php中文网其它相关文章！