Go 任务队列示例

本示例创建一个显示 HTML 表单的应用。在对话框中输入一个字符串,然后点击 Add。接着该应用就会计算您以这种方式输入字符串的次数。

该应用执行以下操作:

  • 当您点击 Add 时,表单使用 HTTP POST 请求将字符串发送到在 App Engine 上运行的应用。接着,应用将字符串捆绑到任务中并将其发送到默认队列。
  • 队列将任务转发到内置的任务处理程序,映射到网址 /worker,后者将字符串异步写入数据存储区。
  • 发送 HTTP GET 请求会显示您输入的字符串列表以及 Add 每个字符串的次数,具体操作是键入请求或在下拉框中点击请求。

要将此应用部署到 App Engine,请执行以下操作:

  1. 将以下内容复制到名为 queue.yaml 的文件中。这会将处理任务的速率从默认的每秒 5 次更改为每秒 3 次。

    queue:
    - name: default
      rate: 3/s
    
  2. 在同一目录中,将以下内容复制到您命名的文件(以 .go 结尾)中。这是应用代码,包括任务处理程序。

    
    package counter
    
    import (
    	"html/template"
    	"net/http"
    
    	"google.golang.org/appengine"
    	"google.golang.org/appengine/datastore"
    	"google.golang.org/appengine/log"
    	"google.golang.org/appengine/taskqueue"
    )
    
    func init() {
    	http.HandleFunc("/", handler)
    	http.HandleFunc("/worker", worker)
    }
    
    type Counter struct {
    	Name  string
    	Count int
    }
    
    func handler(w http.ResponseWriter, r *http.Request) {
    	ctx := appengine.NewContext(r)
    	if name := r.FormValue("name"); name != "" {
    		t := taskqueue.NewPOSTTask("/worker", map[string][]string{"name": {name}})
    		if _, err := taskqueue.Add(ctx, t, ""); err != nil {
    			http.Error(w, err.Error(), http.StatusInternalServerError)
    			return
    		}
    	}
    	q := datastore.NewQuery("Counter")
    	var counters []Counter
    	if _, err := q.GetAll(ctx, &counters); err != nil {
    		http.Error(w, err.Error(), http.StatusInternalServerError)
    		return
    	}
    	if err := handlerTemplate.Execute(w, counters); err != nil {
    		http.Error(w, err.Error(), http.StatusInternalServerError)
    		return
    	}
    	// OK
    }
    
    func worker(w http.ResponseWriter, r *http.Request) {
    	ctx := appengine.NewContext(r)
    	name := r.FormValue("name")
    	key := datastore.NewKey(ctx, "Counter", name, 0, nil)
    	var counter Counter
    	if err := datastore.Get(ctx, key, &counter); err == datastore.ErrNoSuchEntity {
    		counter.Name = name
    	} else if err != nil {
    		log.Errorf(ctx, "%v", err)
    		return
    	}
    	counter.Count++
    	if _, err := datastore.Put(ctx, key, &counter); err != nil {
    		log.Errorf(ctx, "%v", err)
    	}
    }
    
    var handlerTemplate = template.Must(template.New("handler").Parse(handlerHTML))
    
    const handlerHTML = `
    {{range .}}
    <p>{{.Name}}: {{.Count}}</p>
    {{end}}
    <p>Start a new counter:</p>
    <form action="/" method="POST">
    <input type="text" name="name">
    <input type="submit" value="Add">
    </form>
    `
    
  3. 在同一目录中,将以下内容复制到名为 app.yaml 的文件中。这会为 App Engine 配置您的应用:

    runtime: go
    api_version: go1
    
    handlers:
    - url: /worker/.*
      script: _go_app
      login: admin
    - url: /.*
      script: _go_app
  4. 确保您已有部署了 App Engine 应用的 Google Cloud Platform 项目,而且您已初始化并配置了该项目的 gcloud 命令。

  5. 使用 gcloud app deploy 命令将应用部署到 App Engine。

  6. 使用 gcloud app browse 命令查看应用的运行情况。