package net

import (
	"fmt"
	"git.noahlan.cn/northlan/nnet/component"
	"github.com/panjf2000/ants/v2"
)

type Handler struct {
	allServices map[string]*component.Service // 所有注册的Service
	allHandlers map[string]*component.Handler // 所有注册的Handler

	workerSize int64           // 业务工作Worker数量
	taskQueue  []chan *Request // 工作协程的消息队列
}

func NewHandler() *Handler {
	return &Handler{
		allServices: make(map[string]*component.Service),
		allHandlers: make(map[string]*component.Handler),
		// TODO 读取配置
		workerSize: 10,
		taskQueue:  make([]chan *Request, 10),
	}
}

func (h *Handler) register(comp component.Component, opts []component.Option) error {
	s := component.NewService(comp, opts)

	p, _ := ants.NewPool()
	p.Submit(func() {

	})

	if _, ok := h.allServices[s.Name]; ok {
		return fmt.Errorf("handler: service already defined: %s", s.Name)
	}

	if err := s.ExtractHandler(); err != nil {
		return err
	}

	h.allServices[s.Name] = s

	// 拷贝一份所有handlers
	for name, handler := range s.Handlers {
		handleName := fmt.Sprintf("%s.%s", s.Name, name)
		// TODO print log
		h.allHandlers[handleName] = handler
	}

	return nil
}

// DoWorker 将工作交给worker处理
func (h *Handler) DoWorker(request *Request) {
	// 根据sessionID平均分配worker处理
	workerId := request.Session.ID() % h.workerSize
	fmt.Printf("sessionID %d to workerID %d\n", request.Session.ID(), workerId)
	// 入队
	h.taskQueue[workerId] <- request
}

func (h *Handler) handle(request *Request) {

}