Support obproxy ce (#94)

* support obproxy-ce

core.py

@@ -2189,7 +2189,7 @@ class ObdHome(object):
         deploy_config = deploy.deploy_config
 
         if opts.component is None:
-            for component_name in ['obproxy', 'oceanbase', 'oceanbase-ce']:
+            for component_name in ['obproxy', 'obproxy-ce', 'oceanbase', 'oceanbase-ce']:
                 if component_name in deploy_config.components:
                     opts.component = component_name
                     break
@@ -2357,7 +2357,7 @@ class ObdHome(object):
         self._call_stdio('verbose', 'Get deploy configuration')
         deploy_config = deploy.deploy_config
 
-        allow_components = ['obproxy', 'oceanbase', 'oceanbase-ce']
+        allow_components = ['obproxy', 'obproxy-ce', 'oceanbase', 'oceanbase-ce']
         if opts.component is None:
             for component_name in allow_components:
                 if component_name in deploy_config.components:
@@ -2423,7 +2423,7 @@ class ObdHome(object):
         connect_plugin = self.search_py_script_plugin(repositories, 'connect')[repository]
 
         
-        if repository.name == 'obproxy':
+        if repository.name in ['obproxy', 'obproxy-ce']:
             ob_optimization = False
             allow_components = ['oceanbase', 'oceanbase-ce']
             for component_name in deploy_config.components:
@@ -2583,7 +2583,7 @@ class ObdHome(object):
         self._call_stdio('verbose', 'Get deploy configuration')
         deploy_config = deploy.deploy_config
 
-        allow_components = ['obproxy', 'oceanbase', 'oceanbase-ce']
+        allow_components = ['obproxy', 'obproxy-ce', 'oceanbase', 'oceanbase-ce']
         if opts.component is None:
             for component_name in allow_components:
                 if component_name in deploy_config.components:
@@ -2649,7 +2649,7 @@ class ObdHome(object):
 
         connect_plugin = self.search_py_script_plugin(repositories, 'connect')[repository]
 
-        if repository.name == 'obproxy':
+        if repository.name in ['obproxy', 'obproxy-ce']:
             ob_optimization = False
             allow_components = ['oceanbase', 'oceanbase-ce']
             for component in deploy_info.components:

plugins/oceanbase/3.1.0/bootstrap.py

@@ -33,6 +33,11 @@ def bootstrap(plugin_context, cursor, *args, **kwargs):
     inner_config = {
         InnerConfigItem('$_zone_idc'): 'idc'
     }
+    has_obproxy = False
+    for componet_name in ['obproxy', 'obproxy-ce']:
+        if componet_name in plugin_context.components:
+            has_obproxy = True
+            break
     inner_keys = inner_config.keys()
     for server in cluster_config.servers:
         server_config = cluster_config.get_server_conf(server)
@@ -67,7 +72,7 @@ def bootstrap(plugin_context, cursor, *args, **kwargs):
                 stdio.verbose('execute sql: %s' % sql)
                 cursor.execute(sql)
         global_conf = cluster_config.get_global_conf()
-        if 'proxyro_password' in global_conf or 'obproxy' in plugin_context.components:
+        if has_obproxy or 'proxyro_password' in global_conf:
             value = global_conf['proxyro_password'] if global_conf.get('proxyro_password') is not None else ''
             sql = 'create user "proxyro" IDENTIFIED BY %s'
             stdio.verbose(sql)

plugins/oceanbase/3.1.0/generate_config.py

@@ -75,8 +75,9 @@ def generate_config(plugin_context, deploy_config, *args, **kwargs):
 
     if not cluster_config.get_global_conf().get('appname'):
         default_appname = 'obcluster'
-        if 'obproxy' in deploy_config.components:
-            obproxy_cluster_config = deploy_config.components['obproxy']
+        for componet_name in ['obproxy', 'obproxy-ce']:
+            if componet_name in deploy_config.components:
+                obproxy_cluster_config = deploy_config.components[componet_name]
                 cluster_name = obproxy_cluster_config.get_global_conf().get('cluster_name')
                 if not cluster_name:
                     for server in obproxy_cluster_config.servers:
@@ -84,6 +85,7 @@ def generate_config(plugin_context, deploy_config, *args, **kwargs):
                         if server_config.get('cluster_name'):
                             default_appname = server_config['cluster_name']
                             break
+                break
         cluster_config.update_global_conf('appname', default_appname, False)
 
     MIN_MEMORY = 8 << 30