go-micro/server/rpc_handler.go

package server

import (
	"reflect"

	"go-micro.dev/v5/registry"
)

type RpcHandler struct {
	handler   interface{}
	opts      HandlerOptions
	name      string
	endpoints []*registry.Endpoint
}

func NewRpcHandler(handler interface{}, opts ...HandlerOption) Handler {
	options := HandlerOptions{
		Metadata: make(map[string]map[string]string),
	}

	for _, o := range opts {
		o(&options)
	}

	typ := reflect.TypeOf(handler)
	hdlr := reflect.ValueOf(handler)
	name := reflect.Indirect(hdlr).Type().Name()

	// Auto-extract documentation from Go doc comments
	autoMetadata := extractHandlerDocs(handler)

	// Merge auto-extracted metadata with manually provided metadata
	// Manual metadata takes precedence over auto-extracted
	for endpoint, meta := range autoMetadata {
		fullName := name + "." + endpoint
		if options.Metadata[fullName] == nil {
			options.Metadata[fullName] = make(map[string]string)
		}
		// Only add auto-extracted values if not manually provided
		for k, v := range meta {
			if _, exists := options.Metadata[fullName][k]; !exists {
				options.Metadata[fullName][k] = v
			}
		}
	}

	var endpoints []*registry.Endpoint

	for m := 0; m < typ.NumMethod(); m++ {
		if e := extractEndpoint(typ.Method(m)); e != nil {
			e.Name = name + "." + e.Name

			for k, v := range options.Metadata[e.Name] {
				e.Metadata[k] = v
			}

			endpoints = append(endpoints, e)
		}
	}

	return &RpcHandler{
		name:      name,
		handler:   handler,
		endpoints: endpoints,
		opts:      options,
	}
}

func (r *RpcHandler) Name() string {
	return r.name
}

func (r *RpcHandler) Handler() interface{} {
	return r.handler
}

func (r *RpcHandler) Endpoints() []*registry.Endpoint {
	return r.endpoints
}

func (r *RpcHandler) Options() HandlerOptions {
	return r.opts
}