mirror of
https://github.com/go-micro/go-micro.git
synced 2024-12-12 08:23:58 +02:00
02839cfba5
* api/handler: use http.MaxBytesReader protect api handlers from OOM cases Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
102 lines
2.0 KiB
Go
102 lines
2.0 KiB
Go
// Package cloudevents provides a cloudevents handler publishing the event using the go-micro/client
|
|
package cloudevents
|
|
|
|
import (
|
|
"net/http"
|
|
"path"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/micro/go-micro/v2/api/handler"
|
|
"github.com/micro/go-micro/v2/util/ctx"
|
|
)
|
|
|
|
type event struct {
|
|
opts handler.Options
|
|
}
|
|
|
|
var (
|
|
Handler = "cloudevents"
|
|
versionRe = regexp.MustCompilePOSIX("^v[0-9]+$")
|
|
)
|
|
|
|
func eventName(parts []string) string {
|
|
return strings.Join(parts, ".")
|
|
}
|
|
|
|
func evRoute(ns, p string) (string, string) {
|
|
p = path.Clean(p)
|
|
p = strings.TrimPrefix(p, "/")
|
|
|
|
if len(p) == 0 {
|
|
return ns, "event"
|
|
}
|
|
|
|
parts := strings.Split(p, "/")
|
|
|
|
// no path
|
|
if len(parts) == 0 {
|
|
// topic: namespace
|
|
// action: event
|
|
return strings.Trim(ns, "."), "event"
|
|
}
|
|
|
|
// Treat /v[0-9]+ as versioning
|
|
// /v1/foo/bar => topic: v1.foo action: bar
|
|
if len(parts) >= 2 && versionRe.Match([]byte(parts[0])) {
|
|
topic := ns + "." + strings.Join(parts[:2], ".")
|
|
action := eventName(parts[1:])
|
|
return topic, action
|
|
}
|
|
|
|
// /foo => topic: ns.foo action: foo
|
|
// /foo/bar => topic: ns.foo action: bar
|
|
topic := ns + "." + strings.Join(parts[:1], ".")
|
|
action := eventName(parts[1:])
|
|
|
|
return topic, action
|
|
}
|
|
|
|
func (e *event) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
bsize := handler.DefaultMaxRecvSize
|
|
if e.opts.MaxRecvSize > 0 {
|
|
bsize = e.opts.MaxRecvSize
|
|
}
|
|
|
|
r.Body = http.MaxBytesReader(w, r.Body, bsize)
|
|
|
|
// request to topic:event
|
|
// create event
|
|
// publish to topic
|
|
topic, _ := evRoute(e.opts.Namespace, r.URL.Path)
|
|
|
|
// create event
|
|
ev, err := FromRequest(r)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
return
|
|
}
|
|
|
|
// get client
|
|
c := e.opts.Service.Client()
|
|
|
|
// create publication
|
|
p := c.NewMessage(topic, ev)
|
|
|
|
// publish event
|
|
if err := c.Publish(ctx.FromRequest(r), p); err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
return
|
|
}
|
|
}
|
|
|
|
func (e *event) String() string {
|
|
return "cloudevents"
|
|
}
|
|
|
|
func NewHandler(opts ...handler.Option) handler.Handler {
|
|
return &event{
|
|
opts: handler.NewOptions(opts...),
|
|
}
|
|
}
|