milvus/pyengine/engine/controller/meta_manager.py
jinhai 6276006645 Update for storage
Former-commit-id: 42039c22af7c1a141894fa9d214c8b43c7f25847
2019-04-09 11:26:12 +08:00

35 lines
1.1 KiB
Python

from engine.model.group_table import GroupTable
from engine.model.file_table import FileTable
from engine.controller.error_code import ErrorCode
class MetaManager(object):
def Sync(self):
db.session.commit()
def AddGroup(self, group_name, dimension):
group = GroupTable.query.filter(GroupTable.group_name==group_id).first()
if group:
return ErrorCode.ALREADY_EXIST, group_name
else:
new_group = GroupTable(group_name, dimension)
GroupHandler.CreateGroupDirectory(group_id)
# add into database
db.session.add(new_group)
self.Sync()
return ErrorCode.SUCCESS_CODE, group_name
@staticmethod
def GetGroup(group_name):
group = GroupTable.query.filter(GroupTable.group_name==group_name).first()
if group:
return ErrorCode.SUCCESS_CODE, group
else:
return ErrorCode.FAULT_CODE, None
# def DeleteGroup(group_id):
# group = GroupTable.query.filter(GroupTable.group_name==group_id).first()
# if(group):
# db.session.delete(group)
# else: