1. 程式人生 > 其它 >python 指令碼註冊consul

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()