@@ -7,7 +7,6 @@ import com.intellij.openapi.Disposable
77import com.intellij.openapi.application.runInEdt
88import com.intellij.serviceContainer.AlreadyDisposedException
99import com.intellij.util.io.HttpRequests
10- import kotlinx.coroutines.Dispatchers
1110import kotlinx.coroutines.delay
1211import kotlinx.coroutines.withContext
1312import org.apache.commons.codec.digest.DigestUtils
@@ -28,6 +27,7 @@ import software.aws.toolkits.core.utils.exists
2827import software.aws.toolkits.core.utils.getLogger
2928import software.aws.toolkits.core.utils.info
3029import software.aws.toolkits.core.utils.warn
30+ import software.aws.toolkits.jetbrains.core.coroutines.getCoroutineBgContext
3131import software.aws.toolkits.jetbrains.services.codemodernizer.client.GumbyClient
3232import software.aws.toolkits.jetbrains.services.codemodernizer.commands.CodeTransformMessageListener
3333import software.aws.toolkits.jetbrains.services.codemodernizer.model.CodeModernizerException
@@ -383,9 +383,11 @@ class CodeModernizerSession(
383383 * Adapted from [CodeWhispererCodeScanSession]
384384 */
385385 suspend fun uploadPayload (payload : File ): String {
386- val sha256checksum: String = Base64 .getEncoder().encodeToString(withContext(Dispatchers .IO ) {
387- DigestUtils .sha256(FileInputStream (payload))
388- })
386+ val sha256checksum: String = Base64 .getEncoder().encodeToString(
387+ withContext(getCoroutineBgContext()) {
388+ DigestUtils .sha256(FileInputStream (payload))
389+ }
390+ )
389391 if (isDisposed.get()) {
390392 throw AlreadyDisposedException (" Disposed when about to create upload URL" )
391393 }
@@ -401,15 +403,14 @@ class CodeModernizerSession(
401403 throw AlreadyDisposedException (" Disposed when about to upload project artifact to s3" )
402404 }
403405 val uploadStartTime = Instant .now()
404- waitUntil (
406+ waitUntil(
405407 exceptionsToIgnore = setOf (
406408 UnknownHostException ::class ,
407409 SocketTimeoutException ::class ,
408410 HttpRequests .HttpStatusException ::class ,
409411 ConnectException ::class
410412 )
411- )
412- {
413+ ) {
413414 clientAdaptor.uploadArtifactToS3(
414415 createUploadUrlResponse.uploadUrl(),
415416 payload,
0 commit comments