Не удалось выполнить UDF

есть таблица, в которой поле "A" содержит sql запрос. Необходимо добавить дополнительное поле «Б», которое содержало бы время, затрачиваемое на выполнение запроса из поля «А». Я написал UDF, и все работает хорошо, но при кэшировании результирующей таблицы или попытке записать окончательный фрейм данных в физическую таблицу я получил ошибку:

"Не удалось выполнить пользовательскую функцию ($anonfun$1: (string) => нить)"

. В чем может быть проблема? Пример:

val set_time = udf((query: String) => {
val start = new Timestamp(new Date().getDate)
val count = spark.sql(s"${query}").count
val time_query = (new Timestamp(new Date().getTime)).getTime() - start.getTime()
time_query.toString
})

Исходная таблица "источник":

+--------------------+
|          A         |
+--------------------+
|"Select * From ..." |
|"Select * From ..." |
|"Select * From ..." |
|"Select * From ..." |
|"Select * From ..." |
+--------------------+
val result = spark.sql("from source").
withColumn("B", set_time(col("A")))

result.show
+--------------------+------+
|          A         |   B  |
+--------------------+------+
|"Select * From ..." | 356  |
|"Select * From ..." | 642  |
|"Select * From ..." | 2745 |
|"Select * From ..." | 1324 |
|"Select * From ..." | 635  |
+--------------------+------+

Но:

//ERROR
result.write.mode("overwrite").saveAsTable("dbName.result")

//ERROR
val result_cache = result.persist
result_cache.show