multiroute code - prototype

multiroute.go

//Package multiroute defines an interface needed for plugins that support routes with multiple backends.
package plugin

import (
	"net/http"
	"errors"
)

type MultiRouteHandler struct {
	backendMap map[string]http.Handler
}

func NewMultiRouteHandler() *MultiRouteHandler {
	return &MultiRouteHandler{}
}

func (m *MultiRouteHandler) GetHandlerForBackend(backendName string) (http.Handler,error) {
	handler, ok := m.backendMap[backendName]
	if !ok {
		return errors.New("No handler for backend " + backendName)
	}

	return handler
}

func (m *MultiRouteHandler) AddHandlerForBackend(backendName string, handler http.Handler) {
	m.backendMap[backendName] = handler
}