Adding timeout to transactional batch in comsosdb statestore

Signed-off-by: Ryan Lettieri <ryanLettieri@microsoft.com>
This commit is contained in:
Ryan Lettieri 2022-09-15 16:57:46 -06:00
parent b2f4f1fe17
commit 097737f6cd
1 changed files with 3 additions and 1 deletions

View File

@ -400,7 +400,9 @@ func (c *StateStore) Multi(request *state.TransactionalStateRequest) error {
c.logger.Debugf("#operations=%d,partitionkey=%s", numOperations, partitionKey)
batchResponse, err := c.client.ExecuteTransactionalBatch(context.Background(), batch, nil)
ctx, cancel := context.WithTimeout(context.Background(), defaultTimeout)
batchResponse, err := c.client.ExecuteTransactionalBatch(ctx, batch, nil)
cancel()
if err != nil {
return err
}