132 lines
2.9 KiB
Go
132 lines
2.9 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"go.unistack.org/micro/v3/server"
|
|
)
|
|
|
|
// SetError pass error to caller
|
|
func SetError(err interface{}) error {
|
|
return &Error{err: err}
|
|
}
|
|
|
|
// GetError return underline error
|
|
func GetError(err interface{}) interface{} {
|
|
if verr, ok := err.(*Error); ok {
|
|
return verr.err
|
|
}
|
|
return err
|
|
}
|
|
|
|
// Error struct holds error
|
|
type Error struct {
|
|
err interface{}
|
|
}
|
|
|
|
// Error func for error interface
|
|
func (err *Error) Error() string {
|
|
return fmt.Sprintf("%v", err.err)
|
|
}
|
|
|
|
type (
|
|
rspCodeKey struct{}
|
|
rspCodeVal struct {
|
|
code int
|
|
}
|
|
)
|
|
|
|
// SetRspCode saves response code in context, must be used by handler to specify http code
|
|
func SetRspCode(ctx context.Context, code int) {
|
|
if rsp, ok := ctx.Value(rspCodeKey{}).(*rspCodeVal); ok {
|
|
rsp.code = code
|
|
}
|
|
}
|
|
|
|
// GetRspCode used internally by generated http server handler
|
|
func GetRspCode(ctx context.Context) int {
|
|
var code int
|
|
if rsp, ok := ctx.Value(rspCodeKey{}).(*rspCodeVal); ok {
|
|
code = rsp.code
|
|
}
|
|
return code
|
|
}
|
|
|
|
type middlewareKey struct{}
|
|
|
|
// Middleware passes http middlewares
|
|
func Middleware(mw ...func(http.Handler) http.Handler) server.Option {
|
|
return server.SetOption(middlewareKey{}, mw)
|
|
}
|
|
|
|
type serverKey struct{}
|
|
|
|
// Server provide ability to pass *http.Server
|
|
func Server(hs *http.Server) server.Option {
|
|
return server.SetOption(serverKey{}, hs)
|
|
}
|
|
|
|
type errorHandlerKey struct{}
|
|
|
|
// ErrorHandler specifies handler for errors
|
|
func ErrorHandler(fn func(ctx context.Context, s server.Handler, w http.ResponseWriter, r *http.Request, err error, status int)) server.Option {
|
|
return server.SetOption(errorHandlerKey{}, fn)
|
|
}
|
|
|
|
type (
|
|
pathHandlerKey struct{}
|
|
pathHandlerVal struct {
|
|
h map[string]map[string]http.HandlerFunc
|
|
}
|
|
)
|
|
|
|
// PathHandler specifies http handler for path regexp
|
|
func PathHandler(method, path string, handler http.HandlerFunc) server.Option {
|
|
return func(o *server.Options) {
|
|
if o.Context == nil {
|
|
o.Context = context.Background()
|
|
}
|
|
v, ok := o.Context.Value(pathHandlerKey{}).(*pathHandlerVal)
|
|
if !ok {
|
|
v = &pathHandlerVal{h: make(map[string]map[string]http.HandlerFunc)}
|
|
}
|
|
m, ok := v.h[method]
|
|
if !ok {
|
|
m = make(map[string]http.HandlerFunc)
|
|
v.h[method] = m
|
|
}
|
|
m[path] = handler
|
|
o.Context = context.WithValue(o.Context, pathHandlerKey{}, v)
|
|
}
|
|
}
|
|
|
|
type registerRPCHandlerKey struct{}
|
|
|
|
// RegisterRPCHandler registers compatibility endpoints with /ServiceName.ServiceEndpoint method POST
|
|
func RegisterRPCHandler(b bool) server.Option {
|
|
return server.SetOption(registerRPCHandlerKey{}, b)
|
|
}
|
|
|
|
type headerKey struct{}
|
|
|
|
type handlerOptions struct {
|
|
headers []string
|
|
cookies []string
|
|
}
|
|
|
|
type FillRequestOption func(*handlerOptions)
|
|
|
|
func Header(headers ...string) FillRequestOption {
|
|
return func(o *handlerOptions) {
|
|
o.headers = append(o.headers, headers...)
|
|
}
|
|
}
|
|
|
|
func Cookie(cookies ...string) FillRequestOption {
|
|
return func(o *handlerOptions) {
|
|
o.cookies = append(o.cookies, cookies...)
|
|
}
|
|
}
|