This change adds `access` format support for notifications to a Elasticsearch server, and it refactors `namespace` format support. In the case of `access` format, for each event in Minio, a JSON document is inserted into Elasticsearch with its timestamp set to the event's timestamp, and with the ID generated automatically by elasticsearch. No events are modified or deleted in this mode. In the case of `namespace` format, for each event in Minio, a JSON document is keyed together by the bucket and object name is updated in Elasticsearch. In the case of an object being created or over-written in Minio, a new document or an existing document is inserted into the Elasticsearch index. If an object is deleted in Minio, the corresponding document is deleted from the Elasticsearch index. Additionally, this change upgrades Elasticsearch support to the 5.x series. This is a breaking change, and users of previous elasticsearch versions should upgrade. Also updates documentation on Elasticsearch notification target usage and has a link to an elasticsearch upgrade guide. This is the last patch that finally resolves #3928.
123 lines
3.1 KiB
Go
123 lines
3.1 KiB
Go
// Copyright 2012-present Oliver Eilhard. All rights reserved.
|
|
// Use of this source code is governed by a MIT-license.
|
|
// See http://olivere.mit-license.org/license.txt for details.
|
|
|
|
package elastic
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"encoding/json"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"runtime"
|
|
"strings"
|
|
)
|
|
|
|
// Elasticsearch-specific HTTP request
|
|
type Request http.Request
|
|
|
|
// NewRequest is a http.Request and adds features such as encoding the body.
|
|
func NewRequest(method, url string) (*Request, error) {
|
|
req, err := http.NewRequest(method, url, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.Header.Add("User-Agent", "elastic/"+Version+" ("+runtime.GOOS+"-"+runtime.GOARCH+")")
|
|
req.Header.Add("Accept", "application/json")
|
|
return (*Request)(req), nil
|
|
}
|
|
|
|
// SetBasicAuth wraps http.Request's SetBasicAuth.
|
|
func (r *Request) SetBasicAuth(username, password string) {
|
|
((*http.Request)(r)).SetBasicAuth(username, password)
|
|
}
|
|
|
|
// SetBody encodes the body in the request. Optionally, it performs GZIP compression.
|
|
func (r *Request) SetBody(body interface{}, gzipCompress bool) error {
|
|
switch b := body.(type) {
|
|
case string:
|
|
if gzipCompress {
|
|
return r.setBodyGzip(b)
|
|
} else {
|
|
return r.setBodyString(b)
|
|
}
|
|
default:
|
|
if gzipCompress {
|
|
return r.setBodyGzip(body)
|
|
} else {
|
|
return r.setBodyJson(body)
|
|
}
|
|
}
|
|
}
|
|
|
|
// setBodyJson encodes the body as a struct to be marshaled via json.Marshal.
|
|
func (r *Request) setBodyJson(data interface{}) error {
|
|
body, err := json.Marshal(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
r.Header.Set("Content-Type", "application/json")
|
|
r.setBodyReader(bytes.NewReader(body))
|
|
return nil
|
|
}
|
|
|
|
// setBodyString encodes the body as a string.
|
|
func (r *Request) setBodyString(body string) error {
|
|
return r.setBodyReader(strings.NewReader(body))
|
|
}
|
|
|
|
// setBodyGzip gzip's the body. It accepts both strings and structs as body.
|
|
// The latter will be encoded via json.Marshal.
|
|
func (r *Request) setBodyGzip(body interface{}) error {
|
|
switch b := body.(type) {
|
|
case string:
|
|
buf := new(bytes.Buffer)
|
|
w := gzip.NewWriter(buf)
|
|
if _, err := w.Write([]byte(b)); err != nil {
|
|
return err
|
|
}
|
|
if err := w.Close(); err != nil {
|
|
return err
|
|
}
|
|
r.Header.Add("Content-Encoding", "gzip")
|
|
r.Header.Add("Vary", "Accept-Encoding")
|
|
return r.setBodyReader(bytes.NewReader(buf.Bytes()))
|
|
default:
|
|
data, err := json.Marshal(b)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
buf := new(bytes.Buffer)
|
|
w := gzip.NewWriter(buf)
|
|
if _, err := w.Write(data); err != nil {
|
|
return err
|
|
}
|
|
if err := w.Close(); err != nil {
|
|
return err
|
|
}
|
|
r.Header.Add("Content-Encoding", "gzip")
|
|
r.Header.Add("Vary", "Accept-Encoding")
|
|
r.Header.Set("Content-Type", "application/json")
|
|
return r.setBodyReader(bytes.NewReader(buf.Bytes()))
|
|
}
|
|
}
|
|
|
|
// setBodyReader writes the body from an io.Reader.
|
|
func (r *Request) setBodyReader(body io.Reader) error {
|
|
rc, ok := body.(io.ReadCloser)
|
|
if !ok && body != nil {
|
|
rc = ioutil.NopCloser(body)
|
|
}
|
|
r.Body = rc
|
|
if body != nil {
|
|
switch v := body.(type) {
|
|
case *strings.Reader:
|
|
r.ContentLength = int64(v.Len())
|
|
case *bytes.Buffer:
|
|
r.ContentLength = int64(v.Len())
|
|
}
|
|
}
|
|
return nil
|
|
}
|