mirror of
https://github.com/kuaifan/dootask.git
synced 2025-12-14 04:32:49 +00:00
460 lines
14 KiB
PHP
460 lines
14 KiB
PHP
<?php
|
||
|
||
namespace App\Module\ZincSearch;
|
||
|
||
use App\Models\WebSocketDialogMsg;
|
||
use App\Models\WebSocketDialogUser;
|
||
use Illuminate\Support\Facades\Log;
|
||
|
||
/**
|
||
* 对话系统消息索引
|
||
*
|
||
* 使用方法:
|
||
*
|
||
* 1. 索引管理
|
||
* - 创建索引: ZincSearchUserMsg::generateIndex();
|
||
* - 检查索引: ZincSearchBase::indexExists(ZincSearchUserMsg::$indexName);
|
||
* - 清空索引: ZincSearchUserMsg::clear();
|
||
*
|
||
* 2. 会话用户操作
|
||
* - 单个同步: ZincSearchUserMsg::syncUser($dialogUser);
|
||
* - 批量同步: ZincSearchUserMsg::batchSyncUsers($dialogUsers);
|
||
* - 删除用户: ZincSearchUserMsg::deleteUser($dialogUser);
|
||
*
|
||
* 3. 会话消息操作
|
||
* - 单个同步: ZincSearchUserMsg::syncMsg($dialogMsg);
|
||
* - 批量同步: ZincSearchUserMsg::batchSyncMsgs($dialogMsgs);
|
||
* - 删除消息: ZincSearchUserMsg::deleteMsg($dialogMsg);
|
||
*
|
||
* 4. 搜索功能
|
||
* - 关键词搜索: ZincSearchUserMsg::searchByKeyword('用户ID', '关键词');
|
||
*
|
||
* Class ZincSearchUserMsg
|
||
* @package App\Module\ZincSearch
|
||
*/
|
||
class ZincSearchUserMsg
|
||
{
|
||
/**
|
||
* 索引名称
|
||
*/
|
||
protected static string $indexName = 'userMsg';
|
||
|
||
// ==============================
|
||
// 索引管理相关方法
|
||
// ==============================
|
||
|
||
/**
|
||
* 创建聊天系统索引
|
||
*
|
||
* @return array
|
||
*/
|
||
public static function generateIndex(): array
|
||
{
|
||
// 定义映射
|
||
$mappings = [
|
||
'properties' => [
|
||
// 共用字段
|
||
'dialog_id' => ['type' => 'keyword', 'index' => true],
|
||
'created_at' => ['type' => 'date', 'index' => true],
|
||
'updated_at' => ['type' => 'date', 'index' => true],
|
||
|
||
// dialog_users 字段
|
||
'userid' => ['type' => 'keyword', 'index' => true],
|
||
'top_at' => ['type' => 'date', 'index' => true],
|
||
'last_at' => ['type' => 'date', 'index' => true],
|
||
'mark_unread' => ['type' => 'numeric', 'index' => true],
|
||
'silence' => ['type' => 'numeric', 'index' => true],
|
||
'hide' => ['type' => 'numeric', 'index' => true],
|
||
'color' => ['type' => 'keyword', 'index' => true],
|
||
|
||
// dialog_msgs 字段
|
||
'msg_id' => ['type' => 'keyword', 'index' => true],
|
||
'sender_userid' => ['type' => 'keyword', 'index' => true],
|
||
'msg_type' => ['type' => 'keyword', 'index' => true],
|
||
'key' => ['type' => 'text', 'index' => true],
|
||
'bot' => ['type' => 'numeric', 'index' => true],
|
||
|
||
// 关联字段 - ZincSearch不支持父子文档，使用引用字段代替
|
||
'parent_id' => ['type' => 'keyword', 'index' => true],
|
||
'doc_type' => ['type' => 'keyword', 'index' => true] // dialog_user 或 dialog_msg
|
||
]
|
||
];
|
||
|
||
try {
|
||
return ZincSearchBase::createIndex(self::$indexName, $mappings);
|
||
} catch (\Exception $e) {
|
||
Log::error('创建聊天系统索引失败: ' . $e->getMessage());
|
||
return ['success' => false, 'error' => $e->getMessage()];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 清空对话系统索引
|
||
* 删除索引并重新创建一个空索引
|
||
*
|
||
* @return bool 是否清空成功
|
||
*/
|
||
public static function clear(): bool
|
||
{
|
||
try {
|
||
// 检查索引是否存在
|
||
if (!ZincSearchBase::indexExists(self::$indexName)) {
|
||
return true; // 索引不存在视为已清空
|
||
}
|
||
|
||
// 删除索引
|
||
$deleteResult = ZincSearchBase::deleteIndex(self::$indexName);
|
||
if (!($deleteResult['success'] ?? false)) {
|
||
Log::error('清空对话系统索引失败: ' . ($deleteResult['error'] ?? '未知错误'));
|
||
return false;
|
||
}
|
||
|
||
// 重新创建索引
|
||
$createResult = self::generateIndex();
|
||
return $createResult['success'] ?? false;
|
||
} catch (\Exception $e) {
|
||
Log::error('清空对话系统索引异常: ' . $e->getMessage());
|
||
return false;
|
||
}
|
||
}
|
||
|
||
// ==============================
|
||
// 搜索相关方法
|
||
// ==============================
|
||
|
||
/**
|
||
* 构建对话系统特定的搜索 - 根据用户ID和消息关键词搜索会话
|
||
*
|
||
* @param string $userid 用户ID
|
||
* @param string $keyword 消息关键词
|
||
* @param int $from 起始位置
|
||
* @param int $size 返回结果数量
|
||
* @return array
|
||
*/
|
||
public static function searchByKeyword(string $userid, string $keyword, int $from = 0, int $size = 20): array
|
||
{
|
||
// 构建复杂的搜索查询
|
||
$searchParams = [
|
||
'search_type' => 'querystring',
|
||
'query' => [
|
||
'term' => "+userid:{$userid} +key:*{$keyword}*"
|
||
],
|
||
'from' => $from,
|
||
'max_results' => $size,
|
||
'sort_fields' => ["updated_at:desc"]
|
||
];
|
||
|
||
try {
|
||
return ZincSearchBase::advancedSearch(self::$indexName, $searchParams);
|
||
} catch (\Exception $e) {
|
||
Log::error('搜索对话消息失败: ' . $e->getMessage());
|
||
return [
|
||
'success' => false,
|
||
'error' => $e->getMessage(),
|
||
'hits' => ['total' => ['value' => 0], 'hits' => []]
|
||
];
|
||
}
|
||
}
|
||
|
||
// ==============================
|
||
// 会话用户相关方法
|
||
// ==============================
|
||
|
||
/**
|
||
* 会话用户 - 生成文档ID
|
||
*
|
||
* @param WebSocketDialogUser $dialogUser
|
||
* @return string
|
||
*/
|
||
public static function generateUserDocId(WebSocketDialogUser $dialogUser): string
|
||
{
|
||
return "user_{$dialogUser->userid}_dialog_{$dialogUser->dialog_id}";
|
||
}
|
||
|
||
/**
|
||
* 会话用户 - 生成文档格式
|
||
*
|
||
* @param WebSocketDialogUser $dialogUser
|
||
* @return array
|
||
*/
|
||
public static function generateUserFormat(WebSocketDialogUser $dialogUser): array
|
||
{
|
||
return [
|
||
'dialog_id' => $dialogUser->dialog_id,
|
||
'created_at' => $dialogUser->created_at,
|
||
'updated_at' => $dialogUser->updated_at,
|
||
|
||
'userid' => $dialogUser->userid,
|
||
'top_at' => $dialogUser->top_at,
|
||
'last_at' => $dialogUser->last_at,
|
||
'mark_unread' => $dialogUser->mark_unread ?: 0,
|
||
'silence' => $dialogUser->silence ?: 0,
|
||
'hide' => $dialogUser->hide ?: 0,
|
||
'color' => $dialogUser->color,
|
||
|
||
'doc_type' => 'dialog_user',
|
||
'parent_id' => '' // 用户文档没有父文档
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 会话用户 - 同步到ZincSearch
|
||
*
|
||
* @param WebSocketDialogUser $dialogUser
|
||
* @return void
|
||
*/
|
||
public static function syncUser(WebSocketDialogUser $dialogUser): void
|
||
{
|
||
try {
|
||
if (!ZincSearchBase::indexExists(self::$indexName)) {
|
||
self::generateIndex();
|
||
}
|
||
$docFormat = self::generateUserFormat($dialogUser);
|
||
ZincSearchBase::addDoc(self::$indexName, $docFormat);
|
||
} catch (\Exception $e) {
|
||
Log::error('syncUser: ' . $e->getMessage());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 批量同步会话用户
|
||
*
|
||
* @param array|iterable $dialogUsers WebSocketDialogUser对象集合
|
||
* @return int 成功同步的用户数
|
||
*/
|
||
public static function batchSyncUsers($dialogUsers): int
|
||
{
|
||
$count = 0;
|
||
try {
|
||
if (!ZincSearchBase::indexExists(self::$indexName)) {
|
||
self::generateIndex();
|
||
}
|
||
|
||
$docs = [];
|
||
foreach ($dialogUsers as $dialogUser) {
|
||
$docs[] = self::generateUserFormat($dialogUser);
|
||
$count++;
|
||
}
|
||
|
||
if (!empty($docs)) {
|
||
ZincSearchBase::addDocs(self::$indexName, $docs);
|
||
}
|
||
} catch (\Exception $e) {
|
||
Log::error('batchSyncUsers: ' . $e->getMessage());
|
||
}
|
||
|
||
return $count;
|
||
}
|
||
|
||
/**
|
||
* 会话用户 - 从ZincSearch删除
|
||
*
|
||
* @param WebSocketDialogUser $dialogUser
|
||
* @return void
|
||
*/
|
||
public static function deleteUser(WebSocketDialogUser $dialogUser): void
|
||
{
|
||
try {
|
||
$docId = self::generateUserDocId($dialogUser);
|
||
|
||
// 首先查询相关消息
|
||
$searchParams = [
|
||
'search_type' => 'term',
|
||
'query' => [
|
||
'field' => 'parent_id',
|
||
'term' => $docId
|
||
],
|
||
'from' => 0,
|
||
'max_results' => 1000 // 限制一次查询返回的文档数
|
||
];
|
||
|
||
$result = ZincSearchBase::advancedSearch(self::$indexName, $searchParams);
|
||
$hits = $result['data']['hits']['hits'] ?? [];
|
||
|
||
// 批量删除子文档
|
||
$batch = [];
|
||
foreach ($hits as $hit) {
|
||
if (isset($hit['_id'])) {
|
||
ZincSearchBase::deleteDoc(self::$indexName, $hit['_id']);
|
||
}
|
||
}
|
||
|
||
// 删除用户文档
|
||
ZincSearchBase::deleteDoc(self::$indexName, $docId);
|
||
} catch (\Exception $e) {
|
||
Log::error('deleteUser: ' . $e->getMessage());
|
||
}
|
||
}
|
||
|
||
// ==============================
|
||
// 会话消息相关方法
|
||
// ==============================
|
||
|
||
/**
|
||
* 会话消息 - 生成父文档ID
|
||
*
|
||
* @param WebSocketDialogMsg $dialogMsg
|
||
* @param string $userid
|
||
* @return string
|
||
*/
|
||
public static function generateMsgParentId(WebSocketDialogMsg $dialogMsg, string $userid): string
|
||
{
|
||
return "user_{$userid}_dialog_{$dialogMsg->dialog_id}";
|
||
}
|
||
|
||
/**
|
||
* 会话消息 - 生成文档ID
|
||
*
|
||
* @param WebSocketDialogMsg $dialogMsg
|
||
* @param string $userid
|
||
* @return string
|
||
*/
|
||
public static function generateMsgDocId(WebSocketDialogMsg $dialogMsg, string $userid): string
|
||
{
|
||
return "msg_{$dialogMsg->id}_user_{$userid}";
|
||
}
|
||
|
||
/**
|
||
* 会话消息 - 生成文档格式
|
||
*
|
||
* @param WebSocketDialogMsg $dialogMsg
|
||
* @param string $userid
|
||
* @return array
|
||
*/
|
||
public static function generateMsgFormat(WebSocketDialogMsg $dialogMsg, string $userid): array
|
||
{
|
||
return [
|
||
'dialog_id' => $dialogMsg->dialog_id,
|
||
'created_at' => $dialogMsg->created_at,
|
||
'updated_at' => $dialogMsg->updated_at,
|
||
|
||
'msg_id' => $dialogMsg->id,
|
||
'sender_userid' => $dialogMsg->userid,
|
||
'msg_type' => $dialogMsg->type,
|
||
'key' => $dialogMsg->key,
|
||
'bot' => $dialogMsg->bot ? 1 : 0,
|
||
|
||
'doc_type' => 'dialog_msg',
|
||
'parent_id' => self::generateMsgParentId($dialogMsg, $userid)
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 会话消息 - 同步到ZincSearch
|
||
*
|
||
* @param WebSocketDialogMsg $dialogMsg
|
||
* @return void
|
||
*/
|
||
public static function syncMsg(WebSocketDialogMsg $dialogMsg): void
|
||
{
|
||
try {
|
||
// 获取此会话的所有用户
|
||
$dialogUsers = WebSocketDialogUser::whereDialogId($dialogMsg->dialog_id)->get();
|
||
|
||
if ($dialogUsers->isEmpty()) {
|
||
return;
|
||
}
|
||
|
||
$docs = [];
|
||
foreach ($dialogUsers as $dialogUser) {
|
||
$docId = self::generateMsgDocId($dialogMsg, $dialogUser->userid);
|
||
$docFormat = self::generateMsgFormat($dialogMsg, $dialogUser->userid);
|
||
$docs[] = $docFormat;
|
||
}
|
||
|
||
if (!empty($docs)) {
|
||
if (ZincSearchBase::indexExists(self::$indexName)) {
|
||
ZincSearchBase::addDocs(self::$indexName, $docs);
|
||
} else {
|
||
self::generateIndex();
|
||
ZincSearchBase::addDocs(self::$indexName, $docs);
|
||
}
|
||
}
|
||
} catch (\Exception $e) {
|
||
Log::error('syncMsg: ' . $e->getMessage());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 批量同步会话消息
|
||
*
|
||
* @param array|iterable $dialogMsgs WebSocketDialogMsg对象集合
|
||
* @return int 成功同步的消息数
|
||
*/
|
||
public static function batchSyncMsgs($dialogMsgs): int
|
||
{
|
||
$count = 0;
|
||
try {
|
||
$docs = [];
|
||
$userDialogs = [];
|
||
|
||
// 预处理：收集所有涉及的对话ID
|
||
$dialogIds = [];
|
||
foreach ($dialogMsgs as $message) {
|
||
$dialogIds[] = $message->dialog_id;
|
||
}
|
||
$dialogIds = array_unique($dialogIds);
|
||
|
||
// 获取所有相关的用户-对话关系
|
||
if (!empty($dialogIds)) {
|
||
$dialogUsers = WebSocketDialogUser::whereIn('dialog_id', $dialogIds)->get();
|
||
|
||
// 按对话ID组织用户
|
||
foreach ($dialogUsers as $dialogUser) {
|
||
$userDialogs[$dialogUser->dialog_id][] = $dialogUser;
|
||
}
|
||
}
|
||
|
||
// 为每条消息准备所有相关用户的文档
|
||
foreach ($dialogMsgs as $message) {
|
||
if (isset($userDialogs[$message->dialog_id])) {
|
||
foreach ($userDialogs[$message->dialog_id] as $dialogUser) {
|
||
$docFormat = self::generateMsgFormat($message, $dialogUser->userid);
|
||
$docs[] = $docFormat;
|
||
$count++;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 批量写入
|
||
if (!empty($docs)) {
|
||
if (ZincSearchBase::indexExists(self::$indexName)) {
|
||
ZincSearchBase::addDocs(self::$indexName, $docs);
|
||
} else {
|
||
self::generateIndex();
|
||
ZincSearchBase::addDocs(self::$indexName, $docs);
|
||
}
|
||
}
|
||
} catch (\Exception $e) {
|
||
Log::error('batchSyncMsgs: ' . $e->getMessage());
|
||
}
|
||
|
||
return $count;
|
||
}
|
||
|
||
/**
|
||
* 会话消息 - 从ZincSearch删除
|
||
*
|
||
* @param WebSocketDialogMsg $dialogMsg
|
||
* @return void
|
||
*/
|
||
public static function deleteMsg(WebSocketDialogMsg $dialogMsg): void
|
||
{
|
||
try {
|
||
// 获取此会话的所有用户
|
||
$dialogUsers = WebSocketDialogUser::whereDialogId($dialogMsg->dialog_id)->get();
|
||
|
||
if ($dialogUsers->isEmpty()) {
|
||
return;
|
||
}
|
||
|
||
foreach ($dialogUsers as $dialogUser) {
|
||
$docId = self::generateMsgDocId($dialogMsg, $dialogUser->userid);
|
||
ZincSearchBase::deleteDoc(self::$indexName, $docId);
|
||
}
|
||
} catch (\Exception $e) {
|
||
Log::error('deleteMsg: ' . $e->getMessage());
|
||
}
|
||
}
|
||
}
|