70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package monitor
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/influxdb/influxdb/influxql"
|
|
"github.com/influxdb/influxdb/models"
|
|
)
|
|
|
|
// StatementExecutor translates InfluxQL queries to Monitor methods.
|
|
type StatementExecutor struct {
|
|
Monitor interface {
|
|
Statistics(map[string]string) ([]*statistic, error)
|
|
Diagnostics() (map[string]*Diagnostic, error)
|
|
}
|
|
}
|
|
|
|
// ExecuteStatement executes monitor-related query statements.
|
|
func (s *StatementExecutor) ExecuteStatement(stmt influxql.Statement) *influxql.Result {
|
|
switch stmt := stmt.(type) {
|
|
case *influxql.ShowStatsStatement:
|
|
return s.executeShowStatistics(stmt.Module)
|
|
case *influxql.ShowDiagnosticsStatement:
|
|
return s.executeShowDiagnostics()
|
|
default:
|
|
panic(fmt.Sprintf("unsupported statement type: %T", stmt))
|
|
}
|
|
}
|
|
|
|
func (s *StatementExecutor) executeShowStatistics(module string) *influxql.Result {
|
|
stats, err := s.Monitor.Statistics(nil)
|
|
if err != nil {
|
|
return &influxql.Result{Err: err}
|
|
}
|
|
rows := make([]*models.Row, 0)
|
|
|
|
for _, stat := range stats {
|
|
if module != "" && stat.Name != module {
|
|
continue
|
|
}
|
|
row := &models.Row{Name: stat.Name, Tags: stat.Tags}
|
|
|
|
values := make([]interface{}, 0, len(stat.Values))
|
|
for _, k := range stat.valueNames() {
|
|
row.Columns = append(row.Columns, k)
|
|
values = append(values, stat.Values[k])
|
|
}
|
|
row.Values = [][]interface{}{values}
|
|
rows = append(rows, row)
|
|
}
|
|
return &influxql.Result{Series: rows}
|
|
}
|
|
|
|
func (s *StatementExecutor) executeShowDiagnostics() *influxql.Result {
|
|
diags, err := s.Monitor.Diagnostics()
|
|
if err != nil {
|
|
return &influxql.Result{Err: err}
|
|
}
|
|
rows := make([]*models.Row, 0, len(diags))
|
|
|
|
for k, v := range diags {
|
|
row := &models.Row{Name: k}
|
|
|
|
row.Columns = v.Columns
|
|
row.Values = v.Rows
|
|
rows = append(rows, row)
|
|
}
|
|
return &influxql.Result{Series: rows}
|
|
}
|