92 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /**
 | |
|  * Copyright (c) 2021 OceanBase
 | |
|  * OceanBase CE is licensed under Mulan PubL v2.
 | |
|  * You can use this software according to the terms and conditions of the Mulan PubL v2.
 | |
|  * You may obtain a copy of Mulan PubL v2 at:
 | |
|  *          http://license.coscl.org.cn/MulanPubL-2.0
 | |
|  * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 | |
|  * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 | |
|  * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 | |
|  * See the Mulan PubL v2 for more details.
 | |
|  */
 | |
| 
 | |
| #ifndef OCEANBASE_SCHEMA_DB_INITIALIZER_H_
 | |
| #define OCEANBASE_SCHEMA_DB_INITIALIZER_H_
 | |
| 
 | |
| #include "lib/mysqlclient/ob_mysql_proxy.h"
 | |
| #include "lib/mysqlclient/ob_single_mysql_connection_pool.h"
 | |
| #include "share/config/ob_server_config.h"
 | |
| #include "share/inner_table/ob_inner_table_schema.h"
 | |
| #include <string>
 | |
| 
 | |
| namespace oceanbase {
 | |
| namespace share {
 | |
| namespace schema {
 | |
| 
 | |
| // Mysql server backend intialize helper for unittest.
 | |
| //
 | |
| // 1. initialize empty database and mysql client, connect to localhost mysql:
 | |
| //     host: 127.0.0.1
 | |
| //     port: 3306
 | |
| //     user: root
 | |
| //     pass: ""
 | |
| //     database: ${USER//./_}_unittest
 | |
| //
 | |
| // 2. create OceanBase system table schema in mysql server.
 | |
| //
 | |
| class DBInitializer {
 | |
| public:
 | |
|   DBInitializer()
 | |
|   {}
 | |
|   virtual ~DBInitializer();
 | |
| 
 | |
|   inline int init();
 | |
|   inline std::string get_db_name()
 | |
|   {
 | |
|     return db_.db_;
 | |
|   }
 | |
|   inline int create_system_table(bool only_core_tables);
 | |
|   inline int fill_sys_stat_table(const uint64_t tenant_id = common::OB_SYS_TENANT_ID);
 | |
|   inline int create_tenant_space_tables(const uint64_t tenant_id);
 | |
|   inline int create_tenant_space(const uint64_t tenant_id);
 | |
|   inline int create_virtual_table(const schema_create_func create_func);
 | |
|   common::ObMySQLProxy& get_sql_proxy()
 | |
|   {
 | |
|     return sql_proxy_;
 | |
|   }
 | |
|   common::ObServerConfig& get_config()
 | |
|   {
 | |
|     return common::ObServerConfig::get_instance();
 | |
|   }
 | |
| 
 | |
| private:
 | |
|   struct DBConfig {
 | |
|     std::string host_;
 | |
|     int port_;
 | |
|     std::string user_;
 | |
|     std::string pass_;
 | |
|     std::string db_;
 | |
|   };
 | |
| 
 | |
|   std::string to_var_name(const std::string& from) const;
 | |
|   std::string get_env_value(const char* name, const char* def_val);
 | |
|   int db_gc();
 | |
|   int mysql_host_detect();
 | |
| 
 | |
|   DBConfig db_;
 | |
| 
 | |
|   inline int log_init_core_operation();
 | |
|   common::sqlclient::ObSingleMySQLConnectionPool sql_conn_pool_;
 | |
|   common::ObMySQLProxy sql_proxy_;
 | |
|   DISALLOW_COPY_AND_ASSIGN(DBInitializer);
 | |
| };
 | |
| 
 | |
| }  // end namespace schema
 | |
| }  // end namespace share
 | |
| }  // end namespace oceanbase
 | |
| 
 | |
| // all implemented in .h file for convenient: no need to link
 | |
| #include "db_initializer.ipp"
 | |
| 
 | |
| #endif  // OCEANBASE_SCHEMA_DB_INITIALIZER_H_
 | 
