dapr
/
components-contrib
mirror of https://github.com/dapr/components-contrib.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Address review comments 

Signed-off-by: Bernd Verst <github@bernd.dev>
This commit is contained in:
Bernd Verst 2023-02-24 16:23:00 -08:00
parent 358ebbf0bc
commit 4b01912b0f
1 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
bindings/azure/storagequeues/storagequeues.go
View File

@ -98,7 +98,7 @@ func (d *AzureQueueHelper) Init(ctx context.Context, meta bindings.Metadata) (*s
var clientErr error
queueServiceClient, clientErr = azqueue.NewServiceClient(m.GetQueueURL(azEnvSettings), credential, &options)
if clientErr != nil {
return nil, fmt.Errorf("cannot init storage queue client with Azure AD token: %w", err)
return nil, fmt.Errorf("cannot init storage queue client with Azure AD token: %w", clientErr)
}
}
@ -108,7 +108,7 @@ func (d *AzureQueueHelper) Init(ctx context.Context, meta bindings.Metadata) (*s
d.queueClient = queueServiceClient.NewQueueClient(m.QueueName)
createCtx, createCancel := context.WithTimeout(ctx, 2*time.Minute)
_, err = d.queueClient.Create(createCtx, &azqueue.CreateOptions{})
_, err = d.queueClient.Create(createCtx, nil)
createCancel()
if err != nil {
return nil, err
@ -121,6 +121,8 @@ func (d *AzureQueueHelper) Write(ctx context.Context, data []byte, ttl *time.Dur
var ttlSeconds *int32
if ttl != nil {
ttlSeconds = ptr.Of(int32(ttl.Seconds()))
} else {
ttlSeconds = ptr.Of(int32(defaultTTL.Seconds()))
}
s, err := strconv.Unquote(string(data))
@ -132,10 +134,6 @@ func (d *AzureQueueHelper) Write(ctx context.Context, data []byte, ttl *time.Dur
s = base64.StdEncoding.EncodeToString([]byte(s))
}
if ttl == nil {
ttlToUse := defaultTTL
ttl = &ttlToUse
}
_, err = d.queueClient.EnqueueMessage(ctx, s, &azqueue.EnqueueMessageOptions{
TimeToLive: ttlSeconds,
})
@ -161,8 +159,8 @@ func (d *AzureQueueHelper) Read(ctx context.Context, consumer *consumer) error {
}
mt := res.Messages[0].MessageText
var data []byte
data := []byte("")
if mt != nil {
if d.decodeBase64 {
decoded, decodeError := base64.StdEncoding.DecodeString(*mt)
if decodeError != nil {
@ -172,6 +170,7 @@ func (d *AzureQueueHelper) Read(ctx context.Context, consumer *consumer) error {
} else {
data = []byte(*mt)
}
}
_, err = consumer.callback(ctx, &bindings.ReadResponse{
Data: data,
@ -181,12 +180,15 @@ func (d *AzureQueueHelper) Read(ctx context.Context, consumer *consumer) error {
return err
}
_, err = d.queueClient.DeleteMessage(ctx, *res.Messages[0].MessageID, *res.Messages[0].PopReceipt, &azqueue.DeleteMessageOptions{})
if res.Messages[0].MessageID != nil && res.Messages[0].PopReceipt != nil {
_, err = d.queueClient.DeleteMessage(ctx, *res.Messages[0].MessageID, *res.Messages[0].PopReceipt, nil)
if err != nil {
return err
}
return nil
} else {
return fmt.Errorf("could not delete message from queue: message ID or pop receipt is nil")
}
}
func (d *AzureQueueHelper) Close() error {