Add ClickHouse input plugin (#6441)
This commit is contained in:
Daniel Nelson
GitHub
parent
b6892378a0
commit
e4caa347a2
|
|
@ -2,4 +2,4 @@
|
|||
/telegraf
|
||||
/telegraf.exe
|
||||
/telegraf.gz
|
||||
/vendor
|
||||
/vendor
|
||||
|
|
@ -19,6 +19,7 @@ import (
|
|||
_ "github.com/influxdata/telegraf/plugins/inputs/chrony"
|
||||
_ "github.com/influxdata/telegraf/plugins/inputs/cisco_telemetry_gnmi"
|
||||
_ "github.com/influxdata/telegraf/plugins/inputs/cisco_telemetry_mdt"
|
||||
_ "github.com/influxdata/telegraf/plugins/inputs/clickhouse"
|
||||
_ "github.com/influxdata/telegraf/plugins/inputs/cloud_pubsub"
|
||||
_ "github.com/influxdata/telegraf/plugins/inputs/cloud_pubsub_push"
|
||||
_ "github.com/influxdata/telegraf/plugins/inputs/cloudwatch"
|
||||
|
|
|
|||
|
|
@ -0,0 +1,119 @@
|
|||
# Telegraf Input Plugin: ClickHouse
|
||||
|
||||
This plugin gathers the statistic data from [ClickHouse](https://github.com/ClickHouse/ClickHouse) server.
|
||||
|
||||
### Configuration
|
||||
```ini
|
||||
# Read metrics from one or many ClickHouse servers
|
||||
[[inputs.clickhouse]]
|
||||
## Username for authorization on ClickHouse server
|
||||
## example: user = "default"
|
||||
user = "default"
|
||||
|
||||
## Password for authorization on ClickHouse server
|
||||
## example: password = "super_secret"
|
||||
|
||||
## HTTP(s) timeout while getting metrics values
|
||||
## The timeout includes connection time, any redirects, and reading the response body.
|
||||
## example: timeout = 1s
|
||||
# timeout = 5s
|
||||
|
||||
## List of servers for metrics scraping
|
||||
## metrics scrape via HTTP(s) clickhouse interface
|
||||
## https://clickhouse.tech/docs/en/interfaces/http/
|
||||
## example: servers = ["http://127.0.0.1:8123","https://custom-server.mdb.yandexcloud.net"]
|
||||
servers = ["http://127.0.0.1:8123"]
|
||||
|
||||
## If "auto_discovery"" is "true" plugin tries to connect to all servers available in the cluster
|
||||
## with using same "user:password" described in "user" and "password" parameters
|
||||
## and get this server hostname list from "system.clusters" table
|
||||
## see
|
||||
## - https://clickhouse.tech/docs/en/operations/system_tables/#system-clusters
|
||||
## - https://clickhouse.tech/docs/en/operations/server_settings/settings/#server_settings_remote_servers
|
||||
## - https://clickhouse.tech/docs/en/operations/table_engines/distributed/
|
||||
## - https://clickhouse.tech/docs/en/operations/table_engines/replication/#creating-replicated-tables
|
||||
## example: auto_discovery = false
|
||||
# auto_discovery = true
|
||||
|
||||
## Filter cluster names in "system.clusters" when "auto_discovery" is "true"
|
||||
## when this filter present then "WHERE cluster IN (...)" filter will apply
|
||||
## please use only full cluster names here, regexp and glob filters is not allowed
|
||||
## for "/etc/clickhouse-server/config.d/remote.xml"
|
||||
## <yandex>
|
||||
## <remote_servers>
|
||||
## <my-own-cluster>
|
||||
## <shard>
|
||||
## <replica><host>clickhouse-ru-1.local</host><port>9000</port></replica>
|
||||
## <replica><host>clickhouse-ru-2.local</host><port>9000</port></replica>
|
||||
## </shard>
|
||||
## <shard>
|
||||
## <replica><host>clickhouse-eu-1.local</host><port>9000</port></replica>
|
||||
## <replica><host>clickhouse-eu-2.local</host><port>9000</port></replica>
|
||||
## </shard>
|
||||
## </my-onw-cluster>
|
||||
## </remote_servers>
|
||||
##
|
||||
## </yandex>
|
||||
##
|
||||
## example: cluster_include = ["my-own-cluster"]
|
||||
# cluster_include = []
|
||||
|
||||
## Filter cluster names in "system.clusters" when "auto_discovery" is "true"
|
||||
## when this filter present then "WHERE cluster NOT IN (...)" filter will apply
|
||||
## example: cluster_exclude = ["my-internal-not-discovered-cluster"]
|
||||
# cluster_exclude = []
|
||||
|
||||
## Optional TLS Config
|
||||
# tls_ca = "/etc/telegraf/ca.pem"
|
||||
# tls_cert = "/etc/telegraf/cert.pem"
|
||||
# tls_key = "/etc/telegraf/key.pem"
|
||||
## Use TLS but skip chain & host verification
|
||||
# insecure_skip_verify = false
|
||||
```
|
||||
|
||||
### Metrics:
|
||||
- clickhouse_events
|
||||
- tags:
|
||||
- hostname (ClickHouse server hostname)
|
||||
- cluster (Name of the cluster [optional])
|
||||
- shard_num (Shard number in the cluster [optional])
|
||||
- fields:
|
||||
- all rows from system.events, all metrics is COUNTER type, look https://clickhouse.tech/docs/en/operations/system_tables/#system_tables-events
|
||||
|
||||
- clickhouse_metrics
|
||||
- tags:
|
||||
- hostname (ClickHouse server hostname)
|
||||
- cluster (Name of the cluster [optional])
|
||||
- shard_num (Shard number in the cluster [optional])
|
||||
- fields:
|
||||
- all rows from system.metrics, all metrics is GAUGE type, look https://clickhouse.tech/docs/en/operations/system_tables/#system_tables-metrics
|
||||
|
||||
- clickhouse_asynchronous_metrics
|
||||
- tags:
|
||||
- hostname (ClickHouse server hostname)
|
||||
- cluster (Name of the cluster [optional])
|
||||
- shard_num (Shard number in the cluster [optional])
|
||||
- fields:
|
||||
- all rows from system.asynchronous_metrics, all metrics is GAUGE type, look https://clickhouse.tech/docs/en/operations/system_tables/#system_tables-asynchronous_metrics
|
||||
|
||||
- clickhouse_tables
|
||||
- tags:
|
||||
- hostname (ClickHouse server hostname)
|
||||
- table
|
||||
- database
|
||||
- cluster (Name of the cluster [optional])
|
||||
- shard_num (Shard number in the cluster [optional])
|
||||
- fields:
|
||||
- bytes
|
||||
- parts
|
||||
- rows
|
||||
|
||||
### Example Output
|
||||
|
||||
```
|
||||
clickhouse_events,cluster=test_cluster_two_shards_localhost,host=kshvakov,hostname=localhost,shard_num=1 read_compressed_bytes=212i,arena_alloc_chunks=35i,function_execute=85i,merge_tree_data_writer_rows=3i,rw_lock_acquired_read_locks=421i,file_open=46i,io_buffer_alloc_bytes=86451985i,inserted_bytes=196i,regexp_created=3i,real_time_microseconds=116832i,query=23i,network_receive_elapsed_microseconds=268i,merge_tree_data_writer_compressed_bytes=1080i,arena_alloc_bytes=212992i,disk_write_elapsed_microseconds=556i,inserted_rows=3i,compressed_read_buffer_bytes=81i,read_buffer_from_file_descriptor_read_bytes=148i,write_buffer_from_file_descriptor_write=47i,merge_tree_data_writer_blocks=3i,soft_page_faults=896i,hard_page_faults=7i,select_query=21i,merge_tree_data_writer_uncompressed_bytes=196i,merge_tree_data_writer_blocks_already_sorted=3i,user_time_microseconds=40196i,compressed_read_buffer_blocks=5i,write_buffer_from_file_descriptor_write_bytes=3246i,io_buffer_allocs=296i,created_write_buffer_ordinary=12i,disk_read_elapsed_microseconds=59347044i,network_send_elapsed_microseconds=1538i,context_lock=1040i,insert_query=1i,system_time_microseconds=14582i,read_buffer_from_file_descriptor_read=3i 1569421000000000000
|
||||
clickhouse_asynchronous_metrics,cluster=test_cluster_two_shards_localhost,host=kshvakov,hostname=localhost,shard_num=1 jemalloc.metadata_thp=0i,replicas_max_relative_delay=0i,jemalloc.mapped=1803177984i,jemalloc.allocated=1724839256i,jemalloc.background_thread.run_interval=0i,jemalloc.background_thread.num_threads=0i,uncompressed_cache_cells=0i,replicas_max_absolute_delay=0i,mark_cache_bytes=0i,compiled_expression_cache_count=0i,replicas_sum_queue_size=0i,number_of_tables=35i,replicas_max_merges_in_queue=0i,replicas_max_inserts_in_queue=0i,replicas_sum_merges_in_queue=0i,replicas_max_queue_size=0i,mark_cache_files=0i,jemalloc.background_thread.num_runs=0i,jemalloc.active=1726210048i,uptime=158i,jemalloc.retained=380481536i,replicas_sum_inserts_in_queue=0i,uncompressed_cache_bytes=0i,number_of_databases=2i,jemalloc.metadata=9207704i,max_part_count_for_partition=1i,jemalloc.resident=1742442496i 1569421000000000000
|
||||
clickhouse_metrics,cluster=test_cluster_two_shards_localhost,host=kshvakov,hostname=localhost,shard_num=1 replicated_send=0i,write=0i,ephemeral_node=0i,zoo_keeper_request=0i,distributed_files_to_insert=0i,replicated_fetch=0i,background_schedule_pool_task=0i,interserver_connection=0i,leader_replica=0i,delayed_inserts=0i,global_thread_active=41i,merge=0i,readonly_replica=0i,memory_tracking_in_background_schedule_pool=0i,memory_tracking_for_merges=0i,zoo_keeper_session=0i,context_lock_wait=0i,storage_buffer_bytes=0i,background_pool_task=0i,send_external_tables=0i,zoo_keeper_watch=0i,part_mutation=0i,disk_space_reserved_for_merge=0i,distributed_send=0i,version_integer=19014003i,local_thread=0i,replicated_checks=0i,memory_tracking=0i,memory_tracking_in_background_processing_pool=0i,leader_election=0i,revision=54425i,open_file_for_read=0i,open_file_for_write=0i,storage_buffer_rows=0i,rw_lock_waiting_readers=0i,rw_lock_waiting_writers=0i,rw_lock_active_writers=0i,local_thread_active=0i,query_preempted=0i,tcp_connection=1i,http_connection=1i,read=2i,query_thread=0i,dict_cache_requests=0i,rw_lock_active_readers=1i,global_thread=43i,query=1i 1569421000000000000
|
||||
clickhouse_tables,cluster=test_cluster_two_shards_localhost,database=system,host=kshvakov,hostname=localhost,shard_num=1,table=trace_log bytes=754i,parts=1i,rows=1i 1569421000000000000
|
||||
clickhouse_tables,cluster=test_cluster_two_shards_localhost,database=default,host=kshvakov,hostname=localhost,shard_num=1,table=example bytes=326i,parts=2i,rows=2i 1569421000000000000
|
||||
```
|
||||
|
|
@ -0,0 +1,390 @@
|
|||
package clickhouse
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"net"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/internal"
|
||||
"github.com/influxdata/telegraf/internal/tls"
|
||||
"github.com/influxdata/telegraf/plugins/inputs"
|
||||
)
|
||||
|
||||
var defaultTimeout = 5 * time.Second
|
||||
|
||||
var sampleConfig = `
|
||||
## Username for authorization on ClickHouse server
|
||||
## example: user = "default""
|
||||
user = "default"
|
||||
|
||||
## Password for authorization on ClickHouse server
|
||||
## example: password = "super_secret"
|
||||
|
||||
## HTTP(s) timeout while getting metrics values
|
||||
## The timeout includes connection time, any redirects, and reading the response body.
|
||||
## example: timeout = 1s
|
||||
# timeout = 5s
|
||||
|
||||
## List of servers for metrics scraping
|
||||
## metrics scrape via HTTP(s) clickhouse interface
|
||||
## https://clickhouse.tech/docs/en/interfaces/http/
|
||||
## example: servers = ["http://127.0.0.1:8123","https://custom-server.mdb.yandexcloud.net"]
|
||||
servers = ["http://127.0.0.1:8123"]
|
||||
|
||||
## If "auto_discovery"" is "true" plugin tries to connect to all servers available in the cluster
|
||||
## with using same "user:password" described in "user" and "password" parameters
|
||||
## and get this server hostname list from "system.clusters" table
|
||||
## see
|
||||
## - https://clickhouse.tech/docs/en/operations/system_tables/#system-clusters
|
||||
## - https://clickhouse.tech/docs/en/operations/server_settings/settings/#server_settings_remote_servers
|
||||
## - https://clickhouse.tech/docs/en/operations/table_engines/distributed/
|
||||
## - https://clickhouse.tech/docs/en/operations/table_engines/replication/#creating-replicated-tables
|
||||
## example: auto_discovery = false
|
||||
# auto_discovery = true
|
||||
|
||||
## Filter cluster names in "system.clusters" when "auto_discovery" is "true"
|
||||
## when this filter present then "WHERE cluster IN (...)" filter will apply
|
||||
## please use only full cluster names here, regexp and glob filters is not allowed
|
||||
## for "/etc/clickhouse-server/config.d/remote.xml"
|
||||
## <yandex>
|
||||
## <remote_servers>
|
||||
## <my-own-cluster>
|
||||
## <shard>
|
||||
## <replica><host>clickhouse-ru-1.local</host><port>9000</port></replica>
|
||||
## <replica><host>clickhouse-ru-2.local</host><port>9000</port></replica>
|
||||
## </shard>
|
||||
## <shard>
|
||||
## <replica><host>clickhouse-eu-1.local</host><port>9000</port></replica>
|
||||
## <replica><host>clickhouse-eu-2.local</host><port>9000</port></replica>
|
||||
## </shard>
|
||||
## </my-onw-cluster>
|
||||
## </remote_servers>
|
||||
##
|
||||
## </yandex>
|
||||
##
|
||||
## example: cluster_include = ["my-own-cluster"]
|
||||
# cluster_include = []
|
||||
|
||||
## Filter cluster names in "system.clusters" when "auto_discovery" is "true"
|
||||
## when this filter present then "WHERE cluster NOT IN (...)" filter will apply
|
||||
## example: cluster_exclude = ["my-internal-not-discovered-cluster"]
|
||||
# cluster_exclude = []
|
||||
|
||||
## Optional TLS Config
|
||||
# tls_ca = "/etc/telegraf/ca.pem"
|
||||
# tls_cert = "/etc/telegraf/cert.pem"
|
||||
# tls_key = "/etc/telegraf/key.pem"
|
||||
## Use TLS but skip chain & host verification
|
||||
# insecure_skip_verify = false
|
||||
`
|
||||
|
||||
type connect struct {
|
||||
Cluster string `json:"cluster"`
|
||||
ShardNum int `json:"shard_num"`
|
||||
Hostname string `json:"host_name"`
|
||||
url *url.URL
|
||||
}
|
||||
|
||||
func init() {
|
||||
inputs.Add("clickhouse", func() telegraf.Input {
|
||||
return &ClickHouse{
|
||||
AutoDiscovery: true,
|
||||
ClientConfig: tls.ClientConfig{
|
||||
InsecureSkipVerify: false,
|
||||
},
|
||||
Timeout: internal.Duration{Duration: defaultTimeout},
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// ClickHouse Telegraf Input Plugin
|
||||
type ClickHouse struct {
|
||||
User string `toml:"user"`
|
||||
Password string `toml:"password"`
|
||||
Servers []string `toml:"servers"`
|
||||
AutoDiscovery bool `toml:"auto_discovery"`
|
||||
ClusterInclude []string `toml:"cluster_include"`
|
||||
ClusterExclude []string `toml:"cluster_exclude"`
|
||||
Timeout internal.Duration `toml:"timeout"`
|
||||
client http.Client
|
||||
tls.ClientConfig
|
||||
}
|
||||
|
||||
// SampleConfig returns the sample config
|
||||
func (*ClickHouse) SampleConfig() string {
|
||||
return sampleConfig
|
||||
}
|
||||
|
||||
// Description return plugin description
|
||||
func (*ClickHouse) Description() string {
|
||||
return "Read metrics from one or many ClickHouse servers"
|
||||
}
|
||||
|
||||
// Start ClickHouse input service
|
||||
func (ch *ClickHouse) Start(telegraf.Accumulator) error {
|
||||
timeout := defaultTimeout
|
||||
if ch.Timeout.Duration != 0 {
|
||||
timeout = ch.Timeout.Duration
|
||||
}
|
||||
tlsCfg, err := ch.ClientConfig.TLSConfig()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
ch.client = http.Client{
|
||||
Timeout: timeout,
|
||||
Transport: &http.Transport{
|
||||
TLSClientConfig: tlsCfg,
|
||||
Proxy: http.ProxyFromEnvironment,
|
||||
},
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Gather collect data from ClickHouse server
|
||||
func (ch *ClickHouse) Gather(acc telegraf.Accumulator) (err error) {
|
||||
var (
|
||||
connects []connect
|
||||
exists = func(host string) bool {
|
||||
for _, c := range connects {
|
||||
if c.Hostname == host {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
)
|
||||
|
||||
for _, server := range ch.Servers {
|
||||
u, err := url.Parse(server)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
switch {
|
||||
case ch.AutoDiscovery:
|
||||
var conns []connect
|
||||
if err := ch.execQuery(u, "SELECT cluster, shard_num, host_name FROM system.clusters "+ch.clusterIncludeExcludeFilter(), &conns); err != nil {
|
||||
acc.AddError(err)
|
||||
continue
|
||||
}
|
||||
for _, c := range conns {
|
||||
if !exists(c.Hostname) {
|
||||
c.url = &url.URL{
|
||||
Scheme: u.Scheme,
|
||||
Host: net.JoinHostPort(c.Hostname, u.Port()),
|
||||
}
|
||||
connects = append(connects, c)
|
||||
}
|
||||
}
|
||||
default:
|
||||
connects = append(connects, connect{
|
||||
url: u,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
for _, conn := range connects {
|
||||
if err := ch.tables(acc, &conn); err != nil {
|
||||
acc.AddError(err)
|
||||
}
|
||||
for metric := range commonMetrics {
|
||||
if err := ch.commonMetrics(acc, &conn, metric); err != nil {
|
||||
acc.AddError(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (ch *ClickHouse) clusterIncludeExcludeFilter() string {
|
||||
if len(ch.ClusterInclude) == 0 && len(ch.ClusterExclude) == 0 {
|
||||
return ""
|
||||
}
|
||||
var (
|
||||
escape = func(in string) string {
|
||||
return "'" + strings.NewReplacer(`\`, `\\`, `'`, `\'`).Replace(in) + "'"
|
||||
}
|
||||
makeFilter = func(expr string, args []string) string {
|
||||
in := make([]string, 0, len(args))
|
||||
for _, v := range args {
|
||||
in = append(in, escape(v))
|
||||
}
|
||||
return fmt.Sprintf("cluster %s (%s)", expr, strings.Join(in, ", "))
|
||||
}
|
||||
includeFilter, excludeFilter string
|
||||
)
|
||||
|
||||
if len(ch.ClusterInclude) != 0 {
|
||||
includeFilter = makeFilter("IN", ch.ClusterInclude)
|
||||
}
|
||||
if len(ch.ClusterExclude) != 0 {
|
||||
excludeFilter = makeFilter("NOT IN", ch.ClusterExclude)
|
||||
}
|
||||
if includeFilter != "" && excludeFilter != "" {
|
||||
return "WHERE " + includeFilter + " OR " + excludeFilter
|
||||
}
|
||||
if includeFilter == "" && excludeFilter != "" {
|
||||
return "WHERE " + excludeFilter
|
||||
}
|
||||
if includeFilter != "" && excludeFilter == "" {
|
||||
return "WHERE " + includeFilter
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (ch *ClickHouse) commonMetrics(acc telegraf.Accumulator, conn *connect, metric string) error {
|
||||
var result []struct {
|
||||
Metric string `json:"metric"`
|
||||
Value chUInt64 `json:"value"`
|
||||
}
|
||||
if err := ch.execQuery(conn.url, commonMetrics[metric], &result); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tags := map[string]string{
|
||||
"source": conn.Hostname,
|
||||
}
|
||||
if len(conn.Cluster) != 0 {
|
||||
tags["cluster"] = conn.Cluster
|
||||
}
|
||||
if conn.ShardNum != 0 {
|
||||
tags["shard_num"] = strconv.Itoa(conn.ShardNum)
|
||||
}
|
||||
|
||||
fields := make(map[string]interface{})
|
||||
for _, r := range result {
|
||||
fields[internal.SnakeCase(r.Metric)] = uint64(r.Value)
|
||||
}
|
||||
|
||||
acc.AddFields("clickhouse_"+metric, fields, tags)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (ch *ClickHouse) tables(acc telegraf.Accumulator, conn *connect) error {
|
||||
var parts []struct {
|
||||
Database string `json:"database"`
|
||||
Table string `json:"table"`
|
||||
Bytes chUInt64 `json:"bytes"`
|
||||
Parts chUInt64 `json:"parts"`
|
||||
Rows chUInt64 `json:"rows"`
|
||||
}
|
||||
|
||||
if err := ch.execQuery(conn.url, systemParts, &parts); err != nil {
|
||||
return err
|
||||
}
|
||||
tags := map[string]string{
|
||||
"source": conn.Hostname,
|
||||
}
|
||||
if len(conn.Cluster) != 0 {
|
||||
tags["cluster"] = conn.Cluster
|
||||
}
|
||||
if conn.ShardNum != 0 {
|
||||
tags["shard_num"] = strconv.Itoa(conn.ShardNum)
|
||||
}
|
||||
for _, part := range parts {
|
||||
tags["table"] = part.Table
|
||||
tags["database"] = part.Database
|
||||
acc.AddFields("clickhouse_tables",
|
||||
map[string]interface{}{
|
||||
"bytes": uint64(part.Bytes),
|
||||
"parts": uint64(part.Parts),
|
||||
"rows": uint64(part.Rows),
|
||||
},
|
||||
tags,
|
||||
)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type clickhouseError struct {
|
||||
StatusCode int
|
||||
body []byte
|
||||
}
|
||||
|
||||
func (e *clickhouseError) Error() string {
|
||||
return fmt.Sprintf("received error code %d: %s", e.StatusCode, e.body)
|
||||
}
|
||||
|
||||
func (ch *ClickHouse) execQuery(url *url.URL, query string, i interface{}) error {
|
||||
q := url.Query()
|
||||
q.Set("query", query+" FORMAT JSON")
|
||||
url.RawQuery = q.Encode()
|
||||
req, _ := http.NewRequest("GET", url.String(), nil)
|
||||
if ch.User != "" {
|
||||
req.Header.Add("X-ClickHouse-User", ch.User)
|
||||
}
|
||||
if ch.Password != "" {
|
||||
req.Header.Add("X-ClickHouse-Key", ch.Password)
|
||||
}
|
||||
resp, err := ch.client.Do(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode >= 300 {
|
||||
body, _ := ioutil.ReadAll(io.LimitReader(resp.Body, 200))
|
||||
return &clickhouseError{
|
||||
StatusCode: resp.StatusCode,
|
||||
body: body,
|
||||
}
|
||||
}
|
||||
var response struct {
|
||||
Data json.RawMessage
|
||||
}
|
||||
if err := json.NewDecoder(resp.Body).Decode(&response); err != nil {
|
||||
return err
|
||||
}
|
||||
return json.Unmarshal(response.Data, i)
|
||||
}
|
||||
|
||||
// see https://clickhouse.yandex/docs/en/operations/settings/settings/#session_settings-output_format_json_quote_64bit_integers
|
||||
type chUInt64 uint64
|
||||
|
||||
func (i *chUInt64) UnmarshalJSON(b []byte) error {
|
||||
b = bytes.TrimPrefix(b, []byte(`"`))
|
||||
b = bytes.TrimSuffix(b, []byte(`"`))
|
||||
v, err := strconv.ParseUint(string(b), 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*i = chUInt64(v)
|
||||
return nil
|
||||
}
|
||||
|
||||
const (
|
||||
systemEventsSQL = "SELECT event AS metric, CAST(value AS UInt64) AS value FROM system.events"
|
||||
systemMetricsSQL = "SELECT metric, CAST(value AS UInt64) AS value FROM system.metrics"
|
||||
systemAsyncMetricsSQL = "SELECT metric, CAST(value AS UInt64) AS value FROM system.asynchronous_metrics"
|
||||
systemParts = `
|
||||
SELECT
|
||||
database,
|
||||
table,
|
||||
SUM(bytes) AS bytes,
|
||||
COUNT(*) AS parts,
|
||||
SUM(rows) AS rows
|
||||
FROM system.parts
|
||||
WHERE active = 1
|
||||
GROUP BY
|
||||
database, table
|
||||
ORDER BY
|
||||
database, table
|
||||
`
|
||||
)
|
||||
|
||||
var commonMetrics = map[string]string{
|
||||
"events": systemEventsSQL,
|
||||
"metrics": systemMetricsSQL,
|
||||
"asynchronous_metrics": systemAsyncMetricsSQL,
|
||||
}
|
||||
|
||||
var _ telegraf.ServiceInput = &ClickHouse{}
|
||||
|
|
@ -0,0 +1,6 @@
|
|||
// +build !go1.12
|
||||
|
||||
package clickhouse
|
||||
|
||||
// Stop ClickHouse input service
|
||||
func (ch *ClickHouse) Stop() {}
|
||||
|
|
@ -0,0 +1,8 @@
|
|||
// +build go1.12
|
||||
|
||||
package clickhouse
|
||||
|
||||
// Stop ClickHouse input service
|
||||
func (ch *ClickHouse) Stop() {
|
||||
ch.client.CloseIdleConnections()
|
||||
}
|
||||
|
|
@ -0,0 +1,161 @@
|
|||
package clickhouse
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/influxdata/telegraf/testutil"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestClusterIncludeExcludeFilter(t *testing.T) {
|
||||
ch := ClickHouse{}
|
||||
if assert.Equal(t, "", ch.clusterIncludeExcludeFilter()) {
|
||||
ch.ClusterExclude = []string{"test_cluster"}
|
||||
assert.Equal(t, "WHERE cluster NOT IN ('test_cluster')", ch.clusterIncludeExcludeFilter())
|
||||
|
||||
ch.ClusterExclude = []string{"test_cluster"}
|
||||
ch.ClusterInclude = []string{"cluster"}
|
||||
assert.Equal(t, "WHERE cluster IN ('cluster') OR cluster NOT IN ('test_cluster')", ch.clusterIncludeExcludeFilter())
|
||||
|
||||
ch.ClusterExclude = []string{}
|
||||
ch.ClusterInclude = []string{"cluster1", "cluster2"}
|
||||
assert.Equal(t, "WHERE cluster IN ('cluster1', 'cluster2')", ch.clusterIncludeExcludeFilter())
|
||||
|
||||
ch.ClusterExclude = []string{"cluster1", "cluster2"}
|
||||
ch.ClusterInclude = []string{}
|
||||
assert.Equal(t, "WHERE cluster NOT IN ('cluster1', 'cluster2')", ch.clusterIncludeExcludeFilter())
|
||||
}
|
||||
}
|
||||
|
||||
func TestChInt64(t *testing.T) {
|
||||
assets := map[string]uint64{
|
||||
`"1"`: 1,
|
||||
"1": 1,
|
||||
"42": 42,
|
||||
`"42"`: 42,
|
||||
"18446743937525109187": 18446743937525109187,
|
||||
}
|
||||
for src, expected := range assets {
|
||||
var v chUInt64
|
||||
if err := v.UnmarshalJSON([]byte(src)); assert.NoError(t, err) {
|
||||
assert.Equal(t, expected, uint64(v))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGather(t *testing.T) {
|
||||
var (
|
||||
ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
type result struct {
|
||||
Data interface{} `json:"data"`
|
||||
}
|
||||
enc := json.NewEncoder(w)
|
||||
switch query := r.URL.Query().Get("query"); {
|
||||
case strings.Contains(query, "system.parts"):
|
||||
enc.Encode(result{
|
||||
Data: []struct {
|
||||
Database string `json:"database"`
|
||||
Table string `json:"table"`
|
||||
Bytes chUInt64 `json:"bytes"`
|
||||
Parts chUInt64 `json:"parts"`
|
||||
Rows chUInt64 `json:"rows"`
|
||||
}{
|
||||
{
|
||||
Database: "test_database",
|
||||
Table: "test_table",
|
||||
Bytes: 1,
|
||||
Parts: 10,
|
||||
Rows: 100,
|
||||
},
|
||||
},
|
||||
})
|
||||
case strings.Contains(query, "system.events"):
|
||||
enc.Encode(result{
|
||||
Data: []struct {
|
||||
Metric string `json:"metric"`
|
||||
Value chUInt64 `json:"value"`
|
||||
}{
|
||||
{
|
||||
Metric: "TestSystemEvent",
|
||||
Value: 1000,
|
||||
},
|
||||
{
|
||||
Metric: "TestSystemEvent2",
|
||||
Value: 2000,
|
||||
},
|
||||
},
|
||||
})
|
||||
case strings.Contains(query, "system.metrics"):
|
||||
enc.Encode(result{
|
||||
Data: []struct {
|
||||
Metric string `json:"metric"`
|
||||
Value chUInt64 `json:"value"`
|
||||
}{
|
||||
{
|
||||
Metric: "TestSystemMetric",
|
||||
Value: 1000,
|
||||
},
|
||||
{
|
||||
Metric: "TestSystemMetric2",
|
||||
Value: 2000,
|
||||
},
|
||||
},
|
||||
})
|
||||
case strings.Contains(query, "system.asynchronous_metrics"):
|
||||
enc.Encode(result{
|
||||
Data: []struct {
|
||||
Metric string `json:"metric"`
|
||||
Value chUInt64 `json:"value"`
|
||||
}{
|
||||
{
|
||||
Metric: "TestSystemAsynchronousMetric",
|
||||
Value: 1000,
|
||||
},
|
||||
{
|
||||
Metric: "TestSystemAsynchronousMetric2",
|
||||
Value: 2000,
|
||||
},
|
||||
},
|
||||
})
|
||||
}
|
||||
}))
|
||||
ch = &ClickHouse{
|
||||
Servers: []string{
|
||||
ts.URL,
|
||||
},
|
||||
}
|
||||
acc = &testutil.Accumulator{}
|
||||
)
|
||||
defer ts.Close()
|
||||
ch.Gather(acc)
|
||||
|
||||
acc.AssertContainsFields(t, "clickhouse_tables",
|
||||
map[string]interface{}{
|
||||
"bytes": uint64(1),
|
||||
"parts": uint64(10),
|
||||
"rows": uint64(100),
|
||||
},
|
||||
)
|
||||
acc.AssertContainsFields(t, "clickhouse_events",
|
||||
map[string]interface{}{
|
||||
"test_system_event": uint64(1000),
|
||||
"test_system_event2": uint64(2000),
|
||||
},
|
||||
)
|
||||
acc.AssertContainsFields(t, "clickhouse_metrics",
|
||||
map[string]interface{}{
|
||||
"test_system_metric": uint64(1000),
|
||||
"test_system_metric2": uint64(2000),
|
||||
},
|
||||
)
|
||||
acc.AssertContainsFields(t, "clickhouse_asynchronous_metrics",
|
||||
map[string]interface{}{
|
||||
"test_system_asynchronous_metric": uint64(1000),
|
||||
"test_system_asynchronous_metric2": uint64(2000),
|
||||
},
|
||||
)
|
||||
}
|
||||
|
|
@ -0,0 +1,13 @@
|
|||
-----BEGIN DH PARAMETERS-----
|
||||
MIICCAKCAgEAoo1x7wI5K57P1/AkHUmVWzKNfy46b/ni/QtClomTB78Ks1FP8dzs
|
||||
CQBW/pfL8yidxTialNhMRCZO1J+uPjTvd8dG8SFZzVylkF41LBNrUD+MLyh/b6Nr
|
||||
8uWf3tqYCtsiqsQsnq/oU7C29wn6UjhPPVbRRDPGyJUFOgp0ebPR0L2gOc5HhXSF
|
||||
Tt0fuWnvgZJBKGvyodby3p2CSheu8K6ZteVc8ZgHuanhCQA30nVN+yNQzyozlB2H
|
||||
B9jxTDPJy8+/4Mui3iiNyXg6FaiI9lWdH7xgKoZlHi8BWlLz5Se9JVNYg0dPrMTz
|
||||
K0itQyyTKUlK73x+1uPm6q1AJwz08EZiCXNbk58/Sf+pdwDmAO2QSRrERC73vnvc
|
||||
B1+4+Kf7RS7oYpAHknKm/MFnkCJLVIq1b6kikYcIgVCYe+Z1UytSmG1QfwdgL8QQ
|
||||
TVYVHBg4w07+s3/IJ1ekvNhdxpkmmevYt7GjohWu8vKkip4se+reNdo+sqLsgFKf
|
||||
1IuDMD36zn9FVukvs7e3BwZCTkdosGHvHGjA7zm2DwPPO16hCvJ4mE6ULLpp2NEw
|
||||
EBYWm3Tv6M/xtrF5Afyh0gAh7eL767/qsarbx6jlqs+dnh3LptqsE3WerWK54+0B
|
||||
3Hr5CVfgYbeXuW2HeFb+fS6CNUWmiAsq1XRiz5p16hpeMGYN/qyF1IsCAQI=
|
||||
-----END DH PARAMETERS-----
|
||||
|
|
@ -0,0 +1,16 @@
|
|||
version: '3'
|
||||
|
||||
services:
|
||||
clickhouse:
|
||||
image: yandex/clickhouse-server:latest
|
||||
volumes:
|
||||
- ./dhparam.pem:/etc/clickhouse-server/dhparam.pem
|
||||
- ./tls_settings.xml:/etc/clickhouse-server/config.d/00-tls_settings.xml
|
||||
- ../../../../testutil/pki/serverkey.pem:/etc/clickhouse-server/server.key
|
||||
- ../../../../testutil/pki/servercert.pem:/etc/clickhouse-server/server.crt
|
||||
restart: always
|
||||
ports:
|
||||
- 8123:8123
|
||||
- 8443:8443
|
||||
- 9000:9000
|
||||
- 9009:9009
|
||||
|
|
@ -0,0 +1,12 @@
|
|||
### ClickHouse input plugin
|
||||
|
||||
[[inputs.clickhouse]]
|
||||
timeout = 2
|
||||
user = "default"
|
||||
servers = ["http://127.0.0.1:8123"]
|
||||
auto_discovery = true
|
||||
cluster_include = []
|
||||
cluster_exclude = ["test_shard_localhost"]
|
||||
|
||||
[[outputs.file]]
|
||||
files = ["stdout"]
|
||||
|
|
@ -0,0 +1,16 @@
|
|||
### ClickHouse input plugin
|
||||
|
||||
[[inputs.clickhouse]]
|
||||
timeout = 2
|
||||
user = "default"
|
||||
servers = ["https://127.0.0.1:8443"]
|
||||
auto_discovery = true
|
||||
cluster_include = []
|
||||
cluster_exclude = ["test_shard_localhost"]
|
||||
insecure_skip_verify = false
|
||||
tls_cert = "./testutil/pki/clientcert.pem"
|
||||
tls_key = "./testutil/pki/clientkey.pem"
|
||||
tls_ca = "./testutil/pki/cacert.pem"
|
||||
|
||||
[[outputs.file]]
|
||||
files = ["stdout"]
|
||||
|
|
@ -0,0 +1,4 @@
|
|||
<yandex>
|
||||
<https_port>8443</https_port>
|
||||
<tcp_port_secure>9440</tcp_port_secure>
|
||||
</yandex>
|
||||
Loading…
Reference in New Issue