micro-server-http/subscriber.go

211 lines
4.5 KiB
Go
Raw Permalink Normal View History

package http
import (
2018-12-19 10:33:23 +03:00
"bytes"
"context"
"fmt"
"reflect"
"strings"
"go.unistack.org/micro/v3/broker"
"go.unistack.org/micro/v3/codec"
"go.unistack.org/micro/v3/metadata"
"go.unistack.org/micro/v3/options"
"go.unistack.org/micro/v3/register"
"go.unistack.org/micro/v3/server"
)
2018-12-19 10:33:23 +03:00
var typeOfError = reflect.TypeOf((*error)(nil)).Elem()
type handler struct {
reqType reflect.Type
ctxType reflect.Type
method reflect.Value
2018-12-19 10:33:23 +03:00
}
2018-12-19 12:47:03 +03:00
type httpSubscriber struct {
2018-12-19 10:33:23 +03:00
topic string
rcvr reflect.Value
typ reflect.Type
subscriber interface{}
handlers []*handler
endpoints []*register.Endpoint
2018-12-19 10:33:23 +03:00
opts server.SubscriberOptions
}
func newSubscriber(topic string, sub interface{}, opts ...server.SubscriberOption) server.Subscriber {
options := server.NewSubscriberOptions(opts...)
2018-12-19 10:33:23 +03:00
var endpoints []*register.Endpoint
2018-12-19 10:33:23 +03:00
var handlers []*handler
if typ := reflect.TypeOf(sub); typ.Kind() == reflect.Func {
h := &handler{
method: reflect.ValueOf(sub),
}
switch typ.NumIn() {
case 1:
h.reqType = typ.In(0)
case 2:
h.ctxType = typ.In(0)
h.reqType = typ.In(1)
}
handlers = append(handlers, h)
ep := &register.Endpoint{
Name: "Func",
Request: register.ExtractSubValue(typ),
Metadata: metadata.New(2),
}
ep.Metadata.Set("topic", topic)
ep.Metadata.Set("subscriber", "true")
endpoints = append(endpoints, ep)
2018-12-19 10:33:23 +03:00
} else {
hdlr := reflect.ValueOf(sub)
name := reflect.Indirect(hdlr).Type().Name()
for m := 0; m < typ.NumMethod(); m++ {
method := typ.Method(m)
h := &handler{
method: method.Func,
}
switch method.Type.NumIn() {
case 2:
h.reqType = method.Type.In(1)
case 3:
h.ctxType = method.Type.In(1)
h.reqType = method.Type.In(2)
}
handlers = append(handlers, h)
ep := &register.Endpoint{
Name: name + "." + method.Name,
Request: register.ExtractSubValue(method.Type),
Metadata: metadata.New(2),
}
ep.Metadata.Set("topic", topic)
ep.Metadata.Set("subscriber", "true")
endpoints = append(endpoints, ep)
2018-12-19 10:33:23 +03:00
}
}
2018-12-19 12:47:03 +03:00
return &httpSubscriber{
2018-12-19 10:33:23 +03:00
rcvr: reflect.ValueOf(sub),
typ: reflect.TypeOf(sub),
topic: topic,
subscriber: sub,
handlers: handlers,
endpoints: endpoints,
opts: options,
}
}
func (s *Server) createSubHandler(sb *httpSubscriber, opts server.Options) broker.Handler {
2019-07-07 14:46:31 +03:00
return func(p broker.Event) error {
2018-12-19 10:33:23 +03:00
msg := p.Message()
ct := msg.Header["Content-Type"]
cf, err := s.newCodec(ct)
if err != nil {
return err
}
hdr := metadata.Copy(msg.Header)
ctx := metadata.NewIncomingContext(context.Background(), hdr)
2018-12-19 10:33:23 +03:00
results := make(chan error, len(sb.handlers))
for i := 0; i < len(sb.handlers); i++ {
handler := sb.handlers[i]
var isVal bool
var req reflect.Value
if handler.reqType.Kind() == reflect.Ptr {
req = reflect.New(handler.reqType.Elem())
} else {
req = reflect.New(handler.reqType)
isVal = true
}
if isVal {
req = req.Elem()
}
buf := bytes.NewBuffer(msg.Body)
2018-12-19 10:33:23 +03:00
if err := cf.ReadHeader(buf, &codec.Message{}, codec.Event); err != nil {
2018-12-19 10:33:23 +03:00
return err
}
if err := cf.ReadBody(buf, req.Interface()); err != nil {
2018-12-19 10:33:23 +03:00
return err
}
fn := func(ctx context.Context, msg server.Message) error {
var vals []reflect.Value
if sb.typ.Kind() != reflect.Func {
vals = append(vals, sb.rcvr)
}
if handler.ctxType != nil {
vals = append(vals, reflect.ValueOf(ctx))
}
vals = append(vals, reflect.ValueOf(msg.Body()))
2018-12-19 10:33:23 +03:00
returnValues := handler.method.Call(vals)
if err := returnValues[0].Interface(); err != nil {
return err.(error)
}
return nil
}
opts.Hooks.EachNext(func(hook options.Hook) {
if h, ok := hook.(server.HookSubHandler); ok {
fn = h(fn)
}
})
2018-12-19 10:33:23 +03:00
go func() {
2018-12-19 12:47:03 +03:00
results <- fn(ctx, &httpMessage{
2018-12-19 10:33:23 +03:00
topic: sb.topic,
contentType: ct,
payload: req.Interface(),
header: msg.Header,
codec: cf,
2018-12-19 10:33:23 +03:00
})
}()
}
var errors []string
for i := 0; i < len(sb.handlers); i++ {
if err := <-results; err != nil {
errors = append(errors, err.Error())
}
}
if len(errors) > 0 {
return fmt.Errorf("subscriber error: %s", strings.Join(errors, "\n"))
}
return nil
}
}
2018-12-19 12:47:03 +03:00
func (s *httpSubscriber) Topic() string {
2018-12-19 10:33:23 +03:00
return s.topic
}
2018-12-19 12:47:03 +03:00
func (s *httpSubscriber) Subscriber() interface{} {
2018-12-19 10:33:23 +03:00
return s.subscriber
}
func (s *httpSubscriber) Endpoints() []*register.Endpoint {
2018-12-19 10:33:23 +03:00
return s.endpoints
}
2018-12-19 12:47:03 +03:00
func (s *httpSubscriber) Options() server.SubscriberOptions {
2018-12-19 10:33:23 +03:00
return s.opts
}