Google App Engine Go：精确控制延迟任务在指定模块的执行

package myapp

import (
    "context"
    "log"
    "net/http" // 引入http包，因为appengine.Header是http.Header类型

    "google.golang.org/appengine"
    "google.golang.org/appengine/delay"
    "google.golang.org/appengine/taskqueue"
)

// myDelayFunc 是一个示例延迟函数，它将被延迟执行。
// 参数：ctx 是App Engine上下文，param 是传递给延迟函数的参数。
var myDelayFunc = delay.Func("myDelayFunc", func(ctx context.Context, param string) {
    // 在延迟函数中，可以通过appengine.ModuleName(ctx)获取当前执行模块的名称
    log.Printf("Delayed function 'myDelayFunc' executed on module: %s with param: %s", appengine.ModuleName(ctx), param)
})

// AddDelayedTaskToModule 演示如何将延迟任务添加到指定模块。
// ctx: 当前请求上下文
// targetModule: 目标模块的名称（例如 "server"）
// param: 传递给延迟函数的参数
func AddDelayedTaskToModule(ctx context.Context, targetModule, param string) error {
    // 1. 创建延迟任务实例
    // myDelayFunc.Task(param) 会根据延迟函数签名创建一个taskqueue.Task
    t := myDelayFunc.Task(param)

    // 2. 初始化任务头部
    // 确保t.Header是一个非nil的http.Header map，以便可以安全地设置头部。
    if t.Header == nil {
        t.Header = make(http.Header) // http.Header 是 map[string][]string 的别名
    }

    // 3. 获取目标模块主机名
    // appengine.ModuleHostname(ctx, module, version, instance)
    // module: 目标模块名称 (例如 "server")
    // version: 目标模块版本 (空字符串表示当前默认版本)
    // instance: 目标模块实例 (空字符串表示任意实例)
    hostName, err := appengine.ModuleHostname(ctx, targetModule, "", "")
    if err != nil {
        log.Printf("Failed to get hostname for module %s: %v", targetModule, err)
        return err
    }

    // 4. 设置Host头部，确保任务路由到指定模块
    // 任务队列系统会根据此Host头部将任务请求发送到正确的主机。
    t.Header.Set("Host", hostName)

    // 5. 添加任务到队列
    // 将配置好的任务添加到App Engine的任务队列中。
    _, err = taskqueue.Add(ctx, t)
    if err != nil {
        log.Printf("Failed to add delayed task to queue: %v", err)
        return err
    }

    log.Printf("Successfully added delayed task 'myDelayFunc' to module '%s' with host '%s'", targetModule, hostName)
    return nil
}

/*
// 示例用法（在某个HTTP处理器中调用）
// 假设您的default模块有一个HTTP处理函数，需要向名为 "server" 的模块发送延迟任务。
func MyHTTPHandler(w http.ResponseWriter, r *http.Request) {
    ctx := appengine.NewContext(r)
    // 假设我们想将任务发送到名为 "server" 的模块
    err := AddDelayedTaskToModule(ctx, "server", "some_important_data")
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    fmt.Fprintf(w, "Delayed task scheduled for 'server' module successfully.")
}
*/