I have a test
table in MySQL with id and name like below:
+----+-------+
| id | name |
+----+-------+
| 1 | Name1 |
+----+-------+
| 2 | Name
I believe all the steps above are unnecessary. Here's what you need to do:
Create a dataset A
like val A = spark.read.parquet("....")
Read the table to be updated, as dataframe B
. Make sure enable caching is enabled for dataframe B
. val B = spark.read.jdbc("mytable").cache
Force a count
on B
- this will force execution and cache the table depending on the chosen StorageLevel
- B.count
Now, you can do a transformation like val C = A.union(B)
And, then write C
back to the database like C.write.mode(SaveMode.Overwrite).jdbc("mytable")
The problem is in your code. Because you overwrite a table from which you're trying to read you effectively obliterate all data before Spark can actually access it.
Remember that Spark is lazy. When you create a Dataset
Spark fetches required metadata, but doesn't load the data. So there is no magic cache which will preserve original content. Data will be loaded when it is actually required. Here it is when you execute write
action and when you start writing there is no more data to be fetched.
What you need is something like this:
Dataset
.Apply required transformations and write data to an intermediate MySQL table.
TRUNCATE
the original input and INSERT INTO ... SELECT
from the intermediate table or DROP
the original table and RENAME
intermediate table.
Alternative, but less favorable approach, would be:
Dataset
.df.write.saveAsTable(...)
or equivalent)TRUNCATE
the original input.spark.table(...).write.jdbc(...)
)We cannot stress enough that using Spark cache
/ persist
is not the way to go. Even in with the conservative StorageLevel
(MEMORY_AND_DISK_2
/ MEMORY_AND_DISK_SER_2
) cached data can be lost (node failures), leading to silent correctness errors.
Reading and writing to same table.
cols_df = df_2.columns
broad_cast_var = spark_context.broadcast(df_2.collect())
df_3 = sqlContext.createDataFrame(broad_cast_var.value, cols_df)
Reading and writing to same table with some modification.
cols_df = df_2.columns
broad_cast_var = spark_context.broadcast(df_2.collect())
def update_x(x):
y = (x[0] + 311, *x[1:])
return y
rdd_2_1 = spark_context.parallelize(broad_cast_var.value).map(update_x)
df_3 = sqlContext.createDataFrame(rdd_2_1, cols_df)