oceanbase

1
#!/usr/bin/python
2
import sys
3
import re
4
import os
5
import getopt
6
import mysql.connector
7
import argparse
8
from mysql.connector import errorcode
9

10
class SrsDataImporter:
11
    def get_args(self):
12
        parser = argparse.ArgumentParser(conflict_handler='resolve')
13
        parser.add_argument("-h", "--host", help="Connect to host", required=True)
14
        parser.add_argument("-P", "--port", help="Port number to use for connection", required=True)
15
        parser.add_argument("-p", "--password", default = '', help="Password of sys tenant")
16
        parser.add_argument("-f", "--file", help="The sql script for loading default spatial reference system data", required=True)
17
        parser.add_argument("-t", "--tenant", default='sys', help="Tenant for import data if not sys")
18
        args = parser.parse_args()
19
        self.host=args.host
20
        self.port=args.port
21
        self.pwd=args.password
22
        self.file_name=args.file
23
        self.tenant=args.tenant
24
        self.old_result_cnt = [0, 0]
25
        self.new_result_cnt = [0, 0]
26
        self.print_srs_sql = False
27
        print("host:{0} port:{1} pwd:{2} file:{3}".format(self.host, self.port, self.pwd, self.file_name))
28

29
    def generate_sql(self):
30
        self.sql_list = []
31
        with open(self.file_name) as f_read:
32
            for line in f_read:
33
                if not line.isspace():
34
                    self.sql_list.append(line)
35

36
    def connect_server(self):
37
        try:
38
            self.conn = mysql.connector.connect(user='root', password=self.pwd, host=self.host, port=self.port, database='mysql', connection_timeout=10)
39
        except mysql.connector.Error as err:
40
            print(err)
41
            exit("ERROR: failed to connect host")
42
        self.cur = self.conn.cursor(buffered=True)
43
        print("INFO: sucess to connect server {0}:{1}".format(self.host, self.port))
44
        try:
45
            sql = "select value from oceanbase.__all_sys_parameter where name = 'enable_upgrade_mode';"
46
            self.cur.execute(sql)
47
            print("INFO: execute sql -- {0}".format(sql))
48
            result = self.cur.fetchall()
49
            if 1 == len(result) and 1 == result[0][0]:
50
                self.upgrade_mode = True
51
            else:
52
                self.upgrade_mode = False
53
                sql = "select tenant_id from oceanbase.__all_tenant where tenant_name = '{0}';".format(str(self.tenant))
54
                print("INFO: execute sql -- {0}".format(sql))
55
                self.cur.execute(sql)
56
                result = self.cur.fetchall()
57
                if 1 == len(result):
58
                    print("tenant_id = {0}".format(str(result[0][0])))
59
                    self.tenant_id = result[0][0]
60
                else:
61
                    print("multiple tenants with the same name, tenant_name: '{0}'".format(self.tenant))
62
            if False == self.upgrade_mode and self.tenant_id != 1:
63
                sql = "commit"
64
                self.cur.execute(sql)
65
                sql = "alter system change tenant '{0}'".format(str(self.tenant))
66
                print("INFO: execute sql -- {0}".format(sql))
67
                self.cur.execute(sql)
68
        except mysql.connector.Error as err:
69
            print("ERROR: " + sql)
70
            print(err)
71
            exit("ERROR: failed to import srs data")
72
    def execute_sql(self):
73
        try:
74
            for sql in self.sql_list:
75
                if self.print_srs_sql == True:
76
                    print("INFO: execute sql -- {0}".format(sql))
77
                self.cur.execute(sql)
78
        except mysql.connector.Error as err:
79
            print("ERROR: " + sql)
80
            print(err)
81
            self.conn.rollback()
82
            print("ERROR: failed to import srs data")
83
        else:
84
            print("INFO: succeed to import srs data")
85

86

87
    def execute_prepare_execute(self, table_name, idx):
88
        try:
89
            self.cur.execute("select count(*) from {0}".format(table_name))
90
            result = self.cur.fetchone()
91
            self.old_result_cnt[idx] = result[0]
92
            print("INFO: {0} old result rows -- {1}".format(table_name, self.old_result_cnt[idx]))
93
        except mysql.connector.Error as err:
94
            print(err)
95
            exit("ERROR: failed to import srs data")
96

97
    def prepare_execute(self):
98
        self.execute_prepare_execute("oceanbase.__all_spatial_reference_systems", 0)
99

100
    def execute_check_sql(self, table_name, idx):
101
        self.cur.execute("select count(*) from {0}".format(table_name))
102
        result = self.cur.fetchone()
103
        self.new_result_cnt[idx] = result[0]
104
        print("INFO: {0} old result rows -- {1}".format(table_name, self.old_result_cnt[idx]))
105
        print("INFO: {0} new result rows -- {1}".format(table_name, self.new_result_cnt[idx]))
106

107
    def check_result(self):
108
        self.execute_check_sql("oceanbase.__all_spatial_reference_systems", 0)
109

110
def main():
111
    srs_data_importer = SrsDataImporter()
112
    srs_data_importer.get_args()
113
    srs_data_importer.connect_server()
114
    if False == srs_data_importer.upgrade_mode:
115
        srs_data_importer.generate_sql()
116
        srs_data_importer.prepare_execute()
117
        srs_data_importer.execute_sql()
118
        srs_data_importer.check_result()
119
    else:
120
        print("ERROR: cannot import srs data when upgrade_mode is true")
121

122
if __name__ == "__main__":
123
    main()
124