fix: recorder message factory not found

Author: ZhengYa-0110

server/controller/recorder/updater/az.go

@@ -46,6 +46,10 @@ func (f *AZMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedAZFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_AZ_EN, &AZMessageFactory{})
+}
+
 type AZ struct {
 	UpdaterBase[
 		cloudmodel.AZ,
@@ -67,10 +71,6 @@ func NewAZ(wholeCache *cache.Cache, cloudData []cloudmodel.AZ) *AZ {
 	}
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &AZMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/cen.go

@@ -47,6 +47,10 @@ func (f *CENMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedCENFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_CEN_EN, &CENMessageFactory{})
+}
+
 type CEN struct {
 	UpdaterBase[
 		cloudmodel.CEN,
@@ -68,10 +72,6 @@ func NewCEN(wholeCache *cache.Cache, cloudData []cloudmodel.CEN) *CEN {
 	}
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &CENMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/config_map.go

@@ -50,6 +50,10 @@ func (f *ConfigMapMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedConfigMapFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_CONFIG_MAP_EN, &ConfigMapMessageFactory{})
+}
+
 type ConfigMap struct {
 	UpdaterBase[
 		cloudmodel.ConfigMap,
@@ -72,10 +76,6 @@ func NewConfigMap(wholeCache *cache.Cache, cloudData []cloudmodel.ConfigMap) *Co
 	updater.setDataGenerator(updater)
 	updater.toLoggable = true
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &ConfigMapMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/dhcp_port.go

@@ -46,6 +46,10 @@ func (f *DHCPPortMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedDHCPPortFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_DHCP_PORT_EN, &DHCPPortMessageFactory{})
+}
+
 type DHCPPort struct {
 	UpdaterBase[
 		cloudmodel.DHCPPort,
@@ -67,10 +71,6 @@ func NewDHCPPort(wholeCache *cache.Cache, cloudData []cloudmodel.DHCPPort) *DHCP
 	}
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &DHCPPortMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/floating_ip.go

@@ -47,6 +47,10 @@ func (f *FloatingIPMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedFloatingIPFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_FLOATING_IP_EN, &FloatingIPMessageFactory{})
+}
+
 type FloatingIP struct {
 	UpdaterBase[
 		cloudmodel.FloatingIP,
@@ -68,10 +72,6 @@ func NewFloatingIP(wholeCache *cache.Cache, cloudData []cloudmodel.FloatingIP) *
 	}
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &FloatingIPMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/host.go

@@ -46,6 +46,10 @@ func (f *HostMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedHostFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_HOST_EN, &HostMessageFactory{})
+}
+
 type Host struct {
 	UpdaterBase[
 		cloudmodel.Host,
@@ -67,10 +71,6 @@ func NewHost(wholeCache *cache.Cache, cloudData []cloudmodel.Host) *Host {
 	}
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &HostMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/lan_ip.go

@@ -48,6 +48,10 @@ func (f *LANIPMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedLANIPFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_LAN_IP_EN, &LANIPMessageFactory{})
+}
+
 type LANIP struct {
 	UpdaterBase[
 		cloudmodel.IP,
@@ -70,10 +74,6 @@ func NewLANIP(wholeCache *cache.Cache, domainToolDataSet *tool.DataSet) *LANIP {
 	updater.setDomainToolDataSet(domainToolDataSet)
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &LANIPMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/lb.go

@@ -46,6 +46,10 @@ func (f *LBMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedLBFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_LB_EN, &LBMessageFactory{})
+}
+
 type LB struct {
 	UpdaterBase[
 		cloudmodel.LB,
@@ -67,10 +71,6 @@ func NewLB(wholeCache *cache.Cache, cloudData []cloudmodel.LB) *LB {
 	}
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &LBMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/lb_listener.go

@@ -46,6 +46,10 @@ func (f *LBListenerMessageFactory) CreateUpdatedFields() types.UpdatedFields {
 	return &message.UpdatedLBListenerFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_LB_LISTENER_EN, &LBListenerMessageFactory{})
+}
+
 type LBListener struct {
 	UpdaterBase[
 		cloudmodel.LBListener,
@@ -67,10 +71,6 @@ func NewLBListener(wholeCache *cache.Cache, cloudData []cloudmodel.LBListener) *
 	}
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &LBListenerMessageFactory{})
-	}
-
 	return updater
 }
 

server/controller/recorder/updater/lb_target_server.go

@@ -46,6 +46,10 @@ func (f *LBTargetServerMessageFactory) CreateUpdatedFields() types.UpdatedFields
 	return &message.UpdatedLBTargetServerFields{}
 }
 
+func init() {
+	RegisterMessageFactory(ctrlrcommon.RESOURCE_TYPE_LB_TARGET_SERVER_EN, &LBTargetServerMessageFactory{})
+}
+
 type LBTargetServer struct {
 	UpdaterBase[
 		cloudmodel.LBTargetServer,
@@ -67,10 +71,6 @@ func NewLBTargetServer(wholeCache *cache.Cache, cloudData []cloudmodel.LBTargetS
 	}
 	updater.setDataGenerator(updater)
 
-	if !hasMessageFactory(updater.resourceType) {
-		RegisterMessageFactory(updater.resourceType, &LBTargetServerMessageFactory{})
-	}
-
 	return updater
 }