func (r *RouteRegistry) Register(uri route.Uri, endpoint *route.Endpoint) {
t := time.Now()
data := lager.Data{"uri": uri, "backend": endpoint.CanonicalAddr(), "modification_tag": endpoint.ModificationTag}
r.reporter.CaptureRegistryMessage(endpoint)
r.Lock()
uri = uri.RouteKey()
pool := r.byUri.Find(uri)
if pool == nil {
contextPath := parseContextPath(uri)
pool = route.NewPool(r.dropletStaleThreshold/4, contextPath)
r.byUri.Insert(uri, pool)
r.logger.Debug("uri-added", lager.Data{"uri": uri})
}
endpointAdded := pool.Put(endpoint)
r.timeOfLastUpdate = t
r.Unlock()
if endpointAdded {
r.logger.Debug("endpoint-registered", data)
} else {
r.logger.Debug("endpoint-not-registered", data)
}
}
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
}
// Find returns a *route.Pool that matches exactly the URI parameter, nil if no match was found.
func (r *Trie) Find(uri route.Uri) *route.Pool {
key := strings.TrimPrefix(uri.String(), "/")
node := r
for {
pathParts := parts(key)
SegmentValue := pathParts[0]
matchingChild, ok := node.ChildNodes[SegmentValue]
if !ok {
return nil
}
node = matchingChild
if len(pathParts) <= 1 {
break
}
key = pathParts[1]
}
if nil != node.Pool {
return node.Pool
}
return nil
}
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 parseContextPath(uri route.Uri) string {
contextPath := "/"
split := strings.SplitN(strings.TrimPrefix(uri.String(), "/"), "/", 2)
if len(split) > 1 {
contextPath += split[1]
}
return contextPath
}
func (r *RouteRegistry) LookupWithInstance(uri route.Uri, appId string, appIndex string) *route.Pool {
uri = uri.RouteKey()
p := r.Lookup(uri)
var surgicalPool *route.Pool
p.Each(func(e *route.Endpoint) {
if (e.ApplicationId == appId) && (e.PrivateInstanceIndex == appIndex) {
surgicalPool = route.NewPool(0, "")
surgicalPool.Put(e)
}
})
return surgicalPool
}
func (r *RouteRegistry) Lookup(uri route.Uri) *route.Pool {
started := time.Now()
r.RLock()
uri = uri.RouteKey()
var err error
pool := r.byUri.MatchUri(uri)
for pool == nil && err == nil {
uri, err = uri.NextWildcard()
pool = r.byUri.MatchUri(uri)
}
r.RUnlock()
endLookup := time.Now()
r.reporter.CaptureLookupTime(endLookup.Sub(started))
return pool
}
func (r *RouteRegistry) Unregister(uri route.Uri, endpoint *route.Endpoint) {
data := lager.Data{"uri": uri, "backend": endpoint.CanonicalAddr(), "modification_tag": endpoint.ModificationTag}
r.reporter.CaptureRegistryMessage(endpoint)
r.Lock()
uri = uri.RouteKey()
pool := r.byUri.Find(uri)
if pool != nil {
endpointRemoved := pool.Remove(endpoint)
if endpointRemoved {
r.logger.Debug("endpoint-unregistered", data)
} else {
r.logger.Debug("endpoint-not-unregistered", data)
}
if pool.IsEmpty() {
r.byUri.Delete(uri)
}
}
r.Unlock()
}
package route_test
import (
"code.cloudfoundry.org/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"))