esutil

type BulkIndexer interface {
	// Add adds an item to the indexer. It returns an error when the item cannot be added.
	// Use the OnSuccess and OnFailure callbacks to get the operation result for the item.
	//
	// You must call the Close() method after you're done adding items.
	//
	// It is safe for concurrent use. When it's called from goroutines,
	// they must finish before the call to Close, eg. using sync.WaitGroup.
	Add(context.Context, BulkIndexerItem) error

	// Close waits until all added items are flushed and closes the indexer.
	Close(context.Context) error

	// Flush drains all currently queued items, flushes all worker buffers to
	// Elasticsearch, and waits for the flushes to complete. The indexer remains
	// usable after Flush returns; new items may be added with Add.
	//
	// Per-item results are delivered through the OnSuccess and OnFailure
	// callbacks on each BulkIndexerItem, exactly as with automatic flushes.
	// Flush itself only returns an error for transport-level failures or
	// context cancellation.
	//
	// Items added concurrently with Flush may or may not be included in the
	// flush; only items already enqueued at the time each worker receives the
	// flush signal are guaranteed to be sent.
	//
	// Concurrent calls to Flush are serialised. Flush returns ErrIndexerClosed
	// if the indexer has been closed.
	//
	// It is safe for concurrent use with Add.
	Flush(context.Context) error

	// Stats returns indexer statistics.
	Stats() BulkIndexerStats
}

type BulkIndexerConfig struct {
	NumWorkers          int           // The number of workers. Defaults to runtime.NumCPU().
	FlushBytes          int           // The flush threshold in bytes. Defaults to 5MB.
	FlushInterval       time.Duration // The flush threshold as duration. Defaults to 30sec.
	FlushJitter         time.Duration // Max random jitter added to FlushInterval per worker, to avoid lockstep flushes. Defaults to 0 (disabled).
	QueueSizeMultiplier int           // The capacity of each worker's item queue. Total capacity is NumWorkers * QueueSizeMultiplier. Defaults to 1.

	Client      esapi.Transport         // The Elasticsearch client.
	Decoder     BulkResponseJSONDecoder // A custom JSON decoder.
	DebugLogger BulkIndexerDebugLogger  // An optional logger for debugging.

	OnError      func(context.Context, error)          // Called for indexer errors.
	OnFlushStart func(context.Context) context.Context // Called when the flush starts.
	OnFlushEnd   func(context.Context)                 // Called when the flush ends.

	// Parameters of the Bulk API.
	Index               string
	ErrorTrace          bool
	FilterPath          []string
	Header              http.Header
	Human               bool
	Pipeline            string
	Pretty              bool
	Refresh             string
	Routing             string
	RequireAlias        bool
	Source              []string
	SourceExcludes      []string
	SourceIncludes      []string
	Timeout             time.Duration
	WaitForActiveShards string
}

type BulkIndexerDebugLogger interface {
	Printf(string, ...interface{})
}

type BulkIndexerItem struct {
	Index           string
	Action          string
	DocumentID      string
	Routing         string
	RequireAlias    bool
	Version         *int64
	VersionType     string
	Body            io.ReadSeeker
	RetryOnConflict *int
	IfSeqNo         *int64
	IfPrimaryTerm   *int64

	OnSuccess func(context.Context, BulkIndexerItem, BulkIndexerResponseItem)        // Per item
	OnFailure func(context.Context, BulkIndexerItem, BulkIndexerResponseItem, error) // Per item
	// contains filtered or unexported fields
}

type BulkIndexerResponse struct {
	Took      int                                  `json:"took"`
	HasErrors bool                                 `json:"errors"`
	Items     []map[string]BulkIndexerResponseItem `json:"items,omitempty"`
}

type BulkIndexerResponseItem struct {
	Index        string `json:"_index"`
	DocumentID   string `json:"_id"`
	Version      int64  `json:"_version"`
	Result       string `json:"result"`
	Status       int    `json:"status"`
	SeqNo        int64  `json:"_seq_no"`
	PrimTerm     int64  `json:"_primary_term"`
	FailureStore string `json:"failure_store,omitempty"`

	Shards struct {
		Total      int `json:"total"`
		Successful int `json:"successful"`
		Failed     int `json:"failed"`
	} `json:"_shards"`

	Error struct {
		Type   string `json:"type"`
		Reason string `json:"reason"`
		Cause  struct {
			Type   string `json:"type"`
			Reason string `json:"reason"`
		} `json:"caused_by"`
	} `json:"error,omitempty"`
}

type BulkIndexerStats struct {
	NumAdded     uint64
	NumFlushed   uint64
	NumFailed    uint64
	NumIndexed   uint64
	NumCreated   uint64
	NumUpdated   uint64
	NumDeleted   uint64
	NumRequests  uint64
	FlushedBytes uint64
	FlushedMs    uint64
}

type BulkResponseJSONDecoder interface {
	UnmarshalFromReader(io.Reader, *BulkIndexerResponse) error
}

type JSONEncoder interface {
	EncodeJSON(io.Writer) error
}

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