nnet/component/service.go

package component

import (
	"fmt"
	"reflect"
)

type (
	// Handler 消息处理器，当前仅支持单个自定义参数
	Handler struct {
		Receiver reflect.Value  // 方法接收者
		Method   reflect.Method // 方法存根
		Type     reflect.Type   // 方法参数类型
		IsRawArg bool           // 数据是否需要被序列化，true代表不需要
	}

	// Service 服务，绑定消息处理器用以处理发生的消息
	Service struct {
		Name     string              // 服务名称
		Type     reflect.Type        // 接收者类型
		Receiver reflect.Value       // 该服务下所有方法的接收者
		Handlers map[string]*Handler // 该服务下属的所有方法
		Options  options             // options
	}
)

func NewService(comp Component, opts []Option) *Service {
	s := &Service{
		Type:     reflect.TypeOf(comp),
		Receiver: reflect.ValueOf(comp),
	}

	// apply options
	for _, opt := range opts {
		opt(&s.Options)
	}

	if name := s.Options.serviceName; name != "" {
		s.Name = name
	} else {
		s.Name = reflect.Indirect(s.Receiver).Type().Name()
	}
	return s
}

// suitableHandlerMethods 反射装填指定type-service的所有认定为handlerMethod的方法
func (s *Service) suitableHandlerMethods(typ reflect.Type) map[string]*Handler {
	methods := make(map[string]*Handler)
	for m := 0; m < typ.NumMethod(); m++ {
		method := typ.Method(m)
		mt := method.Type
		mn := method.Name
		if isHandlerMethod(method) {
			raw := false
			if mt.In(2) == typeOfBytes {
				raw = true
			}
			// rewrite handler name
			if s.Options.methodNameFunc != nil {
				mn = s.Options.methodNameFunc(mn)
			}
			methods[mn] = &Handler{Method: method, Type: mt.In(2), IsRawArg: raw}
		}
	}
	return methods
}

// ExtractHandler 反射提取满足以下条件的方法
// - 两个显示入参
// - 第一个是 *net.request
// - 另一个是 []byte 或者 任意指针类型 pointer
func (s *Service) ExtractHandler() error {
	typeName := reflect.Indirect(s.Receiver).Type().Name()
	if typeName == "" {
		return fmt.Errorf("no service name for type %s", s.Type.String())
	}
	if !isExported(typeName) {
		return fmt.Errorf("type %s is not exported", typeName)
	}

	// suitable handlers
	s.Handlers = s.suitableHandlerMethods(s.Type)

	if len(s.Handlers) == 0 {
		method := s.suitableHandlerMethods(reflect.PtrTo(s.Type))
		if len(method) == 0 {
			return fmt.Errorf("type %s has no exported methods of suitable type (hint: pass a pointer to value of that type)", s.Name)
		} else {
			return fmt.Errorf("type %s has no exported methods of suitable type")
		}
	}

	for _, handler := range s.Handlers {
		handler.Receiver = s.Receiver
	}

	return nil
}