<?php

namespace Lib\Es;


/**
 * 批量添加数据的类
 * @author：dc
 * @time 2025/8/6 16:37
 * Class BulkData
 * @package Lib\Es
 */
class BulkData {

    /**
     * @var string
     */
    private array $data = [];

    /**
     * 添加数据
     * @param string $index 必须全小写
     * @param string $_id 指定文档id
     * @param array $data
     * @author：dc
     * @time 2025/8/6 16:41
     */
    public function add(string $index,string $_id, array $data){
        $index = strtolower($index);
        if(empty($this->data[$index])){
            $this->data[$index] = [];
        }
        $this->data[$index][] = [
            '_id'   => $_id,
            '_source'   => $data,
        ];
    }


    /**
     * 获取所有索引名称
     * @return array
     * @author：dc
     * @time 2025/8/6 16:45
     */
    public function getIndexs(): array {
        return array_keys($this->data);
    }

    /**
     * 转换成es可识别的数据
     * @return array
     * @author：dc
     * @time 2025/8/6 16:42
     */
    public function toParams():array {
        $params = [];
        foreach ($this->data as $index=>$items){
            foreach ($items as $item){
                $params[] = [
                    'index' =>  [
                        '_index'    =>  $index,
                        '_id'   =>  $item['_id']
                    ]
                ];
                $params[] = $item['_source'];
            }
        }
        return $params;
    }

    /**
     * 清除 数据
     * @author：dc
     * @time 2025/8/7 9:55
     */
    public function clear(){
        $this->data = [];
    }

    /**
     * 是否为空
     * @return bool
     * @author：dc
     * @time 2025/8/7 9:54
     */
    public function isEmpty():bool
    {
        return empty($this->data);
    }

    /**
     * 统计数量
     * @return int
     * @author：dc
     * @time 2025/8/7 9:56
     */
    public function total():int
    {
        return count($this->data);
    }


}