def sageMakerClient_=()

in scala-spark-sdk/src/main/scala/software/amazon/sagemaker/featurestore/sparksdk/helpers/ClientFactory.scala [59:92]

• 24 lines of code
• 2 McCabe index (conditional complexity)

  def sageMakerClient_=(client: SageMakerClient): Unit = _sageMakerClient = Option(client)
  @VisibleForTesting
  def sageMakerFeatureStoreRuntimeClientBuilder_=(
      runtimeClientBuilder: SageMakerFeatureStoreRuntimeClientBuilder
  ): Unit = _sageMakerFeatureStoreRuntimeClientBuilder = Option(runtimeClientBuilder)
  @VisibleForTesting
  def assumeRoleArn_=(roleArn: String): Unit = _assumeRoleArn = Option(roleArn)
  @VisibleForTesting
  def region_=(region: String): Unit = _region = Option(region)
  @VisibleForTesting
  def stsAssumeRoleCredentialsProvider_=(credentialsProvider: StsAssumeRoleCredentialsProvider): Unit =
    _stsAssumeRoleCredentialsProvider = Option(credentialsProvider)
  @VisibleForTesting
  def skipInitialization_=(skipInitialization: Boolean): Unit = _skipInitialization = skipInitialization

  /** Initialize the client factory
   *
   *  @param roleArn
   *    Initialize the client factory with provided role arn
   *  @param region
   *    Initialize the client factory with provided region, region should be the same as what provided in feature group
   *    arn
   */
  def initialize(region: String, roleArn: String = null): Unit = {
    if (skipInitialization) {
      return
    }

    this.assumeRoleArn = roleArn
    this.region = region
    this.stsAssumeRoleCredentialsProvider = getStsAssumeRoleCredentialsProvider
    this.sageMakerClient = getDefaultSageMakerClient
    this.sageMakerFeatureStoreRuntimeClientBuilder = getDefaultFeatureStoreRuntimeClientBuilder
  }