Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
N
nexpie-grafana-theme
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Registry
Registry
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Kornkitt Poolsup
nexpie-grafana-theme
Commits
c863fd3d
Unverified
Commit
c863fd3d
authored
Jul 14, 2020
by
Arve Knudsen
Committed by
GitHub
Jul 14, 2020
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
CloudWatch: Clean up code (#26259)
* CloudWatch: Clean up code
parent
747bdaf2
Expand all
Show whitespace changes
Inline
Side-by-side
Showing
18 changed files
with
197 additions
and
189 deletions
+197
-189
pkg/tsdb/cloudwatch/annotation_query.go
+5
-5
pkg/tsdb/cloudwatch/cloudwatch.go
+109
-39
pkg/tsdb/cloudwatch/credentials.go
+16
-93
pkg/tsdb/cloudwatch/credentials_test.go
+2
-2
pkg/tsdb/cloudwatch/get_metric_data_executor.go
+1
-1
pkg/tsdb/cloudwatch/get_metric_data_executor_test.go
+1
-1
pkg/tsdb/cloudwatch/log_actions.go
+21
-13
pkg/tsdb/cloudwatch/log_actions_test.go
+7
-7
pkg/tsdb/cloudwatch/metric_data_input_builder.go
+2
-1
pkg/tsdb/cloudwatch/metric_data_query_builder.go
+1
-1
pkg/tsdb/cloudwatch/metric_find_query.go
+0
-0
pkg/tsdb/cloudwatch/metric_find_query_test.go
+24
-18
pkg/tsdb/cloudwatch/query_transformer.go
+2
-2
pkg/tsdb/cloudwatch/query_transformer_test.go
+1
-1
pkg/tsdb/cloudwatch/request_parser.go
+1
-1
pkg/tsdb/cloudwatch/response_parser.go
+1
-1
pkg/tsdb/cloudwatch/time_series_query.go
+2
-2
pkg/tsdb/cloudwatch/time_series_query_test.go
+1
-1
No files found.
pkg/tsdb/cloudwatch/annotation_query.go
View file @
c863fd3d
...
@@ -12,7 +12,7 @@ import (
...
@@ -12,7 +12,7 @@ import (
"github.com/grafana/grafana/pkg/util/errutil"
"github.com/grafana/grafana/pkg/util/errutil"
)
)
func
(
e
*
C
loudWatchExecutor
)
executeAnnotationQuery
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
executeAnnotationQuery
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
result
:=
&
tsdb
.
Response
{
result
:=
&
tsdb
.
Response
{
Results
:
make
(
map
[
string
]
*
tsdb
.
QueryResult
),
Results
:
make
(
map
[
string
]
*
tsdb
.
QueryResult
),
}
}
...
@@ -36,7 +36,7 @@ func (e *CloudWatchExecutor) executeAnnotationQuery(ctx context.Context, queryCo
...
@@ -36,7 +36,7 @@ func (e *CloudWatchExecutor) executeAnnotationQuery(ctx context.Context, queryCo
actionPrefix
:=
parameters
.
Get
(
"actionPrefix"
)
.
MustString
(
""
)
actionPrefix
:=
parameters
.
Get
(
"actionPrefix"
)
.
MustString
(
""
)
alarmNamePrefix
:=
parameters
.
Get
(
"alarmNamePrefix"
)
.
MustString
(
""
)
alarmNamePrefix
:=
parameters
.
Get
(
"alarmNamePrefix"
)
.
MustString
(
""
)
svc
,
err
:=
e
.
get
Client
(
region
)
cli
,
err
:=
e
.
getCW
Client
(
region
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
}
}
...
@@ -48,7 +48,7 @@ func (e *CloudWatchExecutor) executeAnnotationQuery(ctx context.Context, queryCo
...
@@ -48,7 +48,7 @@ func (e *CloudWatchExecutor) executeAnnotationQuery(ctx context.Context, queryCo
ActionPrefix
:
aws
.
String
(
actionPrefix
),
ActionPrefix
:
aws
.
String
(
actionPrefix
),
AlarmNamePrefix
:
aws
.
String
(
alarmNamePrefix
),
AlarmNamePrefix
:
aws
.
String
(
alarmNamePrefix
),
}
}
resp
,
err
:=
svc
.
DescribeAlarms
(
params
)
resp
,
err
:=
cli
.
DescribeAlarms
(
params
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
errutil
.
Wrap
(
"failed to call cloudwatch:DescribeAlarms"
,
err
)
return
nil
,
errutil
.
Wrap
(
"failed to call cloudwatch:DescribeAlarms"
,
err
)
}
}
...
@@ -79,7 +79,7 @@ func (e *CloudWatchExecutor) executeAnnotationQuery(ctx context.Context, queryCo
...
@@ -79,7 +79,7 @@ func (e *CloudWatchExecutor) executeAnnotationQuery(ctx context.Context, queryCo
Statistic
:
aws
.
String
(
s
),
Statistic
:
aws
.
String
(
s
),
Period
:
aws
.
Int64
(
period
),
Period
:
aws
.
Int64
(
period
),
}
}
resp
,
err
:=
svc
.
DescribeAlarmsForMetric
(
params
)
resp
,
err
:=
cli
.
DescribeAlarmsForMetric
(
params
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
errutil
.
Wrap
(
"failed to call cloudwatch:DescribeAlarmsForMetric"
,
err
)
return
nil
,
errutil
.
Wrap
(
"failed to call cloudwatch:DescribeAlarmsForMetric"
,
err
)
}
}
...
@@ -106,7 +106,7 @@ func (e *CloudWatchExecutor) executeAnnotationQuery(ctx context.Context, queryCo
...
@@ -106,7 +106,7 @@ func (e *CloudWatchExecutor) executeAnnotationQuery(ctx context.Context, queryCo
EndDate
:
aws
.
Time
(
endTime
),
EndDate
:
aws
.
Time
(
endTime
),
MaxRecords
:
aws
.
Int64
(
100
),
MaxRecords
:
aws
.
Int64
(
100
),
}
}
resp
,
err
:=
svc
.
DescribeAlarmHistory
(
params
)
resp
,
err
:=
cli
.
DescribeAlarmHistory
(
params
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
errutil
.
Wrap
(
"failed to call cloudwatch:DescribeAlarmHistory"
,
err
)
return
nil
,
errutil
.
Wrap
(
"failed to call cloudwatch:DescribeAlarmHistory"
,
err
)
}
}
...
...
pkg/tsdb/cloudwatch/cloudwatch.go
View file @
c863fd3d
...
@@ -9,25 +9,19 @@ import (
...
@@ -9,25 +9,19 @@ import (
"github.com/grafana/grafana-plugin-sdk-go/data"
"github.com/grafana/grafana-plugin-sdk-go/data"
"github.com/aws/aws-sdk-go/aws/request"
"github.com/aws/aws-sdk-go/service/cloudwatch"
"github.com/aws/aws-sdk-go/service/cloudwatchlogs"
"github.com/aws/aws-sdk-go/service/cloudwatchlogs"
"github.com/aws/aws-sdk-go/service/ec2/ec2iface"
"github.com/aws/aws-sdk-go/service/ec2/ec2iface"
"github.com/aws/aws-sdk-go/service/resourcegroupstaggingapi/resourcegroupstaggingapiiface"
"github.com/aws/aws-sdk-go/service/resourcegroupstaggingapi/resourcegroupstaggingapiiface"
"github.com/grafana/grafana/pkg/components/simplejson"
"github.com/grafana/grafana/pkg/components/simplejson"
"github.com/grafana/grafana/pkg/infra/log"
"github.com/grafana/grafana/pkg/infra/log"
"github.com/grafana/grafana/pkg/models"
"github.com/grafana/grafana/pkg/models"
"github.com/grafana/grafana/pkg/setting"
"github.com/grafana/grafana/pkg/tsdb"
"github.com/grafana/grafana/pkg/tsdb"
)
)
type
CloudWatchExecutor
struct
{
type
datasourceInfo
struct
{
*
models
.
DataSource
ec2Svc
ec2iface
.
EC2API
rgtaSvc
resourcegroupstaggingapiiface
.
ResourceGroupsTaggingAPIAPI
logsClientsByRegion
map
[
string
](
*
cloudwatchlogs
.
CloudWatchLogs
)
mux
sync
.
Mutex
}
type
DatasourceInfo
struct
{
Profile
string
Profile
string
Region
string
Region
string
AuthType
string
AuthType
string
...
@@ -40,56 +34,88 @@ type DatasourceInfo struct {
...
@@ -40,56 +34,88 @@ type DatasourceInfo struct {
}
}
const
cloudWatchTSFormat
=
"2006-01-02 15:04:05.000"
const
cloudWatchTSFormat
=
"2006-01-02 15:04:05.000"
const
defaultRegion
=
"default"
// Constants also defined in datasource/cloudwatch/datasource.ts
// Constants also defined in datasource/cloudwatch/datasource.ts
const
logIdentifierInternal
=
"__log__grafana_internal__"
const
logIdentifierInternal
=
"__log__grafana_internal__"
const
logStreamIdentifierInternal
=
"__logstream__grafana_internal__"
const
logStreamIdentifierInternal
=
"__logstream__grafana_internal__"
func
(
e
*
CloudWatchExecutor
)
getLogsClient
(
region
string
)
(
*
cloudwatchlogs
.
CloudWatchLogs
,
error
)
{
var
plog
=
log
.
New
(
"tsdb.cloudwatch"
)
e
.
mux
.
Lock
()
var
aliasFormat
=
regexp
.
MustCompile
(
`\{\{\s*(.+?)\s*\}\}`
)
defer
e
.
mux
.
Unlock
()
if
logsClient
,
ok
:=
e
.
logsClientsByRegion
[
region
];
ok
{
func
init
()
{
return
logsClient
,
nil
tsdb
.
RegisterTsdbQueryEndpoint
(
"cloudwatch"
,
newcloudWatchExecutor
)
}
}
dsInfo
:=
retrieveDsInfo
(
e
.
DataSource
,
region
)
func
newcloudWatchExecutor
(
datasource
*
models
.
DataSource
)
(
tsdb
.
TsdbQueryEndpoint
,
error
)
{
newLogsClient
,
err
:=
retrieveLogsClient
(
dsInfo
)
e
:=
&
cloudWatchExecutor
{
DataSource
:
datasource
,
}
dsInfo
:=
e
.
getDSInfo
(
defaultRegion
)
defaultLogsClient
,
err
:=
retrieveLogsClient
(
dsInfo
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
}
}
e
.
logsClientsByRegion
=
map
[
string
]
*
cloudwatchlogs
.
CloudWatchLogs
{
dsInfo
.
Region
:
defaultLogsClient
,
defaultRegion
:
defaultLogsClient
,
}
e
.
logsClientsByRegion
[
region
]
=
newLogsClient
return
e
,
nil
}
return
newLogsClient
,
nil
// cloudWatchExecutor executes CloudWatch requests.
type
cloudWatchExecutor
struct
{
*
models
.
DataSource
ec2Svc
ec2iface
.
EC2API
rgtaSvc
resourcegroupstaggingapiiface
.
ResourceGroupsTaggingAPIAPI
logsClientsByRegion
map
[
string
](
*
cloudwatchlogs
.
CloudWatchLogs
)
mux
sync
.
Mutex
}
}
func
NewCloudWatchExecutor
(
datasource
*
models
.
DataSource
)
(
tsdb
.
TsdbQueryEndpoint
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
getCWClient
(
region
string
)
(
*
cloudwatch
.
CloudWatch
,
error
)
{
dsInfo
:=
retrieveDsInfo
(
datasource
,
"default"
)
datasourceInfo
:=
e
.
getDSInfo
(
region
)
defaultLogsClient
,
err
:=
retrieveLogsClient
(
dsInfo
)
cfg
,
err
:=
getAwsConfig
(
datasourceInfo
)
if
err
!=
nil
{
return
nil
,
err
}
sess
,
err
:=
newSession
(
cfg
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
}
}
logsClientsByRegion
:=
make
(
map
[
string
](
*
cloudwatchlogs
.
CloudWatchLogs
))
client
:=
cloudwatch
.
New
(
sess
,
cfg
)
logsClientsByRegion
[
dsInfo
.
Region
]
=
defaultLogsClient
logsClientsByRegion
[
"default"
]
=
defaultLogsClient
client
.
Handlers
.
Send
.
PushFront
(
func
(
r
*
request
.
Request
)
{
r
.
HTTPRequest
.
Header
.
Set
(
"User-Agent"
,
fmt
.
Sprintf
(
"Grafana/%s"
,
setting
.
BuildVersion
))
})
return
&
CloudWatchExecutor
{
return
client
,
nil
logsClientsByRegion
:
logsClientsByRegion
,
},
nil
}
}
var
plog
=
log
.
New
(
"tsdb.cloudwatch"
)
func
(
e
*
cloudWatchExecutor
)
getCWLogsClient
(
region
string
)
(
*
cloudwatchlogs
.
CloudWatchLogs
,
error
)
{
var
aliasFormat
=
regexp
.
MustCompile
(
`\{\{\s*(.+?)\s*\}\}`
)
e
.
mux
.
Lock
()
defer
e
.
mux
.
Unlock
()
func
init
()
{
if
logsClient
,
ok
:=
e
.
logsClientsByRegion
[
region
];
ok
{
tsdb
.
RegisterTsdbQueryEndpoint
(
"cloudwatch"
,
NewCloudWatchExecutor
)
return
logsClient
,
nil
}
dsInfo
:=
e
.
getDSInfo
(
region
)
newLogsClient
,
err
:=
retrieveLogsClient
(
dsInfo
)
if
err
!=
nil
{
return
nil
,
err
}
e
.
logsClientsByRegion
[
region
]
=
newLogsClient
return
newLogsClient
,
nil
}
}
func
(
e
*
C
loudWatchExecutor
)
alertQuery
(
ctx
context
.
Context
,
logsClient
*
cloudwatchlogs
.
CloudWatchLogs
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
cloudwatchlogs
.
GetQueryResultsOutput
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
alertQuery
(
ctx
context
.
Context
,
logsClient
*
cloudwatchlogs
.
CloudWatchLogs
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
cloudwatchlogs
.
GetQueryResultsOutput
,
error
)
{
const
maxAttempts
=
8
const
maxAttempts
=
8
const
pollPeriod
=
1000
*
time
.
Millisecond
const
pollPeriod
=
1000
*
time
.
Millisecond
...
@@ -126,7 +152,8 @@ func (e *CloudWatchExecutor) alertQuery(ctx context.Context, logsClient *cloudwa
...
@@ -126,7 +152,8 @@ func (e *CloudWatchExecutor) alertQuery(ctx context.Context, logsClient *cloudwa
return
nil
,
nil
return
nil
,
nil
}
}
func
(
e
*
CloudWatchExecutor
)
Query
(
ctx
context
.
Context
,
dsInfo
*
models
.
DataSource
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
// Query executes a CloudWatch query.
func
(
e
*
cloudWatchExecutor
)
Query
(
ctx
context
.
Context
,
dsInfo
*
models
.
DataSource
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
e
.
DataSource
=
dsInfo
e
.
DataSource
=
dsInfo
/*
/*
...
@@ -163,18 +190,18 @@ func (e *CloudWatchExecutor) Query(ctx context.Context, dsInfo *models.DataSourc
...
@@ -163,18 +190,18 @@ func (e *CloudWatchExecutor) Query(ctx context.Context, dsInfo *models.DataSourc
return
result
,
err
return
result
,
err
}
}
func
(
e
*
C
loudWatchExecutor
)
executeLogAlertQuery
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
executeLogAlertQuery
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
queryParams
:=
queryContext
.
Queries
[
0
]
.
Model
queryParams
:=
queryContext
.
Queries
[
0
]
.
Model
queryParams
.
Set
(
"subtype"
,
"StartQuery"
)
queryParams
.
Set
(
"subtype"
,
"StartQuery"
)
queryParams
.
Set
(
"queryString"
,
queryParams
.
Get
(
"expression"
)
.
MustString
(
""
))
queryParams
.
Set
(
"queryString"
,
queryParams
.
Get
(
"expression"
)
.
MustString
(
""
))
region
:=
queryParams
.
Get
(
"region"
)
.
MustString
(
"default"
)
region
:=
queryParams
.
Get
(
"region"
)
.
MustString
(
defaultRegion
)
if
region
==
"default"
{
if
region
==
defaultRegion
{
region
=
e
.
DataSource
.
JsonData
.
Get
(
"defaultRegion"
)
.
MustString
()
region
=
e
.
DataSource
.
JsonData
.
Get
(
"defaultRegion"
)
.
MustString
()
queryParams
.
Set
(
"region"
,
region
)
queryParams
.
Set
(
"region"
,
region
)
}
}
logsClient
,
err
:=
e
.
getLogsClient
(
region
)
logsClient
,
err
:=
e
.
get
CW
LogsClient
(
region
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
}
}
...
@@ -227,6 +254,49 @@ func (e *CloudWatchExecutor) executeLogAlertQuery(ctx context.Context, queryCont
...
@@ -227,6 +254,49 @@ func (e *CloudWatchExecutor) executeLogAlertQuery(ctx context.Context, queryCont
return
response
,
nil
return
response
,
nil
}
}
func
(
e
*
cloudWatchExecutor
)
getDSInfo
(
region
string
)
*
datasourceInfo
{
if
region
==
defaultRegion
{
region
=
e
.
DataSource
.
JsonData
.
Get
(
"defaultRegion"
)
.
MustString
()
}
authType
:=
e
.
DataSource
.
JsonData
.
Get
(
"authType"
)
.
MustString
()
assumeRoleArn
:=
e
.
DataSource
.
JsonData
.
Get
(
"assumeRoleArn"
)
.
MustString
()
externalID
:=
e
.
DataSource
.
JsonData
.
Get
(
"externalId"
)
.
MustString
()
decrypted
:=
e
.
DataSource
.
DecryptedValues
()
accessKey
:=
decrypted
[
"accessKey"
]
secretKey
:=
decrypted
[
"secretKey"
]
return
&
datasourceInfo
{
Region
:
region
,
Profile
:
e
.
DataSource
.
Database
,
AuthType
:
authType
,
AssumeRoleArn
:
assumeRoleArn
,
ExternalID
:
externalID
,
AccessKey
:
accessKey
,
SecretKey
:
secretKey
,
}
}
func
retrieveLogsClient
(
dsInfo
*
datasourceInfo
)
(
*
cloudwatchlogs
.
CloudWatchLogs
,
error
)
{
cfg
,
err
:=
getAwsConfig
(
dsInfo
)
if
err
!=
nil
{
return
nil
,
err
}
sess
,
err
:=
newSession
(
cfg
)
if
err
!=
nil
{
return
nil
,
err
}
client
:=
cloudwatchlogs
.
New
(
sess
,
cfg
)
client
.
Handlers
.
Send
.
PushFront
(
func
(
r
*
request
.
Request
)
{
r
.
HTTPRequest
.
Header
.
Set
(
"User-Agent"
,
fmt
.
Sprintf
(
"Grafana/%s"
,
setting
.
BuildVersion
))
})
return
client
,
nil
}
func
isTerminated
(
queryStatus
string
)
bool
{
func
isTerminated
(
queryStatus
string
)
bool
{
return
queryStatus
==
"Complete"
||
queryStatus
==
"Cancelled"
||
queryStatus
==
"Failed"
||
queryStatus
==
"Timeout"
return
queryStatus
==
"Complete"
||
queryStatus
==
"Cancelled"
||
queryStatus
==
"Failed"
||
queryStatus
==
"Timeout"
}
}
pkg/tsdb/cloudwatch/credentials.go
View file @
c863fd3d
...
@@ -14,23 +14,18 @@ import (
...
@@ -14,23 +14,18 @@ import (
"github.com/aws/aws-sdk-go/aws/credentials/stscreds"
"github.com/aws/aws-sdk-go/aws/credentials/stscreds"
"github.com/aws/aws-sdk-go/aws/defaults"
"github.com/aws/aws-sdk-go/aws/defaults"
"github.com/aws/aws-sdk-go/aws/ec2metadata"
"github.com/aws/aws-sdk-go/aws/ec2metadata"
"github.com/aws/aws-sdk-go/aws/request"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/cloudwatch"
"github.com/aws/aws-sdk-go/service/cloudwatchlogs"
"github.com/aws/aws-sdk-go/service/sts"
"github.com/aws/aws-sdk-go/service/sts"
"github.com/aws/aws-sdk-go/service/sts/stsiface"
"github.com/aws/aws-sdk-go/service/sts/stsiface"
"github.com/grafana/grafana/pkg/models"
"github.com/grafana/grafana/pkg/setting"
)
)
type
cach
e
struct
{
type
envelop
e
struct
{
credential
*
credentials
.
Credentials
credential
s
*
credentials
.
Credentials
expiration
*
time
.
Time
expiration
*
time
.
Time
}
}
var
awsCred
entialCache
=
make
(
map
[
string
]
cache
)
var
awsCred
sCache
=
map
[
string
]
envelope
{}
var
cred
ential
CacheLock
sync
.
RWMutex
var
cred
s
CacheLock
sync
.
RWMutex
// Session factory.
// Session factory.
// Stubbable by tests.
// Stubbable by tests.
...
@@ -50,18 +45,17 @@ var newEC2Metadata = func(p client.ConfigProvider, cfgs ...*aws.Config) *ec2meta
...
@@ -50,18 +45,17 @@ var newEC2Metadata = func(p client.ConfigProvider, cfgs ...*aws.Config) *ec2meta
return
ec2metadata
.
New
(
p
,
cfgs
...
)
return
ec2metadata
.
New
(
p
,
cfgs
...
)
}
}
func
getCredentials
(
dsInfo
*
D
atasourceInfo
)
(
*
credentials
.
Credentials
,
error
)
{
func
getCredentials
(
dsInfo
*
d
atasourceInfo
)
(
*
credentials
.
Credentials
,
error
)
{
cacheKey
:=
fmt
.
Sprintf
(
"%s:%s:%s:%s"
,
dsInfo
.
AuthType
,
dsInfo
.
AccessKey
,
dsInfo
.
Profile
,
dsInfo
.
AssumeRoleArn
)
cacheKey
:=
fmt
.
Sprintf
(
"%s:%s:%s:%s"
,
dsInfo
.
AuthType
,
dsInfo
.
AccessKey
,
dsInfo
.
Profile
,
dsInfo
.
AssumeRoleArn
)
credentialCacheLock
.
RLock
()
credsCacheLock
.
RLock
()
if
_
,
ok
:=
awsCredentialCache
[
cacheKey
];
ok
{
if
env
,
ok
:=
awsCredsCache
[
cacheKey
];
ok
{
if
awsCredentialCache
[
cacheKey
]
.
expiration
!=
nil
&&
if
env
.
expiration
!=
nil
&&
env
.
expiration
.
After
(
time
.
Now
()
.
UTC
())
{
awsCredentialCache
[
cacheKey
]
.
expiration
.
After
(
time
.
Now
()
.
UTC
())
{
result
:=
env
.
credentials
result
:=
awsCredentialCache
[
cacheKey
]
.
credential
credsCacheLock
.
RUnlock
()
credentialCacheLock
.
RUnlock
()
return
result
,
nil
return
result
,
nil
}
}
}
}
cred
ential
CacheLock
.
RUnlock
()
cred
s
CacheLock
.
RUnlock
()
accessKeyID
:=
""
accessKeyID
:=
""
secretAccessKey
:=
""
secretAccessKey
:=
""
...
@@ -135,12 +129,12 @@ func getCredentials(dsInfo *DatasourceInfo) (*credentials.Credentials, error) {
...
@@ -135,12 +129,12 @@ func getCredentials(dsInfo *DatasourceInfo) (*credentials.Credentials, error) {
remoteCredProvider
(
sess
),
remoteCredProvider
(
sess
),
})
})
cred
ential
CacheLock
.
Lock
()
cred
s
CacheLock
.
Lock
()
awsCred
entialCache
[
cacheKey
]
=
cach
e
{
awsCred
sCache
[
cacheKey
]
=
envelop
e
{
credential
:
creds
,
credential
s
:
creds
,
expiration
:
expiration
,
expiration
:
expiration
,
}
}
cred
ential
CacheLock
.
Unlock
()
cred
s
CacheLock
.
Unlock
()
return
creds
,
nil
return
creds
,
nil
}
}
...
@@ -178,37 +172,7 @@ func ec2RoleProvider(sess *session.Session) credentials.Provider {
...
@@ -178,37 +172,7 @@ func ec2RoleProvider(sess *session.Session) credentials.Provider {
return
&
ec2rolecreds
.
EC2RoleProvider
{
Client
:
newEC2Metadata
(
sess
),
ExpiryWindow
:
5
*
time
.
Minute
}
return
&
ec2rolecreds
.
EC2RoleProvider
{
Client
:
newEC2Metadata
(
sess
),
ExpiryWindow
:
5
*
time
.
Minute
}
}
}
func
(
e
*
CloudWatchExecutor
)
getDsInfo
(
region
string
)
*
DatasourceInfo
{
func
getAwsConfig
(
dsInfo
*
datasourceInfo
)
(
*
aws
.
Config
,
error
)
{
return
retrieveDsInfo
(
e
.
DataSource
,
region
)
}
func
retrieveDsInfo
(
datasource
*
models
.
DataSource
,
region
string
)
*
DatasourceInfo
{
defaultRegion
:=
datasource
.
JsonData
.
Get
(
"defaultRegion"
)
.
MustString
()
if
region
==
"default"
{
region
=
defaultRegion
}
authType
:=
datasource
.
JsonData
.
Get
(
"authType"
)
.
MustString
()
assumeRoleArn
:=
datasource
.
JsonData
.
Get
(
"assumeRoleArn"
)
.
MustString
()
externalID
:=
datasource
.
JsonData
.
Get
(
"externalId"
)
.
MustString
()
decrypted
:=
datasource
.
DecryptedValues
()
accessKey
:=
decrypted
[
"accessKey"
]
secretKey
:=
decrypted
[
"secretKey"
]
datasourceInfo
:=
&
DatasourceInfo
{
Region
:
region
,
Profile
:
datasource
.
Database
,
AuthType
:
authType
,
AssumeRoleArn
:
assumeRoleArn
,
ExternalID
:
externalID
,
AccessKey
:
accessKey
,
SecretKey
:
secretKey
,
}
return
datasourceInfo
}
func
getAwsConfig
(
dsInfo
*
DatasourceInfo
)
(
*
aws
.
Config
,
error
)
{
creds
,
err
:=
getCredentials
(
dsInfo
)
creds
,
err
:=
getCredentials
(
dsInfo
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
...
@@ -221,44 +185,3 @@ func getAwsConfig(dsInfo *DatasourceInfo) (*aws.Config, error) {
...
@@ -221,44 +185,3 @@ func getAwsConfig(dsInfo *DatasourceInfo) (*aws.Config, error) {
return
cfg
,
nil
return
cfg
,
nil
}
}
func
(
e
*
CloudWatchExecutor
)
getClient
(
region
string
)
(
*
cloudwatch
.
CloudWatch
,
error
)
{
datasourceInfo
:=
e
.
getDsInfo
(
region
)
cfg
,
err
:=
getAwsConfig
(
datasourceInfo
)
if
err
!=
nil
{
return
nil
,
err
}
sess
,
err
:=
newSession
(
cfg
)
if
err
!=
nil
{
return
nil
,
err
}
client
:=
cloudwatch
.
New
(
sess
,
cfg
)
client
.
Handlers
.
Send
.
PushFront
(
func
(
r
*
request
.
Request
)
{
r
.
HTTPRequest
.
Header
.
Set
(
"User-Agent"
,
fmt
.
Sprintf
(
"Grafana/%s"
,
setting
.
BuildVersion
))
})
return
client
,
nil
}
func
retrieveLogsClient
(
datasourceInfo
*
DatasourceInfo
)
(
*
cloudwatchlogs
.
CloudWatchLogs
,
error
)
{
cfg
,
err
:=
getAwsConfig
(
datasourceInfo
)
if
err
!=
nil
{
return
nil
,
err
}
sess
,
err
:=
newSession
(
cfg
)
if
err
!=
nil
{
return
nil
,
err
}
client
:=
cloudwatchlogs
.
New
(
sess
,
cfg
)
client
.
Handlers
.
Send
.
PushFront
(
func
(
r
*
request
.
Request
)
{
r
.
HTTPRequest
.
Header
.
Set
(
"User-Agent"
,
fmt
.
Sprintf
(
"Grafana/%s"
,
setting
.
BuildVersion
))
})
return
client
,
nil
}
pkg/tsdb/cloudwatch/credentials_test.go
View file @
c863fd3d
...
@@ -87,7 +87,7 @@ func TestGetCredentials_ARNAuthType(t *testing.T) {
...
@@ -87,7 +87,7 @@ func TestGetCredentials_ARNAuthType(t *testing.T) {
},
nil
)
.
},
nil
)
.
Times
(
1
)
Times
(
1
)
creds
,
err
:=
getCredentials
(
&
D
atasourceInfo
{
creds
,
err
:=
getCredentials
(
&
d
atasourceInfo
{
AuthType
:
"arn"
,
AuthType
:
"arn"
,
})
})
require
.
NoError
(
t
,
err
)
require
.
NoError
(
t
,
err
)
...
@@ -113,7 +113,7 @@ func TestGetCredentials_ARNAuthType(t *testing.T) {
...
@@ -113,7 +113,7 @@ func TestGetCredentials_ARNAuthType(t *testing.T) {
},
nil
)
.
},
nil
)
.
Times
(
1
)
Times
(
1
)
creds
,
err
:=
getCredentials
(
&
D
atasourceInfo
{
creds
,
err
:=
getCredentials
(
&
d
atasourceInfo
{
AuthType
:
"arn"
,
AuthType
:
"arn"
,
ExternalID
:
"external-id"
,
ExternalID
:
"external-id"
,
})
})
...
...
pkg/tsdb/cloudwatch/get_metric_data_executor.go
View file @
c863fd3d
...
@@ -8,7 +8,7 @@ import (
...
@@ -8,7 +8,7 @@ import (
"github.com/grafana/grafana/pkg/infra/metrics"
"github.com/grafana/grafana/pkg/infra/metrics"
)
)
func
(
e
*
C
loudWatchExecutor
)
executeRequest
(
ctx
context
.
Context
,
client
cloudWatchClient
,
metricDataInput
*
cloudwatch
.
GetMetricDataInput
)
([]
*
cloudwatch
.
GetMetricDataOutput
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
executeRequest
(
ctx
context
.
Context
,
client
cloudWatchClient
,
metricDataInput
*
cloudwatch
.
GetMetricDataInput
)
([]
*
cloudwatch
.
GetMetricDataOutput
,
error
)
{
mdo
:=
make
([]
*
cloudwatch
.
GetMetricDataOutput
,
0
)
mdo
:=
make
([]
*
cloudwatch
.
GetMetricDataOutput
,
0
)
nextToken
:=
""
nextToken
:=
""
...
...
pkg/tsdb/cloudwatch/get_metric_data_executor_test.go
View file @
c863fd3d
...
@@ -35,7 +35,7 @@ func (client *cloudWatchFakeClient) GetMetricDataWithContext(ctx aws.Context, in
...
@@ -35,7 +35,7 @@ func (client *cloudWatchFakeClient) GetMetricDataWithContext(ctx aws.Context, in
}
}
func
TestGetMetricDataExecutorTest
(
t
*
testing
.
T
)
{
func
TestGetMetricDataExecutorTest
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
inputs
:=
&
cloudwatch
.
GetMetricDataInput
{
MetricDataQueries
:
[]
*
cloudwatch
.
MetricDataQuery
{}}
inputs
:=
&
cloudwatch
.
GetMetricDataInput
{
MetricDataQueries
:
[]
*
cloudwatch
.
MetricDataQuery
{}}
res
,
err
:=
executor
.
executeRequest
(
context
.
Background
(),
&
cloudWatchFakeClient
{},
inputs
)
res
,
err
:=
executor
.
executeRequest
(
context
.
Background
(),
&
cloudWatchFakeClient
{},
inputs
)
require
.
NoError
(
t
,
err
)
require
.
NoError
(
t
,
err
)
...
...
pkg/tsdb/cloudwatch/log_actions.go
View file @
c863fd3d
...
@@ -15,7 +15,7 @@ import (
...
@@ -15,7 +15,7 @@ import (
"golang.org/x/sync/errgroup"
"golang.org/x/sync/errgroup"
)
)
func
(
e
*
C
loudWatchExecutor
)
executeLogActions
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
executeLogActions
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
resultChan
:=
make
(
chan
*
tsdb
.
QueryResult
,
len
(
queryContext
.
Queries
))
resultChan
:=
make
(
chan
*
tsdb
.
QueryResult
,
len
(
queryContext
.
Queries
))
eg
,
ectx
:=
errgroup
.
WithContext
(
ctx
)
eg
,
ectx
:=
errgroup
.
WithContext
(
ctx
)
...
@@ -73,13 +73,13 @@ func (e *CloudWatchExecutor) executeLogActions(ctx context.Context, queryContext
...
@@ -73,13 +73,13 @@ func (e *CloudWatchExecutor) executeLogActions(ctx context.Context, queryContext
return
response
,
nil
return
response
,
nil
}
}
func
(
e
*
C
loudWatchExecutor
)
executeLogAction
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
,
query
*
tsdb
.
Query
)
(
*
data
.
Frame
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
executeLogAction
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
,
query
*
tsdb
.
Query
)
(
*
data
.
Frame
,
error
)
{
parameters
:=
query
.
Model
parameters
:=
query
.
Model
subType
:=
query
.
Model
.
Get
(
"subtype"
)
.
MustString
()
subType
:=
query
.
Model
.
Get
(
"subtype"
)
.
MustString
()
defaultRegion
:=
e
.
DataSource
.
JsonData
.
Get
(
"defaultRegion"
)
.
MustString
()
defaultRegion
:=
e
.
DataSource
.
JsonData
.
Get
(
"defaultRegion"
)
.
MustString
()
region
:=
parameters
.
Get
(
"region"
)
.
MustString
(
defaultRegion
)
region
:=
parameters
.
Get
(
"region"
)
.
MustString
(
defaultRegion
)
logsClient
,
err
:=
e
.
getLogsClient
(
region
)
logsClient
,
err
:=
e
.
get
CW
LogsClient
(
region
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
}
}
...
@@ -100,7 +100,6 @@ func (e *CloudWatchExecutor) executeLogAction(ctx context.Context, queryContext
...
@@ -100,7 +100,6 @@ func (e *CloudWatchExecutor) executeLogAction(ctx context.Context, queryContext
case
"GetLogEvents"
:
case
"GetLogEvents"
:
data
,
err
=
e
.
handleGetLogEvents
(
ctx
,
logsClient
,
parameters
)
data
,
err
=
e
.
handleGetLogEvents
(
ctx
,
logsClient
,
parameters
)
}
}
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
}
}
...
@@ -108,7 +107,8 @@ func (e *CloudWatchExecutor) executeLogAction(ctx context.Context, queryContext
...
@@ -108,7 +107,8 @@ func (e *CloudWatchExecutor) executeLogAction(ctx context.Context, queryContext
return
data
,
nil
return
data
,
nil
}
}
func
(
e
*
CloudWatchExecutor
)
handleGetLogEvents
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
data
.
Frame
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
handleGetLogEvents
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
data
.
Frame
,
error
)
{
queryRequest
:=
&
cloudwatchlogs
.
GetLogEventsInput
{
queryRequest
:=
&
cloudwatchlogs
.
GetLogEventsInput
{
Limit
:
aws
.
Int64
(
parameters
.
Get
(
"limit"
)
.
MustInt64
(
10
)),
Limit
:
aws
.
Int64
(
parameters
.
Get
(
"limit"
)
.
MustInt64
(
10
)),
StartFromHead
:
aws
.
Bool
(
parameters
.
Get
(
"startFromHead"
)
.
MustBool
(
false
)),
StartFromHead
:
aws
.
Bool
(
parameters
.
Get
(
"startFromHead"
)
.
MustBool
(
false
)),
...
@@ -159,7 +159,8 @@ func (e *CloudWatchExecutor) handleGetLogEvents(ctx context.Context, logsClient
...
@@ -159,7 +159,8 @@ func (e *CloudWatchExecutor) handleGetLogEvents(ctx context.Context, logsClient
return
data
.
NewFrame
(
"logEvents"
,
timestampField
,
messageField
),
nil
return
data
.
NewFrame
(
"logEvents"
,
timestampField
,
messageField
),
nil
}
}
func
(
e
*
CloudWatchExecutor
)
handleDescribeLogGroups
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
data
.
Frame
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
handleDescribeLogGroups
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
data
.
Frame
,
error
)
{
logGroupNamePrefix
:=
parameters
.
Get
(
"logGroupNamePrefix"
)
.
MustString
(
""
)
logGroupNamePrefix
:=
parameters
.
Get
(
"logGroupNamePrefix"
)
.
MustString
(
""
)
var
response
*
cloudwatchlogs
.
DescribeLogGroupsOutput
=
nil
var
response
*
cloudwatchlogs
.
DescribeLogGroupsOutput
=
nil
...
@@ -189,7 +190,8 @@ func (e *CloudWatchExecutor) handleDescribeLogGroups(ctx context.Context, logsCl
...
@@ -189,7 +190,8 @@ func (e *CloudWatchExecutor) handleDescribeLogGroups(ctx context.Context, logsCl
return
frame
,
nil
return
frame
,
nil
}
}
func
(
e
*
CloudWatchExecutor
)
executeStartQuery
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
,
timeRange
*
tsdb
.
TimeRange
)
(
*
cloudwatchlogs
.
StartQueryOutput
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
executeStartQuery
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
,
timeRange
*
tsdb
.
TimeRange
)
(
*
cloudwatchlogs
.
StartQueryOutput
,
error
)
{
startTime
,
err
:=
timeRange
.
ParseFrom
()
startTime
,
err
:=
timeRange
.
ParseFrom
()
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
...
@@ -224,7 +226,8 @@ func (e *CloudWatchExecutor) executeStartQuery(ctx context.Context, logsClient c
...
@@ -224,7 +226,8 @@ func (e *CloudWatchExecutor) executeStartQuery(ctx context.Context, logsClient c
return
logsClient
.
StartQueryWithContext
(
ctx
,
startQueryInput
)
return
logsClient
.
StartQueryWithContext
(
ctx
,
startQueryInput
)
}
}
func
(
e
*
CloudWatchExecutor
)
handleStartQuery
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
,
timeRange
*
tsdb
.
TimeRange
,
refID
string
)
(
*
data
.
Frame
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
handleStartQuery
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
,
timeRange
*
tsdb
.
TimeRange
,
refID
string
)
(
*
data
.
Frame
,
error
)
{
startQueryResponse
,
err
:=
e
.
executeStartQuery
(
ctx
,
logsClient
,
parameters
,
timeRange
)
startQueryResponse
,
err
:=
e
.
executeStartQuery
(
ctx
,
logsClient
,
parameters
,
timeRange
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
...
@@ -244,7 +247,8 @@ func (e *CloudWatchExecutor) handleStartQuery(ctx context.Context, logsClient cl
...
@@ -244,7 +247,8 @@ func (e *CloudWatchExecutor) handleStartQuery(ctx context.Context, logsClient cl
return
dataFrame
,
nil
return
dataFrame
,
nil
}
}
func
(
e
*
CloudWatchExecutor
)
executeStopQuery
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
cloudwatchlogs
.
StopQueryOutput
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
executeStopQuery
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
cloudwatchlogs
.
StopQueryOutput
,
error
)
{
queryInput
:=
&
cloudwatchlogs
.
StopQueryInput
{
queryInput
:=
&
cloudwatchlogs
.
StopQueryInput
{
QueryId
:
aws
.
String
(
parameters
.
Get
(
"queryId"
)
.
MustString
()),
QueryId
:
aws
.
String
(
parameters
.
Get
(
"queryId"
)
.
MustString
()),
}
}
...
@@ -264,7 +268,8 @@ func (e *CloudWatchExecutor) executeStopQuery(ctx context.Context, logsClient cl
...
@@ -264,7 +268,8 @@ func (e *CloudWatchExecutor) executeStopQuery(ctx context.Context, logsClient cl
return
response
,
err
return
response
,
err
}
}
func
(
e
*
CloudWatchExecutor
)
handleStopQuery
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
data
.
Frame
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
handleStopQuery
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
data
.
Frame
,
error
)
{
response
,
err
:=
e
.
executeStopQuery
(
ctx
,
logsClient
,
parameters
)
response
,
err
:=
e
.
executeStopQuery
(
ctx
,
logsClient
,
parameters
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
...
@@ -274,7 +279,8 @@ func (e *CloudWatchExecutor) handleStopQuery(ctx context.Context, logsClient clo
...
@@ -274,7 +279,8 @@ func (e *CloudWatchExecutor) handleStopQuery(ctx context.Context, logsClient clo
return
dataFrame
,
nil
return
dataFrame
,
nil
}
}
func
(
e
*
CloudWatchExecutor
)
executeGetQueryResults
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
cloudwatchlogs
.
GetQueryResultsOutput
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
executeGetQueryResults
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
)
(
*
cloudwatchlogs
.
GetQueryResultsOutput
,
error
)
{
queryInput
:=
&
cloudwatchlogs
.
GetQueryResultsInput
{
queryInput
:=
&
cloudwatchlogs
.
GetQueryResultsInput
{
QueryId
:
aws
.
String
(
parameters
.
Get
(
"queryId"
)
.
MustString
()),
QueryId
:
aws
.
String
(
parameters
.
Get
(
"queryId"
)
.
MustString
()),
}
}
...
@@ -282,7 +288,8 @@ func (e *CloudWatchExecutor) executeGetQueryResults(ctx context.Context, logsCli
...
@@ -282,7 +288,8 @@ func (e *CloudWatchExecutor) executeGetQueryResults(ctx context.Context, logsCli
return
logsClient
.
GetQueryResultsWithContext
(
ctx
,
queryInput
)
return
logsClient
.
GetQueryResultsWithContext
(
ctx
,
queryInput
)
}
}
func
(
e
*
CloudWatchExecutor
)
handleGetQueryResults
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
,
refID
string
)
(
*
data
.
Frame
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
handleGetQueryResults
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
,
refID
string
)
(
*
data
.
Frame
,
error
)
{
getQueryResultsOutput
,
err
:=
e
.
executeGetQueryResults
(
ctx
,
logsClient
,
parameters
)
getQueryResultsOutput
,
err
:=
e
.
executeGetQueryResults
(
ctx
,
logsClient
,
parameters
)
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
...
@@ -299,7 +306,8 @@ func (e *CloudWatchExecutor) handleGetQueryResults(ctx context.Context, logsClie
...
@@ -299,7 +306,8 @@ func (e *CloudWatchExecutor) handleGetQueryResults(ctx context.Context, logsClie
return
dataFrame
,
nil
return
dataFrame
,
nil
}
}
func
(
e
*
CloudWatchExecutor
)
handleGetLogGroupFields
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
,
refID
string
)
(
*
data
.
Frame
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
handleGetLogGroupFields
(
ctx
context
.
Context
,
logsClient
cloudwatchlogsiface
.
CloudWatchLogsAPI
,
parameters
*
simplejson
.
Json
,
refID
string
)
(
*
data
.
Frame
,
error
)
{
queryInput
:=
&
cloudwatchlogs
.
GetLogGroupFieldsInput
{
queryInput
:=
&
cloudwatchlogs
.
GetLogGroupFieldsInput
{
LogGroupName
:
aws
.
String
(
parameters
.
Get
(
"logGroupName"
)
.
MustString
()),
LogGroupName
:
aws
.
String
(
parameters
.
Get
(
"logGroupName"
)
.
MustString
()),
Time
:
aws
.
Int64
(
parameters
.
Get
(
"time"
)
.
MustInt64
()),
Time
:
aws
.
Int64
(
parameters
.
Get
(
"time"
)
.
MustInt64
()),
...
...
pkg/tsdb/cloudwatch/log_actions_test.go
View file @
c863fd3d
...
@@ -21,7 +21,7 @@ import (
...
@@ -21,7 +21,7 @@ import (
//***
//***
func
TestHandleDescribeLogGroups_WhenLogGroupNamePrefixIsEmpty
(
t
*
testing
.
T
)
{
func
TestHandleDescribeLogGroups_WhenLogGroupNamePrefixIsEmpty
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
logsClient
:=
&
FakeLogsClient
{
logsClient
:=
&
FakeLogsClient
{
Config
:
aws
.
Config
{
Config
:
aws
.
Config
{
...
@@ -43,7 +43,7 @@ func TestHandleDescribeLogGroups_WhenLogGroupNamePrefixIsEmpty(t *testing.T) {
...
@@ -43,7 +43,7 @@ func TestHandleDescribeLogGroups_WhenLogGroupNamePrefixIsEmpty(t *testing.T) {
}
}
func
TestHandleDescribeLogGroups_WhenLogGroupNamePrefixIsNotEmpty
(
t
*
testing
.
T
)
{
func
TestHandleDescribeLogGroups_WhenLogGroupNamePrefixIsNotEmpty
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
logsClient
:=
&
FakeLogsClient
{
logsClient
:=
&
FakeLogsClient
{
Config
:
aws
.
Config
{
Config
:
aws
.
Config
{
...
@@ -64,7 +64,7 @@ func TestHandleDescribeLogGroups_WhenLogGroupNamePrefixIsNotEmpty(t *testing.T)
...
@@ -64,7 +64,7 @@ func TestHandleDescribeLogGroups_WhenLogGroupNamePrefixIsNotEmpty(t *testing.T)
}
}
func
TestHandleGetLogGroupFields_WhenLogGroupNamePrefixIsNotEmpty
(
t
*
testing
.
T
)
{
func
TestHandleGetLogGroupFields_WhenLogGroupNamePrefixIsNotEmpty
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
logsClient
:=
&
FakeLogsClient
{
logsClient
:=
&
FakeLogsClient
{
Config
:
aws
.
Config
{
Config
:
aws
.
Config
{
...
@@ -89,7 +89,7 @@ func TestHandleGetLogGroupFields_WhenLogGroupNamePrefixIsNotEmpty(t *testing.T)
...
@@ -89,7 +89,7 @@ func TestHandleGetLogGroupFields_WhenLogGroupNamePrefixIsNotEmpty(t *testing.T)
}
}
func
TestExecuteStartQuery
(
t
*
testing
.
T
)
{
func
TestExecuteStartQuery
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
logsClient
:=
&
FakeLogsClient
{
logsClient
:=
&
FakeLogsClient
{
Config
:
aws
.
Config
{
Config
:
aws
.
Config
{
...
@@ -117,7 +117,7 @@ func TestExecuteStartQuery(t *testing.T) {
...
@@ -117,7 +117,7 @@ func TestExecuteStartQuery(t *testing.T) {
}
}
func
TestHandleStartQuery
(
t
*
testing
.
T
)
{
func
TestHandleStartQuery
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
logsClient
:=
&
FakeLogsClient
{
logsClient
:=
&
FakeLogsClient
{
Config
:
aws
.
Config
{
Config
:
aws
.
Config
{
...
@@ -152,7 +152,7 @@ func TestHandleStartQuery(t *testing.T) {
...
@@ -152,7 +152,7 @@ func TestHandleStartQuery(t *testing.T) {
}
}
func
TestHandleStopQuery
(
t
*
testing
.
T
)
{
func
TestHandleStopQuery
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
logsClient
:=
&
FakeLogsClient
{
logsClient
:=
&
FakeLogsClient
{
Config
:
aws
.
Config
{
Config
:
aws
.
Config
{
...
@@ -174,7 +174,7 @@ func TestHandleStopQuery(t *testing.T) {
...
@@ -174,7 +174,7 @@ func TestHandleStopQuery(t *testing.T) {
}
}
func
TestHandleGetQueryResults
(
t
*
testing
.
T
)
{
func
TestHandleGetQueryResults
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
logsClient
:=
&
FakeLogsClient
{
logsClient
:=
&
FakeLogsClient
{
Config
:
aws
.
Config
{
Config
:
aws
.
Config
{
...
...
pkg/tsdb/cloudwatch/metric_data_input_builder.go
View file @
c863fd3d
...
@@ -7,7 +7,8 @@ import (
...
@@ -7,7 +7,8 @@ import (
"github.com/aws/aws-sdk-go/service/cloudwatch"
"github.com/aws/aws-sdk-go/service/cloudwatch"
)
)
func
(
e
*
CloudWatchExecutor
)
buildMetricDataInput
(
startTime
time
.
Time
,
endTime
time
.
Time
,
queries
map
[
string
]
*
cloudWatchQuery
)
(
*
cloudwatch
.
GetMetricDataInput
,
error
)
{
func
(
e
*
cloudWatchExecutor
)
buildMetricDataInput
(
startTime
time
.
Time
,
endTime
time
.
Time
,
queries
map
[
string
]
*
cloudWatchQuery
)
(
*
cloudwatch
.
GetMetricDataInput
,
error
)
{
metricDataInput
:=
&
cloudwatch
.
GetMetricDataInput
{
metricDataInput
:=
&
cloudwatch
.
GetMetricDataInput
{
StartTime
:
aws
.
Time
(
startTime
),
StartTime
:
aws
.
Time
(
startTime
),
EndTime
:
aws
.
Time
(
endTime
),
EndTime
:
aws
.
Time
(
endTime
),
...
...
pkg/tsdb/cloudwatch/metric_data_query_builder.go
View file @
c863fd3d
...
@@ -10,7 +10,7 @@ import (
...
@@ -10,7 +10,7 @@ import (
"github.com/aws/aws-sdk-go/service/cloudwatch"
"github.com/aws/aws-sdk-go/service/cloudwatch"
)
)
func
(
e
*
C
loudWatchExecutor
)
buildMetricDataQuery
(
query
*
cloudWatchQuery
)
(
*
cloudwatch
.
MetricDataQuery
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
buildMetricDataQuery
(
query
*
cloudWatchQuery
)
(
*
cloudwatch
.
MetricDataQuery
,
error
)
{
mdq
:=
&
cloudwatch
.
MetricDataQuery
{
mdq
:=
&
cloudwatch
.
MetricDataQuery
{
Id
:
aws
.
String
(
query
.
Id
),
Id
:
aws
.
String
(
query
.
Id
),
ReturnData
:
aws
.
Bool
(
query
.
ReturnData
),
ReturnData
:
aws
.
Bool
(
query
.
ReturnData
),
...
...
pkg/tsdb/cloudwatch/metric_find_query.go
View file @
c863fd3d
This diff is collapsed.
Click to expand it.
pkg/tsdb/cloudwatch/metric_find_query_test.go
View file @
c863fd3d
...
@@ -44,13 +44,16 @@ func (m mockedRGTA) GetResourcesPages(in *resourcegroupstaggingapi.GetResourcesI
...
@@ -44,13 +44,16 @@ func (m mockedRGTA) GetResourcesPages(in *resourcegroupstaggingapi.GetResourcesI
func
TestCloudWatchMetrics
(
t
*
testing
.
T
)
{
func
TestCloudWatchMetrics
(
t
*
testing
.
T
)
{
t
.
Run
(
"When calling getMetricsForCustomMetrics"
,
func
(
t
*
testing
.
T
)
{
t
.
Run
(
"When calling getMetricsForCustomMetrics"
,
func
(
t
*
testing
.
T
)
{
dsInfo
:=
&
DatasourceInfo
{
const
region
=
"us-east-1"
Region
:
"us-east-1"
,
e
:=
&
cloudWatchExecutor
{
Namespace
:
"Foo"
,
DataSource
:
&
models
.
DataSource
{
Profile
:
"default"
,
Database
:
"default"
,
AssumeRoleArn
:
""
,
JsonData
:
simplejson
.
NewFromAny
(
map
[
string
]
interface
{}{
"Region"
:
region
,
}),
},
}
}
f
:=
func
(
dsInfo
*
DatasourceInfo
)
(
cloudwatch
.
ListMetricsOutput
,
error
)
{
f
:=
func
(
region
string
)
(
cloudwatch
.
ListMetricsOutput
,
error
)
{
return
cloudwatch
.
ListMetricsOutput
{
return
cloudwatch
.
ListMetricsOutput
{
Metrics
:
[]
*
cloudwatch
.
Metric
{
Metrics
:
[]
*
cloudwatch
.
Metric
{
{
{
...
@@ -64,20 +67,23 @@ func TestCloudWatchMetrics(t *testing.T) {
...
@@ -64,20 +67,23 @@ func TestCloudWatchMetrics(t *testing.T) {
},
},
},
nil
},
nil
}
}
metrics
,
err
:=
getMetricsForCustomMetrics
(
dsInfo
,
f
)
metrics
,
err
:=
e
.
getMetricsForCustomMetrics
(
region
,
f
)
require
.
NoError
(
t
,
err
)
require
.
NoError
(
t
,
err
)
assert
.
Contains
(
t
,
metrics
,
"Test_MetricName"
)
assert
.
Contains
(
t
,
metrics
,
"Test_MetricName"
)
})
})
t
.
Run
(
"When calling getDimensionsForCustomMetrics"
,
func
(
t
*
testing
.
T
)
{
t
.
Run
(
"When calling getDimensionsForCustomMetrics"
,
func
(
t
*
testing
.
T
)
{
dsInfo
:=
&
DatasourceInfo
{
const
region
=
"us-east-1"
Region
:
"us-east-1"
,
e
:=
&
cloudWatchExecutor
{
Namespace
:
"Foo"
,
DataSource
:
&
models
.
DataSource
{
Profile
:
"default"
,
Database
:
"default"
,
AssumeRoleArn
:
""
,
JsonData
:
simplejson
.
NewFromAny
(
map
[
string
]
interface
{}{
"Region"
:
region
,
}),
},
}
}
f
:=
func
(
dsInfo
*
DatasourceInfo
)
(
cloudwatch
.
ListMetricsOutput
,
error
)
{
f
:=
func
(
region
string
)
(
cloudwatch
.
ListMetricsOutput
,
error
)
{
return
cloudwatch
.
ListMetricsOutput
{
return
cloudwatch
.
ListMetricsOutput
{
Metrics
:
[]
*
cloudwatch
.
Metric
{
Metrics
:
[]
*
cloudwatch
.
Metric
{
{
{
...
@@ -91,14 +97,14 @@ func TestCloudWatchMetrics(t *testing.T) {
...
@@ -91,14 +97,14 @@ func TestCloudWatchMetrics(t *testing.T) {
},
},
},
nil
},
nil
}
}
dimensionKeys
,
err
:=
getDimensionsForCustomMetrics
(
dsInfo
,
f
)
dimensionKeys
,
err
:=
e
.
getDimensionsForCustomMetrics
(
region
,
f
)
require
.
NoError
(
t
,
err
)
require
.
NoError
(
t
,
err
)
assert
.
Contains
(
t
,
dimensionKeys
,
"Test_DimensionName"
)
assert
.
Contains
(
t
,
dimensionKeys
,
"Test_DimensionName"
)
})
})
t
.
Run
(
"When calling handleGetRegions"
,
func
(
t
*
testing
.
T
)
{
t
.
Run
(
"When calling handleGetRegions"
,
func
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{
executor
:=
&
c
loudWatchExecutor
{
ec2Svc
:
mockedEc2
{
RespRegions
:
ec2
.
DescribeRegionsOutput
{
ec2Svc
:
mockedEc2
{
RespRegions
:
ec2
.
DescribeRegionsOutput
{
Regions
:
[]
*
ec2
.
Region
{
Regions
:
[]
*
ec2
.
Region
{
{
{
...
@@ -123,7 +129,7 @@ func TestCloudWatchMetrics(t *testing.T) {
...
@@ -123,7 +129,7 @@ func TestCloudWatchMetrics(t *testing.T) {
})
})
t
.
Run
(
"When calling handleGetEc2InstanceAttribute"
,
func
(
t
*
testing
.
T
)
{
t
.
Run
(
"When calling handleGetEc2InstanceAttribute"
,
func
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{
executor
:=
&
c
loudWatchExecutor
{
ec2Svc
:
mockedEc2
{
Resp
:
ec2
.
DescribeInstancesOutput
{
ec2Svc
:
mockedEc2
{
Resp
:
ec2
.
DescribeInstancesOutput
{
Reservations
:
[]
*
ec2
.
Reservation
{
Reservations
:
[]
*
ec2
.
Reservation
{
{
{
...
@@ -156,7 +162,7 @@ func TestCloudWatchMetrics(t *testing.T) {
...
@@ -156,7 +162,7 @@ func TestCloudWatchMetrics(t *testing.T) {
})
})
t
.
Run
(
"When calling handleGetEbsVolumeIds"
,
func
(
t
*
testing
.
T
)
{
t
.
Run
(
"When calling handleGetEbsVolumeIds"
,
func
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{
executor
:=
&
c
loudWatchExecutor
{
ec2Svc
:
mockedEc2
{
Resp
:
ec2
.
DescribeInstancesOutput
{
ec2Svc
:
mockedEc2
{
Resp
:
ec2
.
DescribeInstancesOutput
{
Reservations
:
[]
*
ec2
.
Reservation
{
Reservations
:
[]
*
ec2
.
Reservation
{
{
{
...
@@ -217,7 +223,7 @@ func TestCloudWatchMetrics(t *testing.T) {
...
@@ -217,7 +223,7 @@ func TestCloudWatchMetrics(t *testing.T) {
})
})
t
.
Run
(
"When calling handleGetResourceArns"
,
func
(
t
*
testing
.
T
)
{
t
.
Run
(
"When calling handleGetResourceArns"
,
func
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{
executor
:=
&
c
loudWatchExecutor
{
rgtaSvc
:
mockedRGTA
{
rgtaSvc
:
mockedRGTA
{
Resp
:
resourcegroupstaggingapi
.
GetResourcesOutput
{
Resp
:
resourcegroupstaggingapi
.
GetResourcesOutput
{
ResourceTagMappingList
:
[]
*
resourcegroupstaggingapi
.
ResourceTagMapping
{
ResourceTagMappingList
:
[]
*
resourcegroupstaggingapi
.
ResourceTagMapping
{
...
...
pkg/tsdb/cloudwatch/query_transformer.go
View file @
c863fd3d
...
@@ -13,7 +13,7 @@ import (
...
@@ -13,7 +13,7 @@ import (
// has more than one statistic defined, one cloudwatchQuery will be created for each statistic.
// has more than one statistic defined, one cloudwatchQuery will be created for each statistic.
// If the query doesn't have an Id defined by the user, we'll give it an with format `query[RefId]`. In the case
// If the query doesn't have an Id defined by the user, we'll give it an with format `query[RefId]`. In the case
// the incoming query had more than one stat, it will ge an id like `query[RefId]_[StatName]`, eg queryC_Average
// the incoming query had more than one stat, it will ge an id like `query[RefId]_[StatName]`, eg queryC_Average
func
(
e
*
C
loudWatchExecutor
)
transformRequestQueriesToCloudWatchQueries
(
requestQueries
[]
*
requestQuery
)
(
map
[
string
]
*
cloudWatchQuery
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
transformRequestQueriesToCloudWatchQueries
(
requestQueries
[]
*
requestQuery
)
(
map
[
string
]
*
cloudWatchQuery
,
error
)
{
cloudwatchQueries
:=
make
(
map
[
string
]
*
cloudWatchQuery
)
cloudwatchQueries
:=
make
(
map
[
string
]
*
cloudWatchQuery
)
for
_
,
requestQuery
:=
range
requestQueries
{
for
_
,
requestQuery
:=
range
requestQueries
{
for
_
,
stat
:=
range
requestQuery
.
Statistics
{
for
_
,
stat
:=
range
requestQuery
.
Statistics
{
...
@@ -50,7 +50,7 @@ func (e *CloudWatchExecutor) transformRequestQueriesToCloudWatchQueries(requestQ
...
@@ -50,7 +50,7 @@ func (e *CloudWatchExecutor) transformRequestQueriesToCloudWatchQueries(requestQ
return
cloudwatchQueries
,
nil
return
cloudwatchQueries
,
nil
}
}
func
(
e
*
C
loudWatchExecutor
)
transformQueryResponseToQueryResult
(
cloudwatchResponses
[]
*
cloudwatchResponse
)
map
[
string
]
*
tsdb
.
QueryResult
{
func
(
e
*
c
loudWatchExecutor
)
transformQueryResponseToQueryResult
(
cloudwatchResponses
[]
*
cloudwatchResponse
)
map
[
string
]
*
tsdb
.
QueryResult
{
responsesByRefID
:=
make
(
map
[
string
][]
*
cloudwatchResponse
)
responsesByRefID
:=
make
(
map
[
string
][]
*
cloudwatchResponse
)
for
_
,
res
:=
range
cloudwatchResponses
{
for
_
,
res
:=
range
cloudwatchResponses
{
responsesByRefID
[
res
.
RefId
]
=
append
(
responsesByRefID
[
res
.
RefId
],
res
)
responsesByRefID
[
res
.
RefId
]
=
append
(
responsesByRefID
[
res
.
RefId
],
res
)
...
...
pkg/tsdb/cloudwatch/query_transformer_test.go
View file @
c863fd3d
...
@@ -10,7 +10,7 @@ import (
...
@@ -10,7 +10,7 @@ import (
func
TestQueryTransformer
(
t
*
testing
.
T
)
{
func
TestQueryTransformer
(
t
*
testing
.
T
)
{
Convey
(
"TestQueryTransformer"
,
t
,
func
()
{
Convey
(
"TestQueryTransformer"
,
t
,
func
()
{
Convey
(
"when transforming queries"
,
func
()
{
Convey
(
"when transforming queries"
,
func
()
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
Convey
(
"one cloudwatchQuery is generated when its request query has one stat"
,
func
()
{
Convey
(
"one cloudwatchQuery is generated when its request query has one stat"
,
func
()
{
requestQueries
:=
[]
*
requestQuery
{
requestQueries
:=
[]
*
requestQuery
{
{
{
...
...
pkg/tsdb/cloudwatch/request_parser.go
View file @
c863fd3d
...
@@ -15,7 +15,7 @@ import (
...
@@ -15,7 +15,7 @@ import (
)
)
// Parses the json queries and returns a requestQuery. The requestQuery has a 1 to 1 mapping to a query editor row
// Parses the json queries and returns a requestQuery. The requestQuery has a 1 to 1 mapping to a query editor row
func
(
e
*
C
loudWatchExecutor
)
parseQueries
(
queryContext
*
tsdb
.
TsdbQuery
,
startTime
time
.
Time
,
endTime
time
.
Time
)
(
map
[
string
][]
*
requestQuery
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
parseQueries
(
queryContext
*
tsdb
.
TsdbQuery
,
startTime
time
.
Time
,
endTime
time
.
Time
)
(
map
[
string
][]
*
requestQuery
,
error
)
{
requestQueries
:=
make
(
map
[
string
][]
*
requestQuery
)
requestQueries
:=
make
(
map
[
string
][]
*
requestQuery
)
for
i
,
model
:=
range
queryContext
.
Queries
{
for
i
,
model
:=
range
queryContext
.
Queries
{
queryType
:=
model
.
Model
.
Get
(
"type"
)
.
MustString
()
queryType
:=
model
.
Model
.
Get
(
"type"
)
.
MustString
()
...
...
pkg/tsdb/cloudwatch/response_parser.go
View file @
c863fd3d
...
@@ -12,7 +12,7 @@ import (
...
@@ -12,7 +12,7 @@ import (
"github.com/grafana/grafana/pkg/tsdb"
"github.com/grafana/grafana/pkg/tsdb"
)
)
func
(
e
*
C
loudWatchExecutor
)
parseResponse
(
metricDataOutputs
[]
*
cloudwatch
.
GetMetricDataOutput
,
queries
map
[
string
]
*
cloudWatchQuery
)
([]
*
cloudwatchResponse
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
parseResponse
(
metricDataOutputs
[]
*
cloudwatch
.
GetMetricDataOutput
,
queries
map
[
string
]
*
cloudWatchQuery
)
([]
*
cloudwatchResponse
,
error
)
{
mdr
:=
make
(
map
[
string
]
map
[
string
]
*
cloudwatch
.
MetricDataResult
)
mdr
:=
make
(
map
[
string
]
map
[
string
]
*
cloudwatch
.
MetricDataResult
)
for
_
,
mdo
:=
range
metricDataOutputs
{
for
_
,
mdo
:=
range
metricDataOutputs
{
requestExceededMaxLimit
:=
false
requestExceededMaxLimit
:=
false
...
...
pkg/tsdb/cloudwatch/time_series_query.go
View file @
c863fd3d
...
@@ -9,7 +9,7 @@ import (
...
@@ -9,7 +9,7 @@ import (
"golang.org/x/sync/errgroup"
"golang.org/x/sync/errgroup"
)
)
func
(
e
*
C
loudWatchExecutor
)
executeTimeSeriesQuery
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
func
(
e
*
c
loudWatchExecutor
)
executeTimeSeriesQuery
(
ctx
context
.
Context
,
queryContext
*
tsdb
.
TsdbQuery
)
(
*
tsdb
.
Response
,
error
)
{
startTime
,
err
:=
queryContext
.
TimeRange
.
ParseFrom
()
startTime
,
err
:=
queryContext
.
TimeRange
.
ParseFrom
()
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
err
return
nil
,
err
...
@@ -50,7 +50,7 @@ func (e *CloudWatchExecutor) executeTimeSeriesQuery(ctx context.Context, queryCo
...
@@ -50,7 +50,7 @@ func (e *CloudWatchExecutor) executeTimeSeriesQuery(ctx context.Context, queryCo
}
}
}()
}()
client
,
err
:=
e
.
getClient
(
region
)
client
,
err
:=
e
.
getC
WC
lient
(
region
)
if
err
!=
nil
{
if
err
!=
nil
{
return
err
return
err
}
}
...
...
pkg/tsdb/cloudwatch/time_series_query_test.go
View file @
c863fd3d
...
@@ -9,7 +9,7 @@ import (
...
@@ -9,7 +9,7 @@ import (
)
)
func
TestTimeSeriesQuery
(
t
*
testing
.
T
)
{
func
TestTimeSeriesQuery
(
t
*
testing
.
T
)
{
executor
:=
&
C
loudWatchExecutor
{}
executor
:=
&
c
loudWatchExecutor
{}
t
.
Run
(
"End time before start time should result in error"
,
func
(
t
*
testing
.
T
)
{
t
.
Run
(
"End time before start time should result in error"
,
func
(
t
*
testing
.
T
)
{
_
,
err
:=
executor
.
executeTimeSeriesQuery
(
context
.
TODO
(),
&
tsdb
.
TsdbQuery
{
TimeRange
:
tsdb
.
NewTimeRange
(
"now-1h"
,
"now-2h"
)})
_
,
err
:=
executor
.
executeTimeSeriesQuery
(
context
.
TODO
(),
&
tsdb
.
TsdbQuery
{
TimeRange
:
tsdb
.
NewTimeRange
(
"now-1h"
,
"now-2h"
)})
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment