ConsulManager/flask-consul/units/consul_manager.py

import requests,json
import sys 
sys.path.append("..") 
from config import consul_token,consul_url

headers = {'X-Consul-Token': consul_token}

def get_hosts():
    url = f'{consul_url}/agent/host'
    response = requests.get(url, headers=headers)
    if response.status_code == 200:
        info = response.json()
        pmem = round(info["Memory"]["usedPercent"])
        pdisk = round(info["Disk"]["usedPercent"])
        host = {'hostname':info["Host"]["hostname"],'uptime':f'{round(info["Host"]["uptime"]/3600/24)}天',
                 'os':f'{info["Host"]["platform"]} {info["Host"]["platformVersion"]}','kernel':info["Host"]["kernelVersion"]}
        cpu = {'cores':f'{len(info["CPU"])}核','vendorId':info["CPU"][0]["vendorId"],'modelName':info["CPU"][0]["modelName"]}
        memory = {'total':f'{round(info["Memory"]["total"]/1024**3)}GB','available':f'{round(info["Memory"]["available"]/1024**3)}GB',
                  'used':f'{round(info["Memory"]["used"]/1024**3)}GB','usedPercent':f'{pmem}%'}
        disk = {'path':info["Disk"]["path"],'fstype':info["Disk"]["fstype"],'total':f'{round(info["Disk"]["total"]/1024**3)}GB',
                'free':f'{round(info["Disk"]["free"]/1024**3)}GB','used':f'{round(info["Disk"]["used"]/1024**3)}GB','usedPercent':f'{pdisk}%'}
        return {'code': 20000,'host':host,'cpu':cpu,'memory':memory,'disk':disk, 'pmem':pmem, 'pdisk':pdisk}
    else:
        return {'code': 50000, 'data': f'{response.status_code}:{response.text}'}
def get_services():
    url = f'{consul_url}/internal/ui/services'
    response = requests.get(url, headers=headers)
    if response.status_code == 200:
        info = response.json()
        services_list = [{'Name':i['Name'],'Datacenter':i['Datacenter'],'InstanceCount':i['InstanceCount'],'ChecksCritical':i['ChecksCritical'],'ChecksPassing':i['ChecksPassing'],'Tags':i['Tags'],'Nodes':list(set(i['Nodes']))} for i in info if i['Name'] != 'consul']
        return {'code': 20000,'services':services_list}
    else:
        return {'code': 50000, 'data': f'{response.status_code}:{response.text}'}
def get_services_nameonly():
    url = f'{consul_url}/catalog/services'
    response = requests.get(url, headers=headers)
    if response.status_code == 200:
        info = response.json()
        info.pop('consul')
        return {'code': 20000,'services_name':list(info.keys())}
    else:
        return {'code': 50000, 'data': f'{response.status_code}:{response.text}'}
def get_instances(service_name):
    url = f'{consul_url}/health/service/{service_name}'
    response = requests.get(url, headers=headers)
    if response.status_code == 200:
        info = response.json()
        instances_list = []
        for i in info:
            instance_dict = {}
            instance_dict['ID'] = i['Service']['ID']
            instance_dict['name'] = i['Service']['Service']
            instance_dict['tags'] = '无' if i['Service']['Tags'] == [] else i['Service']['Tags']
            instance_dict['address'] = i['Service'].get('Address')
            instance_dict['port'] = i['Service'].get('Port')
            if i['Service']['Meta'] == {} or i['Service']['Meta'] is None:
                instance_dict['meta'] = '无'
            else:
                instance_dict['meta'] = [i['Service']['Meta']]
                instance_dict['meta_label'] = [{'prop': x, 'label': x} for x in i['Service']['Meta'].keys()]
            if len(i['Checks']) ==2:
                instance_dict['status'] = i['Checks'][1]['Status']
                instance_dict['output'] = i['Checks'][1]['Output']
            else:
                instance_dict['status'] = '无'
                instance_dict['output'] = '无'
            instances_list.append(instance_dict)
        return {'code': 20000,'instances':instances_list}
    else:
        return {'code': 50000, 'data': f'{response.status_code}:{response.text}'}

def del_instance(service_id):
    reg = requests.put(f'{consul_url}/agent/service/deregister/{service_id}', headers=headers)
    if reg.status_code == 200:
        return {"code": 20000, "data": f"【{service_id}】删除成功！"}
    else:
        return {"code": 50000, "data": f"{reg.status_code}【{service_id}】{reg.text}"}

def add_instance(instance_dict):
    sid = instance_dict['ID']
    if '//' in sid or sid.startswith('/') or sid.endswith('/'):
        return {"code": 50000, "data": f"服务ID【{sid}】首尾不能包含'/'，并且不能包含两个连续的'/'"}
    isMeta = instance_dict['metaInfo']['isMeta']
    isCheck = instance_dict['checkInfo']['isCheck']
    address = instance_dict['address']
    port = None if (instance_dict['port'] == '' or instance_dict['port'] is None) else int(instance_dict['port'])
    instance_dict['port'] = port
    if isMeta:
        try:
            metaJson = json.loads(instance_dict['metaInfo']['metaJson'])
            instance_dict['meta'] = metaJson
        except:
            return {"code": 50000, "data": "Meta必须JSON字符串格式！"}
    if isCheck:
        ctype = instance_dict['checkInfo']['ctype']
        interval = instance_dict['checkInfo']['interval']
        timeout = instance_dict['checkInfo']['timeout']
        if instance_dict['checkInfo']['isAddress'] == 'true':
            if port is not None and address != '' and ctype != 'HTTP':
                checkaddress = f'{address}:{port}'
            elif port is not None and address != '' and ctype == 'HTTP':
                checkaddress = f'http://{address}:{port}'
            else:
                return {"code": 50000, "data": "健康检查地址使用与实例IP端口一致时，地址/端口不可为空！"}
        else:
            checkaddress = instance_dict['checkInfo']['caddress']
            if checkaddress == '':
                return {"code": 50000, "data": "自定义健康检查，地址信息不可为空！"}

        check = {ctype: checkaddress,"Interval": interval,"Timeout": timeout}
        instance_dict['check'] = check
            
    del instance_dict['metaInfo']
    del instance_dict['checkInfo']
    print(instance_dict, flush=True)

    reg = requests.put(f'{consul_url}/agent/service/register', headers=headers, data=json.dumps(instance_dict))
    if reg.status_code == 200:
        return {"code": 20000, "data": f"【{sid}】增加成功！"}
    else:
        return {"code": 50000, "data": f"{reg.status_code}【{sid}】{reg.text}"}