mirror of
https://github.com/1f349/dendrite.git
synced 2024-11-15 08:11:32 +00:00
51ab0a8ccf
This is #2819 but rebased on latest `main`. This PR is against main too as opposed to the `moderncsqlite` branch. The main change here is simply: ```go // add query parameters to the dsn if strings.Contains(dsn, "?") { dsn += "&" } else { dsn += "?" } // wait some time before erroring if the db is locked // https://gitlab.com/cznic/sqlite/-/issues/106#note_1058094993 dsn += "_pragma=busy_timeout%3d10000" ``` ### Pull Request Checklist <!-- Please read https://matrix-org.github.io/dendrite/development/contributing before submitting your pull request --> * [x] I have added tests for PR _or_ I have justified why this PR doesn't need tests. * [x] Pull request includes a [sign off below using a legally identifiable name](https://matrix-org.github.io/dendrite/development/contributing#sign-off) _or_ I have already signed off privately Signed off privately. Co-authored-by: Neil Alexander <neilalexander@users.noreply.github.com>
78 lines
2.3 KiB
Go
78 lines
2.3 KiB
Go
// Copyright 2020 The Matrix.org Foundation C.I.C.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/matrix-org/dendrite/cmd/dendrite-polylith-multi/personalities"
|
|
"github.com/matrix-org/dendrite/setup"
|
|
"github.com/matrix-org/dendrite/setup/base"
|
|
"github.com/matrix-org/dendrite/setup/config"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type entrypoint func(base *base.BaseDendrite, cfg *config.Dendrite)
|
|
|
|
func main() {
|
|
cfg := setup.ParseFlags(false)
|
|
|
|
component := ""
|
|
if flag.NFlag() > 0 {
|
|
component = flag.Arg(0) // ./dendrite-polylith-multi --config=... clientapi
|
|
} else if len(os.Args) > 1 {
|
|
component = os.Args[1] // ./dendrite-polylith-multi clientapi
|
|
}
|
|
|
|
components := map[string]entrypoint{
|
|
"appservice": personalities.Appservice,
|
|
"clientapi": personalities.ClientAPI,
|
|
"federationapi": personalities.FederationAPI,
|
|
"keyserver": personalities.KeyServer,
|
|
"mediaapi": personalities.MediaAPI,
|
|
"roomserver": personalities.RoomServer,
|
|
"syncapi": personalities.SyncAPI,
|
|
"userapi": personalities.UserAPI,
|
|
}
|
|
|
|
start, ok := components[component]
|
|
if !ok {
|
|
if component == "" {
|
|
logrus.Errorf("No component specified")
|
|
logrus.Info("The first argument on the command line must be the name of the component to run")
|
|
} else {
|
|
logrus.Errorf("Unknown component %q specified", component)
|
|
}
|
|
|
|
var list []string
|
|
for c := range components {
|
|
list = append(list, c)
|
|
}
|
|
logrus.Infof("Valid components: %s", strings.Join(list, ", "))
|
|
|
|
os.Exit(1)
|
|
}
|
|
|
|
logrus.Infof("Starting %q component", component)
|
|
|
|
base := base.NewBaseDendrite(cfg, component, base.PolylithMode) // TODO
|
|
defer base.Close() // nolint: errcheck
|
|
|
|
go start(base, cfg)
|
|
base.WaitForShutdown()
|
|
}
|