fix: add more api

apache · Oct 4, 2021 · f7ecfb6 · f7ecfb6
1 parent b1fa50a
commit f7ecfb6
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/config/reference_config.go b/config/reference_config.go
@@ -90,6 +90,9 @@ func (rc *ReferenceConfig) Init(root *RootConfig) error {
 	if root.Application != nil {
 		rc.metaDataType = root.Application.MetadataType
 	}
+	if rc.Cluster == "" {
+		rc.Cluster = "failover"
+	}
 	rc.RegistryIDs = translateRegistryIds(rc.RegistryIDs)
 	if len(rc.RegistryIDs) <= 0 {
 		rc.RegistryIDs = root.Consumer.RegistryIDs
@@ -349,6 +352,20 @@ func (pcb *ReferenceConfigBuilder) SetRegistryIDs(registryIDs ...string) *Refere
 	return pcb
 }
 
+func (pcb *ReferenceConfigBuilder) SetGeneric(generic bool) *ReferenceConfigBuilder {
+	if generic {
+		pcb.referenceConfig.Generic = "true"
+	} else {
+		pcb.referenceConfig.Generic = "false"
+	}
+	return pcb
+}
+
+func (pcb *ReferenceConfigBuilder) SetCluster(cluster string) *ReferenceConfigBuilder {
+	pcb.referenceConfig.Cluster = cluster
+	return pcb
+}
+
 func (pcb *ReferenceConfigBuilder) SetSerialization(serialization string) *ReferenceConfigBuilder {
 	pcb.referenceConfig.Serialization = serialization
 	return pcb

diff --git a/config/root_config.go b/config/root_config.go
@@ -235,13 +235,13 @@ func (rb *RootConfigBuilder) SetApplication(application *ApplicationConfig) *Roo
 	return rb
 }
 
-func (rb *RootConfigBuilder) AddProtocol(protocolKey string, protocolConfig *ProtocolConfig) *RootConfigBuilder {
-	rb.rootConfig.Protocols[protocolKey] = protocolConfig
+func (rb *RootConfigBuilder) AddProtocol(protocolID string, protocolConfig *ProtocolConfig) *RootConfigBuilder {
+	rb.rootConfig.Protocols[protocolID] = protocolConfig
 	return rb
 }
 
-func (rb *RootConfigBuilder) AddRegistry(registryKey string, registryConfig *RegistryConfig) *RootConfigBuilder {
-	rb.rootConfig.Registries[registryKey] = registryConfig
+func (rb *RootConfigBuilder) AddRegistry(registryID string, registryConfig *RegistryConfig) *RootConfigBuilder {
+	rb.rootConfig.Registries[registryID] = registryConfig
 	return rb
 }