#!/usr/bin/python

#coding:utf8

import sys

import time

import smtplib

import logging

from email.mime.text import MIMEText

from email.message import Message

from email.header import Header

alarm_mail =['xxxxxx@163.com']

def main():

    failover_time=time.strftime("%Y-%m-%d %H:%M:%S")

    logging.basicConfig(level=logging.DEBUG,

                format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',

                datefmt='%Y-%m-%d %H:%M:%S',

                filename='/sh/redis/failover.log',

                filemode='a')

    console = logging.StreamHandler()

    console.setLevel(logging.INFO)

    formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')

    console.setFormatter(formatter)

    logging.getLogger('').addHandler(console)

    mail_host='xxxxx'

    mail_port=25

    mail_user='xxxxxxx'

    mail_pass='xxxxxxxx'

    mail_send_from = 'xxxxxxx'

    def send_mail(to_list,sub,content):

        me=mail_send_from

        msg = MIMEText(content, _subtype='html', _charset='utf-8')

        msg['Subject'] = Header(sub,'utf-8')

        msg['From'] = Header(me,'utf-8')

        msg['To'] = ";".join(to_list)

        try:

            smtp = smtplib.SMTP()

            smtp.connect(mail_host,mail_port)

            smtp.login(mail_user,mail_pass)

            smtp.sendmail(me,to_list, msg.as_string())

            smtp.close()

            return True

        except Exception as error:

            logging.error("邮件发送失败: %s" % (error))

            return False

    try:

        master_name = sys.argv[1]

        role = sys.argv[2]

        from_ip = sys.argv[4]

        from_port = sys.argv[5]

        to_ip = sys.argv[6]

        to_port = sys.argv[7]

    except Exception as error:

        logging.error('从 Sentinel 获取参数错误: %s ' % (error))

        sys.exit(1)

    sub='redis %s faiover' % (master_name)

    nodify_message = "%s %s is failover end. sentinel find redis master %s:%s is down. failover to slave %s:%s" % (failover_time,master_name,from_ip,from_port,to_ip,to_port)

    if role == 'leader':

        logging.info(nodify_message)

        send_mail(alarm_mail,sub,nodify_message)

if __name__ == "__main__":

    main()

#!/usr/bin/python

import redis

import os

sentinel_server=['10.36.30.203:26379','10.36.30.204:26379','10.37.124.202:26379','10.37.124.203:26379','10.37.124.204:26379']

def queue(host,port):

    str=''.join(map(lambda xx:(hex(ord(xx))[2:]),os.urandom(16)))

    pool = redis.ConnectionPool(host=host, port=port, db=0)

    r = redis.Redis(connection_pool=pool)

    r.lpush('low_task_queue',str)

def get_sentinel():

    global master_host

    global master_port

    for info in sentinel_server:

        host=info.split(':')[0]

        port=info.split(':')[1]

        try:

            r = redis.Redis(host=host, port=port)

            info=r.info('sentinel')['master0']['address'].split(':')

            master_host=info[0]

            master_port=info[1]

        except Exception as error:

            print 'concat to sentinel error: %s' % (error)

            pass

        else:

            break 

if __name__ == "__main__":

    get_sentinel()

    while True:

        try:

            queue(master_host,master_port)

        except Exception as error:

            print 'conct redis error %s' % (error)

            get_sentinel()

            continue