From fda7529b3f30944aa896d87f9d12f1bf1fbb71da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=B1=88=E8=BD=A9?= Date: Thu, 25 Jul 2024 11:19:20 +0800 Subject: [PATCH] fix(cloudid): cloudid resource sync worker (#20878) --- pkg/cloudid/tasks/cloudaccount_sync_resources_task.go | 8 +++++++- pkg/cloudid/tasks/cloudprovider_sync_resources_task.go | 8 +++++++- 2 files changed, 14 insertions(+), 2 deletions(-) diff --git a/pkg/cloudid/tasks/cloudaccount_sync_resources_task.go b/pkg/cloudid/tasks/cloudaccount_sync_resources_task.go index 161a8d5209..2548e0a806 100644 --- a/pkg/cloudid/tasks/cloudaccount_sync_resources_task.go +++ b/pkg/cloudid/tasks/cloudaccount_sync_resources_task.go @@ -21,6 +21,7 @@ import ( "yunion.io/x/jsonutils" "yunion.io/x/pkg/errors" + "yunion.io/x/onecloud/pkg/appsrv" "yunion.io/x/onecloud/pkg/cloudcommon/db" "yunion.io/x/onecloud/pkg/cloudcommon/db/taskman" "yunion.io/x/onecloud/pkg/cloudid/models" @@ -30,8 +31,13 @@ type CloudaccountSyncResourcesTask struct { taskman.STask } +var ( + CloudaccountSyncWorkerManager *appsrv.SWorkerManager +) + func init() { - taskman.RegisterTask(CloudaccountSyncResourcesTask{}) + CloudaccountSyncWorkerManager = appsrv.NewWorkerManager("CloudaccountSyncWorkerManager", 10, 1024, false) + taskman.RegisterTaskAndWorker(CloudaccountSyncResourcesTask{}, CloudaccountSyncWorkerManager) } func (self *CloudaccountSyncResourcesTask) taskFailed(ctx context.Context, cloudaccount *models.SCloudaccount, err error) { diff --git a/pkg/cloudid/tasks/cloudprovider_sync_resources_task.go b/pkg/cloudid/tasks/cloudprovider_sync_resources_task.go index 3d44d3a626..35a181407f 100644 --- a/pkg/cloudid/tasks/cloudprovider_sync_resources_task.go +++ b/pkg/cloudid/tasks/cloudprovider_sync_resources_task.go @@ -21,6 +21,7 @@ import ( "yunion.io/x/jsonutils" "yunion.io/x/pkg/errors" + "yunion.io/x/onecloud/pkg/appsrv" "yunion.io/x/onecloud/pkg/cloudcommon/db" "yunion.io/x/onecloud/pkg/cloudcommon/db/taskman" "yunion.io/x/onecloud/pkg/cloudid/models" @@ -30,8 +31,13 @@ type CloudproviderSyncResourcesTask struct { taskman.STask } +var ( + CloudproviderSyncWorkerManager *appsrv.SWorkerManager +) + func init() { - taskman.RegisterTask(CloudproviderSyncResourcesTask{}) + CloudproviderSyncWorkerManager = appsrv.NewWorkerManager("CloudproviderSyncWorkerManager", 10, 1024, false) + taskman.RegisterTaskAndWorker(CloudproviderSyncResourcesTask{}, CloudproviderSyncWorkerManager) } func (self *CloudproviderSyncResourcesTask) taskFailed(ctx context.Context, cp *models.SCloudprovider, err error) {