#!/usr/bin/env python3
# -*- coding:utf-8 -*-

import logging, time
import pymysql


class MysqLock:
    def __init__(self, db):
        self.db = db

    def _execute(self, sql):
        cursor = self.db.cursor()
        try:
            ret = None
            cursor.execute(sql)
            if cursor.rowcount != 1:
                logging.error("Multiple rows returned in mysql lock function.")
                ret = None
            else:
                ret = cursor.fetchone()
            cursor.close()
            return ret
        except Exception as ex:
            logging.error("Execute sql \"%s\" failed! Exception: %s", sql, str(ex))
            cursor.close()
            return None

    def get_lock(self, lockstr, timeout):
        """
        设置锁
        """
        sql = "SELECT GET_LOCK('%s', %s)" % (lockstr, timeout)
        ret = self._execute(sql)

        if ret[0] == 0:
            logging.debug("Another client has previously locked '%s'.", lockstr)
            return False
        elif ret[0] == 1:
            logging.debug("The lock '%s' was obtained successfully.", lockstr)
            return True
        else:
            logging.error("Error occurred!")
            return None

    def release_lock(self, lockstr):
        """
        释放锁
        """
        sql = "SELECT RELEASE_LOCK('%s')" % lockstr
        ret = self._execute(sql)
        if ret[0] == 0:
            # 线程尚未被创建锁
            logging.debug("The lock '%s' the lock is not released(the lock was not established by this thread).",
                          lockstr)
            return False
        elif ret[0] == 1:
            # 锁被解开
            logging.debug("The lock '%s' the lock was released.", lockstr)
            return True
        else:
            # 锁不存在
            logging.error("The lock '%s' did not exist.", lockstr)
            return None
        # Init logging


def init_logging():
    sh = logging.StreamHandler()
    logger = logging.getLogger()
    logger.setLevel(logging.DEBUG)
    formatter = logging.Formatter('%(asctime)s -%(module)s:%(filename)s-L%(lineno)d-%(levelname)s: %(message)s')
    sh.setFormatter(formatter)
    logger.addHandler(sh)
    logging.info("Current log level is : %s", logging.getLevelName(logger.getEffectiveLevel()))


def main():
    init_logging()
    db = pymysql.connect(host='localhost', user='root', passwd='123456')
    lock_name = 'queue'

    lock = MysqLock(db)
    ret = lock.get_lock(lock_name, 10)
    if not ret:
        logging.error("Can't get lock! exit!")
        quit()
    time.sleep(10)
    logging.info("You can do some synchronization work across processes!")

    # TODO
    # 可以处理一些事情
    lock.release_lock(lock_name)


if __name__ == "__main__":
    main()