Skip to content

Commit d4533c7

Browse files
authored
Add event for the last call in polling (Azure#948)
1 parent 6bdc4a5 commit d4533c7

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

powershell/llcsharp/operation/method.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -541,12 +541,12 @@ if( ${response.value}.StatusCode == ${System.Net.HttpStatusCode.OK})
541541
case 'azure-async-operation':
542542
case 'location':
543543
// perform a final GET on the specified final URI.
544-
yield $this.finalGet(finalUri, reqParameter, response);
544+
yield $this.finalGet(eventListener, finalUri, reqParameter, response);
545545
break;
546546

547547
default:
548548
yield If(`!string.IsNullOrWhiteSpace(${finalUri})`, function* () {
549-
yield $this.finalGet(finalUri, reqParameter, response);
549+
yield $this.finalGet(eventListener, finalUri, reqParameter, response);
550550
});
551551
break;
552552
}
@@ -572,7 +572,7 @@ if( ${response.value}.StatusCode == ${System.Net.HttpStatusCode.OK})
572572
this.opMethod.emitCall($this.returnNull);
573573
}
574574

575-
private * finalGet(finalLocation: ExpressionOrLiteral, reqParameter: Variable, response: Variable) {
575+
private * finalGet(eventListener: EventListener, finalLocation: ExpressionOrLiteral, reqParameter: Variable, response: Variable) {
576576
yield '// create a new request with the final uri';
577577
yield reqParameter.assign(`${valueOf(reqParameter)}.CloneAndDispose(${System.Uri.new(finalLocation)}, ${ClientRuntime.Method.Get})`);
578578

@@ -583,6 +583,7 @@ if( ${response.value}.StatusCode == ${System.Net.HttpStatusCode.OK})
583583
yield EOL;
584584
yield '// make the final call';
585585
yield response.assign(`await ${this.opMethod.senderParameter}.SendAsync(${valueOf(reqParameter)}, ${this.opMethod.contextParameter})`);
586+
yield eventListener.signal(ClientRuntime.Events.Polling, response.value);
586587

587588
// make sure we're not polling anymore.
588589
yield 'break;';

0 commit comments

Comments
 (0)