hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[GIS] fix import srs data to tenant whose name is the keyword

Browse Source
This commit is contained in:
lyxiong0
2024-01-09 12:47:43 +00:00
committed by ob-robot
parent f2bccfded9
commit 875dc52f26
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
tools/import_srs_data.py
View File

@ -23,8 +23,8 @@ class SrsDataImporter:
self.tenant=args.tenant self.tenant=args.tenant
self.old_result_cnt = [0, 0] self.old_result_cnt = [0, 0]
self.new_result_cnt = [0, 0] self.new_result_cnt = [0, 0]
self.print_srs_sql = False; self.print_srs_sql = False
print "host:{0} port:{1} pwd:{2} file:{3}".format(self.host, self.port, self.pwd, self.file_name) print("host:{0} port:{1} pwd:{2} file:{3}".format(self.host, self.port, self.pwd, self.file_name))
def generate_sql(self): def generate_sql(self):
self.sql_list = [] self.sql_list = []
@ -40,44 +40,44 @@ class SrsDataImporter:
print(err) print(err)
exit("ERROR: failed to connect host") exit("ERROR: failed to connect host")
self.cur = self.conn.cursor(buffered=True) self.cur = self.conn.cursor(buffered=True)
print "INFO: sucess to connect server {0}:{1}".format(self.host, self.port) print("INFO: sucess to connect server {0}:{1}".format(self.host, self.port))
try: try:
sql = "select value from oceanbase.__all_sys_parameter where name = 'enable_upgrade_mode';" sql = "select value from oceanbase.__all_sys_parameter where name = 'enable_upgrade_mode';"
self.cur.execute(sql) self.cur.execute(sql)
print "INFO: execute sql -- {0}".format(sql) print("INFO: execute sql -- {0}".format(sql))
result = self.cur.fetchall() result = self.cur.fetchall()
if 1 == len(result) and 1 == result[0][0]: if 1 == len(result) and 1 == result[0][0]:
self.upgrade_mode = True self.upgrade_mode = True
else: else:
self.upgrade_mode = False self.upgrade_mode = False
sql = "select tenant_id from oceanbase.__all_tenant where tenant_name = '{0}';".format(str(self.tenant)) sql = "select tenant_id from oceanbase.__all_tenant where tenant_name = '{0}';".format(str(self.tenant))
print "INFO: execute sql -- {0}".format(sql) print("INFO: execute sql -- {0}".format(sql))
self.cur.execute(sql) self.cur.execute(sql)
result = self.cur.fetchall() result = self.cur.fetchall()
if 1 == len(result): if 1 == len(result):
print "tenant_id = {0}".format(str(result[0][0])) print("tenant_id = {0}".format(str(result[0][0])))
self.tenant_id = result[0][0] self.tenant_id = result[0][0]
else: else:
print "multiple tenants with the same name, tenant_name: '{0}'".format(self.tenant) print("multiple tenants with the same name, tenant_name: '{0}'".format(self.tenant))
if False == self.upgrade_mode and self.tenant_id != 1: if False == self.upgrade_mode and self.tenant_id != 1:
sql = "commit" sql = "commit"
self.cur.execute(sql) self.cur.execute(sql)
sql = "alter system change tenant " + str(self.tenant) sql = "alter system change tenant '{0}'".format(str(self.tenant))
print "INFO: execute sql -- {0}".format(sql) print("INFO: execute sql -- {0}".format(sql))
self.cur.execute(sql) self.cur.execute(sql)
except mysql.connector.Error as err: except mysql.connector.Error as err:
print("ERROR: " + sql) print("ERROR: " + sql)
print(err); print(err)
exit("ERROR: failed to import srs data") exit("ERROR: failed to import srs data")
def execute_sql(self): def execute_sql(self):
try: try:
for sql in self.sql_list: for sql in self.sql_list:
if self.print_srs_sql == True: if self.print_srs_sql == True:
print "INFO: execute sql -- {0}".format(sql) print("INFO: execute sql -- {0}".format(sql))
self.cur.execute(sql) self.cur.execute(sql)
except mysql.connector.Error as err: except mysql.connector.Error as err:
print("ERROR: " + sql) print("ERROR: " + sql)
print(err); print(err)
self.conn.rollback() self.conn.rollback()
print("ERROR: failed to import srs data") print("ERROR: failed to import srs data")
else: else:
@ -89,9 +89,9 @@ class SrsDataImporter:
self.cur.execute("select count(*) from {0}".format(table_name)) self.cur.execute("select count(*) from {0}".format(table_name))
result = self.cur.fetchone() result = self.cur.fetchone()
self.old_result_cnt[idx] = result[0] self.old_result_cnt[idx] = result[0]
print "INFO: {0} old result rows -- {1}".format(table_name, self.old_result_cnt[idx]) print("INFO: {0} old result rows -- {1}".format(table_name, self.old_result_cnt[idx]))
except mysql.connector.Error as err: except mysql.connector.Error as err:
print(err); print(err)
exit("ERROR: failed to import srs data") exit("ERROR: failed to import srs data")
def prepare_execute(self): def prepare_execute(self):
@ -101,8 +101,8 @@ class SrsDataImporter:
self.cur.execute("select count(*) from {0}".format(table_name)) self.cur.execute("select count(*) from {0}".format(table_name))
result = self.cur.fetchone() result = self.cur.fetchone()
self.new_result_cnt[idx] = result[0] self.new_result_cnt[idx] = result[0]
print "INFO: {0} old result rows -- {1}".format(table_name, self.old_result_cnt[idx]) print("INFO: {0} old result rows -- {1}".format(table_name, self.old_result_cnt[idx]))
print "INFO: {0} new result rows -- {1}".format(table_name, self.new_result_cnt[idx]) print("INFO: {0} new result rows -- {1}".format(table_name, self.new_result_cnt[idx]))
def check_result(self): def check_result(self):
self.execute_check_sql("oceanbase.__all_spatial_reference_systems", 0) self.execute_check_sql("oceanbase.__all_spatial_reference_systems", 0)