mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-29 11:02:08 +00:00
109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package clickhouse
|
|
|
|
import (
|
|
"github.com/ClickHouse/clickhouse-diagnostics/internal/collectors"
|
|
"github.com/ClickHouse/clickhouse-diagnostics/internal/platform"
|
|
"github.com/ClickHouse/clickhouse-diagnostics/internal/platform/config"
|
|
"github.com/ClickHouse/clickhouse-diagnostics/internal/platform/data"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type DBLogTable struct {
|
|
orderBy data.OrderBy
|
|
excludeColumns []string
|
|
}
|
|
|
|
var DbLogTables = map[string]DBLogTable{
|
|
"query_log": {
|
|
orderBy: data.OrderBy{
|
|
Column: "event_time_microseconds",
|
|
Order: data.Asc,
|
|
},
|
|
excludeColumns: []string{},
|
|
},
|
|
"query_thread_log": {
|
|
orderBy: data.OrderBy{
|
|
Column: "event_time_microseconds",
|
|
Order: data.Asc,
|
|
},
|
|
excludeColumns: []string{},
|
|
},
|
|
"text_log": {
|
|
orderBy: data.OrderBy{
|
|
Column: "event_time_microseconds",
|
|
Order: data.Asc,
|
|
},
|
|
excludeColumns: []string{},
|
|
},
|
|
}
|
|
|
|
// This collector collects db logs
|
|
|
|
type DBLogsCollector struct {
|
|
resourceManager *platform.ResourceManager
|
|
}
|
|
|
|
func NewDBLogsCollector(m *platform.ResourceManager) *DBLogsCollector {
|
|
return &DBLogsCollector{
|
|
resourceManager: m,
|
|
}
|
|
}
|
|
|
|
func (dc *DBLogsCollector) Collect(conf config.Configuration) (*data.DiagnosticBundle, error) {
|
|
conf, err := conf.ValidateConfig(dc.Configuration())
|
|
if err != nil {
|
|
return &data.DiagnosticBundle{}, err
|
|
}
|
|
rowLimit, err := config.ReadIntValue(conf, "row_limit")
|
|
if err != nil {
|
|
return &data.DiagnosticBundle{}, err
|
|
}
|
|
|
|
frames := make(map[string]data.Frame)
|
|
var frameErrors []error
|
|
for logTable, tableConfig := range DbLogTables {
|
|
frame, err := dc.resourceManager.DbClient.ReadTable("system", logTable, tableConfig.excludeColumns, tableConfig.orderBy, rowLimit)
|
|
if err != nil {
|
|
frameErrors = append(frameErrors, errors.Wrapf(err, "Unable to collect %s", logTable))
|
|
} else {
|
|
frames[logTable] = frame
|
|
}
|
|
}
|
|
|
|
fErrors := data.FrameErrors{
|
|
Errors: frameErrors,
|
|
}
|
|
return &data.DiagnosticBundle{
|
|
Frames: frames,
|
|
Errors: fErrors,
|
|
}, nil
|
|
}
|
|
|
|
func (dc *DBLogsCollector) Configuration() config.Configuration {
|
|
return config.Configuration{
|
|
Params: []config.ConfigParam{
|
|
config.IntParam{
|
|
Value: 100000,
|
|
Param: config.NewParam("row_limit", "Maximum number of log rows to collect. Negative values mean unlimited", false),
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (dc *DBLogsCollector) IsDefault() bool {
|
|
return true
|
|
}
|
|
|
|
func (dc DBLogsCollector) Description() string {
|
|
return "Collects the ClickHouse logs directly from the database."
|
|
}
|
|
|
|
// here we register the collector for use
|
|
func init() {
|
|
collectors.Register("db_logs", func() (collectors.Collector, error) {
|
|
return &DBLogsCollector{
|
|
resourceManager: platform.GetResourceManager(),
|
|
}, nil
|
|
})
|
|
}
|