def get_room(self, room_id):
room = self._rooms.get(room_id)
if not room:
try:
room = self.query_room_from_db(room_id)
except Exception as e:
room = None
func.log_exception('[RoomManager] get_room room_id: {} do not exist, error: {}'.format(
room_id, e.message
))
return room
def execute_sql(self, sql):
conn = self.get_conn(5)
cursor = conn.cursor()
try:
count = cursor.execute(sql)
conn.commit()
cursor.close()
conn.close()
return count
except Exception as e:
cursor.close()
conn.close()
func.log_exception('[DBPool] execute_sql exception: {}, sql: {}'.format(e.message, sql))
return None
def execute_sql(self, sql):
conn = self.get_conn(5)
cursor = conn.cursor()
try:
count = cursor.execute(sql)
conn.commit()
cursor.close()
conn.close()
return count
except Exception as e:
cursor.close()
conn.close()
func.log_exception(
'[DBPool] execute_sql exception: {}, sql: {}'.format(
e.message, sql))
return None
def query_one(self, sql, dict_cursor=False):
conn = self.get_conn(0)
if dict_cursor:
cursor = conn.cursor(cursorclass=MySQLdb.cursors.DictCursor)
else:
cursor = conn.cursor()
try:
cursor.execute(sql)
results = cursor.fetchone()
cursor.close()
conn.close()
return results
except Exception as e:
cursor.close()
conn.cursor()
func.log_exception('[DBPool] query_one exception: {}, sql: {}'.format(e.message, sql))
return None
def execute_auto_increment(self, sql):
conn = self.get_conn(5)
cursor = conn.cursor()
try:
count = cursor.execute(sql)
conn.commit()
if count > 0:
increment_id = cursor.lastrowid
else:
increment_id = 0
cursor.close()
conn.close()
return increment_id
except Exception as e:
cursor.close()
conn.close()
func.log_exception('[DBPool] execute_auto_increment exception: {}, sql: {}'.format(e.message, sql))
return None
def query_one(self, sql, dict_cursor=False):
conn = self.get_conn(0)
if dict_cursor:
cursor = conn.cursor(cursorclass=MySQLdb.cursors.DictCursor)
else:
cursor = conn.cursor()
try:
cursor.execute(sql)
results = cursor.fetchone()
cursor.close()
conn.close()
return results
except Exception as e:
cursor.close()
conn.cursor()
func.log_exception(
'[DBPool] query_one exception: {}, sql: {}'.format(
e.message, sql))
return None
def execute_auto_increment(self, sql):
conn = self.get_conn(5)
cursor = conn.cursor()
try:
count = cursor.execute(sql)
conn.commit()
if count > 0:
increment_id = cursor.lastrowid
else:
increment_id = 0
cursor.close()
conn.close()
return increment_id
except Exception as e:
cursor.close()
conn.close()
func.log_exception(
'[DBPool] execute_auto_increment exception: {}, sql: {}'.
format(e.message, sql))
return None
def execute_multi_sql(self, sql_list):
conn = self.get_conn(5)
cursor = conn.cursor()
try:
conn.autocommit(False)
for sql in sql_list:
count = cursor.execute(sql)
if count < 0:
raise
else:
conn.commit()
conn.autocommit(True)
cursor.close()
conn.close()
return True
except Exception as e:
conn.rollback()
conn.autocommit(True)
cursor.close()
conn.close()
func.log_exception('[DBPool] execute_sql exception: {}, sql_list: {}'.format(e.message, sql_list))
return False
def execute_multi_sql(self, sql_list):
conn = self.get_conn(5)
cursor = conn.cursor()
try:
conn.autocommit(False)
for sql in sql_list:
count = cursor.execute(sql)
if count < 0:
raise
else:
conn.commit()
conn.autocommit(True)
cursor.close()
conn.close()
return True
except Exception as e:
conn.rollback()
conn.autocommit(True)
cursor.close()
conn.close()
func.log_exception(
'[DBPool] execute_sql exception: {}, sql_list: {}'.format(
e.message, sql_list))
return False
def rollback(self):
try:
self._cur_conn.rollback()
except Exception as e:
func.log_exception('[DBPool] rollback exception: {}'.format(e.message))
def commit(self):
try:
self._cur_conn.commit()
except Exception as e:
func.log_exception('[DBPool] commit exception: {}'.format(e.message))
def rollback(self):
try:
self._cur_conn.rollback()
except Exception as e:
func.log_exception('[DBPool] rollback exception: {}'.format(
e.message))
def commit(self):
try:
self._cur_conn.commit()
except Exception as e:
func.log_exception('[DBPool] commit exception: {}'.format(
e.message))