from .Framework import SocketService


class MySQLSocket(SocketService):

    def __init__(self, socket='/var/lib/mysql/mysql.sock', port=3306):
        SocketService.__init__(self)

        self.unix_socket = socket
        # self.port = port
        self.request = 'Random string just to get a response\r\n'.encode()

    def _get_data(self):
        # For our Zabbix Triggers, 0 is good, 1 is bad.
        data = self._get_raw_data()
        if data and data.endswith('Got packets out of order'):
            return 0
        return 1

    def check(self):
        data = self.get_data()
        if data == 0:
            return True
        return False