mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
31 lines
661 B
Go
31 lines
661 B
Go
package server // import "github.com/docker/docker/api/server"
|
|
|
|
import (
|
|
"net/http"
|
|
"sync"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
// routerSwapper is an http.Handler that allows you to swap
|
|
// mux routers.
|
|
type routerSwapper struct {
|
|
mu sync.Mutex
|
|
router *mux.Router
|
|
}
|
|
|
|
// Swap changes the old router with the new one.
|
|
func (rs *routerSwapper) Swap(newRouter *mux.Router) {
|
|
rs.mu.Lock()
|
|
rs.router = newRouter
|
|
rs.mu.Unlock()
|
|
}
|
|
|
|
// ServeHTTP makes the routerSwapper to implement the http.Handler interface.
|
|
func (rs *routerSwapper) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
rs.mu.Lock()
|
|
router := rs.router
|
|
rs.mu.Unlock()
|
|
router.ServeHTTP(w, r)
|
|
}
|