@@ -3,27 +3,26 @@ package dynamodb
3
3
import (
4
4
"context"
5
5
"errors"
6
- "time"
7
-
8
6
"github.com/aws/aws-sdk-go/service/dynamodb"
7
+ "time"
9
8
)
10
9
11
- type DynamoDBHealthChecker struct {
10
+ type HealthChecker struct {
12
11
db * dynamodb.DynamoDB
13
12
name string
14
13
timeout time.Duration
15
14
}
16
15
17
- func NewDynamoDBHealthChecker (db * dynamodb.DynamoDB , name string , timeouts ... time.Duration ) * DynamoDBHealthChecker {
16
+ func NewDynamoDBHealthChecker (db * dynamodb.DynamoDB , name string , timeouts ... time.Duration ) * HealthChecker {
18
17
var timeout time.Duration
19
18
if len (timeouts ) >= 1 {
20
19
timeout = timeouts [0 ]
21
20
} else {
22
21
timeout = 4 * time .Second
23
22
}
24
- return & DynamoDBHealthChecker {db , name , timeout }
23
+ return & HealthChecker {db , name , timeout }
25
24
}
26
- func NewHealthChecker (db * dynamodb.DynamoDB , options ... string ) * DynamoDBHealthChecker {
25
+ func NewHealthChecker (db * dynamodb.DynamoDB , options ... string ) * HealthChecker {
27
26
var name string
28
27
if len (options ) > 0 && len (options [0 ]) > 0 {
29
28
name = options [0 ]
@@ -33,11 +32,11 @@ func NewHealthChecker(db *dynamodb.DynamoDB, options ...string) *DynamoDBHealthC
33
32
return NewDynamoDBHealthChecker (db , name , 4 * time .Second )
34
33
}
35
34
36
- func (s * DynamoDBHealthChecker ) Name () string {
35
+ func (s * HealthChecker ) Name () string {
37
36
return s .name
38
37
}
39
38
40
- func (s * DynamoDBHealthChecker ) Check (ctx context.Context ) (map [string ]interface {}, error ) {
39
+ func (s * HealthChecker ) Check (ctx context.Context ) (map [string ]interface {}, error ) {
41
40
res := make (map [string ]interface {}, 0 )
42
41
if s .timeout > 0 {
43
42
ctx , _ = context .WithTimeout (ctx , s .timeout )
@@ -61,7 +60,7 @@ func (s *DynamoDBHealthChecker) Check(ctx context.Context) (map[string]interface
61
60
}
62
61
}
63
62
64
- func (s * DynamoDBHealthChecker ) Build (ctx context.Context , data map [string ]interface {}, err error ) map [string ]interface {} {
63
+ func (s * HealthChecker ) Build (ctx context.Context , data map [string ]interface {}, err error ) map [string ]interface {} {
65
64
if data == nil {
66
65
data = make (map [string ]interface {}, 0 )
67
66
}
0 commit comments