mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
47 lines
1.9 KiB
Go
47 lines
1.9 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"Wavelet/core"
|
|
"Wavelet/core/extpoints"
|
|
cf "Wavelet/openflare/plugins/server/domain/cloudflare"
|
|
"Wavelet/openflare/plugins/server/domain/fleet"
|
|
"Wavelet/openflare/plugins/server/domain/observability"
|
|
"Wavelet/openflare/plugins/server/domain/pages"
|
|
"Wavelet/openflare/plugins/server/domain/tls"
|
|
oftask "Wavelet/openflare/plugins/server/kernel/task"
|
|
)
|
|
|
|
func registerOpenFlareTasks(ctx *core.Context) {
|
|
registerOFTask(ctx, fleet.SSLRenewTask, &fleet.SSLRenewHandler{}, fleet.SSLRenewMeta)
|
|
registerOFTask(ctx, fleet.WAFIPGroupSyncTask, &fleet.WAFIPGroupSyncHandler{}, fleet.WAFIPGroupSyncMeta)
|
|
registerOFTask(ctx, fleet.UptimeKumaSyncTask, &fleet.UptimeKumaSyncHandler{}, fleet.UptimeKumaSyncMeta)
|
|
registerOFTask(ctx, fleet.LogDBSwitchTask, &observability.LogDBSwitchHandler{}, fleet.LogDBSwitchMeta)
|
|
|
|
registerOFTask(ctx, cf.SyncMemberTask, &cf.SyncMemberTaskHandler{}, cf.SyncMemberMeta)
|
|
registerOFTask(ctx, cf.SyncGroupTask, &cf.SyncGroupTaskHandler{}, cf.SyncGroupMeta)
|
|
registerOFTask(ctx, cf.SyncByNodeTask, &cf.SyncByNodeTaskHandler{}, cf.SyncByNodeMeta)
|
|
|
|
registerOFTask(ctx, pages.PagesSourceScanTask, &pages.SourceScanHandler{}, pages.PagesSourceScanMeta)
|
|
registerOFTask(ctx, pages.PagesSourceActionTask, &pages.SourceActionHandler{}, pages.PagesSourceActionMeta)
|
|
|
|
registerOFTask(ctx, tls.SSLSingleRenewTask, &tls.SSLSingleRenewHandler{}, tls.SSLSingleRenewMeta)
|
|
}
|
|
|
|
func registerOFTask(ctx *core.Context, pattern string, handler oftask.TaskHandler, meta oftask.TaskMeta) {
|
|
opts := []extpoints.TaskOption{
|
|
extpoints.WithTaskMeta(meta.ToDTO()),
|
|
}
|
|
if meta.MaxRetry > 0 {
|
|
opts = append(opts, extpoints.WithTaskRetry(meta.MaxRetry))
|
|
}
|
|
ctx.Task().Register(pattern, func(c context.Context, payload []byte) error {
|
|
_, err := handler.Execute(c, payload)
|
|
return err
|
|
}, opts...)
|
|
}
|