go asynq异步任务

1.第三方库

https://github.com/hibiken/asynq

准备条件:
需要安装redis
2.client

package main

import (
	"encoding/json"
	"fmt"

	"github.com/hibiken/asynq"
)

func main() {
	client := asynq.NewClient(
		asynq.RedisClientOpt{
			Addr:     "127.0.0.1:6379",
			Password: "***", 
			DB: 0,
		})
    go startUpdate()              // 定时触发
}

//触发任务
func startUpdate() {
	updateTask()

	ticker := time.NewTicker(time.Minute*10)
	defer ticker.Stop()

	for {
		select {
		case <-ticker.C:
			updateTask()
		case <-chExit.Wait():
			return
		}
	}
}

func updateTask(){
	data := map[string]string{
		"aa": "11",
		"bb": "22",
	}

	payload, _ := json.Marshal(data)

	t1 := asynq.NewTask("test99", payload)

	info, err := client.Enqueue(t1, asynq.TaskID("sdf11111"))//添加taskID,如果任务没有被消费,则入队列失败
	if err != nil {
		fmt.Printf("err:%v,info:%v\n", err, info)
	}

	fmt.Printf(" [*] Successfully enqueued task: %+v", info)
}

3.server

package main

import (
	"fmt"
	"encoding/json"

	"github.com/hibiken/asynq"
)

var AsynqServer *asynq.Server // kodo中的异步任务server

func main() {
	initTaskServer()
	mux := asynq.NewServeMux()

	mux.HandleFunc("test99", HandleUpdateTask)
	// ...register other handlers...

	if err := AsynqServer.Run(mux); err != nil {
		fmt.Printf("could not run asynq server: %v", err)
	}
}

func HandleUpdateTask(ctx context.Context, t *asynq.Task) error {
	res := make(map[string]string)

	err := json.Unmarshal(t.Payload(), &res)
	if err != nil {
		fmt.Printf("rum session, can not parse payload: %s,  err: %v", t.Payload(), err)
		return nil
	}
//--------具体处理逻辑------------//
	fmt.Printf("setting:%+#v\n", res)
	return nil
}


func initTaskServer() error {
	// 初始化异步任务服务端
	AsynqServer = asynq.NewServer(
		asynq.RedisClientOpt{
			Addr:     "127.0.0.1:6379",
			Password: "***", //与client对应
			DB:       0,
		},
		asynq.Config{
			// Specify how many concurrent workers to use
			Concurrency: 100,
			// Optionally specify multiple queues with different priority.
			Queues: map[string]int{
				"critical": 6,
				"default":  3,
				"low":      1,
			},
			// See the godoc for other configuration options
		},
	)
	return nil
}

4.UI
可以通过Asynqmon查看任务量和占用内存
在这里插入图片描述

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Machinery 是一个 Go 语言的异步任务队列和作业队列,基于分布式消息传递。类似 Python 的 Celery 框架。Machinery 中的任务(或者作业)可通过多个 worker 在很多服务器上并发的执行,或者可在单个服务器上利用 Go 的协程运行多个 worker 进程。任务的定义:type AddTask struct{} func (t AddTask) Run(args []interface{}) (interface{}, error) {     parsedArgs, err := machinery.ParseNumberArgs(args)     if err != nil {         return nil, err     }     add := func(args []float64) float64 {         sum := 0.0         for _, arg := range args {             sum  = arg         }         return sum     }     return add(parsedArgs), nil } type MultiplyTask struct{} func (t MultiplyTask) Run(args []interface{}) (interface{}, error) {     parsedArgs, err := machinery.ParseNumberArgs(args)     if err != nil {         return nil, err     }     multiply := func(args []float64) float64 {         sum := 1.0         for _, arg := range args {             sum *= arg         }         return sum     }     return multiply(parsedArgs), nil } // ... more tasks任务注册:tasks := map[string]machinery.Task{     "add":      AddTask{},     "multiply": MultiplyTask{}, } app.RegisterTasks(tasks) 标签:GoMachinery
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值