83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
#!/usr/bin/python3
|
|
import requests,json
|
|
from units import consul_kv
|
|
from config import consul_token,consul_url,vendors,regions
|
|
headers = {'X-Consul-Token': consul_token}
|
|
geturl = f'{consul_url}/agent/services'
|
|
delurl = f'{consul_url}/agent/service/deregister'
|
|
puturl = f'{consul_url}/agent/service/register'
|
|
def w2consul(vendor,account,region,ecs_dict):
|
|
service_name = f'{vendor}_{account}_ecs'
|
|
params = {'filter': f'Service == "{service_name}" and "{region}" in Tags and Meta.account == "{account}"'}
|
|
try:
|
|
consul_ecs_iid_list = requests.get(geturl, headers=headers, params=params).json().keys()
|
|
except:
|
|
consul_ecs_iid_list = []
|
|
|
|
#在consul中删除云厂商不存在的ecs
|
|
for del_ecs in [x for x in consul_ecs_iid_list if x not in ecs_dict.keys()]:
|
|
dereg = requests.put(f'{delurl}/{del_ecs}', headers=headers)
|
|
if dereg.status_code == 200:
|
|
print({"code": 20000,"data": f"{account}-删除成功！"}, flush=True)
|
|
else:
|
|
print({"code": 50000,"data": f'{dereg.status_code}:{dereg.text}'}, flush=True)
|
|
off,on = 0,0
|
|
for k,v in ecs_dict.items():
|
|
iid = k
|
|
#对consul中关机的ecs做标记。
|
|
if v['status'] in ['SHUTOFF','Stopped','STOPPED']:
|
|
off = off + 1
|
|
tags = ['shutoff', region]
|
|
stat = 'off'
|
|
else:
|
|
on = on + 1
|
|
tags = [v['ostype'],region]
|
|
stat = 'on'
|
|
custom_ecs = consul_kv.get_value(f'ConsulManager/assets/sync_ecs_custom/{iid}')
|
|
port = custom_ecs.get('port')
|
|
ip = custom_ecs.get('ip')
|
|
if port == None:
|
|
port = 9100 if v['ostype'] == 'linux' else 9182
|
|
if ip == None:
|
|
ip = v['ip'] if isinstance(v['ip'],list) is False else v['ip'][0]
|
|
instance = f'{ip}:{port}'
|
|
data = {
|
|
'id': iid,
|
|
'name': service_name,
|
|
'Address': ip,
|
|
'port': port,
|
|
'tags': tags,
|
|
'Meta': {
|
|
'iid': iid,
|
|
'name': v['name'],
|
|
'region': regions[vendor].get(region,'未找到'),
|
|
'group': v['group'],
|
|
'instance': instance,
|
|
'account': account,
|
|
'vendor': vendors.get(vendor,'未找到'),
|
|
'os': v['ostype'],
|
|
'cpu': v['cpu'],
|
|
'mem': v['mem'],
|
|
'exp': v['exp'],
|
|
'stat': stat
|
|
},
|
|
"check": {
|
|
"tcp": f"{ip}:{port}",
|
|
"interval": "60s"
|
|
}
|
|
}
|
|
if vendor == 'alicloud' and v['ecstag'] != []:
|
|
ecstag_dict = {}
|
|
for ecstag in v['ecstag']:
|
|
if ecstag['TagKey'].encode().isalnum():
|
|
ecstag_dict[ecstag['TagKey']] = ecstag['TagValue']
|
|
data['Meta'].update(ecstag_dict)
|
|
reg = requests.put(puturl, headers=headers, data=json.dumps(data))
|
|
if reg.status_code == 200:
|
|
pass
|
|
#print({f"{account}:code": 20000,"data": "增加成功！"}, flush=True)
|
|
else:
|
|
print({f"{account}:code": 50000,"data": f'{reg.status_code}:{reg.text}'}, flush=True)
|
|
#return {"code": 50000,"data": f'{reg.status_code}:{reg.text}'}
|
|
return off,on
|