mirror of https://github.com/midoks/mdserver-web
parent
c94a662235
commit
b4acb2942f
@ -0,0 +1,215 @@ |
|||||||
|
# coding:utf-8 |
||||||
|
|
||||||
|
import sys |
||||||
|
import io |
||||||
|
import os |
||||||
|
import time |
||||||
|
import re |
||||||
|
import pymongo |
||||||
|
import json |
||||||
|
|
||||||
|
|
||||||
|
sys.path.append(os.getcwd() + "/class/core") |
||||||
|
import mw |
||||||
|
|
||||||
|
def singleton(cls): |
||||||
|
_instance = {} |
||||||
|
|
||||||
|
def inner(): |
||||||
|
if cls not in _instance: |
||||||
|
_instance[cls] = cls() |
||||||
|
return _instance[cls] |
||||||
|
return inner |
||||||
|
|
||||||
|
|
||||||
|
@singleton |
||||||
|
class nosqlMySQL(): |
||||||
|
|
||||||
|
__DB_PASS = None |
||||||
|
__DB_USER = None |
||||||
|
__DB_PORT = 6379 |
||||||
|
__DB_HOST = '127.0.0.1' |
||||||
|
__DB_CONN = None |
||||||
|
__DB_ERR = None |
||||||
|
|
||||||
|
__DB_LOCAL = None |
||||||
|
|
||||||
|
def __init__(self): |
||||||
|
self.__config = self.get_options(None) |
||||||
|
|
||||||
|
|
||||||
|
def mgdb_conn(self): |
||||||
|
|
||||||
|
if self.__DB_HOST in ['127.0.0.1', 'localhost']: |
||||||
|
my_path = "{}/mysql".format(mw.getServerDir()) |
||||||
|
if not os.path.exists(my_path): return False |
||||||
|
|
||||||
|
if not self.__DB_LOCAL: |
||||||
|
self.__DB_PORT = int(self.__config['port']) |
||||||
|
|
||||||
|
# print(self.__DB_HOST,self.__DB_PORT, self.__DB_PASS) |
||||||
|
try: |
||||||
|
self.__DB_CONN = pymongo.MongoClient(host=self.__DB_HOST, port=self.__DB_PORT, maxPoolSize=10) |
||||||
|
self.__DB_CONN.admin.command('ping') |
||||||
|
return self.__DB_CONN |
||||||
|
except pymongo.errors.ConnectionFailure: |
||||||
|
return False |
||||||
|
except Exception: |
||||||
|
self.__DB_ERR = mw.get_error_info() |
||||||
|
return False |
||||||
|
|
||||||
|
# 获取配置项 |
||||||
|
def get_options(self, get=None): |
||||||
|
result = {} |
||||||
|
mgdb_content = mw.readFile("{}/mysql/my.conf".format(mw.getServerDir())) |
||||||
|
if not mgdb_content: return False |
||||||
|
|
||||||
|
keys = ["bind_ip", "port"] |
||||||
|
|
||||||
|
result['host'] = '127.0.0.1' |
||||||
|
rep = 'port\s*=\s*(.*)' |
||||||
|
ip_re = re.search(rep, mgdb_content) |
||||||
|
if ip_re: |
||||||
|
result['port'] = int(ip_re.groups()[0].strip()) |
||||||
|
else: |
||||||
|
result['port'] = 27017 |
||||||
|
return result |
||||||
|
|
||||||
|
def set_host(self, host, port, name, username, password, prefix=''): |
||||||
|
self.__DB_HOST = host |
||||||
|
self.__DB_PORT = int(port) |
||||||
|
self.__DB_NAME = name |
||||||
|
if self.__DB_NAME: self.__DB_NAME = str(self.__DB_NAME) |
||||||
|
self.__DB_USER = str(username) |
||||||
|
self._USER = str(username) |
||||||
|
self.__DB_PASS = str(password) |
||||||
|
self.__DB_PREFIX = prefix |
||||||
|
self.__DB_LOCAL = 1 |
||||||
|
return self |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@singleton |
||||||
|
class nosqlMySQLCtr(): |
||||||
|
|
||||||
|
def __init__(self): |
||||||
|
pass |
||||||
|
|
||||||
|
def getInstanceBySid(self, sid = 0): |
||||||
|
instance = nosqlMySQL() |
||||||
|
return instance |
||||||
|
|
||||||
|
def getDbList(self, args): |
||||||
|
sid = args['sid'] |
||||||
|
mgdb_instance = self.getInstanceBySid(sid).mgdb_conn() |
||||||
|
if mgdb_instance is False: |
||||||
|
return mw.returnData(False,'无法链接') |
||||||
|
|
||||||
|
result = {} |
||||||
|
doc_list = mgdb_instance.list_database_names() |
||||||
|
rlist = [] |
||||||
|
for x in doc_list: |
||||||
|
if not x in ['admin', 'config', 'local']: |
||||||
|
rlist.append(x) |
||||||
|
result['list'] = rlist |
||||||
|
return mw.returnData(True,'ok', result) |
||||||
|
|
||||||
|
def getCollectionsList(self, args): |
||||||
|
sid = args['sid'] |
||||||
|
name = args['name'] |
||||||
|
|
||||||
|
mgdb_instance = self.getInstanceBySid(sid).mgdb_conn() |
||||||
|
if mgdb_instance is False: |
||||||
|
return mw.returnData(False,'无法链接') |
||||||
|
|
||||||
|
result = {} |
||||||
|
collections = mgdb_instance[name].list_collection_names() |
||||||
|
result['collections'] = collections |
||||||
|
return mw.returnData(True,'ok', result) |
||||||
|
|
||||||
|
def getDataList(self, args): |
||||||
|
sid = args['sid'] |
||||||
|
db = args['db'] |
||||||
|
collection = args['collection'] |
||||||
|
p = 1 |
||||||
|
size = 10 |
||||||
|
if 'p' in args: |
||||||
|
p = args['p'] |
||||||
|
|
||||||
|
if 'size' in args: |
||||||
|
size = args['size'] |
||||||
|
|
||||||
|
mgdb_instance = self.getInstanceBySid(sid).mgdb_conn() |
||||||
|
if mgdb_instance is False: |
||||||
|
return mw.returnData(False,'无法链接') |
||||||
|
|
||||||
|
db_instance = mgdb_instance[db] |
||||||
|
collection_instance = db_instance[collection] |
||||||
|
|
||||||
|
start_index = (p - 1) * size |
||||||
|
end_index = p * size |
||||||
|
args_where = args['where'] |
||||||
|
|
||||||
|
where = {} |
||||||
|
if 'field' in args_where: |
||||||
|
mg_field = args_where['field'] |
||||||
|
|
||||||
|
if mg_field == '_id': |
||||||
|
mg_value = ObjectId(args_where['value']) |
||||||
|
where[mg_field] = mg_value |
||||||
|
else: |
||||||
|
mg_value = args_where['value'] |
||||||
|
where[mg_field] = re.compile(mg_value) |
||||||
|
|
||||||
|
# print(where) |
||||||
|
result = collection_instance.find(where).skip(start_index).limit(size).sort({'_id':-1}) |
||||||
|
count = collection_instance.count_documents(where) |
||||||
|
d = [] |
||||||
|
for document in result: |
||||||
|
d.append(document) |
||||||
|
|
||||||
|
doc_str_json = dumps(d) |
||||||
|
result = json.loads(doc_str_json) |
||||||
|
|
||||||
|
|
||||||
|
page_args = {} |
||||||
|
page_args['count'] = count |
||||||
|
page_args['tojs'] = 'mongodbDataList' |
||||||
|
page_args['p'] = p |
||||||
|
page_args['row'] = size |
||||||
|
|
||||||
|
rdata = {} |
||||||
|
rdata['page'] = mw.getPage(page_args) |
||||||
|
rdata['list'] = result |
||||||
|
rdata['count'] = count |
||||||
|
|
||||||
|
rdata['soso_field'] = '' |
||||||
|
if 'field' in args_where: |
||||||
|
rdata['soso_field'] = args_where['field'] |
||||||
|
|
||||||
|
|
||||||
|
return mw.returnData(True,'ok', rdata) |
||||||
|
|
||||||
|
|
||||||
|
# ---------------------------------- run ---------------------------------- |
||||||
|
# 获取 mysql 列表 |
||||||
|
def get_db_list(args): |
||||||
|
t = nosqlMySQLCtr() |
||||||
|
return t.getDbList(args) |
||||||
|
|
||||||
|
# 获取 mysql 列表 |
||||||
|
def get_collections_list(args): |
||||||
|
t = nosqlMySQLCtr() |
||||||
|
return t.getCollectionsList(args) |
||||||
|
|
||||||
|
def get_data_list(args): |
||||||
|
t = nosqlMySQLCtr() |
||||||
|
return t.getDataList(args) |
||||||
|
|
||||||
|
# 测试 |
||||||
|
def test(args): |
||||||
|
sid = args['sid'] |
||||||
|
t = nosqlMySQLCtr() |
||||||
|
print(t.get_options()) |
||||||
|
print("test") |
||||||
|
return 'ok' |
||||||
Loading…
Reference in new issue