github.com/influxdata/influxdb/v2@v2.7.6/influxql/mock/proxy_query_service.go (about)

     1  package mock
     2  
     3  import (
     4  	"context"
     5  	"io"
     6  
     7  	"github.com/influxdata/influxdb/v2/influxql"
     8  	"github.com/influxdata/influxdb/v2/kit/check"
     9  )
    10  
    11  var _ influxql.ProxyQueryService = (*ProxyQueryService)(nil)
    12  
    13  // ProxyQueryService mocks the InfluxQL QueryService for testing.
    14  type ProxyQueryService struct {
    15  	QueryF func(ctx context.Context, w io.Writer, req *influxql.QueryRequest) (influxql.Statistics, error)
    16  }
    17  
    18  func (s *ProxyQueryService) Query(ctx context.Context, w io.Writer, req *influxql.QueryRequest) (influxql.Statistics, error) {
    19  	return s.QueryF(ctx, w, req)
    20  }
    21  
    22  func (s *ProxyQueryService) Check(ctx context.Context) check.Response {
    23  	return check.Response{Name: "Mock InfluxQL Proxy Query Service", Status: check.StatusPass}
    24  }