Nginx Upstream 模块开发：自定义后端负载均衡调度算法-Nginx-PHP中文网

开发Nginx自定义upstream负载均衡算法需编写C模块，实现init_upstream、init_peer、get_peer等回调函数，编译时静态链接进Nginx，无法热替换；示例随机算法演示了核心流程，须注意内存安全、随机数初始化及配置生效验证。

nginx upstream 模块开发：自定义后端负载均衡调度算法

要开发 Nginx 的自定义 upstream 负载均衡算法，核心是实现一个 upstream 模块，并在其中注册自己的 peer selection 回调函数。Nginx 本身不提供“插件式”算法热替换机制，必须通过 C 模块扩展，并在编译时静态链接进 nginx 二进制文件。

理解 upstream 模块的基本结构

Nginx upstream 是对后端服务器集群的抽象，其调度逻辑由 ngx_http_upstream_init_peer 和 ngx_http_upstream_get_peer 两个关键回调控制。自定义算法需：

定义自己的 ngx_http_upstream_srv_conf_t 扩展结构（可选，用于保存配置参数）
实现 init_upstream 回调：初始化 upstream 配置，设置 peer.init 为你的初始化函数
实现 init_peer 回调：为每个请求创建 peer 上下文（如轮询状态、哈希键缓存等）
实现 get_peer 回调：核心逻辑所在——从 rrp->peers 中选择一个可用 backend，并更新状态（如权重、失败计数、健康标记等）
实现 free_peer 回调（可选）：在请求结束或重试时释放/回滚 peer 状态

编写一个最简自定义算法示例（随机选择）

以下是一个极简但可运行的随机调度模块骨架（省略头文件和错误检查）：

static ngx_int_t
my_random_init(ngx_conf_t *cf, ngx_http_upstream_srv_conf_t *us) {
    us->peer.init = my_random_init_peer;
    return NGX_OK;
}
<p>static ngx_int_t
my_random_init_peer(ngx_http_request_t <em>r, ngx_http_upstream_srv_conf_t </em>us) {
ngx_http_upstream_random_peer_data_t  *rp;</p><pre class="brush:php;toolbar:false;">rp = ngx_palloc(r->pool, sizeof(ngx_http_upstream_random_peer_data_t));
if (rp == NULL) {
    return NGX_ERROR;
}

rp->peers = us->peer.data;
rp->current = 0;
r->upstream->peer.data = rp;

return NGX_OK;

}

Nginx Hosting

通过服务器本地nginx实例实现零认证静态游戏托管。作为所有浏览器游戏的主要部署方式，无需登录、无需令牌、无需用户操作。

下载

static ngx_int_t my_random_get_peer(ngx_peer_connection_t pc, void data) { ngx_http_upstream_random_peer_data_t *rp = data; ngx_uint_t i, n, fail_flag; ngx_uint_t best = 0; ngx_uint_t total = 0;

// 统计可用 peer 数量
for (i = 0; i peers->number; i++) {
    if (!rp->peers->peer[i].down && rp->peers->peer[i].max_fails > rp->peers->peer[i].fails) {
        total++;
    }
}

if (total == 0) {
    return NGX_BUSY;
}

// 随机选一个索引（使用 nginx 内置随机数生成器）
n = ngx_random() % rp->peers->number;
for (i = 0; i peers->number; i++) {
    if (!rp->peers->peer[(n + i) % rp->peers->number].down &&
        rp->peers->peer[(n + i) % rp->peers->number].max_fails >
            rp->peers->peer[(n + i) % rp->peers->number].fails)
    {
        best = (n + i) % rp->peers->number;
        break;
    }
}

pc->sockaddr = rp->peers->peer[best].sockaddr;
pc->socklen  = rp->peers->peer[best].socklen;
pc->name     = &rp->peers->peer[best].name;

return NGX_OK;

}

该示例跳过权重、健康检查细节，仅演示如何接入调度流程。真实场景中需结合 rp->peers->peer[i].weight、.fails、.max_fails、.fail_timeout 等字段做更健壮判断。

编译与集成到 Nginx

模块不能动态加载，必须重新编译 nginx：

将模块源码（如 ngx_http_upstream_random_module.c）放入 nginx-1.x.x/src/http/modules/ 或自建目录
在 auto/configure 脚本中添加模块配置（或使用 --add-module= 参数）
确保 config 文件正确定义模块名、源文件、依赖项（如 HTTP_UPSTREAM）
执行 ./configure --add-module=/path/to/your/module && make && sudo make install

成功后，在 nginx.conf 中即可使用：
upstream backend {
  random; # 你注册的指令
  server 192.168.1.10:8080;
  server 192.168.1.11:8080;
}

调试与验证要点

开发过程中常见问题及应对方式：

段错误或崩溃：多数因访问未初始化的 rp->peers 或越界读取 peer[i]；务必检查 peers->number > 0 和循环边界
始终选同一个 backend：可能是 ngx_random() 未正确初始化（需在 master 进程中调用 ngx_srand(ngx_pid)），或未考虑失败标记导致其他节点被跳过
无法生效：确认模块已编译进 nginx（nginx -V 2>&1 | grep -o "random"），且配置中使用了你注册的指令（如 random），而非误写为 least_conn 等内置指令
日志无输出：使用 ngx_log_debug 系列宏需开启 error_log ... debug; 并确保 configure 时带 --with-debug