evalphobia /
aws-sdk-go-v2-wrapper
| Total Lines | 58 |
| Duplicated Lines | 0 % |
| Changes | 0 | ||
| 1 | package sqs |
||
| 2 | |||
| 3 | import ( |
||
| 4 | SDK "github.com/aws/aws-sdk-go-v2/service/sqs" |
||
| 5 | |||
| 6 | "github.com/evalphobia/aws-sdk-go-v2-wrapper/config" |
||
| 7 | "github.com/evalphobia/aws-sdk-go-v2-wrapper/errors" |
||
| 8 | "github.com/evalphobia/aws-sdk-go-v2-wrapper/log" |
||
| 9 | ) |
||
| 10 | |||
| 11 | const ( |
||
| 12 | serviceName = "SQS" |
||
| 13 | ) |
||
| 14 | |||
| 15 | // SQS has SQS client. |
||
| 16 | type SQS struct { |
||
| 17 | client *SDK.Client |
||
| 18 | |||
| 19 | logger log.Logger |
||
| 20 | errWrap func(errors.ErrorData) error |
||
| 21 | } |
||
| 22 | |||
| 23 | // New returns initialized *SQS. |
||
| 24 | func New(conf config.Config) (*SQS, error) { |
||
| 25 | c, err := conf.AWSConfig() |
||
| 26 | if err != nil { |
||
| 27 | return nil, err |
||
| 28 | } |
||
| 29 | if conf.Endpoints.HasSQS() { |
||
| 30 | c.EndpointResolver = conf.Endpoints.GetSQS() |
||
| 31 | } |
||
| 32 | |||
| 33 | svc := &SQS{ |
||
| 34 | client: SDK.New(c), |
||
| 35 | logger: conf.GetLogger(), |
||
| 36 | errWrap: conf.GetErrWrap(), |
||
| 37 | } |
||
| 38 | return svc, nil |
||
| 39 | } |
||
| 40 | |||
| 41 | // GetClient gets original SDK client. |
||
| 42 | func (svc *SQS) GetClient() *SDK.Client { |
||
| 43 | return svc.client |
||
| 44 | } |
||
| 45 | |||
| 46 | // SetLogger sets logger. |
||
| 47 | func (svc *SQS) SetLogger(logger log.Logger) { |
||
| 48 | svc.logger = logger |
||
| 49 | } |
||
| 50 | |||
| 51 | // Infof logging information. |
||
| 52 | func (svc *SQS) Infof(format string, v ...interface{}) { |
||
| 53 | svc.logger.Infof(serviceName, format, v...) |
||
| 54 | } |
||
| 55 | |||
| 56 | // Errorf logging error information. |
||
| 57 | func (svc *SQS) Errorf(format string, v ...interface{}) { |
||
| 58 | svc.logger.Errorf(serviceName, format, v...) |
||
|
introduced
by
|
|||
| 59 | } |
||
| 60 |
Loading history...