Function useWorker

znet/msghandler.go:162–217

Use worker ID 占用workerID

(conn ziface.IConnection)

// Use worker ID
// 占用workerID
func useWorker(conn ziface.IConnection) uint32 {
	var workerId uint32

	mh, _ := conn.GetMsgHandler().(*MsgHandle)
	if mh == nil {
		zlog.Ins().ErrorF("useWorker failed, mh is nil")
		return 0
	}

	if zconf.GlobalObject.WorkerMode == zconf.WorkerModeBind {
		mh.freeWorkerMu.Lock()
		defer mh.freeWorkerMu.Unlock()

		for k := range mh.freeWorkers {
			delete(mh.freeWorkers, k)
			return k
		}
	}

	if zconf.GlobalObject.WorkerMode == zconf.WorkerModeDynamicBind {
		mh.freeWorkerMu.Lock()
		// try to get workerID from workerPool first
		// 首先尝试从工作线程池里获取一个空闲的workerID
		for workerID := range mh.freeWorkers {
			delete(mh.freeWorkers, workerID)
			mh.freeWorkerMu.Unlock()
			return workerID
		}
		mh.freeWorkerMu.Unlock()

		// 工作池的worker用完了，临时从extraFreeWorkers取一个额外的workerID, 并相应启动一个临时的worker
		mh.extraFreeWorkerMu.Lock()
		defer mh.extraFreeWorkerMu.Unlock()
		for workerID := range mh.extraFreeWorkers {
			zlog.Ins().DebugF("start extra worker, workerID=%d", workerID)
			mh.TaskQueue[workerID] = make(chan ziface.IRequest, zconf.GlobalObject.MaxWorkerTaskLen)
			go mh.StartOneWorker(int(workerID), mh.TaskQueue[workerID])
			return workerID
		}
	}

	//Compatible with the situation where the client has no worker, and solve the situation divide 0
	//(兼容client没有worker情况，解决除0的情况)
	if mh.WorkerPoolSize == 0 {
		workerId = 0
	} else {
		// Assign the worker responsible for processing the current connection based on the ConnID
		// Using a round-robin average allocation rule to get the workerID that needs to process this connection
		// (根据ConnID来分配当前的连接应该由哪个worker负责处理
		// 轮询的平均分配法则
		// 得到需要处理此条连接的workerID)
		workerId = uint32(conn.GetConnID() % uint64(mh.WorkerPoolSize))
	}

	return workerId
}

// Free worker ID