routing.go 1.7 KB
Newer Older
1 2 3 4 5 6
package routing

import (
	"net/http"

	"github.com/gorilla/mux"
7
	"github.com/matrix-org/dendrite/clientapi/config"
8 9
	"github.com/matrix-org/dendrite/clientapi/readers"
	"github.com/matrix-org/dendrite/clientapi/writers"
10
	"github.com/matrix-org/dendrite/roomserver/api"
11 12
	"github.com/matrix-org/util"
	"github.com/prometheus/client_golang/prometheus"
K
Kegsay 已提交
13
	sarama "gopkg.in/Shopify/sarama.v1"
14 15 16 17 18 19
)

const pathPrefixR0 = "/_matrix/client/r0"

// Setup registers HTTP handlers with the given ServeMux. It also supplies the given http.Client
// to clients which need to make outbound HTTP requests.
20
func Setup(servMux *http.ServeMux, httpClient *http.Client, cfg config.ClientAPI, producer sarama.SyncProducer, queryAPI api.RoomserverQueryAPI) {
21 22
	apiMux := mux.NewRouter()
	r0mux := apiMux.PathPrefix(pathPrefixR0).Subrouter()
23
	r0mux.Handle("/createRoom", make("createRoom", util.NewJSONRequestHandler(func(req *http.Request) util.JSONResponse {
K
Kegsay 已提交
24
		return writers.CreateRoom(req, cfg, producer)
25
	})))
26
	r0mux.Handle("/sync", make("sync", util.NewJSONRequestHandler(func(req *http.Request) util.JSONResponse {
27 28
		return readers.Sync(req)
	})))
29
	r0mux.Handle("/rooms/{roomID}/send/{eventType}/{txnID}",
30
		make("send_message", util.NewJSONRequestHandler(func(req *http.Request) util.JSONResponse {
31
			vars := mux.Vars(req)
32
			return writers.SendMessage(req, vars["roomID"], vars["eventType"], vars["txnID"], cfg, queryAPI, producer)
33 34 35 36 37 38 39 40 41 42 43
		})),
	)

	servMux.Handle("/metrics", prometheus.Handler())
	servMux.Handle("/api/", http.StripPrefix("/api", apiMux))
}

// make a util.JSONRequestHandler into an http.Handler
func make(metricsName string, h util.JSONRequestHandler) http.Handler {
	return prometheus.InstrumentHandler(metricsName, util.MakeJSONAPI(h))
}