func (r *Trie) Insert(uri route.Uri, value *route.Pool) *Trie {
key := strings.TrimPrefix(uri.String(), "/")
node := r
for {
pathParts := parts(key)
SegmentValue := pathParts[0]
matchingChild, ok := node.ChildNodes[SegmentValue]
if !ok {
matchingChild = NewTrie()
matchingChild.Segment = SegmentValue
matchingChild.Parent = node
node.ChildNodes[SegmentValue] = matchingChild
}
node = matchingChild
if len(pathParts) != 2 {
break
}
key = pathParts[1]
}
node.Pool = value
return node
}
func (r *Trie) Delete(uri route.Uri) bool {
key := strings.TrimPrefix(uri.String(), "/")
node := r
initialKey := key
for {
pathParts := parts(key)
SegmentValue := pathParts[0]
// It is currently impossible to Delete a non-existent path. This invariant is
// provided by the fact that a call to Find is done before Delete in the registry.
matchingChild, _ := node.ChildNodes[SegmentValue]
node = matchingChild
if len(pathParts) <= 1 {
break
}
key = pathParts[1]
}
node.Pool = nil
r.deleteEmptyNodes(initialKey)
return true
}
func (r *Trie) Find(uri route.Uri) (*route.Pool, bool) {
key := strings.TrimPrefix(uri.String(), "/")
node := r
for {
pathParts := parts(key)
SegmentValue := pathParts[0]
matchingChild, ok := node.ChildNodes[SegmentValue]
if !ok {
return nil, false
}
node = matchingChild
if len(pathParts) <= 1 {
break
}
key = pathParts[1]
}
if nil != node.Pool {
return node.Pool, true
}
return nil, false
}
func parseContextPath(uri route.Uri) string {
contextPath := "/"
split := strings.SplitN(strings.TrimPrefix(uri.String(), "/"), "/", 2)
if len(split) > 1 {
contextPath += split[1]
}
return contextPath
}
import (
"github.com/cloudfoundry/gorouter/route"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("URIs", func() {
Context("RouteKey", func() {
var key route.Uri
It("creates a route key based on uri", func() {
key = route.Uri("dora.app.com").RouteKey()
Expect(key.String()).To(Equal("dora.app.com"))
key = route.Uri("dora.app.com/").RouteKey()
Expect(key.String()).To(Equal("dora.app.com"))
key = route.Uri("dora.app.com/v1").RouteKey()
Expect(key.String()).To(Equal("dora.app.com/v1"))
})
Context("has a context path", func() {
It("creates route key with context path", func() {
key = route.Uri("dora.app.com/v1").RouteKey()
Expect(key.String()).To(Equal("dora.app.com/v1"))