aboutsummaryrefslogtreecommitdiff
path: root/appservice/appservice.go
diff options
context:
space:
mode:
authorNeil Alexander <neilalexander@users.noreply.github.com>2022-09-01 09:20:40 +0100
committerGitHub <noreply@github.com>2022-09-01 09:20:40 +0100
commitad6b902b8462adb568d799c69a74b60d69574d0c (patch)
tree9037eb130a47c25cb320116758baa6ee265e89b6 /appservice/appservice.go
parent175f65407a7f684753334022e66b8209f3db7396 (diff)
Refactor appservices component (#2687)
This PR refactors the app services component. It makes the following changes: * Each appservice now gets its own NATS JetStream consumer * The appservice database is now removed entirely, since we just use JetStream as a data source instead * The entire component is now much simpler and we deleted lots of lines of code 💅 The result is that it should be much lighter and hopefully much more performant.
Diffstat (limited to 'appservice/appservice.go')
-rw-r--r--appservice/appservice.go55
1 files changed, 17 insertions, 38 deletions
diff --git a/appservice/appservice.go b/appservice/appservice.go
index 8fe1b2fc..9000adb1 100644
--- a/appservice/appservice.go
+++ b/appservice/appservice.go
@@ -18,7 +18,6 @@ import (
"context"
"crypto/tls"
"net/http"
- "sync"
"time"
"github.com/gorilla/mux"
@@ -28,9 +27,6 @@ import (
"github.com/matrix-org/dendrite/appservice/consumers"
"github.com/matrix-org/dendrite/appservice/inthttp"
"github.com/matrix-org/dendrite/appservice/query"
- "github.com/matrix-org/dendrite/appservice/storage"
- "github.com/matrix-org/dendrite/appservice/types"
- "github.com/matrix-org/dendrite/appservice/workers"
roomserverAPI "github.com/matrix-org/dendrite/roomserver/api"
"github.com/matrix-org/dendrite/setup/base"
"github.com/matrix-org/dendrite/setup/config"
@@ -59,57 +55,40 @@ func NewInternalAPI(
Proxy: http.ProxyFromEnvironment,
},
}
- js, _ := base.NATS.Prepare(base.ProcessContext, &base.Cfg.Global.JetStream)
+ // Create appserivce query API with an HTTP client that will be used for all
+ // outbound and inbound requests (inbound only for the internal API)
+ appserviceQueryAPI := &query.AppServiceQueryAPI{
+ HTTPClient: client,
+ Cfg: &base.Cfg.AppServiceAPI,
+ }
- // Create a connection to the appservice postgres DB
- appserviceDB, err := storage.NewDatabase(base, &base.Cfg.AppServiceAPI.Database)
- if err != nil {
- logrus.WithError(err).Panicf("failed to connect to appservice db")
+ if len(base.Cfg.Derived.ApplicationServices) == 0 {
+ return appserviceQueryAPI
}
// Wrap application services in a type that relates the application service and
// a sync.Cond object that can be used to notify workers when there are new
// events to be sent out.
- workerStates := make([]types.ApplicationServiceWorkerState, len(base.Cfg.Derived.ApplicationServices))
- for i, appservice := range base.Cfg.Derived.ApplicationServices {
- m := sync.Mutex{}
- ws := types.ApplicationServiceWorkerState{
- AppService: appservice,
- Cond: sync.NewCond(&m),
- }
- workerStates[i] = ws
-
+ for _, appservice := range base.Cfg.Derived.ApplicationServices {
// Create bot account for this AS if it doesn't already exist
- if err = generateAppServiceAccount(userAPI, appservice); err != nil {
+ if err := generateAppServiceAccount(userAPI, appservice); err != nil {
logrus.WithFields(logrus.Fields{
"appservice": appservice.ID,
}).WithError(err).Panicf("failed to generate bot account for appservice")
}
}
- // Create appserivce query API with an HTTP client that will be used for all
- // outbound and inbound requests (inbound only for the internal API)
- appserviceQueryAPI := &query.AppServiceQueryAPI{
- HTTPClient: client,
- Cfg: base.Cfg,
- }
-
// Only consume if we actually have ASes to track, else we'll just chew cycles needlessly.
// We can't add ASes at runtime so this is safe to do.
- if len(workerStates) > 0 {
- consumer := consumers.NewOutputRoomEventConsumer(
- base.ProcessContext, base.Cfg, js, appserviceDB,
- rsAPI, workerStates,
- )
- if err := consumer.Start(); err != nil {
- logrus.WithError(err).Panicf("failed to start appservice roomserver consumer")
- }
+ js, _ := base.NATS.Prepare(base.ProcessContext, &base.Cfg.Global.JetStream)
+ consumer := consumers.NewOutputRoomEventConsumer(
+ base.ProcessContext, &base.Cfg.AppServiceAPI,
+ client, js, rsAPI,
+ )
+ if err := consumer.Start(); err != nil {
+ logrus.WithError(err).Panicf("failed to start appservice roomserver consumer")
}
- // Create application service transaction workers
- if err := workers.SetupTransactionWorkers(client, appserviceDB, workerStates); err != nil {
- logrus.WithError(err).Panicf("failed to start app service transaction workers")
- }
return appserviceQueryAPI
}