mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-11-25 13:16:40 +00:00
[chore] little startup tweaks (#2941)
* [chore] little startup tweaks * go fmt
This commit is contained in:
parent
2fd69ec58b
commit
4db596b8b9
|
@ -78,7 +78,7 @@
|
||||||
// depending on what services were
|
// depending on what services were
|
||||||
// managed to be started.
|
// managed to be started.
|
||||||
|
|
||||||
state state.State
|
state = new(state.State)
|
||||||
route *router.Router
|
route *router.Router
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -137,7 +137,7 @@
|
||||||
state.Caches.Start()
|
state.Caches.Start()
|
||||||
|
|
||||||
// Open connection to the database now caches started.
|
// Open connection to the database now caches started.
|
||||||
dbService, err := bundb.NewBunDBService(ctx, &state)
|
dbService, err := bundb.NewBunDBService(ctx, state)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("error creating dbservice: %s", err)
|
return fmt.Errorf("error creating dbservice: %s", err)
|
||||||
}
|
}
|
||||||
|
@ -177,14 +177,14 @@
|
||||||
})
|
})
|
||||||
|
|
||||||
// Build handlers used in later initializations.
|
// Build handlers used in later initializations.
|
||||||
mediaManager := media.NewManager(&state)
|
mediaManager := media.NewManager(state)
|
||||||
oauthServer := oauth.New(ctx, dbService)
|
oauthServer := oauth.New(ctx, dbService)
|
||||||
typeConverter := typeutils.NewConverter(&state)
|
typeConverter := typeutils.NewConverter(state)
|
||||||
visFilter := visibility.NewFilter(&state)
|
visFilter := visibility.NewFilter(state)
|
||||||
spamFilter := spam.NewFilter(&state)
|
spamFilter := spam.NewFilter(state)
|
||||||
federatingDB := federatingdb.New(&state, typeConverter, visFilter, spamFilter)
|
federatingDB := federatingdb.New(state, typeConverter, visFilter, spamFilter)
|
||||||
transportController := transport.NewController(&state, federatingDB, &federation.Clock{}, client)
|
transportController := transport.NewController(state, federatingDB, &federation.Clock{}, client)
|
||||||
federator := federation.NewFederator(&state, federatingDB, transportController, typeConverter, visFilter, mediaManager)
|
federator := federation.NewFederator(state, federatingDB, transportController, typeConverter, visFilter, mediaManager)
|
||||||
|
|
||||||
// Decide whether to create a noop email
|
// Decide whether to create a noop email
|
||||||
// sender (won't send emails) or a real one.
|
// sender (won't send emails) or a real one.
|
||||||
|
@ -205,18 +205,18 @@
|
||||||
|
|
||||||
// Initialize both home / list timelines.
|
// Initialize both home / list timelines.
|
||||||
state.Timelines.Home = timeline.NewManager(
|
state.Timelines.Home = timeline.NewManager(
|
||||||
tlprocessor.HomeTimelineGrab(&state),
|
tlprocessor.HomeTimelineGrab(state),
|
||||||
tlprocessor.HomeTimelineFilter(&state, visFilter),
|
tlprocessor.HomeTimelineFilter(state, visFilter),
|
||||||
tlprocessor.HomeTimelineStatusPrepare(&state, typeConverter),
|
tlprocessor.HomeTimelineStatusPrepare(state, typeConverter),
|
||||||
tlprocessor.SkipInsert(),
|
tlprocessor.SkipInsert(),
|
||||||
)
|
)
|
||||||
if err := state.Timelines.Home.Start(); err != nil {
|
if err := state.Timelines.Home.Start(); err != nil {
|
||||||
return fmt.Errorf("error starting home timeline: %s", err)
|
return fmt.Errorf("error starting home timeline: %s", err)
|
||||||
}
|
}
|
||||||
state.Timelines.List = timeline.NewManager(
|
state.Timelines.List = timeline.NewManager(
|
||||||
tlprocessor.ListTimelineGrab(&state),
|
tlprocessor.ListTimelineGrab(state),
|
||||||
tlprocessor.ListTimelineFilter(&state, visFilter),
|
tlprocessor.ListTimelineFilter(state, visFilter),
|
||||||
tlprocessor.ListTimelineStatusPrepare(&state, typeConverter),
|
tlprocessor.ListTimelineStatusPrepare(state, typeConverter),
|
||||||
tlprocessor.SkipInsert(),
|
tlprocessor.SkipInsert(),
|
||||||
)
|
)
|
||||||
if err := state.Timelines.List.Start(); err != nil {
|
if err := state.Timelines.List.Start(); err != nil {
|
||||||
|
@ -230,17 +230,19 @@
|
||||||
// Add a task to the scheduler to sweep caches.
|
// Add a task to the scheduler to sweep caches.
|
||||||
// Frequency = 1 * minute
|
// Frequency = 1 * minute
|
||||||
// Threshold = 60% capacity
|
// Threshold = 60% capacity
|
||||||
_ = state.Workers.Scheduler.AddRecurring(
|
if !state.Workers.Scheduler.AddRecurring(
|
||||||
"@cachesweep", // id
|
"@cachesweep", // id
|
||||||
time.Time{}, // start
|
time.Time{}, // start
|
||||||
time.Minute, // freq
|
time.Minute, // freq
|
||||||
func(context.Context, time.Time) {
|
func(context.Context, time.Time) {
|
||||||
state.Caches.Sweep(60)
|
state.Caches.Sweep(60)
|
||||||
},
|
},
|
||||||
)
|
) {
|
||||||
|
return fmt.Errorf("error scheduling cache sweep: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
// Create background cleaner.
|
// Create background cleaner.
|
||||||
cleaner := cleaner.New(&state)
|
cleaner := cleaner.New(state)
|
||||||
|
|
||||||
// Now schedule background cleaning tasks.
|
// Now schedule background cleaning tasks.
|
||||||
if err := cleaner.ScheduleJobs(); err != nil {
|
if err := cleaner.ScheduleJobs(); err != nil {
|
||||||
|
@ -255,7 +257,7 @@ func(context.Context, time.Time) {
|
||||||
federator,
|
federator,
|
||||||
oauthServer,
|
oauthServer,
|
||||||
mediaManager,
|
mediaManager,
|
||||||
&state,
|
state,
|
||||||
emailSender,
|
emailSender,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -308,7 +310,7 @@ func(context.Context, time.Time) {
|
||||||
// note: hooks adding ctx fields must be ABOVE
|
// note: hooks adding ctx fields must be ABOVE
|
||||||
// the logger, otherwise won't be accessible.
|
// the logger, otherwise won't be accessible.
|
||||||
middleware.Logger(config.GetLogClientIP()),
|
middleware.Logger(config.GetLogClientIP()),
|
||||||
middleware.HeaderFilter(&state),
|
middleware.HeaderFilter(state),
|
||||||
middleware.UserAgent(),
|
middleware.UserAgent(),
|
||||||
middleware.CORS(),
|
middleware.CORS(),
|
||||||
middleware.ExtraHeaders(),
|
middleware.ExtraHeaders(),
|
||||||
|
@ -366,7 +368,7 @@ func(context.Context, time.Time) {
|
||||||
|
|
||||||
var (
|
var (
|
||||||
authModule = api.NewAuth(dbService, processor, idp, routerSession, sessionName) // auth/oauth paths
|
authModule = api.NewAuth(dbService, processor, idp, routerSession, sessionName) // auth/oauth paths
|
||||||
clientModule = api.NewClient(&state, processor) // api client endpoints
|
clientModule = api.NewClient(state, processor) // api client endpoints
|
||||||
metricsModule = api.NewMetrics() // Metrics endpoints
|
metricsModule = api.NewMetrics() // Metrics endpoints
|
||||||
healthModule = api.NewHealth(dbService.Ready) // Health check endpoints
|
healthModule = api.NewHealth(dbService.Ready) // Health check endpoints
|
||||||
fileserverModule = api.NewFileserver(processor) // fileserver endpoints
|
fileserverModule = api.NewFileserver(processor) // fileserver endpoints
|
||||||
|
|
|
@ -64,7 +64,7 @@
|
||||||
// depending on what services were
|
// depending on what services were
|
||||||
// managed to be started.
|
// managed to be started.
|
||||||
|
|
||||||
state state.State
|
state = new(state.State)
|
||||||
route *router.Router
|
route *router.Router
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -85,7 +85,7 @@
|
||||||
// Stop any currently running
|
// Stop any currently running
|
||||||
// worker processes / scheduled
|
// worker processes / scheduled
|
||||||
// tasks from being executed.
|
// tasks from being executed.
|
||||||
testrig.StopWorkers(&state)
|
testrig.StopWorkers(state)
|
||||||
|
|
||||||
if state.Timelines.Home != nil {
|
if state.Timelines.Home != nil {
|
||||||
// Home timeline mgr was setup, ensure it gets stopped.
|
// Home timeline mgr was setup, ensure it gets stopped.
|
||||||
|
@ -130,7 +130,7 @@
|
||||||
}
|
}
|
||||||
|
|
||||||
// Initialize caches and database
|
// Initialize caches and database
|
||||||
state.DB = testrig.NewTestDB(&state)
|
state.DB = testrig.NewTestDB(state)
|
||||||
|
|
||||||
// New test db inits caches so we don't need to do
|
// New test db inits caches so we don't need to do
|
||||||
// that twice, we can just start the initialized caches.
|
// that twice, we can just start the initialized caches.
|
||||||
|
@ -156,11 +156,11 @@
|
||||||
testrig.StandardStorageSetup(state.Storage, "./testrig/media")
|
testrig.StandardStorageSetup(state.Storage, "./testrig/media")
|
||||||
|
|
||||||
// Initialize workers.
|
// Initialize workers.
|
||||||
testrig.StartNoopWorkers(&state)
|
testrig.StartNoopWorkers(state)
|
||||||
defer testrig.StopWorkers(&state)
|
defer testrig.StopWorkers(state)
|
||||||
|
|
||||||
// build backend handlers
|
// build backend handlers
|
||||||
transportController := testrig.NewTestTransportController(&state, testrig.NewMockHTTPClient(func(req *http.Request) (*http.Response, error) {
|
transportController := testrig.NewTestTransportController(state, testrig.NewMockHTTPClient(func(req *http.Request) (*http.Response, error) {
|
||||||
r := io.NopCloser(bytes.NewReader([]byte{}))
|
r := io.NopCloser(bytes.NewReader([]byte{}))
|
||||||
return &http.Response{
|
return &http.Response{
|
||||||
StatusCode: 200,
|
StatusCode: 200,
|
||||||
|
@ -170,34 +170,34 @@
|
||||||
},
|
},
|
||||||
}, nil
|
}, nil
|
||||||
}, ""))
|
}, ""))
|
||||||
mediaManager := testrig.NewTestMediaManager(&state)
|
mediaManager := testrig.NewTestMediaManager(state)
|
||||||
federator := testrig.NewTestFederator(&state, transportController, mediaManager)
|
federator := testrig.NewTestFederator(state, transportController, mediaManager)
|
||||||
|
|
||||||
emailSender := testrig.NewEmailSender("./web/template/", nil)
|
emailSender := testrig.NewEmailSender("./web/template/", nil)
|
||||||
typeConverter := typeutils.NewConverter(&state)
|
typeConverter := typeutils.NewConverter(state)
|
||||||
filter := visibility.NewFilter(&state)
|
filter := visibility.NewFilter(state)
|
||||||
|
|
||||||
// Initialize both home / list timelines.
|
// Initialize both home / list timelines.
|
||||||
state.Timelines.Home = timeline.NewManager(
|
state.Timelines.Home = timeline.NewManager(
|
||||||
tlprocessor.HomeTimelineGrab(&state),
|
tlprocessor.HomeTimelineGrab(state),
|
||||||
tlprocessor.HomeTimelineFilter(&state, filter),
|
tlprocessor.HomeTimelineFilter(state, filter),
|
||||||
tlprocessor.HomeTimelineStatusPrepare(&state, typeConverter),
|
tlprocessor.HomeTimelineStatusPrepare(state, typeConverter),
|
||||||
tlprocessor.SkipInsert(),
|
tlprocessor.SkipInsert(),
|
||||||
)
|
)
|
||||||
if err := state.Timelines.Home.Start(); err != nil {
|
if err := state.Timelines.Home.Start(); err != nil {
|
||||||
return fmt.Errorf("error starting home timeline: %s", err)
|
return fmt.Errorf("error starting home timeline: %s", err)
|
||||||
}
|
}
|
||||||
state.Timelines.List = timeline.NewManager(
|
state.Timelines.List = timeline.NewManager(
|
||||||
tlprocessor.ListTimelineGrab(&state),
|
tlprocessor.ListTimelineGrab(state),
|
||||||
tlprocessor.ListTimelineFilter(&state, filter),
|
tlprocessor.ListTimelineFilter(state, filter),
|
||||||
tlprocessor.ListTimelineStatusPrepare(&state, typeConverter),
|
tlprocessor.ListTimelineStatusPrepare(state, typeConverter),
|
||||||
tlprocessor.SkipInsert(),
|
tlprocessor.SkipInsert(),
|
||||||
)
|
)
|
||||||
if err := state.Timelines.List.Start(); err != nil {
|
if err := state.Timelines.List.Start(); err != nil {
|
||||||
return fmt.Errorf("error starting list timeline: %s", err)
|
return fmt.Errorf("error starting list timeline: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
processor := testrig.NewTestProcessor(&state, federator, emailSender, mediaManager)
|
processor := testrig.NewTestProcessor(state, federator, emailSender, mediaManager)
|
||||||
|
|
||||||
// Initialize metrics.
|
// Initialize metrics.
|
||||||
if err := metrics.Initialize(state.DB); err != nil {
|
if err := metrics.Initialize(state.DB); err != nil {
|
||||||
|
@ -222,7 +222,7 @@
|
||||||
|
|
||||||
middlewares = append(middlewares, []gin.HandlerFunc{
|
middlewares = append(middlewares, []gin.HandlerFunc{
|
||||||
middleware.Logger(config.GetLogClientIP()),
|
middleware.Logger(config.GetLogClientIP()),
|
||||||
middleware.HeaderFilter(&state),
|
middleware.HeaderFilter(state),
|
||||||
middleware.UserAgent(),
|
middleware.UserAgent(),
|
||||||
middleware.CORS(),
|
middleware.CORS(),
|
||||||
middleware.ExtraHeaders(),
|
middleware.ExtraHeaders(),
|
||||||
|
@ -280,7 +280,7 @@
|
||||||
|
|
||||||
var (
|
var (
|
||||||
authModule = api.NewAuth(state.DB, processor, idp, routerSession, sessionName) // auth/oauth paths
|
authModule = api.NewAuth(state.DB, processor, idp, routerSession, sessionName) // auth/oauth paths
|
||||||
clientModule = api.NewClient(&state, processor) // api client endpoints
|
clientModule = api.NewClient(state, processor) // api client endpoints
|
||||||
metricsModule = api.NewMetrics() // Metrics endpoints
|
metricsModule = api.NewMetrics() // Metrics endpoints
|
||||||
healthModule = api.NewHealth(state.DB.Ready) // Health check endpoints
|
healthModule = api.NewHealth(state.DB.Ready) // Health check endpoints
|
||||||
fileserverModule = api.NewFileserver(processor) // fileserver endpoints
|
fileserverModule = api.NewFileserver(processor) // fileserver endpoints
|
||||||
|
@ -304,7 +304,7 @@
|
||||||
webModule.Route(route)
|
webModule.Route(route)
|
||||||
|
|
||||||
// Create background cleaner.
|
// Create background cleaner.
|
||||||
cleaner := cleaner.New(&state)
|
cleaner := cleaner.New(state)
|
||||||
|
|
||||||
// Now schedule background cleaning tasks.
|
// Now schedule background cleaning tasks.
|
||||||
if err := cleaner.ScheduleJobs(); err != nil {
|
if err := cleaner.ScheduleJobs(); err != nil {
|
||||||
|
|
Loading…
Reference in a new issue