如何在Bash脚本中控制ZooKeeper临时节点仅在任务完成后释放？

你的核心问题在于对临时节点的使用逻辑理解有误：直接执行zookeeper-client create -e命令，执行完客户端就会退出，会话直接断开，临时节点会立刻被删除，根本起不到互斥作用。临时节点的核心特性是与会话绑定——只要会话保持存活，节点就存在；会话终止（主动退出或进程意外崩溃），节点才会自动删除。

要实现「任务执行期间锁生效，任务完成后释放锁」，必须让创建临时节点的ZooKeeper客户端会话在任务全程保持连接，任务结束后再主动关闭会话或删除节点。下面给你两种可行的Bash实现方案：

方案1：用协进程（coproc）维持会话 #

这是Bash 4.0+支持的特性，无需创建临时文件，简洁高效：

# 1. 从队列获取任务ID，这里模拟为task1
task_id="task1"

# 启动ZooKeeper协进程，维持会话
coproc ZK_CLIENT { zookeeper-client; }

# 发送创建临时节点的命令
echo "create -e /tasks/${task_id}" >&${ZK_CLIENT[1]}

# 读取客户端响应，检查是否创建成功
read -u ${ZK_CLIENT[0]} response
if echo "$response" | grep -q "Node already exists"; then
  echo "任务${task_id}已被其他主机锁定，放弃执行"
  echo "quit" >&${ZK_CLIENT[1]}
  wait ${ZK_CLIENT_PID}
  exit 1
fi

echo "成功锁定任务${task_id}，开始执行..."
# 2. 替换为你的实际任务执行命令，比如调用脚本、运行程序
sleep 30  # 模拟任务耗时

echo "任务${task_id}执行完成，释放锁..."
# 3. 主动删除节点并关闭会话
echo "delete /tasks/${task_id}" >&${ZK_CLIENT[1]}
echo "quit" >&${ZK_CLIENT[1]}

# 等待协进程退出
wait ${ZK_CLIENT_PID}

方案2：用FIFO管道维持会话 #

如果你的Bash版本不支持coproc，用FIFO也能实现：

# 1. 获取任务ID
task_id="task1"

# 创建临时FIFO管道，用于和ZooKeeper客户端通信
mkfifo /tmp/zk_task_lock_fifo

# 启动后台ZooKeeper客户端，从FIFO读取命令
zookeeper-client < /tmp/zk_task_lock_fifo &
zk_pid=$!

# 发送创建临时节点的命令
echo "create -e /tasks/${task_id}" > /tmp/zk_task_lock_fifo

# 检查节点是否创建成功（处理竞争场景）
node_status=$(echo "get /tasks/${task_id}" | zookeeper-client 2>&1)
if echo "$node_status" | grep -q "Node does not exist"; then
  echo "任务${task_id}已被锁定，退出"
  echo "quit" > /tmp/zk_task_lock_fifo
  wait $zk_pid
  rm /tmp/zk_task_lock_fifo
  exit 1
fi

echo "锁定成功，开始执行任务..."
# 执行实际任务
sleep 30

echo "任务完成，释放锁..."
# 删除节点并关闭客户端
echo "delete /tasks/${task_id}" > /tmp/zk_task_lock_fifo
echo "quit" > /tmp/zk_task_lock_fifo

# 等待客户端进程退出
wait $zk_pid

# 清理临时FIFO
rm /tmp/zk_task_lock_fifo

关键注意点 #

• 会话维持是核心：必须保证创建临时节点的客户端会话在任务执行期间一直存活，否则锁会提前失效。
• 竞争处理：多个主机同时抢同一个任务时，只有一个能创建节点成功，其他必须放弃任务，避免重复执行。
• 自动兜底：如果任务执行过程中主机崩溃或脚本意外退出，ZooKeeper客户端会话会自动断开，临时节点会被删除，不会出现死锁。

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