コード例 #1
0
func (m *Matcher) Match(path string) (string, string, error) {

	// TODO:
	// match with regex

	// The list of path prefixes is in reverse order by string length. We want
	// to return the first (most specific) match we come accross.
	for _, prefix := range m.prefixesList {
		if strings.HasPrefix(path, prefix) {
			return m.prefixes[prefix], prefix, nil
		}
	}

	datadog.Count("no_matching_prefix_error", 1, nil, 1.0)
	return "", "", noMatchingPrefixError
}
コード例 #2
0
// NewKubernetesRouter gives you a router instance.
func NewKubernetesRouter(config *Config) (*http.Server, error) {

	log.Infoln("Domain suffixes:", config.DomainSuffixes())
	log.Infoln("Kubernetes service domain suffix:", config.KubernetesServiceDomainSuffix())

	// Create a new director object.
	dir := director.NewDirector()

	// Check for a routes JSON file.
	if config.ValidateRoutes || config.RoutesFilename != "" {

		routesFile, err := os.Open(config.RoutesFilename)
		if err != nil {
			return nil, err
		}

		routesJSON, err := ioutil.ReadAll(routesFile)
		if err != nil {
			return nil, err
		}

		if err := routesFile.Close(); err != nil {
			return nil, err
		}

		var routes map[string]map[string]string
		if err := json.Unmarshal(routesJSON, &routes); err != nil {
			return nil, err
		}

		for domain, prefixMap := range routes {
			for prefix, service := range prefixMap {
				dir.SetService(domain, prefix, service)
			}
		}
	}

	// Build the reverse proxy HTTP handler.
	reverseProxy := &httputil.ReverseProxy{
		// Specify a custom transport which rate limits requests and compresses responses.
		Transport: &httpwrapper.Transport{
			MaxConcurrencyPerHost: config.Concurrency,
			CompressionLevel:      config.CompressionLevel,
			Transport: &http.Transport{
				MaxIdleConnsPerHost: config.Concurrency,
				DisableKeepAlives:   true,
				DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
					return net.DialTimeout(network, addr, config.Timeout)
				},
			},
		},
		// The Director has the opportunity to modify the HTTP request before it
		// is handed off to the Transport.
		Director: func(req *http.Request) {
			// empty director atm
		},
	}

	return &http.Server{
			Addr: config.Address,
			Handler: accesslog.CustomLoggingHandler(
				os.Stdout,
				http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
					// Drop the connection header to ensure keepalives are maintained.
					req.Header.Del("connection")

					if root, prefix, err := dir.Service(req.Host, req.URL.Path); err != nil {
						// The director didn't find a match, handle it gracefully.

						if err != director.NoMatchingServiceError {
							log.Errorln("Error:", req.Host, req.URL.Path, err)
						} else {

							// If NoMatchingServiceError is thrown, check against the domain suffixes, e.g. {service}.local
							for _, domainSuffix := range config.DomainSuffixes() {
								if root := strings.TrimSuffix(req.Host, domainSuffix); root != req.Host {
									req.URL.Scheme = "http"
									req.URL.Host = root + config.KubernetesServiceDomainSuffix()
									log.Debug("Domain Suffix Match:", req.Host, req.URL.Host, req.URL.Path)
									reverseProxy.ServeHTTP(w, req)
									return
								}
							}

							// Otherwise, send traffic to the fallback.
							if config.Fallback.Enable {

								// Set the URL scheme, host, and path.
								req.URL.Scheme = config.Fallback.Scheme
								req.URL.Host = config.Fallback.Host
								req.URL.Path = path.Join(config.Fallback.Path, req.URL.Path)
								datadog.Count("fallback", 1, nil, 1.0)
								log.Debug("Fallback:", req.Host, req.URL.Path, " to ", req.URL.Host)
							} else {
								datadog.Count("no_route_matched_no_fallback_enabled", 1, nil, 1.0)
								log.Errorln("No route matched and fallback not enabled for", req.Host, req.URL.Path)
							}

						}

					} else {
						// The director found a match.

						if config.Static.Enable && strings.HasPrefix(root, "/") {
							// Handle static file requests.

							// we need to modify response
							// with equivalent of nginx
							// proxy_redirect /<%= application.name %>/ /;
							// http://nginx.org/en/docs/http/ngx_http_proxy_module.html#proxy_redirect
							// // Sets the text that should be changed in the “Location” and “Refresh” header
							// // fields of a proxied server response.
							// Otherwise, AWS returned redirects will have wrong paths
							//
							// for example
							// curl -v http://well.127.0.0.1.xip.io:8080/projects/workouts
							// Location: /well_workout/projects/workouts/
							// needs to get rewritten to
							// Location: /projects/workouts/
							// so
							// here we set headers so that
							// in httpwrapper.Transport.RoundTrip we know what's needed to  be replaced
							req.Header.Add("x-static-root", path.Join(config.Static.Path, root)+"/")
							req.Header.Add("x-original-url", req.Host+req.URL.String())

							// Set the URL scheme, host, and path.
							req.URL.Scheme = config.Static.Scheme
							req.URL.Host = config.Static.Host

							log.Debugln("Path: ", req.URL.Path)
							trailing := strings.HasSuffix(req.URL.Path, "/")

							req.URL.Path = path.Join(config.Static.Path, root, req.URL.Path)
							if trailing && !strings.HasSuffix(req.URL.Path, "/") {
								req.URL.Path += "/"
							}

							// Set the request host (used as the "Host" header value).
							req.Host = config.Static.Host

							// Drop cookies given that the response should not vary.
							req.Header.Del("cookie")

							log.Debugln("Static:", req.Header.Get("x-original-url"), "to", req.URL.Host+req.URL.Path)

						} else if redirectURLString := strings.TrimPrefix(root, ">"); redirectURLString != root {
							// log.Println("Path:", req.URL.Path)
							redirectURL, err := url.Parse(redirectURLString)
							if err != nil {
								log.Error(err)
							}
							redirectURL.Path = path.Join(redirectURL.Path, strings.Replace(req.URL.Path, prefix, "/", 1))
							if req.URL.RawQuery != "" {
								redirectURL.RawQuery = req.URL.RawQuery
							}
							datadog.Count("redirect_301", 1, nil, 1.0)
							log.Debug("Redirect: %s%s to %s", req.Host, req.URL.Path, redirectURL.String())
							http.Redirect(w, req, redirectURL.String(), 301)
							return
						} else {
							// Handle an arbitrary URL routing to a service.

							req.URL.Scheme = "http"
							req.URL.Host = root + config.KubernetesServiceDomainSuffix()
							log.Debugln("Proxy:", req.Host+req.URL.Path, "to", req.URL.Host)
						}
					}

					reverseProxy.ServeHTTP(w, req)
				}),
			),
		},
		nil
}