mirror of
https://github.com/kuaifan/dootask.git
synced 2025-12-11 02:12:53 +00:00
309 lines
7.9 KiB
PHP
309 lines
7.9 KiB
PHP
<?php
|
||
|
||
namespace App\Module\ElasticSearch;
|
||
|
||
use Elastic\Elasticsearch\ClientBuilder;
|
||
use Elastic\Elasticsearch\Exception\MissingParameterException;
|
||
use Illuminate\Support\Facades\Log;
|
||
|
||
/**
|
||
* Elasticsearch基础类
|
||
*
|
||
* Class ElasticSearchBase
|
||
* @package App\Module\ElasticSearch
|
||
*/
|
||
class ElasticSearchBase
|
||
{
|
||
/**
|
||
* Elasticsearch客户端实例
|
||
*
|
||
* @var \Elastic\Elasticsearch\Client
|
||
*/
|
||
protected $client;
|
||
|
||
/**
|
||
* 当前操作的索引名称
|
||
*
|
||
* @var string
|
||
*/
|
||
protected $index;
|
||
|
||
/**
|
||
* 构造函数
|
||
*
|
||
* @param null $index 默认索引名称
|
||
* @throws \Elastic\Elasticsearch\Exception\ConfigException
|
||
*/
|
||
public function __construct($index = null)
|
||
{
|
||
$host = env('ELASTICSEARCH_HOST', 'es');
|
||
$port = env('ELASTICSEARCH_PORT', '9200');
|
||
$scheme = env('ELASTICSEARCH_SCHEME', 'http');
|
||
$user = env('ELASTICSEARCH_USER', '');
|
||
$pass = env('ELASTICSEARCH_PASS', '');
|
||
$verifi = env('ELASTICSEARCH_VERIFI', false);
|
||
$ca = env('ELASTICSEARCH_CA', '');
|
||
$key = env('ELASTICSEARCH_KEY', '');
|
||
$cert = env('ELASTICSEARCH_CERT', '');
|
||
// 为8.x版本客户端配置连接
|
||
$config = [
|
||
'hosts' => ["{$scheme}://{$host}:{$port}"]
|
||
];
|
||
|
||
// 如果设置了用户名和密码
|
||
if (!empty($user)) {
|
||
$config['basicAuthentication'] = [$user, $pass];
|
||
}
|
||
|
||
$config['SSLVerification'] = $verifi;
|
||
if ($verifi) {
|
||
$config['SSLCert'] = $cert;
|
||
$config['CABundle'] = $ca;
|
||
$config['SSLKey'] = $key;
|
||
}
|
||
// 8.x版本使用ClientBuilder::fromConfig创建客户端
|
||
$this->client = ClientBuilder::fromConfig($config);
|
||
|
||
if ($index) {
|
||
$this->index = $index;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 设置索引名称
|
||
*
|
||
* @param string $index
|
||
* @return $this
|
||
*/
|
||
public function setIndex($index)
|
||
{
|
||
$this->index = $index;
|
||
return $this;
|
||
}
|
||
|
||
/**
|
||
* 检查索引是否存在
|
||
*
|
||
* @return bool
|
||
* @throws \Exception
|
||
*/
|
||
public function indexExists()
|
||
{
|
||
$params = ['index' => $this->index];
|
||
return $this->client->indices()->exists($params)->asBool();
|
||
}
|
||
|
||
/**
|
||
* 创建索引
|
||
*
|
||
* @param array $settings 索引设置
|
||
* @param array $mappings 字段映射
|
||
* @return array
|
||
*/
|
||
public function createIndex($settings = [], $mappings = [])
|
||
{
|
||
$params = [
|
||
'index' => $this->index
|
||
];
|
||
|
||
$body = [];
|
||
if (!empty($settings)) {
|
||
$body['settings'] = $settings;
|
||
}
|
||
|
||
if (!empty($mappings)) {
|
||
$body['mappings'] = $mappings;
|
||
}
|
||
|
||
if (!empty($body)) {
|
||
$params['body'] = $body;
|
||
}
|
||
|
||
try {
|
||
// 在8.x中，索引操作位于indices()命名空间
|
||
return $this->client->indices()->create($params)->asArray();
|
||
} catch (\Exception $e) {
|
||
Log::error('创建Elasticsearch索引失败: ' . $e->getMessage());
|
||
return ['error' => $e->getMessage()];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 删除索引
|
||
* @return array
|
||
*/
|
||
public function deleteIndex()
|
||
{
|
||
try {
|
||
$params = ['index' => $this->index];
|
||
return $this->client->indices()->delete($params)->asArray();
|
||
} catch (\Exception $e) {
|
||
Log::error('删除Elasticsearch索引失败: ' . $e->getMessage());
|
||
return ['error' => $e->getMessage()];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 批量操作（批量添加/更新/删除文档）
|
||
*
|
||
* @param array $operations 批量操作的数据
|
||
* @return array
|
||
*/
|
||
public function bulk($operations)
|
||
{
|
||
try {
|
||
// 在8.x中，批量操作API签名相同，但内部实现有所变化
|
||
return $this->client->bulk($operations)->asArray();
|
||
} catch (\Exception $e) {
|
||
Log::error('批量操作失败: ' . $e->getMessage());
|
||
return ['error' => $e->getMessage()];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 索引单个文档
|
||
*
|
||
* @param array $document 文档数据
|
||
* @param string $id 文档ID
|
||
* @param string|null $routing 路由值，用于父子文档
|
||
* @return array
|
||
*/
|
||
public function indexDocument($document, $id, $routing = null)
|
||
{
|
||
$params = [
|
||
'index' => $this->index,
|
||
'id' => $id,
|
||
'body' => $document
|
||
];
|
||
|
||
if ($routing) {
|
||
$params['routing'] = $routing;
|
||
}
|
||
|
||
try {
|
||
return $this->client->index($params)->asArray();
|
||
} catch (\Exception $e) {
|
||
Log::error('索引文档失败: ' . $e->getMessage());
|
||
return ['error' => $e->getMessage()];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 删除文档
|
||
*
|
||
* @param string $id 文档ID
|
||
* @param string|null $routing 路由值，用于父子文档
|
||
* @return array
|
||
*/
|
||
public function deleteDocument($id, $routing = null)
|
||
{
|
||
$params = [
|
||
'index' => $this->index,
|
||
'id' => $id
|
||
];
|
||
|
||
if ($routing) {
|
||
$params['routing'] = $routing;
|
||
}
|
||
|
||
try {
|
||
return $this->client->delete($params)->asArray();
|
||
} catch (MissingParameterException $e) {
|
||
// 文档不存在时返回成功
|
||
return ['result' => 'not_found', 'error' => $e->getMessage()];
|
||
} catch (\Exception $e) {
|
||
Log::error('删除文档失败: ' . $e->getMessage());
|
||
return ['error' => $e->getMessage()];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 刷新索引
|
||
* @return array
|
||
*/
|
||
public function refreshIndex()
|
||
{
|
||
$params = [
|
||
'index' => $this->index
|
||
];
|
||
|
||
try {
|
||
return $this->client->indices()->refresh($params)->asArray();
|
||
} catch (\Exception $e) {
|
||
Log::error('刷新索引失败: ' . $e->getMessage());
|
||
return ['error' => $e->getMessage()];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 检查索引映射
|
||
* @return array
|
||
*/
|
||
public function checkIndexMapping()
|
||
{
|
||
try {
|
||
return $this->client->indices()->getMapping(['index' => $this->index])->asArray();
|
||
} catch (\Exception $e) {
|
||
return ['error' => $e->getMessage()];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 通用搜索方法
|
||
*
|
||
* @param array $query 搜索查询
|
||
* @param int $from 起始位置
|
||
* @param int $size 返回结果数量
|
||
* @param array $sort 排序规则
|
||
* @return array
|
||
*/
|
||
public function search($query, $from = 0, $size = 10, $sort = [])
|
||
{
|
||
$params = [
|
||
'index' => $this->index,
|
||
'body' => [
|
||
'query' => $query,
|
||
'from' => $from,
|
||
'size' => $size
|
||
]
|
||
];
|
||
|
||
if (!empty($sort)) {
|
||
$params['body']['sort'] = $sort;
|
||
}
|
||
|
||
try {
|
||
return $this->client->search($params)->asArray();
|
||
} catch (\Exception $e) {
|
||
Log::error('搜索失败: ' . $e->getMessage());
|
||
return ['error' => $e->getMessage(), 'hits' => ['total' => ['value' => 0], 'hits' => []]];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 索引名称
|
||
*/
|
||
const indexName = 'default';
|
||
|
||
/**
|
||
* 获取索引名称
|
||
* @param string $index 索引名称
|
||
* @param string|null $prefix 索引前缀
|
||
* @param string|null $subfix 索引后缀
|
||
* @return string
|
||
*/
|
||
public static function indexName($index = '', $prefix = '', $subfix = '')
|
||
{
|
||
$index = $index ?: static::indexName;
|
||
$prefix = $prefix ?: env('ES_INDEX_PREFIX', '');
|
||
$subfix = $subfix ?: env('ES_INDEX_SUFFIX', '');
|
||
if ($prefix) {
|
||
$index = rtrim($prefix, '_') . '_' . $index;
|
||
}
|
||
if ($subfix) {
|
||
$index = $index . '_' . ltrim($subfix, '_');
|
||
}
|
||
return $index;
|
||
}
|
||
}
|