qing
4275f002e0
- 新增 --host/--port/--redirect-port/--redirect-code CLI 参数（覆盖 confs+ENV） - 新增 --log-level debug|info|warn|error（log_client 加级别过滤） - 新增 --instance 实例名（注册中心用） - 自定义 --help 用法文本（argparse，confs 不可用时也能 --help） - log_client: 新增 set_log_level() + push() 级别过滤 - 修复: redirect_code 配置项原加载但未使用，do_GET/do_HEAD 改为用 _config['redirect_code'] - 配置优先级: CLI > ENV > confs > 默认值 - services.yaml: redirect80 声明 --instance 和 --log-level
259 lines
9.4 KiB
Python
259 lines
9.4 KiB
Python
"""
|
||
redirect80 — HTTP 80 端口重定向服务。
|
||
|
||
监听 80 端口，所有 HTTP 请求返回 301/308 重定向到 HTTPS (443)，
|
||
保留原始 Host 和路径。为搜索引擎和浏览器提供 HTTP→HTTPS 无缝升级。
|
||
|
||
技术选型：Python 内置 http.server + 多线程混合模式，无外部 Web 框架依赖。
|
||
日志通过 log_client 异步推送到 db 的 Logs 表（fail-safe）。
|
||
|
||
配置（三层加载）：
|
||
1. confs 强依赖拉取 redirect80.ini（retry_on_refused 阻塞等待）
|
||
2. 环境变量覆盖（Docker/k8s 兼容）
|
||
3. 命令行参数（最高优先级）
|
||
配置不可用时直接退出，不做本地回退。
|
||
"""
|
||
|
||
import argparse
|
||
import configparser
|
||
import os
|
||
import sys as _sys
|
||
from http.server import HTTPServer, BaseHTTPRequestHandler
|
||
from socketserver import ThreadingMixIn
|
||
|
||
# ── 路径设置 ──────────────────────────────────────────────────────────
|
||
_here = os.path.dirname(os.path.abspath(__file__))
|
||
_sys.path.insert(0, os.path.join(_here, ".."))
|
||
from log_client import logs, set_log_level # noqa: E402
|
||
|
||
_sys.path.insert(0, os.path.join(_here, "..", "pymsg"))
|
||
from pymsg import RpcClient # noqa: E402
|
||
|
||
|
||
# ── 配置加载 ─────────────────────────────────────────────────────────
|
||
|
||
def load_config(argv):
|
||
"""三层加载: confs → ENV → CLI（最高优先级）。"""
|
||
|
||
# argparse 先行：--help 在任何网络调用前处理
|
||
parser = argparse.ArgumentParser(
|
||
prog="redirect80",
|
||
description="HTTP 80 端口重定向服务",
|
||
add_help=False,
|
||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||
epilog="配置优先级: CLI > 环境变量 > confs > 默认值",
|
||
)
|
||
parser.add_argument("--host", default=None,
|
||
help="绑定地址，默认 0.0.0.0")
|
||
parser.add_argument("--port", type=int, default=None,
|
||
help="监听端口，默认 80")
|
||
parser.add_argument("--redirect-port", type=int, default=None,
|
||
help="重定向目标端口，默认 443")
|
||
parser.add_argument("--redirect-code", type=int, default=None,
|
||
help="GET/HEAD 重定向状态码，默认 301")
|
||
parser.add_argument("--log-level", default=None,
|
||
help="debug|info|warn|error，默认 info")
|
||
parser.add_argument("--instance", default=None,
|
||
help="实例名（注册中心用），默认 redirect80")
|
||
parser.add_argument("--help", "-h", action="help",
|
||
help="显示此帮助并退出")
|
||
args = parser.parse_args(argv)
|
||
|
||
# 默认值
|
||
config = {
|
||
"host": "0.0.0.0",
|
||
"port": 80,
|
||
"redirect_port": 443,
|
||
"redirect_code": 301,
|
||
"log_level": "info",
|
||
"instance": "redirect80",
|
||
}
|
||
|
||
# 1. confs 强依赖拉取配置（阻塞等待 confs 就绪，失败则退出）
|
||
confs_host = os.environ.get("CONFS_HOST", "127.0.0.1")
|
||
confs_port = int(os.environ.get("CONFS_PORT", "8010"))
|
||
try:
|
||
client = RpcClient(
|
||
confs_host, confs_port, "redirect80",
|
||
retry_on_refused=True, timeout=5,
|
||
)
|
||
data = client.call("get", {"name": "redirect80.ini"})
|
||
ini_text = data.get("content", "")
|
||
ini = configparser.ConfigParser(interpolation=None)
|
||
ini.read_string(ini_text)
|
||
|
||
_merge = {
|
||
("server", "host"): "host",
|
||
("server", "port"): "port",
|
||
("server", "redirect_port"): "redirect_port",
|
||
("server", "redirect_code"): "redirect_code",
|
||
}
|
||
for (sec, key), var in _merge.items():
|
||
if ini.has_option(sec, key):
|
||
val = ini.get(sec, key)
|
||
config[var] = type(config[var])(val)
|
||
logs.info("redirect80", "配置已从 confs 加载")
|
||
except Exception as e:
|
||
print(f"[redirect80] confs 获取失败: {e}", file=_sys.stderr)
|
||
_sys.exit(1)
|
||
|
||
# 2. 环境变量覆盖
|
||
_env_map = {
|
||
"REDIRECT80_HOST": "host",
|
||
"REDIRECT80_PORT": ("port", int),
|
||
"REDIRECT80_REDIRECT_PORT": ("redirect_port", int),
|
||
"REDIRECT80_REDIRECT_CODE": ("redirect_code", int),
|
||
}
|
||
for env_var, mapping in _env_map.items():
|
||
val = os.environ.get(env_var)
|
||
if val is not None:
|
||
if isinstance(mapping, tuple):
|
||
config[mapping[0]] = mapping[1](val)
|
||
else:
|
||
config[mapping] = val
|
||
|
||
# 3. CLI 参数（最高优先级）
|
||
if args.host is not None:
|
||
config["host"] = args.host
|
||
if args.port is not None:
|
||
config["port"] = args.port
|
||
if args.redirect_port is not None:
|
||
config["redirect_port"] = args.redirect_port
|
||
if args.redirect_code is not None:
|
||
config["redirect_code"] = args.redirect_code
|
||
if args.log_level is not None:
|
||
config["log_level"] = args.log_level
|
||
if args.instance is not None:
|
||
config["instance"] = args.instance
|
||
|
||
# log_level 合法性校验
|
||
lv = config["log_level"].lower()
|
||
if lv == "warning":
|
||
lv = "warn"
|
||
if lv not in ("debug", "info", "warn", "error"):
|
||
print(f"[redirect80] invalid log_level: {config['log_level']} "
|
||
f"(must be debug|info|warn|error)", file=_sys.stderr)
|
||
_sys.exit(1)
|
||
config["log_level"] = lv
|
||
|
||
return config
|
||
|
||
|
||
# ── 模块级配置（main() 中赋值）──────────────────────────────────────
|
||
_config = {}
|
||
|
||
|
||
# ── 请求处理器 ──────────────────────────────────────────────────────────
|
||
|
||
|
||
class RedirectHandler(BaseHTTPRequestHandler):
|
||
"""HTTP 请求处理器：将所有请求重定向到 HTTPS。
|
||
|
||
GET/HEAD → 301 (Moved Permanently)，搜索引擎友好的永久重定向。
|
||
POST/PUT/DELETE/PATCH → 308 (Permanent Redirect)，保留请求方法和 body。
|
||
"""
|
||
|
||
server_version = "UniServe/redirect80"
|
||
sys_version = ""
|
||
|
||
def _redirect(self, code: int):
|
||
"""构造 HTTPS 重定向 URL 并返回。"""
|
||
host = self.headers.get("Host", "")
|
||
# 剥离端口号（如果客户端带了 :80）
|
||
hostname = host.split(":")[0] if host else _config["host"]
|
||
target_port = _config["redirect_port"]
|
||
# 标准 443 端口不显式写 :443
|
||
if target_port == 443:
|
||
location = f"https://{hostname}{self.path}"
|
||
else:
|
||
location = f"https://{hostname}:{target_port}{self.path}"
|
||
|
||
self.send_response(code)
|
||
self.send_header("Location", location)
|
||
self.send_header("Connection", "close")
|
||
self.send_header("Cache-Control", "max-age=31536000")
|
||
self.end_headers()
|
||
|
||
# 日志：仅记录摘要（不刷屏）
|
||
logs.info("redirect80",
|
||
f"{code} {self.command} {host} → {location}")
|
||
|
||
def do_GET(self):
|
||
self._redirect(_config["redirect_code"])
|
||
|
||
def do_HEAD(self):
|
||
self._redirect(_config["redirect_code"])
|
||
|
||
def do_POST(self):
|
||
self._redirect(308)
|
||
|
||
def do_PUT(self):
|
||
self._redirect(308)
|
||
|
||
def do_DELETE(self):
|
||
self._redirect(308)
|
||
|
||
def do_PATCH(self):
|
||
self._redirect(308)
|
||
|
||
def do_OPTIONS(self):
|
||
self.send_response(204)
|
||
self.send_header("Allow", "GET, HEAD, POST, PUT, DELETE, PATCH, OPTIONS")
|
||
self.send_header("Connection", "close")
|
||
self.end_headers()
|
||
|
||
def log_message(self, format, *args):
|
||
"""抑制默认的 stderr 日志（已通过 log_client 记录）。"""
|
||
pass
|
||
|
||
|
||
# ── 多线程 HTTP 服务器 ──────────────────────────────────────────────────
|
||
|
||
|
||
class ThreadingRedirectServer(ThreadingMixIn, HTTPServer):
|
||
"""多线程混合 HTTP 服务器，处理并发连接。
|
||
|
||
ThreadingMixIn 为每个请求创建新线程，适合 redirect80
|
||
这种请求处理极快（无 I/O 阻塞）的场景。
|
||
"""
|
||
daemon_threads = True
|
||
allow_reuse_address = True
|
||
|
||
def handle_error(self, request, client_address):
|
||
"""覆盖默认错误处理，异常写入 log_client 而非 stderr。"""
|
||
logs.error("redirect80",
|
||
f"处理 {client_address} 请求时异常",
|
||
extra={"client": str(client_address)})
|
||
|
||
|
||
def main():
|
||
global _config
|
||
_config = load_config(_sys.argv[1:])
|
||
|
||
# 设置日志级别过滤
|
||
set_log_level(_config["log_level"])
|
||
|
||
host = _config["host"]
|
||
port = int(_config["port"])
|
||
|
||
server = ThreadingRedirectServer((host, port), RedirectHandler)
|
||
|
||
logs.info("redirect80",
|
||
f"服务启动 instance={_config['instance']} "
|
||
f"http://{host}:{port} → https://*:{_config['redirect_port']}")
|
||
|
||
print(f"[redirect80] instance={_config['instance']} log-level={_config['log_level']} "
|
||
f"监听 http://{host}:{port} → 重定向到 :{_config['redirect_port']}",
|
||
file=_sys.stderr)
|
||
|
||
try:
|
||
server.serve_forever()
|
||
except KeyboardInterrupt:
|
||
logs.info("redirect80", "收到 SIGINT，正在关闭...")
|
||
server.shutdown()
|
||
logs.info("redirect80", "服务已停止")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|