@@ -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