mirror of
https://github.com/tnb-labs/panel.git
synced 2026-09-01 14:55:12 +08:00
耗子
121 lines
3.2 KiB
Go
121 lines
3.2 KiB
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"github.com/leonelquinteros/gotext"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/acepanel/panel/v3/internal/biz"
|
|
"github.com/acepanel/panel/v3/internal/request"
|
|
"github.com/acepanel/panel/v3/pkg/db"
|
|
)
|
|
|
|
type databaseElasticsearchRepo struct {
|
|
t *gotext.Locale
|
|
orm *gorm.DB
|
|
log *slog.Logger
|
|
}
|
|
|
|
func NewDatabaseElasticsearchRepo(db *gorm.DB, t *gotext.Locale, log *slog.Logger) biz.DatabaseElasticsearchRepo {
|
|
return &databaseElasticsearchRepo{
|
|
t: t,
|
|
orm: db,
|
|
log: log,
|
|
}
|
|
}
|
|
|
|
func (r *databaseElasticsearchRepo) Indices(ctx context.Context, req *request.DatabaseESIndices) ([]db.ESIndex, error) {
|
|
client, err := r.getClient(ctx, req.ServerID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer client.Close()
|
|
|
|
return client.Indices()
|
|
}
|
|
|
|
func (r *databaseElasticsearchRepo) IndexCreate(ctx context.Context, req *request.DatabaseESIndexCreate) error {
|
|
client, err := r.getClient(ctx, req.ServerID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer client.Close()
|
|
|
|
return client.IndexCreate(req.Name)
|
|
}
|
|
|
|
func (r *databaseElasticsearchRepo) IndexDelete(ctx context.Context, req *request.DatabaseESIndexDelete) error {
|
|
client, err := r.getClient(ctx, req.ServerID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer client.Close()
|
|
|
|
return client.IndexDelete(req.Name)
|
|
}
|
|
|
|
func (r *databaseElasticsearchRepo) Data(ctx context.Context, req *request.DatabaseESData) ([]db.ESDocument, int64, error) {
|
|
client, err := r.getClient(ctx, req.ServerID)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
defer client.Close()
|
|
|
|
return client.Search(req.Index, req.Search, int(req.Page), int(req.Limit))
|
|
}
|
|
|
|
func (r *databaseElasticsearchRepo) DocumentGet(ctx context.Context, req *request.DatabaseESDocumentGet) (*db.ESDocument, error) {
|
|
client, err := r.getClient(ctx, req.ServerID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer client.Close()
|
|
|
|
return client.DocumentGet(req.Index, req.ID)
|
|
}
|
|
|
|
func (r *databaseElasticsearchRepo) DocumentSet(ctx context.Context, req *request.DatabaseESDocumentSet) error {
|
|
client, err := r.getClient(ctx, req.ServerID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer client.Close()
|
|
|
|
if req.ID == "" {
|
|
return client.DocumentCreate(req.Index, req.Body)
|
|
}
|
|
return client.DocumentUpdate(req.Index, req.ID, req.Body)
|
|
}
|
|
|
|
func (r *databaseElasticsearchRepo) DocumentDelete(ctx context.Context, req *request.DatabaseESDocumentDelete) error {
|
|
client, err := r.getClient(ctx, req.ServerID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer client.Close()
|
|
|
|
return client.DocumentDelete(req.Index, req.ID)
|
|
}
|
|
|
|
// getClient 根据服务器 ID 创建 Elasticsearch 客户端
|
|
func (r *databaseElasticsearchRepo) getClient(ctx context.Context, serverID uint) (*db.Elasticsearch, error) {
|
|
server := new(biz.DatabaseServer)
|
|
if err := r.orm.Where("id = ?", serverID).First(server).Error; err != nil {
|
|
return nil, errors.New(r.t.Get("server not found"))
|
|
}
|
|
if server.Type != biz.DatabaseTypeElasticsearch {
|
|
return nil, errors.New(r.t.Get("server is not Elasticsearch type"))
|
|
}
|
|
|
|
client, err := db.NewElasticsearch(ctx, fmt.Sprintf("%s:%d", server.Host, server.Port), server.Username, server.Password)
|
|
if err != nil {
|
|
return nil, errors.New(r.t.Get("failed to connect to Elasticsearch: %v", err))
|
|
}
|
|
|
|
return client, nil
|
|
}
|