OCaml使用Unix.establish_server时调用exit 0无法终止程序如何解决

问题根源 #

你遇到的问题核心是Unix.establish_server的运行机制导致的：

The function Unix.establish_server never returns normally.

官方文档的这句话是指，这个函数内部已经封装了「无限监听端口 + 每收到一个客户端连接就fork子进程处理连接逻辑」的逻辑，你写的所有handle_service中的代码全部运行在fork出来的子进程中，子进程内抛出的Sys.Break异常只会终止当前处理连接的子进程，根本无法传递到负责监听端口的主进程中，所以你写在主进程的异常捕获分支永远不会触发，主进程会持续运行等待新连接。

解决方案 #

方案1：不更换establish_server的临时实现 #

你可以通过给主进程发信号的方式触发主进程终止，修改逻辑如下：

1. 程序启动时先保存主进程的PID，这一步要在establish_server调用前执行，保证拿到的是主进程ID：

let main_pid = Unix.getpid ()

2. 修改handle_service的异常处理逻辑，不需要抛异常到上层，直接给主进程发终止信号即可：

let handle_service ic oc =
  try while true do
        ...
        if ... then raise Finish_interaction
      done ;
  with
  | Finish_interaction -> 
    (* 给主进程发SIGINT信号，触发主进程的Sys.Break异常 *)
    Unix.kill main_pid Sys.sigint;
    (* 终止当前子进程 *)
    exit 0
  | _ -> ...

这种方案下主进程捕获到Sys.Break后执行的exit 0就会正常生效，整个服务端进程会退出。

方案2：自行实现服务端监听逻辑（更可控，推荐） #

Unix.establish_server本身就是为永久运行的极简服务设计的，可控性非常差，你可以自己实现监听逻辑，主动控制服务终止，核心代码逻辑如下：

let main_server serv_fun =
  if Array.length Sys.argv < 4 then ...
  else 
    let port = int_of_string Sys.argv.(1) in
    let my_address = Unix.inet_addr_loopback in
    let running = ref true in
    (* 创建监听socket *)
    let sock = Unix.socket Unix.PF_INET Unix.SOCK_STREAM 0 in
    Unix.setsockopt sock Unix.SO_REUSEADDR true;
    Unix.bind sock (Unix.ADDR_INET(my_address, port));
    Unix.listen sock 5;
    (* 自行控制监听循环 *)
    while !running do
      try
        let (client_sock, _) = Unix.accept sock in
        let ic = Unix.in_channel_of_descr client_sock in
        let oc = Unix.out_channel_of_descr client_sock in
        (* 处理连接，你可以选择fork子进程或者单线程处理 *)
        serv_fun ic oc;
        (* 处理完关闭客户端连接 *)
        Unix.close client_sock;
        (* 如果你需要终止服务，直接把running设为false即可 *)
        if 某终止条件 then running := false
      with
      | Sys.Break -> running := false
    done;
    (* 关闭监听socket，退出程序 *)
    Unix.close sock;
    exit 0

这种实现完全没有establish_server的限制，你可以在任意时机终止服务端运行。

内容的提问来源于stack exchange，提问作者user2347300