chore: move context.Context to first argument in methods per convention
parent
092c7a9976
commit
a8b2129df5
|
@ -61,10 +61,10 @@ type StatementExecutor struct {
|
|||
}
|
||||
|
||||
// ExecuteStatement executes the given statement with the given execution context.
|
||||
func (e *StatementExecutor) ExecuteStatement(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
func (e *StatementExecutor) ExecuteStatement(ctx *query.ExecutionContext, stmt influxql.Statement) error {
|
||||
// Select statements are handled separately so that they can be streamed.
|
||||
if stmt, ok := stmt.(*influxql.SelectStatement); ok {
|
||||
return e.executeSelectStatement(stmt, ctx)
|
||||
return e.executeSelectStatement(ctx, stmt)
|
||||
}
|
||||
|
||||
var rows models.Rows
|
||||
|
@ -145,9 +145,9 @@ func (e *StatementExecutor) ExecuteStatement(stmt influxql.Statement, ctx *query
|
|||
err = e.executeDropUserStatement(stmt)
|
||||
case *influxql.ExplainStatement:
|
||||
if stmt.Analyze {
|
||||
rows, err = e.executeExplainAnalyzeStatement(stmt, ctx)
|
||||
rows, err = e.executeExplainAnalyzeStatement(ctx, stmt)
|
||||
} else {
|
||||
rows, err = e.executeExplainStatement(stmt, ctx)
|
||||
rows, err = e.executeExplainStatement(ctx, stmt)
|
||||
}
|
||||
case *influxql.GrantStatement:
|
||||
if ctx.ReadOnly {
|
||||
|
@ -172,19 +172,19 @@ func (e *StatementExecutor) ExecuteStatement(stmt influxql.Statement, ctx *query
|
|||
case *influxql.ShowContinuousQueriesStatement:
|
||||
rows, err = e.executeShowContinuousQueriesStatement(stmt)
|
||||
case *influxql.ShowDatabasesStatement:
|
||||
rows, err = e.executeShowDatabasesStatement(stmt, ctx)
|
||||
rows, err = e.executeShowDatabasesStatement(ctx, stmt)
|
||||
case *influxql.ShowDiagnosticsStatement:
|
||||
rows, err = e.executeShowDiagnosticsStatement(stmt)
|
||||
case *influxql.ShowGrantsForUserStatement:
|
||||
rows, err = e.executeShowGrantsForUserStatement(stmt)
|
||||
case *influxql.ShowMeasurementsStatement:
|
||||
return e.executeShowMeasurementsStatement(stmt, ctx)
|
||||
return e.executeShowMeasurementsStatement(ctx, stmt)
|
||||
case *influxql.ShowMeasurementCardinalityStatement:
|
||||
rows, err = e.executeShowMeasurementCardinalityStatement(stmt, ctx)
|
||||
rows, err = e.executeShowMeasurementCardinalityStatement(ctx, stmt)
|
||||
case *influxql.ShowRetentionPoliciesStatement:
|
||||
rows, err = e.executeShowRetentionPoliciesStatement(stmt)
|
||||
case *influxql.ShowSeriesCardinalityStatement:
|
||||
rows, err = e.executeShowSeriesCardinalityStatement(stmt, ctx)
|
||||
rows, err = e.executeShowSeriesCardinalityStatement(ctx, stmt)
|
||||
case *influxql.ShowShardsStatement:
|
||||
rows, err = e.executeShowShardsStatement(stmt)
|
||||
case *influxql.ShowShardGroupsStatement:
|
||||
|
@ -194,9 +194,9 @@ func (e *StatementExecutor) ExecuteStatement(stmt influxql.Statement, ctx *query
|
|||
case *influxql.ShowSubscriptionsStatement:
|
||||
rows, err = e.executeShowSubscriptionsStatement(stmt)
|
||||
case *influxql.ShowTagKeysStatement:
|
||||
return e.executeShowTagKeys(stmt, ctx)
|
||||
return e.executeShowTagKeys(ctx, stmt)
|
||||
case *influxql.ShowTagValuesStatement:
|
||||
return e.executeShowTagValues(stmt, ctx)
|
||||
return e.executeShowTagValues(ctx, stmt)
|
||||
case *influxql.ShowUsersStatement:
|
||||
rows, err = e.executeShowUsersStatement(stmt)
|
||||
case *influxql.SetPasswordUserStatement:
|
||||
|
@ -206,7 +206,7 @@ func (e *StatementExecutor) ExecuteStatement(stmt influxql.Statement, ctx *query
|
|||
err = e.executeSetPasswordUserStatement(stmt)
|
||||
case *influxql.ShowQueriesStatement, *influxql.KillQueryStatement:
|
||||
// Send query related statements to the task manager.
|
||||
return e.TaskManager.ExecuteStatement(stmt, ctx)
|
||||
return e.TaskManager.ExecuteStatement(ctx, stmt)
|
||||
default:
|
||||
return query.ErrInvalidQuery
|
||||
}
|
||||
|
@ -411,7 +411,7 @@ func (e *StatementExecutor) executeDropUserStatement(q *influxql.DropUserStateme
|
|||
return e.MetaClient.DropUser(q.Name)
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeExplainStatement(q *influxql.ExplainStatement, ctx *query.ExecutionContext) (models.Rows, error) {
|
||||
func (e *StatementExecutor) executeExplainStatement(ctx *query.ExecutionContext, q *influxql.ExplainStatement) (models.Rows, error) {
|
||||
opt := query.SelectOptions{
|
||||
NodeID: ctx.ExecutionOptions.NodeID,
|
||||
MaxSeriesN: e.MaxSelectSeriesN,
|
||||
|
@ -442,7 +442,7 @@ func (e *StatementExecutor) executeExplainStatement(q *influxql.ExplainStatement
|
|||
return models.Rows{row}, nil
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeExplainAnalyzeStatement(q *influxql.ExplainStatement, ectx *query.ExecutionContext) (models.Rows, error) {
|
||||
func (e *StatementExecutor) executeExplainAnalyzeStatement(ectx *query.ExecutionContext, q *influxql.ExplainStatement) (models.Rows, error) {
|
||||
stmt := q.Statement
|
||||
t, span := tracing.NewTrace("select")
|
||||
ctx := tracing.NewContextWithTrace(ectx, t)
|
||||
|
@ -541,7 +541,7 @@ func (e *StatementExecutor) executeSetPasswordUserStatement(q *influxql.SetPassw
|
|||
return e.MetaClient.UpdateUser(q.Name, q.Password)
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeSelectStatement(stmt *influxql.SelectStatement, ctx *query.ExecutionContext) error {
|
||||
func (e *StatementExecutor) executeSelectStatement(ctx *query.ExecutionContext, stmt *influxql.SelectStatement) error {
|
||||
cur, err := e.createIterators(ctx, stmt, ctx.ExecutionOptions)
|
||||
if err != nil {
|
||||
return err
|
||||
|
@ -659,7 +659,7 @@ func (e *StatementExecutor) executeShowContinuousQueriesStatement(stmt *influxql
|
|||
return rows, nil
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeShowDatabasesStatement(q *influxql.ShowDatabasesStatement, ctx *query.ExecutionContext) (models.Rows, error) {
|
||||
func (e *StatementExecutor) executeShowDatabasesStatement(ctx *query.ExecutionContext, q *influxql.ShowDatabasesStatement) (models.Rows, error) {
|
||||
dis := e.MetaClient.Databases()
|
||||
a := ctx.ExecutionOptions.Authorizer
|
||||
|
||||
|
@ -714,7 +714,7 @@ func (e *StatementExecutor) executeShowGrantsForUserStatement(q *influxql.ShowGr
|
|||
return []*models.Row{row}, nil
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeShowMeasurementsStatement(q *influxql.ShowMeasurementsStatement, ctx *query.ExecutionContext) error {
|
||||
func (e *StatementExecutor) executeShowMeasurementsStatement(ctx *query.ExecutionContext, q *influxql.ShowMeasurementsStatement) error {
|
||||
if q.Database == "" {
|
||||
return ErrDatabaseNameRequired
|
||||
}
|
||||
|
@ -758,7 +758,7 @@ func (e *StatementExecutor) executeShowMeasurementsStatement(q *influxql.ShowMea
|
|||
})
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeShowMeasurementCardinalityStatement(stmt *influxql.ShowMeasurementCardinalityStatement, ctx *query.ExecutionContext) (models.Rows, error) {
|
||||
func (e *StatementExecutor) executeShowMeasurementCardinalityStatement(ctx *query.ExecutionContext, stmt *influxql.ShowMeasurementCardinalityStatement) (models.Rows, error) {
|
||||
if stmt.Database == "" {
|
||||
return nil, ErrDatabaseNameRequired
|
||||
}
|
||||
|
@ -829,7 +829,7 @@ func (e *StatementExecutor) executeShowShardsStatement(stmt *influxql.ShowShards
|
|||
return rows, nil
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeShowSeriesCardinalityStatement(stmt *influxql.ShowSeriesCardinalityStatement, ctx *query.ExecutionContext) (models.Rows, error) {
|
||||
func (e *StatementExecutor) executeShowSeriesCardinalityStatement(ctx *query.ExecutionContext, stmt *influxql.ShowSeriesCardinalityStatement) (models.Rows, error) {
|
||||
if stmt.Database == "" {
|
||||
return nil, ErrDatabaseNameRequired
|
||||
}
|
||||
|
@ -929,7 +929,7 @@ func (e *StatementExecutor) executeShowSubscriptionsStatement(stmt *influxql.Sho
|
|||
return rows, nil
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeShowTagKeys(q *influxql.ShowTagKeysStatement, ctx *query.ExecutionContext) error {
|
||||
func (e *StatementExecutor) executeShowTagKeys(ctx *query.ExecutionContext, q *influxql.ShowTagKeysStatement) error {
|
||||
if q.Database == "" {
|
||||
return ErrDatabaseNameRequired
|
||||
}
|
||||
|
@ -1016,7 +1016,7 @@ func (e *StatementExecutor) executeShowTagKeys(q *influxql.ShowTagKeysStatement,
|
|||
return nil
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) executeShowTagValues(q *influxql.ShowTagValuesStatement, ctx *query.ExecutionContext) error {
|
||||
func (e *StatementExecutor) executeShowTagValues(ctx *query.ExecutionContext, q *influxql.ShowTagValuesStatement) error {
|
||||
if q.Database == "" {
|
||||
return ErrDatabaseNameRequired
|
||||
}
|
||||
|
|
|
@ -160,7 +160,7 @@ func NewContextWithIterators(ctx context.Context, itr *Iterators) context.Contex
|
|||
type StatementExecutor interface {
|
||||
// ExecuteStatement executes a statement. Results should be sent to the
|
||||
// results channel in the ExecutionContext.
|
||||
ExecuteStatement(stmt influxql.Statement, ctx *ExecutionContext) error
|
||||
ExecuteStatement(ctx *ExecutionContext, stmt influxql.Statement) error
|
||||
}
|
||||
|
||||
// StatementNormalizer normalizes a statement before it is executed.
|
||||
|
@ -332,7 +332,7 @@ LOOP:
|
|||
}
|
||||
|
||||
// Send any other statements to the underlying statement executor.
|
||||
err = e.StatementExecutor.ExecuteStatement(stmt, ctx)
|
||||
err = e.StatementExecutor.ExecuteStatement(ctx, stmt)
|
||||
if err == ErrQueryInterrupted {
|
||||
// Query was interrupted so retrieve the real interrupt error from
|
||||
// the query task if there is one.
|
||||
|
|
|
@ -17,7 +17,7 @@ type StatementExecutor struct {
|
|||
ExecuteStatementFn func(stmt influxql.Statement, ctx *query.ExecutionContext) error
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) ExecuteStatement(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
func (e *StatementExecutor) ExecuteStatement(ctx *query.ExecutionContext, stmt influxql.Statement) error {
|
||||
return e.ExecuteStatementFn(stmt, ctx)
|
||||
}
|
||||
|
||||
|
@ -57,7 +57,7 @@ func TestQueryExecutor_KillQuery(t *testing.T) {
|
|||
ExecuteStatementFn: func(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
switch stmt.(type) {
|
||||
case *influxql.KillQueryStatement:
|
||||
return e.TaskManager.ExecuteStatement(stmt, ctx)
|
||||
return e.TaskManager.ExecuteStatement(ctx, stmt)
|
||||
}
|
||||
|
||||
qid <- ctx.QueryID
|
||||
|
@ -98,7 +98,7 @@ func TestQueryExecutor_KillQuery_Zombie(t *testing.T) {
|
|||
ExecuteStatementFn: func(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
switch stmt.(type) {
|
||||
case *influxql.KillQueryStatement, *influxql.ShowQueriesStatement:
|
||||
return e.TaskManager.ExecuteStatement(stmt, ctx)
|
||||
return e.TaskManager.ExecuteStatement(ctx, stmt)
|
||||
}
|
||||
|
||||
qid <- ctx.QueryID
|
||||
|
@ -167,7 +167,7 @@ func TestQueryExecutor_KillQuery_CloseTaskManager(t *testing.T) {
|
|||
ExecuteStatementFn: func(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
switch stmt.(type) {
|
||||
case *influxql.KillQueryStatement, *influxql.ShowQueriesStatement:
|
||||
return e.TaskManager.ExecuteStatement(stmt, ctx)
|
||||
return e.TaskManager.ExecuteStatement(ctx, stmt)
|
||||
}
|
||||
|
||||
qid <- ctx.QueryID
|
||||
|
@ -224,7 +224,7 @@ func TestQueryExecutor_KillQuery_AlreadyKilled(t *testing.T) {
|
|||
ExecuteStatementFn: func(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
switch stmt.(type) {
|
||||
case *influxql.KillQueryStatement, *influxql.ShowQueriesStatement:
|
||||
return e.TaskManager.ExecuteStatement(stmt, ctx)
|
||||
return e.TaskManager.ExecuteStatement(ctx, stmt)
|
||||
}
|
||||
|
||||
qid <- ctx.QueryID
|
||||
|
@ -314,7 +314,7 @@ func TestQueryExecutor_ShowQueries(t *testing.T) {
|
|||
ExecuteStatementFn: func(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
switch stmt.(type) {
|
||||
case *influxql.ShowQueriesStatement:
|
||||
return e.TaskManager.ExecuteStatement(stmt, ctx)
|
||||
return e.TaskManager.ExecuteStatement(ctx, stmt)
|
||||
}
|
||||
|
||||
t.Errorf("unexpected statement: %s", stmt)
|
||||
|
|
|
@ -93,7 +93,7 @@ func NewTaskManager() *TaskManager {
|
|||
}
|
||||
|
||||
// ExecuteStatement executes a statement containing one of the task management queries.
|
||||
func (t *TaskManager) ExecuteStatement(stmt influxql.Statement, ctx *ExecutionContext) error {
|
||||
func (t *TaskManager) ExecuteStatement(ctx *ExecutionContext, stmt influxql.Statement) error {
|
||||
switch stmt := stmt.(type) {
|
||||
case *influxql.ShowQueriesStatement:
|
||||
rows, err := t.executeShowQueriesStatement(stmt)
|
||||
|
|
|
@ -825,7 +825,7 @@ type StatementExecutor struct {
|
|||
ExecuteStatementFn func(stmt influxql.Statement, ctx *query.ExecutionContext) error
|
||||
}
|
||||
|
||||
func (e *StatementExecutor) ExecuteStatement(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
func (e *StatementExecutor) ExecuteStatement(ctx *query.ExecutionContext, stmt influxql.Statement) error {
|
||||
return e.ExecuteStatementFn(stmt, ctx)
|
||||
}
|
||||
|
||||
|
|
|
@ -2187,7 +2187,7 @@ type HandlerStatementExecutor struct {
|
|||
ExecuteStatementFn func(stmt influxql.Statement, ctx *query.ExecutionContext) error
|
||||
}
|
||||
|
||||
func (e *HandlerStatementExecutor) ExecuteStatement(stmt influxql.Statement, ctx *query.ExecutionContext) error {
|
||||
func (e *HandlerStatementExecutor) ExecuteStatement(ctx *query.ExecutionContext, stmt influxql.Statement) error {
|
||||
return e.ExecuteStatementFn(stmt, ctx)
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue