chore: import upstream snapshot with attribution
govulncheck / govulncheck (push) Has been cancelled
Lint / golangci-lint (push) Has been cancelled
Run Tests / Unit Tests (push) Has been cancelled
Run Tests / Etcd Integration Tests (push) Has been cancelled
Harness (E2E) / Harnesses (mock LLM) (push) Has been cancelled
Harness (E2E) / Provider harnesses (live LLM conformance) (push) Has been cancelled
govulncheck / govulncheck (push) Has been cancelled
Lint / golangci-lint (push) Has been cancelled
Run Tests / Unit Tests (push) Has been cancelled
Run Tests / Etcd Integration Tests (push) Has been cancelled
Harness (E2E) / Harnesses (mock LLM) (push) Has been cancelled
Harness (E2E) / Provider harnesses (live LLM conformance) (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,83 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
|
||||
"go-micro.dev/v6/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
|
||||
}
|
||||
Reference in New Issue
Block a user