79 lines
2.5 KiB
Go
79 lines
2.5 KiB
Go
package zincx
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
)
|
|
|
|
// 创建索引（幂等操作，若存在则跳过）
|
|
func (zc *ZincClient) CreateFileUploadIndex(indexName string) error {
|
|
exists, err := zc.IndexExists(indexName)
|
|
if err != nil {
|
|
return fmt.Errorf("检查索引失败: %w", err)
|
|
}
|
|
if exists {
|
|
return nil // 索引已存在则跳过
|
|
}
|
|
|
|
// 定义完整的索引映射
|
|
mapping := map[string]interface{}{
|
|
"mappings": map[string]interface{}{
|
|
"properties": map[string]interface{}{
|
|
// 基础信息
|
|
"storage_id": map[string]string{"type": "numeric"},
|
|
"face_id": map[string]string{"type": "numeric"},
|
|
"file_name": map[string]string{"type": "keyword"},
|
|
"file_size": map[string]string{"type": "numeric"},
|
|
"uid": map[string]string{"type": "keyword"},
|
|
"file_path": map[string]string{"type": "text"},
|
|
"thumb_path": map[string]string{"type": "text"},
|
|
"created_at": map[string]string{
|
|
"type": "date"},
|
|
|
|
// 文件元数据
|
|
"provider": map[string]string{"type": "keyword"},
|
|
"bucket": map[string]string{"type": "keyword"},
|
|
"file_type": map[string]string{"type": "keyword"},
|
|
"is_anime": map[string]string{"type": "boolean"},
|
|
"tag_name": map[string]string{"type": "keyword"},
|
|
"landscape": map[string]string{"type": "keyword"},
|
|
"top_category": map[string]string{"type": "keyword"},
|
|
"is_screenshot": map[string]string{"type": "boolean"},
|
|
|
|
// 媒体属性
|
|
"width": map[string]string{"type": "numeric"},
|
|
"height": map[string]string{"type": "numeric"},
|
|
"thumb_w": map[string]string{"type": "numeric"},
|
|
"thumb_h": map[string]string{"type": "numeric"},
|
|
"thumb_size": map[string]string{"type": "numeric"},
|
|
|
|
// 地理信息
|
|
"longitude": map[string]string{"type": "numeric"},
|
|
"latitude": map[string]string{"type": "numeric"},
|
|
"country": map[string]string{"type": "keyword"},
|
|
"province": map[string]string{"type": "keyword"},
|
|
"city": map[string]string{"type": "keyword"},
|
|
|
|
// 其他
|
|
"album_id": map[string]string{"type": "numeric"},
|
|
"has_qrcode": map[string]string{"type": "boolean"},
|
|
},
|
|
},
|
|
}
|
|
|
|
resp, err := zc.Client.R().
|
|
SetBasicAuth(zc.Username, zc.Password).
|
|
SetHeader("Content-Type", "application/json").
|
|
SetBody(mapping).
|
|
Put(zc.BaseURL + "/api/index/" + indexName)
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("创建索引请求失败: %w", err)
|
|
}
|
|
|
|
if resp.StatusCode() != http.StatusOK {
|
|
return fmt.Errorf("创建索引失败 (状态码 %d): %s", resp.StatusCode(), resp.String())
|
|
}
|
|
return nil
|
|
}
|