micro/api/resolver/vpath/vpath.go

73 lines
1.4 KiB
Go
Raw Normal View History

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