golang队列任务处理

2020-09-12 17:34:04 阅读：193 来源： 互联网

标签：task taskNum 队列 chan 工厂 golang 任务 time id

package main

import (
	"crypto/rand"
	"fmt"
	"math/big"
	"strconv"
	"strings"
	"time"
)

type task struct {
	id       uint32   // 订单编号
	callback chan int // 工厂编号
}

const taskNum = 16                          // 准备找16个工厂
var chTaskList = make([]chan task, taskNum) // 工厂列表

var arr = [10]int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}

func task_init() {
	for i := 0; i < taskNum; i++ {
		chTaskList[i] = make(chan task)            // 工厂登记
		go working(chTaskList[i], i) // 工厂开门大吉，待命中
	}
}

func main() {
	task_init() // 工厂就绪
	time.Sleep(1 * time.Second)

	for i := 0; i < 100; i++ {
		go order() // 给工厂派活
	}
	time.Sleep(5 * time.Second)
}

// 出任务的，甲方爸爸
func order() {
	id := randomString(10)                       // 来个订单号
	callack := make(chan int)                    // 把返程的招呼打好
	t := task{id: uint32(id), callback: callack} // 订单起草完毕

	chTasks := chTaskList[id%taskNum] // 随机选择一个工厂,因为id都是随机生成的，取余数也是一个随机值
	chTasks <- t                      // 把订单交给小弟让他送工厂去

	msg := <-callack
	fmt.Printf("给%v号工厂下了订单，%v号工厂拿到了订单\n", id%taskNum, msg)
}

// 工厂，接收小弟(chan)，返回工厂编号
func working(chTasks chan task, num int) {
	fmt.Printf("第%v号工厂待命中...\n", num)
	for {
		t := <-chTasks    // 卸货
		id := t.id        // 订单编号
		_ = id            // 此处生产产品（业务逻辑）
		t.callback <- num // 返回工厂编号
	}
}

// 生成订单号
func randomString(n int) uint32 {
	var b []string
	for i := 0; i < n; i++ {
		num, _ := rand.Int(rand.Reader, big.NewInt(int64(len(arr))))
		// 1.将 *big.Int 转化为 int64
		// 2.将 int 64 转化为 int
		// 3.将 int 转化为 string
		b = append(b, strconv.Itoa(arr[int(num.Int64())]))
	}

	str := strings.Join(b, "") // 拼接字符串
	a, _ := strconv.Atoi(str)  // string 转 int
	yes := uint32(a)           // int 转 uint32
	return yes
}

标签：task,taskNum,队列,chan,工厂,golang,任务,time,id
来源： https://www.cnblogs.com/longzhankunlun/p/13657757.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

golang队列任务处理