wait4it

概述

一个简单的Go应用程序，用于测试端口是否准备好接受连接，或检查MySQL、PostgreSQL、MongoDB、Redis等服务是否就绪。还可进行HTTP调用并检查响应码和响应文本，支持超时设置，可在等待特定时间后失败。

支持的模块

• TCP端口
• MySQL
• PostgreSQL
• HTTP
• MongoDB
• Redis
• RabbitMQ
• Memcached
• ElasticSearch
• Aerospike

安装

可下载最新发布版，或自行构建。构建命令：go build

配置

环境变量

支持以下环境变量：

• W4IT_TYPE：检查类型（http, tcp, mysql, postgres等）
• W4IT_TIMEOUT：超时时间（秒）
• W4IT_HOST：目标主机
• W4IT_PORT：目标端口
• W4IT_USERNAME：服务认证用户名
• W4IT_PASSWORD：服务认证密码
• W4IT_DBNAME：MySQL/PostgreSQL数据库名
• W4IT_SSL_MODE：PostgreSQL SSL模式（disable/enable）
• W4IT_OPERATION_MODE：Redis操作模式（standalone/cluster）
• W4IT_HTTP_STATUS_CODE：HTTP检查期望状态码
• W4IT_HTTP_TEXT：HTTP响应中需包含的子字符串
• W4IT_HTTP_FOLLOW_REDIRECT：HTTP是否跟随重定向（true/false，默认true）

命令行参数

支持以下命令行标志：

• -type：检查类型（http, tcp, mysql, postgres等）
• -h：目标主机（默认127.0.0.1）
• -p：目标端口
• -t：超时时间（秒，默认30）
• -u：服务认证用户名
• -P：服务认证密码
• -n：MySQL/PostgreSQL数据库名
• -ssl：PostgreSQL SSL模式（disable/enable）
• -operation-mode：Redis操作模式（standalone/cluster）
• -http-status：HTTP检查期望状态码
• -http-text：HTTP响应中需包含的子字符串
• -http-follow-redirect：HTTP是否跟随重定向（true/false，默认true）

使用示例

检查TCP端口

./wait4it -type=tcp -h=127.0.0.1 -p=8080 -t=60

检查MySQL实例

./wait4it -type=mysql -h=127.0.0.1 -p=3306 -t=60 -u=root -P=secret -n=app 

检查PostgreSQL实例

./wait4it -type=postgres -h=127.0.0.1 -p=5432 -t=60 -u=postgres -P=secret -ssl=disable

检查HTTP响应和文本

./wait4it -type=http -h=[***] -t=60 -http-status=200 -http-text="Software Engineer" 
./wait4it -type=http -h=[***] -t=60 -http-status=301 -http-follow-redirect=false 

检查MongoDB实例

./wait4it -type=mongo -p=27017 -t=60 -u=mongoadmin -P=secret -h=127.0.0.1

检查Redis实例

# 带密码的Redis
./wait4it -type=redis -p=6379 -t=60 -P=secret -h=127.0.0.1

# 无密码的Redis
./wait4it -type=redis -p=6379 -t=60 -h=127.0.0.1

# Redis集群
./wait4it -type=redis -p=6379 -t=60 -P=secret -operation-mode=cluster -h=127.0.0.1

检查其他服务

# RabbitMQ
./wait4it -type=rabbitmq -p=5267 -t=60 -u=guest -P=guest -h=127.0.0.1

# Memcached
./wait4it -type=memcached -h=127.0.0.1 -p=11211 -t=60

# ElasticSearch
./wait4it -type=elasticsearch -h=[***] -p=9200 -t=60

# Aerospike
./wait4it -type=aerospike -h=127.0.0.1 -p=3000 -t=60

Docker使用

可在Docker容器中运行，也可作为K8s/OpenShift初始化容器。镜像地址：docker pull ph4r5h4d/wait4it

Docker示例

检查TCP端口

docker run ph4r5h4d/wait4it -type=tcp -h=127.0.0.1 -p=8080 -t=60

检查MySQL实例

docker run ph4r5h4d/wait4it -type=mysql -h=127.0.0.1 -p=3306 -t=60 -u=root -P=secret -n=app 

检查PostgreSQL实例

docker run ph4r5h4d/wait4it -type=postgres -h=127.0.0.1 -p=5432 -t=60 -u=postgres -P=secret -ssl=disable

检查HTTP响应

docker run ph4r5h4d/wait4it -type=http -h=[***] -t=60 -http-status=200 -http-text="Software Engineer" 
docker run ph4r5h4d/wait4it -type=http -h=[***] -t=60 -http-status=301 -http-follow-redirect=false  

检查其他服务

# MongoDB
docker run ph4r5h4d/wait4it -type=mongo -p=32768 -t=60 -u=mongoadmin -P=secret -h=127.0.0.1

# Redis集群
docker run ph4r5h4d/wait4it -type=redis -p=6379 -t=60 -P=secret -operation-mode=cluster -h=127.0.0.1

# ElasticSearch
docker run ph4r5h4d/wait4it -type=elasticsearch -h=[***] -p=9200

注意事项

配置优先级

环境变量优先级高于命令行参数，若同时设置，以环境变量为准。

退出码

• 0：连接成功建立
• 1：超时
• 2：验证错误或输入数据错误

服务检查细节

• HTTP检查：默认检查200状态码；未指定http-text时仅检查状态码
• PostgreSQL：默认禁用SSL模式（disable）
• MongoDB：仅支持用户名/密码认证
• Redis：默认单机模式，暂不支持Sentinel；集群模式仅检查单个主机
• ElasticSearch：无需用户名/密码（除非启用X-Pack）；暂不支持集群检查
• Memcached：暂不支持多主机和集群检查