Github user tdas commented on a diff in the pull request:
https://github.com/apache/spark/pull/11645#discussion_r56918468
--- Diff:
sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/state/HDFSBackedStateStoreProvider.scala
---
@@ -0,0 +1,597 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.spark.sql.execution.streaming.state
+
+import java.io.{DataInputStream, DataOutputStream}
+
+import scala.collection.JavaConverters._
+import scala.collection.mutable
+import scala.util.Random
+import scala.util.control.NonFatal
+
+import com.google.common.io.ByteStreams
+import org.apache.hadoop.conf.Configuration
+import org.apache.hadoop.fs.{FileStatus, Path}
+
+import org.apache.spark.SparkConf
+import org.apache.spark.internal.Logging
+import org.apache.spark.io.LZ4CompressionCodec
+import org.apache.spark.serializer.KryoSerializer
+import org.apache.spark.sql.catalyst.expressions.UnsafeRow
+import org.apache.spark.sql.types.StructType
+import org.apache.spark.util.Utils
+
+
+/**
+ * An implementation of [[StateStoreProvider]] and [[StateStore]] in which
all the data is backed
+ * by files in a HDFS-compatible file system. All updates to the store has
to be done in sets
+ * transactionally, and each set of updates increments the store's
version. These versions can
+ * be used to re-execute the updates (by retries in RDD operations) on the
correct version of
+ * the store, and regenerate the store version.
+ *
+ * Usage:
+ * To update the data in the state store, the following order of
operations are needed.
+ *
+ * - val store = StateStore.get(operatorId, partitionId, version) // to
get the right store
+ * - store.update(...)
+ * - store.remove(...)
+ * - store.commit() // commits all the updates to made with version
number
+ * - store.iterator() // key-value data after last commit as an iterator
+ * - store.updates() // updates made in the last as an iterator
+ *
+ * Fault-tolerance model:
+ * - Every set of updates is written to a delta file before committing.
+ * - The state store is responsible for managing, collapsing and cleaning
up of delta files.
+ * - Multiple attempts to commit the same version of updates may overwrite
each other.
+ * Consistency guarantees depend on whether multiple attempts have the
same updates and
+ * the overwrite semantics of underlying file system.
+ * - Background maintenance of files ensures that last versions of the
store is always recoverable
+ * to ensure re-executed RDD operations re-apply updates on the correct
past version of the
+ * store.
+ */
+private[state] class HDFSBackedStateStoreProvider(
+ val id: StateStoreId,
+ keySchema: StructType,
+ valueSchema: StructType,
+ sparkConf: SparkConf,
+ hadoopConf: Configuration
+ ) extends StateStoreProvider with Logging {
+
+ import HDFSBackedStateStoreProvider._
+
+ type MapType = java.util.HashMap[UnsafeRow, UnsafeRow]
+
+ /** Implementation of [[StateStore]] API which is backed by a
HDFS-compatible file system */
+ class HDFSBackedStateStore(val version: Long, mapToUpdate: MapType)
+ extends StateStore {
+
+ /** Trait and classes representing the internal state of the store */
+ trait STATE
+ case object UPDATING extends STATE
+ case object COMMITTED extends STATE
+ case object CANCELLED extends STATE
+
+ private val newVersion = version + 1
+ private val tempDeltaFile = new Path(baseDir,
s"temp-${Random.nextLong}")
+ private val tempDeltaFileStream =
compressStream(fs.create(tempDeltaFile, true))
+
+ // private val tempDeltaFileStream = fs.create(tempDeltaFile, true)
--- End diff --
dang! removed.
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at [email protected] or file a JIRA ticket
with INFRA.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
