Пример #1
0
func main() {
	port, consulURL, pathToCheckARecord := parseCommandLineFlags()
	proxyURL, err := url.Parse(consulURL)
	if err != nil {
		log.Fatal(err)
	}

	if pathToCheckARecord == "" {
		log.Fatal("--path-to-check-a-record is required")
	}

	mux := http.NewServeMux()
	logBuffer := bytes.NewBuffer([]byte{})
	healthCheckHandler := handlers.NewHealthCheckHandler()
	dnsHandler := handlers.NewDNSHandler(pathToCheckARecord)

	proxy := httputil.NewSingleHostReverseProxy(proxyURL)
	director := proxy.Director
	proxy.Director = func(request *http.Request) {
		director(request)
		request.URL.Path = strings.TrimPrefix(request.URL.Path, "/consul")
		request.Host = request.URL.Host
	}
	proxy.ErrorLog = log.New(logBuffer, "", 0)

	mux.HandleFunc("/consul/", func(w http.ResponseWriter, req *http.Request) {
		bufferedRW := buffered.NewResponseWriter(w, logBuffer)
		proxy.ServeHTTP(bufferedRW, req)
		bufferedRW.Copy()
	})

	mux.HandleFunc("/health_check", func(w http.ResponseWriter, req *http.Request) {
		healthCheckHandler.ServeHTTP(w, req)
	})

	mux.HandleFunc("/dns", func(w http.ResponseWriter, req *http.Request) {
		dnsHandler.ServeHTTP(w, req)
	})

	log.Fatal(http.ListenAndServe(fmt.Sprintf("0.0.0.0:%s", port), mux))
}
package buffered_test

import (
	"bytes"
	"net/http"
	"net/http/httptest"
	"strconv"

	"github.com/cloudfoundry-incubator/consul-release/src/acceptance-tests/testing/testconsumer/buffered"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ http.ResponseWriter = buffered.NewResponseWriter(nil, nil)

var _ = Describe("ResponseBuffer", func() {
	Describe("Copy", func() {
		var (
			bufferedWriter *buffered.ResponseWriter
			actualWriter   *httptest.ResponseRecorder
			logBuffer      *bytes.Buffer
		)

		BeforeEach(func() {
			logBuffer = bytes.NewBuffer([]byte{})
			actualWriter = httptest.NewRecorder()
			bufferedWriter = buffered.NewResponseWriter(actualWriter, logBuffer)
		})

		It("copies all the headers to the real response writer", func() {