107 lines
3.0 KiB
Python
107 lines
3.0 KiB
Python
# -*- coding:utf8 -*-
|
|
# pip install clickhouse-driver
|
|
from clickhouse_driver import Client
|
|
from appPublic.argsConvert import ArgsConvert, ConditionConvert
|
|
from .sor import SQLor
|
|
from .ddl_template_clickhouse import clickhouse_ddl_tmpl
|
|
from .const import ROWS
|
|
|
|
class ClickHouseor(SQLor):
|
|
ddl_template = clickhouse_ddl_tmpl
|
|
|
|
db2modelTypeMapping = {
|
|
'int8': 'short',
|
|
'int16': 'short',
|
|
'int32': 'long',
|
|
'int64': 'long',
|
|
'float32': 'float',
|
|
'float64': 'float',
|
|
'decimal': 'float',
|
|
'string': 'str',
|
|
'date': 'date',
|
|
'datetime': 'datetime',
|
|
'uuid': 'str',
|
|
'bool': 'short',
|
|
}
|
|
|
|
model2dbTypemapping = {
|
|
'short': 'Int32',
|
|
'long': 'Int64',
|
|
'float': 'Float64',
|
|
'str': 'String',
|
|
'char': 'String',
|
|
'date': 'Date',
|
|
'datetime': 'DateTime',
|
|
'timestamp': 'DateTime',
|
|
'text': 'String',
|
|
'bin': 'String',
|
|
'file': 'String',
|
|
}
|
|
|
|
@classmethod
|
|
def isMe(self, name):
|
|
return name.lower() in ('clickhouse', 'clickhouse_driver')
|
|
|
|
def placeHolder(self, varname, pos=None):
|
|
if varname == '__mainsql__':
|
|
return ''
|
|
return '%s'
|
|
|
|
def dataConvert(self, dataList):
|
|
if isinstance(dataList, dict):
|
|
d = [i for i in dataList.values()]
|
|
else:
|
|
d = [i['value'] for i in dataList]
|
|
return tuple(d)
|
|
|
|
def pagingSQLmodel(self):
|
|
# ClickHouse 支持 LIMIT offset, size
|
|
return """SELECT * FROM (%s) AS A LIMIT $[from_line]$, $[rows]$"""
|
|
|
|
def tablesSQL(self):
|
|
return """SELECT name, comment AS title
|
|
FROM system.tables
|
|
WHERE database = '%s'""" % self.dbdesc.get('dbname', 'default')
|
|
|
|
def fieldsSQL(self, tablename=None):
|
|
sql = """SELECT
|
|
name AS name,
|
|
type AS type,
|
|
NULL AS length,
|
|
NULL AS dec,
|
|
'yes' AS nullable,
|
|
comment AS title,
|
|
'%s' AS table_name
|
|
FROM system.columns
|
|
WHERE database = '%s'
|
|
""" % (tablename or '', self.dbdesc.get('dbname', 'default'))
|
|
if tablename:
|
|
sql += " AND table = '%s';" % tablename
|
|
return sql
|
|
|
|
def pkSQL(self, tablename=None):
|
|
# ClickHouse 没有 system.keys 表，用 order_by_keys 替代
|
|
sql = """SELECT name FROM system.columns
|
|
WHERE database = '%s' AND table = '%s' AND is_in_primary_key = 1;
|
|
""" % (self.dbdesc.get('dbname', 'default'), tablename.lower())
|
|
return sql
|
|
|
|
def indexesSQL(self, tablename=None):
|
|
# ClickHouse 没有传统索引
|
|
return """SELECT name, 'order_by' AS index_name, 'primary' AS is_unique, name AS column_name
|
|
FROM system.columns
|
|
WHERE database = '%s' AND table = '%s' AND is_in_primary_key = 1;
|
|
""" % (self.dbdesc.get('dbname', 'default'), tablename.lower())
|
|
|
|
def fkSQL(self, tablename=None):
|
|
# ClickHouse 不支持外键
|
|
return "SELECT 'ClickHouse does not support foreign keys' AS msg;"
|
|
|
|
async def connect(self):
|
|
self.conn = Client(**self.dbdesc)
|
|
self.cur = self.conn
|
|
|
|
async def close(self):
|
|
self.conn.close()
|
|
|