def close()

in core/src/main/scala/org/apache/spark/sql/eventhubs/EventHubsWriteTask.scala [69:93]

• 22 lines of code
• 4 McCabe index (conditional complexity)

  def close(): Unit = {
    log.info(s"close is called. ${EventHubsUtils.getTaskContextSlim}")

    var success = false
    if (sender != null) {
      try {
        sender.close()
        success = true
      } catch {
        case e: Exception =>
          log.warn(s"an error occurred. eventhub name = ${ehConf.name}, error = ${e.getMessage}")
          throw e
      }
      sender = null
    }

    metricPlugin.foreach(
      _.onSendMetric(EventHubsUtils.getTaskContextSlim,
                     ehConf.name,
                     totalMessageCount,
                     totalMessageSizeInBytes,
                     System.currentTimeMillis() - writerOpenTime,
                     isSuccess = success))

  }