71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
// Package vpath resolves using http path and recognised versioned urls
|
|
package vpath
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/micro/go-micro/v2/api/resolver"
|
|
)
|
|
|
|
func NewResolver(opts ...resolver.Option) resolver.Resolver {
|
|
return &Resolver{opts: resolver.NewOptions(opts...)}
|
|
}
|
|
|
|
type Resolver struct {
|
|
opts resolver.Options
|
|
}
|
|
|
|
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")
|
|
}
|
|
|
|
fmt.Println(req.URL.Path)
|
|
|
|
parts := strings.Split(req.URL.Path[1:], "/")
|
|
if len(parts) == 1 {
|
|
return &resolver.Endpoint{
|
|
Name: addNamespace(r.opts.Namespace, parts...),
|
|
Host: req.Host,
|
|
Method: req.Method,
|
|
Path: req.URL.Path,
|
|
}, nil
|
|
}
|
|
|
|
// /v1/foo
|
|
if re.MatchString(parts[0]) {
|
|
return &resolver.Endpoint{
|
|
Name: addNamespace(r.opts.Namespace, parts[0:2]...),
|
|
Host: req.Host,
|
|
Method: req.Method,
|
|
Path: req.URL.Path,
|
|
}, nil
|
|
}
|
|
|
|
return &resolver.Endpoint{
|
|
Name: addNamespace(r.opts.Namespace, parts[0]),
|
|
Host: req.Host,
|
|
Method: req.Method,
|
|
Path: req.URL.Path,
|
|
}, nil
|
|
}
|
|
|
|
func (r *Resolver) String() string {
|
|
return "path"
|
|
}
|
|
|
|
func addNamespace(ns string, parts ...string) string {
|
|
if len(ns) == 0 {
|
|
return strings.Join(parts, ".")
|
|
}
|
|
return strings.Join(append([]string{ns}, parts...), ".")
|
|
}
|