youtubebeat/vendor/github.com/elastic/beats/libbeat/dashboards/kibana_loader.go

140 lines
4.2 KiB
Go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package dashboards
import (
"context"
"encoding/json"
"fmt"
"io/ioutil"
"net/url"
"time"
"github.com/elastic/beats/libbeat/common"
"github.com/elastic/beats/libbeat/kibana"
"github.com/elastic/beats/libbeat/logp"
)
var importAPI = "/api/kibana/dashboards/import"
type KibanaLoader struct {
client *kibana.Client
config *Config
version string
hostname string
msgOutputter MessageOutputter
}
func NewKibanaLoader(ctx context.Context, cfg *common.Config, dashboardsConfig *Config, hostname string, msgOutputter MessageOutputter) (*KibanaLoader, error) {
if cfg == nil || !cfg.Enabled() {
return nil, fmt.Errorf("Kibana is not configured or enabled")
}
client, err := getKibanaClient(ctx, cfg, dashboardsConfig.Retry, 0)
if err != nil {
return nil, fmt.Errorf("Error creating Kibana client: %v", err)
}
loader := KibanaLoader{
client: client,
config: dashboardsConfig,
version: client.GetVersion(),
hostname: hostname,
msgOutputter: msgOutputter,
}
loader.statusMsg("Initialize the Kibana %s loader", client.GetVersion())
return &loader, nil
}
func getKibanaClient(ctx context.Context, cfg *common.Config, retryCfg *Retry, retryAttempt uint) (*kibana.Client, error) {
client, err := kibana.NewKibanaClient(cfg)
if err != nil {
if retryCfg.Enabled && (retryCfg.Maximum == 0 || retryCfg.Maximum > retryAttempt) {
select {
case <-ctx.Done():
return nil, err
case <-time.After(retryCfg.Interval):
return getKibanaClient(ctx, cfg, retryCfg, retryAttempt+1)
}
}
return nil, fmt.Errorf("Error creating Kibana client: %v", err)
}
return client, nil
}
func (loader KibanaLoader) ImportIndex(file string) error {
params := url.Values{}
params.Set("force", "true") //overwrite the existing dashboards
// read json file
reader, err := ioutil.ReadFile(file)
if err != nil {
return fmt.Errorf("fail to read index-pattern from file %s: %v", file, err)
}
var indexContent common.MapStr
err = json.Unmarshal(reader, &indexContent)
if err != nil {
return fmt.Errorf("fail to unmarshal the index content from file %s: %v", file, err)
}
indexContent = ReplaceIndexInIndexPattern(loader.config.Index, indexContent)
return loader.client.ImportJSON(importAPI, params, indexContent)
}
func (loader KibanaLoader) ImportDashboard(file string) error {
params := url.Values{}
params.Set("force", "true") //overwrite the existing dashboards
params.Add("exclude", "index-pattern") //don't import the index pattern from the dashboards
// read json file
reader, err := ioutil.ReadFile(file)
if err != nil {
return fmt.Errorf("fail to read dashboard from file %s: %v", file, err)
}
var content common.MapStr
err = json.Unmarshal(reader, &content)
if err != nil {
return fmt.Errorf("fail to unmarshal the dashboard content from file %s: %v", file, err)
}
content = ReplaceIndexInDashboardObject(loader.config.Index, content)
content, err = ReplaceStringInDashboard("CHANGEME_HOSTNAME", loader.hostname, content)
if err != nil {
return fmt.Errorf("fail to replace the hostname in dashboard %s: %v", file, err)
}
return loader.client.ImportJSON(importAPI, params, content)
}
func (loader KibanaLoader) Close() error {
return loader.client.Close()
}
func (loader KibanaLoader) statusMsg(msg string, a ...interface{}) {
if loader.msgOutputter != nil {
loader.msgOutputter(msg, a...)
} else {
logp.Debug("dashboards", msg, a...)
}
}