es

type ApplicationPrivileges struct {
	Application string   `json:"application"`
	Privileges  []string `json:"privileges"`
	Resources   []string `json:"resources,omitempty"`
}

type Bulk struct {
	// contains filtered or unexported fields
}

type BulkOperation struct {
	Index   string      `json:"_index"`
	ID      string      `json:"_id,omitempty"`
	Action  string      `json:"-"` // index, create, update, delete
	Payload interface{} `json:"-"`
}

type Client struct {
	*elasticsearch.Client
	// contains filtered or unexported fields
}

type Config struct {
	Addresses []string      `json:"addresses"`
	Username  string        `json:"username"`
	Password  string        `json:"password"`
	APIKey    string        `json:"api_key"`
	Timeout   time.Duration `json:"timeout"`

	// 连接池配置
	MaxIdleConns          int           `json:"max_idle_conns"`
	MaxIdleConnsPerHost   int           `json:"max_idle_conns_per_host"`
	MaxConnsPerHost       int           `json:"max_conns_per_host"`
	IdleConnTimeout       time.Duration `json:"idle_conn_timeout"`
	TLSHandshakeTimeout   time.Duration `json:"tls_handshake_timeout"`
	ExpectContinueTimeout time.Duration `json:"expect_continue_timeout"`
	ResponseHeaderTimeout time.Duration `json:"response_header_timeout"`
	ConnectionTimeout     time.Duration `json:"connection_timeout"`
	ConnectionKeepAlive   time.Duration `json:"connection_keep_alive"`

	// 重试配置
	RetryOnStatus []int         `json:"retry_on_status"`
	MaxRetries    int           `json:"max_retries"`
	RetryBackoff  time.Duration `json:"retry_backoff"` // 基础重试间隔
}

type Document struct {
	// contains filtered or unexported fields
}

type FieldSecurity struct {
	Grant  []string `json:"grant,omitempty"`
	Except []string `json:"except,omitempty"`
}

type Options func(*esOptions)

type PaginatedResult struct {
	SearchResult
	Pagination PaginationResult `json:"pagination"`
}

type PaginatedSearchRequest struct {
	Query      interface{} `json:"query,omitempty"`
	Pagination Pagination  `json:"pagination,omitempty"`
	Sort       interface{} `json:"sort,omitempty"`
	Source     interface{} `json:"_source,omitempty"`
}

type Pagination struct {
	Page     int `json:"page"`      // 页码，从1开始
	PageSize int `json:"page_size"` // 每页大小
}

type PaginationResult struct {
	Page       int `json:"page"`        // 当前页码
	PageSize   int `json:"page_size"`   // 每页大小
	Total      int `json:"total"`       // 总记录数
	TotalPages int `json:"total_pages"` // 总页数
}

type Role struct {
	Cluster           []string                 `json:"cluster,omitempty"`
	Indices           []RoleIndicesPermissions `json:"indices,omitempty"`
	Applications      []ApplicationPrivileges  `json:"applications,omitempty"`
	Global            interface{}              `json:"global,omitempty"`
	Metadata          map[string]interface{}   `json:"metadata,omitempty"`
	TransientMetadata map[string]interface{}   `json:"transient_metadata,omitempty"`
}

type RoleIndicesPermissions struct {
	Names                  []string       `json:"names"`
	Privileges             []string       `json:"privileges"`
	FieldSecurity          *FieldSecurity `json:"field_security,omitempty"`
	Query                  *string        `json:"query,omitempty"`
	AllowRestrictedIndices *bool          `json:"allow_restricted_indices,omitempty"`
}

type ScrollSearchResult struct {
	SearchResult
	ScrollID string `json:"_scroll_id,omitempty"`
}

type SearchRequest struct {
	Query  interface{} `json:"query,omitempty"`
	Size   int         `json:"size,omitempty"`
	From   int         `json:"from,omitempty"`
	Sort   interface{} `json:"sort,omitempty"`
	Source interface{} `json:"_source,omitempty"`
}

type SearchRequestWithSearchAfter struct {
	SearchRequest
	SearchAfter []interface{} `json:"search_after,omitempty"`
}

type SearchResult struct {
	Took     int  `json:"took"`
	TimedOut bool `json:"timed_out"`
	Hits     struct {
		Total struct {
			Value    int    `json:"value"`
			Relation string `json:"relation"`
		} `json:"total"`
		Hits []struct {
			Index  string          `json:"_index"`
			ID     string          `json:"_id"`
			Score  float64         `json:"_score"`
			Source json.RawMessage `json:"_source"`
		} `json:"hits"`
	} `json:"hits"`
}

type SearchService interface {
	Search(ctx context.Context, index string, req SearchRequest) (*SearchResult, error)
	SearchWithRawQuery(ctx context.Context, index string, query []byte) (*SearchResult, error)
	SearchWithPagination(ctx context.Context, index string, req PaginatedSearchRequest) (*PaginatedResult, error)
	ScrollSearch(ctx context.Context, index string, req SearchRequest, scrollTime time.Duration) (*ScrollSearchResult, error)
	ScrollContinue(ctx context.Context, scrollID string, scrollTime time.Duration) (*ScrollSearchResult, error)
	ScrollClear(ctx context.Context, scrollIDs []string) error
	SearchWithSearchAfter(ctx context.Context, index string, req SearchRequestWithSearchAfter) (*SearchResult, error)
}

type User struct {
	Username     string                 `json:"username,omitempty"`
	FullName     string                 `json:"full_name,omitempty"`
	Email        string                 `json:"email,omitempty"`
	Roles        []string               `json:"roles"`
	Password     string                 `json:"password,omitempty"`
	PasswordHash string                 `json:"password_hash,omitempty"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
	Enabled      bool                   `json:"enabled"`
}