python 指令碼註冊consul
註冊http
import requests headers = { "contentType": "application/json" } def register(name, id, address, port): url = "http://192.168.190.129:8500/v1/agent/service/register" rsp = requests.put(url, headers=headers, json={ "Name": name, "ID": id, "Tags": ["mxshop", "bobby", "imocc", "web"], "Address": address, "Port": port, "Check": { "HTTP": f"http://{address}:{port}/health", "Timeout": "5s", "Interval": "5s", "DeregisterCriticalServiceAfter": "5s", } }) if rsp.status_code == 200: print("註冊成功") else: print(f"註冊失敗:{rsp.status_code}") def deregister(id): url = f"http://192.168.190.129:8500/v1/agent/service/deregister/{id}" rsp = requests.put(url, headers=headers) if rsp.status_code == 200: print("登出成功") else: print(f"登出失敗:{rsp.status_code}") if __name__ == '__main__': register("mxshop-web", "mxshop-web", "192.168.230.1", 8021) # deregister("mxshop-web") pass
註冊grpc
import socket
import sys
from concurrent import futures
import signal
import argparse
import os
import grpc
from loguru import logger
from user_srv.settings.settings import client
BASE_DIR = os.path.dirname(os.path.abspath(os.path.dirname(file)))
sys.path.insert(0, BASE_DIR)
from user_srv.proto import user_pb2_grpc
from user_srv.handler.user import UserServicer
from common.grpc_health.v1 import health,health_pb2_grpc
from common.register import consul
from user_srv.settings import settings
def on_exit(signo, frame):
logger.info("程序中端")
sys.exit(0)
def get_free_tcp_port():
tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM) tcp.bind(("", 0)) _, port = tcp.getsockname() tcp.close() return port
def serve():
parse = argparse.ArgumentParser()
parse.add_argument("--ip",
nargs="?",
type=str,
default="192.168.230.1",
help="binding ip"
)
parse.add_argument("--port",
nargs="?",
type=int,
default=0,
help="the listening port"
)
args = parse.parse_args()
if args.port == 0:
args.port = get_free_tcp_port()
# logger.add("logs/user_srv_{time}.log")
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
# 註冊使用者服務
user_pb2_grpc.add_UserServicer_to_server(UserServicer(), server)
# 註冊健康檢查
health_pb2_grpc.add_HealthServicer_to_server(health.HealthServicer(), server)
server.add_insecure_port(f"{args.ip}:{(args.port)}")
server.start()
logger.info(f"啟動服務: {args.ip}:{args.port}]")
"""
SIGINT ctrl+C
SIGTERM kill 發出的終止
"""
signal.signal(signal.SIGINT, on_exit)
signal.signal(signal.SIGTERM, on_exit)
logger.info(f"服務註冊開始")
register = consul.ConsulRegister(settings.CONSUL_HOST, settings.CONSUL_PORT)
if not register.register(name=settings.SERVICE_NAME, id=settings.SERVICE_ID, address=args.ip, port=args.port, tags=settings.SERVICE_TAGS, check=None):
logger.info(f"服務註冊失敗")
sys.exit(0)
logger.info(f"服務註冊成功")
server.wait_for_termination()
def test_cb(args):
print("配置檔案產生變化")
print(args)
if name == 'main':
# print(get_free_tcp_port())
client.add_config_watcher(settings.NACOS["DataId"], settings.NACOS["Group"], test_cb)
serve()