@@ -81,11 +81,8 @@ internal class WorkerContextImpl(
8181 /* * The service for accessing secrets. */
8282 private val secretService : SecretService
8383) : WorkerContext {
84- /* * The object for accessing secrets. */
85- private val secretStorage by lazy { SecretStorage .createStorage(configManager) }
86-
8784 /* * A cache for the secrets that have already been loaded. */
88- private val secretsCache = ConcurrentHashMap <String , Deferred <String >>()
85+ private val secretsCache = ConcurrentHashMap <Secret , Deferred <String >>()
8986
9087 /* * A cache for the configuration secrets that have already been loaded. */
9188 private val configSecretsCache = ConcurrentHashMap <String , Deferred <String >>()
@@ -130,10 +127,10 @@ internal class WorkerContextImpl(
130127 private val currentContext by lazy { ortRun.resolvedJobConfigContext?.let (::Context ) }
131128
132129 override suspend fun resolveSecret (secret : Secret ): String =
133- singleTransform(secret, secretsCache, this ::resolveSecret, ::extractSecretKey)
130+ singleTransform(secret, secretsCache, ::resolveSecretValue) { it }
134131
135132 override suspend fun resolveSecrets (vararg secrets : Secret ): Map <Secret , String > =
136- parallelTransform(secrets.toList(), secretsCache, this ::resolveSecret, ::extractSecretKey)
133+ parallelTransform(secrets.toList(), secretsCache, ::resolveSecretValue) { it }
137134
138135 override suspend fun resolvePluginConfigSecrets (
139136 config : Map <String , ResolvablePluginConfig >?
@@ -263,10 +260,10 @@ internal class WorkerContextImpl(
263260 }
264261
265262 /* *
266- * Resolve the secret with the given [path ] using the [SecretStorage] owned by this instance .
263+ * Resolve the value of the provided [secret ] using the [secretService] .
267264 */
268- private fun resolveSecret ( path : String ): String =
269- secretStorage.getSecret( Path (path)) .value
265+ private fun resolveSecretValue ( secret : Secret ): String =
266+ secretService.getSecretValue(secret)? .value ? : error( " Could not resolve secret at path ' ${secret.path} ' " )
270267
271268 /* *
272269 * Resolve the given [secret] from the configuration manager.
@@ -287,11 +284,6 @@ internal class WorkerContextImpl(
287284 }
288285}
289286
290- /* *
291- * A key extraction function for [Secret]s. As key for the given [secret] its path is used.
292- */
293- private fun extractSecretKey (secret : Secret ): String = secret.path
294-
295287/* *
296288 * Return a key extraction function for configuration files that are downloaded to the given [directory] and optionally
297289 * renamed to the given [targetName]. The resulting function ensures that all relevant components are reflected in
0 commit comments