thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-22 15:42:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #31724 from ClickHouse/add-another-ci-database

Browse Source 
Try to push data into another ci database.
This commit is contained in:
Nikolai Kochetov 2021-11-26 11:22:56 +03:00 committed by GitHub
commit 260928cb95
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/ci/clickhouse_helper.py
View File

@ -8,8 +8,16 @@ from get_robot_token import get_parameter_from_ssm
class ClickHouseHelper:
def __init__(self, url=None, user=None, password=None):
self.url2 = None
self.auth2 = None
if url is None:
url = get_parameter_from_ssm("clickhouse-test-stat-url")
self.url2 = get_parameter_from_ssm("clickhouse-test-stat-url2")
self.auth2 = {
'X-ClickHouse-User': get_parameter_from_ssm("clickhouse-test-stat-login2"),
'X-ClickHouse-Key': ''
}
self.url = url
self.auth = {
@ -17,7 +25,8 @@ class ClickHouseHelper:
'X-ClickHouse-Key': password if password is not None else get_parameter_from_ssm("clickhouse-test-stat-password")
}
def _insert_json_str_info(self, db, table, json_str):
@staticmethod
def _insert_json_str_info_impl(url, auth, db, table, json_str):
params = {
'database': db,
'query': 'INSERT INTO {table} FORMAT JSONEachRow'.format(table=table),
@ -26,7 +35,7 @@ class ClickHouseHelper:
}
for i in range(5):
response = requests.post(self.url, params=params, data=json_str, headers=self.auth, verify=False)
response = requests.post(url, params=params, data=json_str, headers=auth, verify=False)
logging.info("Response content '%s'", response.content)
@ -49,6 +58,11 @@ class ClickHouseHelper:
else:
raise Exception(error)
def _insert_json_str_info(self, db, table, json_str):
self._insert_json_str_info_impl(self.url, self.auth, db, table, json_str)
if self.url2:
self._insert_json_str_info_impl(self.url2, self.auth2, db, table, json_str)
def insert_event_into(self, db, table, event):
event_str = json.dumps(event)
self._insert_json_str_info(db, table, event_str)