Esempio n. 1
0
def tick_to_check_cert_online():
    print('tick_to_check_cert_online now active' +
          str(datetime.datetime.now()))
    cert_collect = []
    dic = {}
    if record_mode == "0":
        tb_vip_certification_list = db_test.read_information_from_temp(2)
    elif record_mode == "1":
        tb_vip_certification_list = db_test.DB_select_connect(
            "select * from dfh.tb_vip_certification")
    for r in tb_vip_certification_list:
        if int(r[4]) >= 2:
            cert_collect.append(r[3])
        elif int(r[4]) < 2:
            if r[1] == 'slogin':
                logout(r[3])
            else:
                if (len(find_cert(r[1], r[2])) > 0):
                    #logout(find_cert(r[1],r[2]))
                    logout(r[3])
                    DB_log("DEBUG", "tick_to_check_cert_online force logout",
                           "id:" + r[1] + ",pwd:" + r[2] + ",cert:" + r[3])
    if len(cert_collect) > 0:
        for item in cert_collect:
            dic.update({'login_cert': item})  #update cert to 1
            db_test.write_temp(dic, 2, 'set')
Esempio n. 2
0
def del_vip(dic_, write_file_TF):
    if write_file_TF:
        txb_sn = db_test.key_exist_in_object('txb_sn', dic_)
        txb_ip = db_test.key_exist_in_object('txb_ip', dic_)
        txb_ver = db_test.key_exist_in_object('txb_ver', dic_)
        txb_area = db_test.key_exist_in_object('txb_area', dic_)
        curr_link = db_test.key_exist_in_object('curr_link', dic_)
        txb_max_link = db_test.key_exist_in_object('txb_max_link', dic_)
        txb_pro = db_test.key_exist_in_object('txb_pro', dic_)
        txb_act = db_test.key_exist_in_object('txb_act', dic_)

        og_txb_ip = db_test.key_exist_in_object('og_txb_ip', dic_)
        og_txb_ver = db_test.key_exist_in_object('og_txb_ver', dic_)
        og_txb_area = db_test.key_exist_in_object('og_txb_area', dic_)
        if record_mode == "0":  # update file
            db_test.write_temp(dic_, 0, "del")
        elif record_mode == "1":
            if len(og_txb_ip) > 0:
                sqlcmd = "delete from dfh.tb_vip_relation where ip = \'" + txb_ip + "\'"
                db_test.DB_update_connect(sqlcmd)
            if len(og_txb_ver) > 0:
                sqlcmd = "update dfh.tb_vip_relation set version = \'\',last_update = timezone(\'CCT\'::text, now()) where version = \'" + txb_ver + "\'"
                db_test.DB_update_connect(sqlcmd)
            if len(og_txb_area) > 0:
                sqlcmd = "update dfh.tb_vip_relation set area = \'\',last_update = timezone(\'CCT\'::text, now()) where area = \'" + txb_area + "\'"
                db_test.DB_update_connect(sqlcmd)
        db_test.write_config(dic_)
        DB_log('DEBUG', 'del_vip', "==== del_vip complete ====")
Esempio n. 3
0
def vip_distribution(ver_, area_):
    group_vip_list = []
    lowest = 999
    vip_lowest = ""
    if record_mode == "0":
        tb_vip_status_list = db_test.read_information_from_temp(0)
        tb_vip_relation_list = db_test.read_information_from_temp(1)
    elif record_mode == "1":
        tb_vip_status_list = DB_select_connect(
            "select * from dfh.tb_vip_status")
        tb_vip_relation_list = DB_select_connect(
            "select * from dfh.tb_vip_relation")
    for r in tb_vip_relation_list:
        if r[2] == ver_ and r[3] == area_ and r[6] == 'T':
            group_vip_list.append(r[1])
    #find lowest
    dic_ = {}
    for r in group_vip_list:
        for rr in tb_vip_status_list:
            if r == rr[0]:
                if int(rr[1]) < int(lowest):
                    lowest = int(rr[1])
                    vip_lowest = rr[0]
    #curr_link = lowest+1#new login vip distribution, so count +1
    curr_link = lowest  #20171108 count +0
    dic_.update({'txb_ip': vip_lowest})
    dic_.update({'curr_link': str(curr_link)})
    if vip_distribution_lock.acquire(1):
        db_test.write_temp(dic_, 0, 'update_relation')
        vip_distribution_lock.release()
    #load>80%, 20151224
    #    curr_link/nax_link
    return vip_lowest
Esempio n. 4
0
def set_config(dic_, write_file_TF):
    if write_file_TF:
        txb_sn = db_test.key_exist_in_object('txb_sn', dic_)
        txb_ip = db_test.key_exist_in_object('txb_ip', dic_)
        txb_ver = db_test.key_exist_in_object('txb_ver', dic_)
        txb_area = db_test.key_exist_in_object('txb_area', dic_)
        curr_link = db_test.key_exist_in_object('curr_link', dic_)
        txb_max_link = db_test.key_exist_in_object('txb_max_link', dic_)
        txb_pro = db_test.key_exist_in_object('txb_pro', dic_)
        txb_act = db_test.key_exist_in_object('txb_act', dic_)

        og_txb_ip = db_test.key_exist_in_object('og_txb_ip', dic_)
        og_txb_ver = db_test.key_exist_in_object('og_txb_ver', dic_)
        og_txb_area = db_test.key_exist_in_object('og_txb_area', dic_)
        if record_mode == "0":  # update file
            db_test.write_temp(dic_, 0, 'set')  # 0 : Overwrite status
            db_test.write_temp(dic_, 1, 'set')  # 1 : Overwrite relation
        elif record_mode == "1":  # update postgraSQL
            if len(og_txb_ip) > 0:
                sqlcmd = "update dfh.tb_vip_relation set ip = \'" + txb_ip + "\',last_update = timezone(\'CCT\'::text, now()) where ip = \'" + og_txb_ip + "\'"
                db_test.DB_update_connect(sqlcmd)
            if len(og_txb_ver) > 0:
                sqlcmd = "update dfh.tb_vip_relation set version = \'" + txb_ver + "\',last_update = timezone(\'CCT\'::text, now()) where version = \'" + og_txb_ver + "\'"
                db_test.DB_update_connect(sqlcmd)
            if len(og_txb_area) > 0:
                sqlcmd = "update dfh.tb_vip_relation set area = \'" + txb_area + "\',last_update = timezone(\'CCT\'::text, now()) where area = \'" + og_txb_area + "\'"
                db_test.DB_update_connect(sqlcmd)
        db_test.write_config(dic_)
        DB_log('DEBUG', 'broker_set_config', '==== set_config complete ====')
Esempio n. 5
0
def count_vip_socket(vip, og_curr_link, int_):
    curr_link_result = 0
    viplist = vip.split(':')
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.connect((viplist[0], 4300))  #60.250.174.141:4300
        sock.settimeout(10)
        #sock.connect(('192.168.12.112', 4300))
        if int_ == 73:
            sock.send(chr(0x73).encode('ascii'))
            data = sock.recv(32)
            curr_link = int.from_bytes(data[2:3], byteorder='little')
        elif int_ == 75:
            #version
            sock.send(chr(0x75).encode('ascii'))
            data = sock.recv(32)
            ver1 = int.from_bytes(data[2:3], byteorder='little')
            ver2 = int.from_bytes(data[3:4], byteorder='little')
            ver3 = int.from_bytes(data[4:5], byteorder='little')
            ver4 = int.from_bytes(data[5:6], byteorder='little')
        elif int_ == 98:
            #sock.connect(('192.168.12.122', 4300))
            sock.send(chr(98).encode('ascii'))
            data = sock.recv(32)
            curr_link = int.from_bytes(data[8:9], byteorder='little')
            #print(curr_link)
        DB_log('DEBUG', 'count_vip_socket sock.recv', str(data))
        sock.close()
    except:
        curr_link = 0
        DB_log('ERROR', 'count_vip_socket', viplist[0] + " socket fail")

    #if int(og_curr_link) >= curr_link:
    #curr_link_result = int(og_curr_link)
    #else:#curr_link > og_curr_link
    curr_link_result = curr_link
    if record_mode == "0":  # update file
        dic = {}
        dic.update({'txb_ip': vip})
        dic.update({'curr_link': curr_link_result})
        if count_vip_socket_lock.acquire(1):
            db_test.write_temp(dic, 0, 'update_relation')  # 0 = status
            count_vip_socket_lock.release()
    elif record_mode == "1":  # update DB
        sqlcmd = "update dfh.tb_vip_status set curr_link = \'" + str(
            curr_link_result
        ) + "\',last_update = timezone(\'CCT\'::text, now()) where ip = \'" + str(
            vip) + "\'"
        DB_update_connect(sqlcmd)
    DB_log(
        "DEBUG", "tick_to_update_curr_link", "ip:" + vip + ",og_curr_link:" +
        str(og_curr_link) + ",curr_link:" + str(curr_link))
Esempio n. 6
0
def mod_list(dic_, write_file_TF):
    if write_file_TF:
        txb_sn = db_test.key_exist_in_object('txb_sn', dic_)
        txb_ip = db_test.key_exist_in_object('txb_ip', dic_)
        txb_ver = db_test.key_exist_in_object('txb_ver', dic_)
        txb_area = db_test.key_exist_in_object('txb_area', dic_)
        curr_link = db_test.key_exist_in_object('curr_link', dic_)
        txb_max_link = db_test.key_exist_in_object('txb_max_link', dic_)
        if txb_ver.split(',')[0] == 'Thin':
            txb_pro = 'D'
        else:
            txb_pro = 'S'
        dic_.update({'txb_pro': txb_pro})
        #txb_pro = db_test.key_exist_in_object('txb_pro',dic_)
        txb_act = db_test.key_exist_in_object('txb_act', dic_)
        if record_mode == "0":
            db_test.write_temp(dic_, 0,
                               'update_relation')  # 0 : Overwrite status
            db_test.write_temp(dic_, 1,
                               'update_relation')  # 1 : Overwrite relation
        elif record_mode == "1":
            sqlcmd = 'update dfh.tb_vip_relation set '
            if len(txb_ver) > 0:
                sqlcmd = sqlcmd + 'version = \'' + txb_ver + '\','
            if len(txb_area) > 0:
                sqlcmd = sqlcmd + 'area = \'' + txb_area + '\','
            if txb_max_link != "--" and len(txb_max_link) > 0:
                sqlcmd = sqlcmd + 'max_link = \'' + txb_max_link + '\','
            if txb_pro != "--" and len(txb_pro) > 0:
                sqlcmd = sqlcmd + 'pro = \'' + txb_pro + '\','
            if txb_act != "--" and len(txb_act) > 0:
                sqlcmd = sqlcmd + 'act = \'' + txb_act + '\','
            sqlcmd = sqlcmd + 'last_update = timezone(\'CCT\'::text, now()) where sn =\'' + txb_sn + '\';'
            db_test.DB_update_connect(sqlcmd)
        DB_log('DEBUG', 'mod_list', '==== mod_list complete ====')
    if record_mode == "0":
        results = db_test.read_information_from_temp(
            1)  #0: tb_vip_status, 1: tb_vip_relation
    elif record_mode == "1":
        sqlcmd = 'select sn,ip,version,area,max_link,pro,act,last_update from dfh.tb_vip_relation ORDER BY sn;'
        results = db_test.DB_select_connect(sqlcmd)
    return results
Esempio n. 7
0
def add_vip(dic_, write_file_TF):
    if write_file_TF:
        txb_sn = db_test.key_exist_in_object('txb_sn', dic_)
        txb_ip = db_test.key_exist_in_object('txb_ip', dic_)
        txb_ver = db_test.key_exist_in_object('txb_ver', dic_)
        txb_area = db_test.key_exist_in_object('txb_area', dic_)
        curr_link = db_test.key_exist_in_object('curr_link', dic_)
        txb_max_link = db_test.key_exist_in_object('txb_max_link', dic_)
        txb_pro = db_test.key_exist_in_object('txb_pro', dic_)
        txb_act = db_test.key_exist_in_object('txb_act', dic_)

        og_txb_ip = db_test.key_exist_in_object('og_txb_ip', dic_)
        og_txb_ver = db_test.key_exist_in_object('og_txb_ver', dic_)
        og_txb_area = db_test.key_exist_in_object('og_txb_area', dic_)

        if record_mode == "0":  # update file
            db_test.write_temp(
                dic_, 0,
                "add")  # add only call once, Overwrite status & relation
        elif record_mode == "1":
            sqlcmd = "insert into dfh.tb_vip_relation (ip,version,area,max_link,pro,act)values(\'" + txb_ip + "\',\'\',\'\',\'100\',\'S\',\'F\');"
            db_test.DB_update_connect(sqlcmd)
        db_test.write_config(dic_)
        DB_log('DEBUG', 'add_vip', "==== add_vip complete ====")
Esempio n. 8
0
def logout(cert_):
    have_cert = ""
    have_id = ""
    if record_mode == "0":
        tb_vip_status_list = db_test.read_information_from_temp(0)
        tb_vip_certification_list = db_test.read_information_from_temp(
            2)  #tb_vip_certification
    elif record_mode == "1":
        # select where have some problem WTF??? 20151229
        tb_vip_status_list = db_test.DB_select_connect(
            "select * from dfh.tb_vip_status")
        tb_vip_certification_list = db_test.DB_select_connect(
            "select ip,account,pwd,cert from dfh.tb_vip_certification")
    for r in tb_vip_certification_list:
        if cert_ == r[3]:
            vip_ip = r[0]
            have_id = r[1]
            have_cert = r[3]
    if len(have_cert) > 0:
        if record_mode == "0":
            dic = {}
            dic.update({'login_cert': have_cert})
            #db_test.write_temp(dic,0,'del_cert')
            # curr_link -1
            for s in tb_vip_status_list:
                if vip_ip == s[0]:
                    if int(s[1]) > 0:
                        curr_link = int(s[1]) - 1
            dic.update({'txb_ip': vip_ip})
            dic.update({'curr_link': str(curr_link)})
            db_test.write_temp(dic, 0, 'update_relation')
        elif record_mode == "1":
            db_test.DB_update_connect(
                "delete from dfh.tb_vip_certification where cert = \'" +
                have_cert + "\'")
        DB_log("DEBUG", "logout", "id = " + have_id + ", cert = " + cert_)