micro/api/resolver/vpath/vpath.go

76 lines
1.5 KiB
Go
Raw Normal View History

2019-06-03 20:44:43 +03:00
// Package vpath resolves using http path and recognised versioned urls
package vpath
import (
"errors"
"net/http"
"regexp"
"strings"
"github.com/micro/go-micro/v2/api/resolver"
2019-06-03 20:44:43 +03:00
)
2020-04-09 12:28:38 +03:00
func NewResolver(opts ...resolver.Option) resolver.Resolver {
return &Resolver{opts: resolver.NewOptions(opts...)}
}
type Resolver struct {
opts resolver.Options
}
2019-06-03 20:44:43 +03:00
var (
re = regexp.MustCompile("^v[0-9]+$")
)
func (r *Resolver) Resolve(req *http.Request, opts ...resolver.ResolveOption) (*resolver.Endpoint, error) {
2019-06-03 20:44:43 +03:00
if req.URL.Path == "/" {
return nil, errors.New("unknown name")
}
options := resolver.NewResolveOptions(opts...)
2020-04-09 12:28:38 +03:00
parts := strings.Split(req.URL.Path[1:], "/")
2019-06-03 20:44:43 +03:00
if len(parts) == 1 {
return &resolver.Endpoint{
Name: r.withPrefix(parts...),
2019-06-03 20:44:43 +03:00
Host: req.Host,
Method: req.Method,
Path: req.URL.Path,
Domain: options.Domain,
2019-06-03 20:44:43 +03:00
}, nil
}
// /v1/foo
if re.MatchString(parts[0]) {
return &resolver.Endpoint{
Name: r.withPrefix(parts[0:2]...),
2019-06-03 20:44:43 +03:00
Host: req.Host,
Method: req.Method,
Path: req.URL.Path,
Domain: options.Domain,
2019-06-03 20:44:43 +03:00
}, nil
}
return &resolver.Endpoint{
Name: r.withPrefix(parts[0]),
2019-06-03 20:44:43 +03:00
Host: req.Host,
Method: req.Method,
Path: req.URL.Path,
Domain: options.Domain,
2019-06-03 20:44:43 +03:00
}, nil
}
func (r *Resolver) String() string {
return "path"
}
// withPrefix transforms "foo" into "go.micro.api.foo"
func (r *Resolver) withPrefix(parts ...string) string {
p := r.opts.ServicePrefix
if len(p) > 0 {
parts = append([]string{p}, parts...)
2020-04-09 12:28:38 +03:00
}
2020-04-09 13:03:33 +03:00
return strings.Join(parts, ".")
2019-06-03 20:44:43 +03:00
}