mirror of
https://github.com/matrix-org/dendrite.git
synced 2025-12-29 09:43:10 -06:00
81 lines
2.8 KiB
Go
81 lines
2.8 KiB
Go
package pushserver
|
|
|
|
import (
|
|
"github.com/gorilla/mux"
|
|
"github.com/matrix-org/dendrite/internal/pushgateway"
|
|
"github.com/matrix-org/dendrite/pushserver/api"
|
|
"github.com/matrix-org/dendrite/pushserver/consumers"
|
|
"github.com/matrix-org/dendrite/pushserver/internal"
|
|
"github.com/matrix-org/dendrite/pushserver/inthttp"
|
|
"github.com/matrix-org/dendrite/pushserver/producers"
|
|
"github.com/matrix-org/dendrite/pushserver/storage"
|
|
roomserverAPI "github.com/matrix-org/dendrite/roomserver/api"
|
|
"github.com/matrix-org/dendrite/setup/config"
|
|
"github.com/matrix-org/dendrite/setup/kafka"
|
|
"github.com/matrix-org/dendrite/setup/process"
|
|
uapi "github.com/matrix-org/dendrite/userapi/api"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// AddInternalRoutes registers HTTP handlers for the internal API. Invokes functions
|
|
// on the given input API.
|
|
func AddInternalRoutes(router *mux.Router, intAPI api.PushserverInternalAPI) {
|
|
inthttp.AddRoutes(intAPI, router)
|
|
}
|
|
|
|
// NewInternalAPI returns a concerete implementation of the internal API. Callers
|
|
// can call functions directly on the returned API or via an HTTP interface using AddInternalRoutes.
|
|
func NewInternalAPI(
|
|
cfg *config.PushServer,
|
|
process *process.ProcessContext,
|
|
pgClient pushgateway.Client,
|
|
rsAPI roomserverAPI.RoomserverInternalAPI,
|
|
userAPI uapi.UserInternalAPI,
|
|
) api.PushserverInternalAPI {
|
|
consumer, _ := kafka.SetupConsumerProducer(&cfg.Matrix.Kafka)
|
|
|
|
db, err := storage.Open(&cfg.Database)
|
|
if err != nil {
|
|
logrus.WithError(err).Panicf("failed to connect to push server db")
|
|
}
|
|
|
|
_, producer := kafka.SetupConsumerProducer(&cfg.Matrix.Kafka)
|
|
syncProducer := producers.NewSyncAPI(
|
|
db,
|
|
producer,
|
|
// TODO: user API should handle syncs for account data. Right now,
|
|
// it's handled by clientapi, and hence uses its topic. When user
|
|
// API handles it for all account data, we can remove it from
|
|
// here.
|
|
cfg.Matrix.Kafka.TopicFor(config.TopicOutputClientData),
|
|
cfg.Matrix.Kafka.TopicFor(config.TopicOutputNotificationData),
|
|
)
|
|
|
|
psAPI := internal.NewPushserverAPI(
|
|
cfg, db, userAPI, syncProducer,
|
|
)
|
|
|
|
caConsumer := consumers.NewOutputClientDataConsumer(
|
|
process, cfg, consumer, db, pgClient, userAPI, syncProducer,
|
|
)
|
|
if err := caConsumer.Start(); err != nil {
|
|
logrus.WithError(err).Panic("failed to start push server clientapi consumer")
|
|
}
|
|
|
|
eduConsumer := consumers.NewOutputReceiptEventConsumer(
|
|
process, cfg, consumer, db, pgClient, syncProducer,
|
|
)
|
|
if err := eduConsumer.Start(); err != nil {
|
|
logrus.WithError(err).Panic("failed to start push server EDU consumer")
|
|
}
|
|
|
|
rsConsumer := consumers.NewOutputRoomEventConsumer(
|
|
process, cfg, consumer, db, pgClient, psAPI, rsAPI, syncProducer,
|
|
)
|
|
if err := rsConsumer.Start(); err != nil {
|
|
logrus.WithError(err).Panic("failed to start push server room server consumer")
|
|
}
|
|
|
|
return psAPI
|
|
}
|