基于Golang 高并发问题的解决方案

//==================================
//  * Name：Jerry
//  * Tel：18017448610
//  * DateTime：2019/2/24 14:02
//==================================
package main
import (
	"github.com/lunny/log"
	"runtime"
	"sync"
	"time"
)
//工厂模型
type Factory struct {
	Wg        *sync.WaitGroup //任务监控系统
	MaxWorker int             //最大机器数
	MaxJobs   int             //最大工作数量
	JobQueue  chan int        //工作队列管道
	Quit      chan bool       //是否关闭机器
}
//创建工厂模型
func NewFactory(maxWorker int, wg *sync.WaitGroup) Factory {
	return Factory{
		Wg:        wg,                        //引用任务监控系统
		MaxWorker: maxWorker,                 //机器数量（数量多少，根据服务器性能而定）
		JobQueue:  make(chan int, maxWorker), //工作管道，数量大于等于机器数
		Quit:      make(chan bool),
	}
}
//设置最大订单数量
func (f *Factory) SetMaxJobs(taskNum int) {
	f.MaxJobs = taskNum
}
//开始上班
func (f *Factory) Start() {
	//机器开机，MaxWorker
	for i := 0; i < f.MaxWorker; i++ {
		//每一台机器开启后，去工作吧
		go func() {
			//等待下发命令
			for {
				select {
				case i := <-f.JobQueue:
					//接到工作，开工！
					f.doWork(i)
				case <-f.Quit:
					log.Println("机器关机")
					return
				}
			}
		}()
	}
}
//分配每个任务到管道中
func (f *Factory) AddTask(taskNum int) {
	//系统监控任务 +1
	f.Wg.Add(1)
	//分配任务到管道中
	f.JobQueue <- taskNum
}
//模拟耗时工作
func (f *Factory) doWork(taskNum int) {
	//生产产品的工作
	time.Sleep(200 * time.Millisecond)
	//完成工作报告
	f.Wg.Done()
	//log.Println("完工：", taskNum)
}
//创建工厂
func Begin() {
	//配置工作核数
	gomaxprocs := runtime.GOMAXPROCS(runtime.NumCPU())
	log.Println("核数：", gomaxprocs)
	//配置监控系统
	wg := new(sync.WaitGroup)
	//开工厂
	factory := NewFactory(1000, wg)
	//订单量
	factory.SetMaxJobs(10000)
	//开始上班
	factory.Start()
	log.Println("开始生产")
	//讲所有的订单，添加到任务队列
	for i := 0; i < factory.MaxJobs; i++ {
		factory.AddTask(i)
	}
	factory.Wg.Wait()
	log.Println("所有订单任务生产完成")
}

func Benchmark_Begin(b *testing.B) {
 Begin()
}

//任务的请求
type MtaskRequest struct {
    Ceshi int
    // [redacted]
}
 
//job队列+work池
var (
    MaxWorker = os.Getenv("MAX_WORKERS")
    MaxQueue  = os.Getenv("MAX_QUEUE")
)
 
// Job represents the job to be run
type Job struct {
    MtaskRequest MtaskRequest
}
 
// A buffered channel that we can send work requests on.
 
// var JobQueue chan Job ---这样申明会卡主，没有初始化
var JobQueue = make(chan Job)
 
// Worker represents the worker that executes the job
type Worker struct {
    WorkerPool chan chan Job
    JobChannel chan Job
    quit       chan bool
}
 
func NewWorker(workerPool chan chan Job) Worker {
    return Worker{
        WorkerPool: workerPool,
        JobChannel: make(chan Job),
        quit:       make(chan bool)}
}
 
// Stop signals the worker to stop listening for work requests.
func (w Worker) Stop() {
    go func() {
        w.quit <- true
    }()
}
 
type Dispatcher struct {
    // A pool of workers channels that are registered with the dispatcher
    WorkerPool chan chan Job
    maxWorkers int
}
 
func NewDispatcher(maxWorkers int) *Dispatcher {
    pool := make(chan chan Job, maxWorkers)
    return &Dispatcher{WorkerPool: pool, maxWorkers: maxWorkers}
}
 
// Start method starts the run loop for the worker, listening for a quit channel in
// case we need to stop it
func (w Worker) Start() {
    go func() {
        for {
            // register the current worker into the worker queue.
            w.WorkerPool <- w.JobChannel
            select {
            case <-w.JobChannel:
                time.Sleep(5 * time.Second)
                // we have received a work request.
                fmt.Println("调起worker")
            case <-w.quit:
                // we have received a signal to stop
                return
                //不能写default
            }
        }
    }()
}
 
func (d *Dispatcher) Run() {
    //启动一定数量的worker
    fmt.Println("启动一定数量的worker")
    for i := 0; i < d.maxWorkers; i++ {
        worker := NewWorker(d.WorkerPool)
        worker.Start()
    }
 
    go d.dispatch()
}
 
//分派任务
func (d *Dispatcher) dispatch() {
    for {
        select {
        case job := <-JobQueue: //接收一个job请求
            fmt.Println("JobQueue 收到请求")
 
            go func(job Job) {
                // try to obtain a worker job channel that is available.
                // this will block until a worker is idle
                jobChannel := <-d.WorkerPool
                // dispatch the job to the worker job channel
                jobChannel <- job
            }(job)
        }
    }
}
 
//接收到红包数据
func (this *TaskRedbao) UserGetRedbao(red_id, uid, shop_id, rand_arr, Amoney string) error {
    fmt.Println("收到 接收到红包数据 http请求")
    mtaskRequest := MtaskRequest{67}
    work := Job{MtaskRequest: mtaskRequest}
 
    JobQueue <- work
    return nil
}