admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

support sqlite

Browse Source
This commit is contained in:
peng.xu 2019-09-21 13:32:29 +08:00
parent 6acddae130
commit ce95b50143
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mishards/db_base.py
View File

@ -1,5 +1,6 @@
import logging
from sqlalchemy import create_engine
from sqlalchemy.engine.url import make_url
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, scoped_session
@ -12,7 +13,11 @@ class DB:
uri and self.init_db(uri, echo)
def init_db(self, uri, echo=False):
self.engine = create_engine(uri, pool_size=100, pool_recycle=5, pool_timeout=30,
url = make_url(uri)
if url.get_backend_name() == 'sqlite':
self.engine = create_engine(url)
else:
self.engine = create_engine(uri, pool_size=100, pool_recycle=5, pool_timeout=30,
pool_pre_ping=True,
echo=echo,
max_overflow=0)