Java - FAILED_WITH_ERROR when saving to snowflake

[orso]

I'm trying to move data from database A to B on Snowflake. There's no permission issue since using the Python package snowflake.connector  works 

Databricks runtime version: 12.2 LTS (includes Apache Spark 3.3.2, Scala 2.12)

Insert into database B fails with the code below. It works when trying to insert into database A

data.write \
.format('net.snowflake.spark.snowflake) \
.options({...}) \
.option('dbtable', 'MY_TABLE') \
.mode('Append') \
.save()​

Error:

java.sql.SQLException: Status of query associated with resultSet is FAILED_WITH_ERROR. Results not generated.

Py4JJavaError                             Traceback (most recent call last)

File <command-4230377357719442>:1

----> 1 snowflake_source_provider.save_table(a, schema, table_name, save_query_context)

File <command-2628261293911194>:45, in SnowflakeProvider.save_table(self, data, schema, table_name, query_context)

     44 def save_table(self, data, schema, table_name, query_context):

---> 45     data.write \

     46         .format(self.SNOWFLAKE_FORMAT) \

     47         .options(**self._get_connection_properties(schema)) \

     48         .option('dbtable', table_name) \

     49         .option('QUERY_TAG', self._get_query_tag(query_context)) \

     50         .mode('Append') \

     51         .save()

File /databricks/spark/python/pyspark/instrumentation_utils.py:48, in _wrap_function.<locals>.wrapper(*args, **kwargs)

     46 start = time.perf_counter()

     47 try:

---> 48     res = func(*args, **kwargs)

     49     logger.log_success(

     50         module_name, class_name, function_name, time.perf_counter() - start, signature

     51     )

     52     return res

File /databricks/spark/python/pyspark/sql/readwriter.py:1395, in DataFrameWriter.save(self, path, format, mode, partitionBy, **options)

   1393     self.format(format)

   1394 if path is None:

-> 1395     self._jwrite.save()

   1396 else:

   1397     self._jwrite.save(path)

File /databricks/spark/python/lib/py4j-0.10.9.5-src.zip/py4j/java_gateway.py:1321, in JavaMember.__call__(self, *args)

   1315 command = proto.CALL_COMMAND_NAME +\

   1316     self.command_header +\

   1317     args_command +\

   1318     proto.END_COMMAND_PART

   1320 answer = self.gateway_client.send_command(command)

-> 1321 return_value = get_return_value(

   1322     answer, self.gateway_client, self.target_id, self.name)

   1324 for temp_arg in temp_args:

   1325     temp_arg._detach()

File /databricks/spark/python/pyspark/errors/exceptions.py:228, in capture_sql_exception.<locals>.deco(*a, **kw)

    226 def deco(*a: Any, **kw: Any) -> Any:

    227     try:

--> 228         return f(*a, **kw)

    229     except Py4JJavaError as e:

    230         converted = convert_exception(e.java_exception)

File /databricks/spark/python/lib/py4j-0.10.9.5-src.zip/py4j/protocol.py:326, in get_return_value(answer, gateway_client, target_id, name)

    324 value = OUTPUT_CONVERTER[type](answer[2:], gateway_client)

    325 if answer[1] == REFERENCE_TYPE:

--> 326     raise Py4JJavaError(

    327         "An error occurred while calling {0}{1}{2}.\n".

    328         format(target_id, ".", name), value)

    329 else:

    330     raise Py4JError(

    331         "An error occurred while calling {0}{1}{2}. Trace:\n{3}\n".

    332         format(target_id, ".", name, value))

Py4JJavaError: An error occurred while calling o3841.save.

: java.sql.SQLException: Status of query associated with resultSet is FAILED_WITH_ERROR. Results not generated.

at net.snowflake.client.jdbc.SFAsyncResultSet.getRealResults(SFAsyncResultSet.java:128)

at net.snowflake.client.jdbc.SFAsyncResultSet.getMetaData(SFAsyncResultSet.java:265)

at net.snowflake.spark.snowflake.io.StageWriter$.executeCopyIntoTable(StageWriter.scala:603)

at net.snowflake.spark.snowflake.io.StageWriter$.writeToTableWithStagingTable(StageWriter.scala:471)

at net.snowflake.spark.snowflake.io.StageWriter$.writeToTable(StageWriter.scala:299)

at net.snowflake.spark.snowflake.io.StageWriter$.writeToStage(StageWriter.scala:238)

at net.snowflake.spark.snowflake.io.package$.writeRDD(package.scala:106)

at net.snowflake.spark.snowflake.SnowflakeWriter.save(SnowflakeWriter.scala:91)

at net.snowflake.spark.snowflake.DefaultSource.createRelation(DefaultSource.scala:156)

at org.apache.spark.sql.execution.datasources.SaveIntoDataSourceCommand.run(SaveIntoDataSourceCommand.scala:49)

at org.apache.spark.sql.execution.command.ExecutedCommandExec.$anonfun$sideEffectResult$1(commands.scala:82)

at com.databricks.spark.util.FrameProfiler$.record(FrameProfiler.scala:80)

at org.apache.spark.sql.execution.command.ExecutedCommandExec.sideEffectResult$lzycompute(commands.scala:80)

at org.apache.spark.sql.execution.command.ExecutedCommandExec.sideEffectResult(commands.scala:79)

at org.apache.spark.sql.execution.command.ExecutedCommandExec.executeCollect(commands.scala:91)

at org.apache.spark.sql.execution.QueryExecution$$anonfun$$nestedInanonfun$eagerlyExecuteCommands$1$1.$anonfun$applyOrElse$3(QueryExecution.scala:238)

at org.apache.spark.sql.catalyst.QueryPlanningTracker$.withTracker(QueryPlanningTracker.scala:165)

at org.apache.spark.sql.execution.QueryExecution$$anonfun$$nestedInanonfun$eagerlyExecuteCommands$1$1.$anonfun$applyOrElse$2(QueryExecution.scala:238)

at org.apache.spark.sql.execution.SQLExecution$.$anonfun$withCustomExecutionEnv$8(SQLExecution.scala:233)

at org.apache.spark.sql.execution.SQLExecution$.withSQLConfPropagated(SQLExecution.scala:417)

at org.apache.spark.sql.execution.SQLExecution$.$anonfun$withCustomExecutionEnv$1(SQLExecution.scala:178)

at org.apache.spark.sql.SparkSession.withActive(SparkSession.scala:1038)

at org.apache.spark.sql.execution.SQLExecution$.withCustomExecutionEnv(SQLExecution.scala:128)

at org.apache.spark.sql.execution.SQLExecution$.withNewExecutionId(SQLExecution.scala:367)

at org.apache.spark.sql.execution.QueryExecution$$anonfun$$nestedInanonfun$eagerlyExecuteCommands$1$1.$anonfun$applyOrElse$1(QueryExecution.scala:237)

at org.apache.spark.sql.execution.QueryExecution.org$apache$spark$sql$execution$QueryExecution$$withMVTagsIfNecessary(QueryExecution.scala:220)

at org.apache.spark.sql.execution.QueryExecution$$anonfun$$nestedInanonfun$eagerlyExecuteCommands$1$1.applyOrElse(QueryExecution.scala:233)

at org.apache.spark.sql.execution.QueryExecution$$anonfun$$nestedInanonfun$eagerlyExecuteCommands$1$1.applyOrElse(QueryExecution.scala:226)

at org.apache.spark.sql.catalyst.trees.TreeNode.$anonfun$transformDownWithPruning$1(TreeNode.scala:519)

at org.apache.spark.sql.catalyst.trees.CurrentOrigin$.withOrigin(TreeNode.scala:106)

at org.apache.spark.sql.catalyst.trees.TreeNode.transformDownWithPruning(TreeNode.scala:519)

at org.apache.spark.sql.catalyst.plans.logical.LogicalPlan.org$apache$spark$sql$catalyst$plans$logical$AnalysisHelper$$super$transformDownWithPruning(LogicalPlan.scala:32)

at org.apache.spark.sql.catalyst.plans.logical.AnalysisHelper.transformDownWithPruning(AnalysisHelper.scala:316)

at org.apache.spark.sql.catalyst.plans.logical.AnalysisHelper.transformDownWithPruning$(AnalysisHelper.scala:312)

at org.apache.spark.sql.catalyst.plans.logical.LogicalPlan.transformDownWithPruning(LogicalPlan.scala:32)

at org.apache.spark.sql.catalyst.plans.logical.LogicalPlan.transformDownWithPruning(LogicalPlan.scala:32)

at org.apache.spark.sql.catalyst.trees.TreeNode.transformDown(TreeNode.scala:495)

at org.apache.spark.sql.execution.QueryExecution.$anonfun$eagerlyExecuteCommands$1(QueryExecution.scala:226)

at org.apache.spark.sql.catalyst.plans.logical.AnalysisHelper$.allowInvokingTransformsInAnalyzer(AnalysisHelper.scala:372)

at org.apache.spark.sql.execution.QueryExecution.eagerlyExecuteCommands(QueryExecution.scala:226)

at org.apache.spark.sql.execution.QueryExecution.commandExecuted$lzycompute(QueryExecution.scala:180)

at org.apache.spark.sql.execution.QueryExecution.commandExecuted(QueryExecution.scala:171)

at org.apache.spark.sql.execution.QueryExecution.assertCommandExecuted(QueryExecution.scala:287)

at org.apache.spark.sql.DataFrameWriter.runCommand(DataFrameWriter.scala:964)

at org.apache.spark.sql.DataFrameWriter.saveToV1Source(DataFrameWriter.scala:429)

at org.apache.spark.sql.DataFrameWriter.saveInternal(DataFrameWriter.scala:396)

at org.apache.spark.sql.DataFrameWriter.save(DataFrameWriter.scala:258)

at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)

at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)

at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)

at java.lang.reflect.Method.invoke(Method.java:498)

at py4j.reflection.MethodInvoker.invoke(MethodInvoker.java:244)

at py4j.reflection.ReflectionEngine.invoke(ReflectionEngine.java:380)

at py4j.Gateway.invoke(Gateway.java:306)

at py4j.commands.AbstractCommand.invokeMethod(AbstractCommand.java:132)

at py4j.commands.CallCommand.execute(CallCommand.java:79)

at py4j.ClientServerConnection.waitForCommands(ClientServerConnection.java:195)

at py4j.ClientServerConnection.run(ClientServerConnection.java:115)

at java.lang.Thread.run(Thread.java:750)

Any idea is appreciated

[orso · Accepted solution]

Found the problem. 
The sub-roles didn't have grants to the warehouse.

I hope it will help someone one day 🙂