✨ add search function
This commit is contained in:
5
common/constant/zinc_index_name.go
Normal file
5
common/constant/zinc_index_name.go
Normal file
@@ -0,0 +1,5 @@
|
||||
package constant
|
||||
|
||||
const (
|
||||
ZincIndexNameStorageInfo = "storage_info"
|
||||
)
|
||||
78
common/zincx/special.go
Normal file
78
common/zincx/special.go
Normal file
@@ -0,0 +1,78 @@
|
||||
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
|
||||
}
|
||||
201
common/zincx/zincx.go
Normal file
201
common/zincx/zincx.go
Normal file
@@ -0,0 +1,201 @@
|
||||
package zincx
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/go-resty/resty/v2"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type ZincClient struct {
|
||||
Client *resty.Client
|
||||
BaseURL string
|
||||
Username string
|
||||
Password string
|
||||
}
|
||||
|
||||
func NewZincClient(BaseURL, Username, Password string) *ZincClient {
|
||||
Client := resty.New().
|
||||
SetDebug(false).
|
||||
SetDisableWarn(true)
|
||||
return &ZincClient{
|
||||
Client: Client,
|
||||
BaseURL: BaseURL,
|
||||
Username: Username,
|
||||
Password: Password,
|
||||
}
|
||||
}
|
||||
|
||||
// 检查索引是否存在 (内部方法)
|
||||
func (zc *ZincClient) IndexExists(indexName string) (bool, error) {
|
||||
resp, err := zc.Client.R().
|
||||
SetBasicAuth(zc.Username, zc.Password).
|
||||
Head(zc.BaseURL + "/api/index/" + indexName)
|
||||
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
return resp.StatusCode() == http.StatusOK, nil
|
||||
}
|
||||
|
||||
type IndexMapping struct {
|
||||
Mappings struct {
|
||||
Properties map[string]interface{} `json:"properties"`
|
||||
} `json:"mappings"`
|
||||
}
|
||||
|
||||
func (zc *ZincClient) CreateIndex(indexName string, mapping *IndexMapping) error {
|
||||
url := fmt.Sprintf("%s/api/index/%s", zc.BaseURL, indexName)
|
||||
resp, err := zc.Client.R().
|
||||
SetBasicAuth(zc.Username, zc.Password).
|
||||
SetHeader("Content-Type", "application/json").
|
||||
SetBody(mapping).
|
||||
Put(url)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if resp.StatusCode() != 200 {
|
||||
return fmt.Errorf("创建索引失败: %s", resp.String())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (zc *ZincClient) IndexDocument(indexName, documentID string, doc interface{}) error {
|
||||
url := fmt.Sprintf("%s/api/%s/_doc/%s", zc.BaseURL, indexName, documentID)
|
||||
resp, err := zc.Client.R().
|
||||
SetBasicAuth(zc.Username, zc.Password).
|
||||
SetHeader("Content-Type", "application/json").
|
||||
SetBody(doc).
|
||||
Put(url)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if resp.StatusCode() != 200 {
|
||||
return fmt.Errorf("插入文档失败: %s", resp.String())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type SearchRequest struct {
|
||||
Query struct {
|
||||
Match map[string]interface{} `json:"match"`
|
||||
} `json:"query"`
|
||||
}
|
||||
|
||||
// 泛型响应结构
|
||||
type ZincSearchResponse[T any] struct {
|
||||
Hits struct {
|
||||
Total struct {
|
||||
Value int `json:"value"`
|
||||
} `json:"total"`
|
||||
Hits []struct {
|
||||
ID string `json:"_id"`
|
||||
Source T `json:"_source"`
|
||||
Score float64 `json:"_score"`
|
||||
} `json:"hits"`
|
||||
} `json:"hits"`
|
||||
}
|
||||
|
||||
// 修改 Search 方法签名
|
||||
func (zc *ZincClient) Search(indexName string, query interface{}, resultType interface{}) (*ZincSearchResponse[interface{}], error) {
|
||||
url := fmt.Sprintf("%s/api/%s/_search", zc.BaseURL, indexName)
|
||||
resp, err := zc.Client.R().
|
||||
SetBasicAuth(zc.Username, zc.Password).
|
||||
SetHeader("Content-Type", "application/json").
|
||||
SetBody(query).
|
||||
SetResult(&ZincSearchResponse[interface{}]{}).
|
||||
Post(url)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if resp.StatusCode() != 200 {
|
||||
return nil, fmt.Errorf("搜索失败: %s", resp.String())
|
||||
}
|
||||
|
||||
// 手动反序列化以支持动态类型
|
||||
var rawResponse ZincSearchResponse[interface{}]
|
||||
if err := json.Unmarshal(resp.Body(), &rawResponse); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 将 Source 转换为目标类型
|
||||
for i := range rawResponse.Hits.Hits {
|
||||
data, _ := json.Marshal(rawResponse.Hits.Hits[i].Source)
|
||||
_ = json.Unmarshal(data, &resultType)
|
||||
rawResponse.Hits.Hits[i].Source = resultType
|
||||
}
|
||||
|
||||
return &rawResponse, nil
|
||||
}
|
||||
|
||||
type BulkRequest struct {
|
||||
Index string `json:"index"`
|
||||
ID string `json:"id"`
|
||||
Source interface{} `json:"source"`
|
||||
}
|
||||
|
||||
func (zc *ZincClient) BulkIndex(indexName string, docs []BulkRequest) error {
|
||||
url := fmt.Sprintf("%s/api/_bulk", zc.BaseURL)
|
||||
body := ""
|
||||
for _, doc := range docs {
|
||||
action := fmt.Sprintf(`{ "index": { "_index": "%s", "_id": "%s" } }`, indexName, doc.ID)
|
||||
source, _ := json.Marshal(doc.Source)
|
||||
body += action + "\n" + string(source) + "\n"
|
||||
}
|
||||
|
||||
resp, err := zc.Client.R().
|
||||
SetBasicAuth(zc.Username, zc.Password).
|
||||
SetHeader("Content-Type", "application/json").
|
||||
SetBody(body).
|
||||
Post(url)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if resp.StatusCode() != 200 {
|
||||
return fmt.Errorf("批量操作失败: %s", resp.String())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 删除文档
|
||||
func (zc *ZincClient) DeleteDocument(indexName, documentID string) error {
|
||||
url := fmt.Sprintf("%s/api/%s/_doc/%s", zc.BaseURL, indexName, documentID)
|
||||
resp, err := zc.Client.R().
|
||||
SetBasicAuth(zc.Username, zc.Password).
|
||||
Delete(url)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if resp.StatusCode() != 200 {
|
||||
return fmt.Errorf("删除文档失败: %s", resp.String())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 删除索引
|
||||
func (zc *ZincClient) DeleteIndex(indexName string) error {
|
||||
url := fmt.Sprintf("%s/api/index/%s", zc.BaseURL, indexName)
|
||||
resp, err := zc.Client.R().
|
||||
SetBasicAuth(zc.Username, zc.Password).
|
||||
Delete(url)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if resp.StatusCode() != 200 {
|
||||
return fmt.Errorf("删除索引失败: %s", resp.String())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user