<?php

namespace Kylin\Activity;

use Closure;
use Ramsey\Uuid\Uuid;

/**
 * 日志批次管理
 *
 * 将同一请求/操作中的多条活动日志归入同一批次（UUID）。
 * 支持：
 *  - 手动设置批次 UUID（setBatch）
 *  - 闭包内批量操作归入同一批次（withinBatch）
 *  - 事务感知（startTransaction/endTransaction）
 *  - 自动生成 UUID（generateUuid）
 */
class LogBatch
{
    /** 当前批次 UUID */
    public ?string $uuid = null;

    /** 事务嵌套计数 */
    public int $transactions = 0;

    /**
     * 生成 UUID v4
     */
    protected function generateUuid(): string
    {
        return Uuid::uuid4()->toString();
    }

    /**
     * 获取当前批次 UUID
     */
    public function getUuid(): ?string
    {
        return $this->uuid;
    }

    /**
     * 设置批次 UUID
     *
     * @param  string  $uuid  批次 UUID
     */
    public function setBatch(string $uuid): void
    {
        $this->uuid = $uuid;
        $this->transactions = 1;
    }

    /**
     * 在批次内执行闭包
     *
     * 闭包内所有日志操作将共享同一批次 UUID。
     *
     * @param  Closure  $callback  批次内执行的闭包
     * @return mixed                闭包返回值
     */
    public function withinBatch(Closure $callback): mixed
    {
        $this->startBatch();

        try {
            $result = $callback($this->getUuid());
        } finally {
            $this->endBatch();
        }

        return $result;
    }

    public function startBatch(): void
    {
        if (! $this->isOpen()) {
            $this->uuid = $this->generateUuid();
        }

        $this->transactions++;
    }

    public function isOpen(): bool
    {
        return $this->transactions > 0;
    }

    public function endBatch(): void
    {
        $this->transactions = max(0, $this->transactions - 1);

        if ($this->transactions === 0) {
            $this->uuid = null;
        }
    }
}
