Skip to content

Commit b5988f9

Browse files
committed
Move files back to arrow package.
1 parent e3a4fc0 commit b5988f9

File tree

5 files changed

+17
-18
lines changed

5 files changed

+17
-18
lines changed

sql/core/src/main/scala/org/apache/spark/sql/execution/arrow/ArrowConverters.scala

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@ import org.apache.arrow.vector.util.ByteArrayReadableSeekableByteChannel
2828

2929
import org.apache.spark.TaskContext
3030
import org.apache.spark.sql.catalyst.InternalRow
31-
import org.apache.spark.sql.execution.vectorized.{ArrowUtils, ArrowWriter}
3231
import org.apache.spark.sql.types._
3332
import org.apache.spark.util.Utils
3433

sql/core/src/main/scala/org/apache/spark/sql/execution/vectorized/ArrowUtils.scala renamed to sql/core/src/main/scala/org/apache/spark/sql/execution/arrow/ArrowUtils.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
* limitations under the License.
1616
*/
1717

18-
package org.apache.spark.sql.execution.vectorized
18+
package org.apache.spark.sql.execution.arrow
1919

2020
import scala.collection.JavaConverters._
2121

sql/core/src/main/scala/org/apache/spark/sql/execution/vectorized/ArrowWriter.scala renamed to sql/core/src/main/scala/org/apache/spark/sql/execution/arrow/ArrowWriter.scala

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
* limitations under the License.
1616
*/
1717

18-
package org.apache.spark.sql.execution.vectorized
18+
package org.apache.spark.sql.execution.arrow
1919

2020
import scala.collection.JavaConverters._
2121

@@ -111,7 +111,7 @@ class ArrowWriter(
111111
}
112112
}
113113

114-
private[sql] abstract class ArrowFieldWriter {
114+
private[arrow] abstract class ArrowFieldWriter {
115115

116116
def valueVector: ValueVector
117117
def valueMutator: ValueVector.Mutator
@@ -150,7 +150,7 @@ private[sql] abstract class ArrowFieldWriter {
150150
}
151151
}
152152

153-
private[sql] class BooleanWriter(val valueVector: NullableBitVector) extends ArrowFieldWriter {
153+
private[arrow] class BooleanWriter(val valueVector: NullableBitVector) extends ArrowFieldWriter {
154154

155155
override def valueMutator: NullableBitVector#Mutator = valueVector.getMutator()
156156

@@ -167,7 +167,7 @@ private[sql] class BooleanWriter(val valueVector: NullableBitVector) extends Arr
167167
}
168168
}
169169

170-
private[sql] class ByteWriter(val valueVector: NullableTinyIntVector) extends ArrowFieldWriter {
170+
private[arrow] class ByteWriter(val valueVector: NullableTinyIntVector) extends ArrowFieldWriter {
171171

172172
override def valueMutator: NullableTinyIntVector#Mutator = valueVector.getMutator()
173173

@@ -184,7 +184,7 @@ private[sql] class ByteWriter(val valueVector: NullableTinyIntVector) extends Ar
184184
}
185185
}
186186

187-
private[sql] class ShortWriter(val valueVector: NullableSmallIntVector) extends ArrowFieldWriter {
187+
private[arrow] class ShortWriter(val valueVector: NullableSmallIntVector) extends ArrowFieldWriter {
188188

189189
override def valueMutator: NullableSmallIntVector#Mutator = valueVector.getMutator()
190190

@@ -201,7 +201,7 @@ private[sql] class ShortWriter(val valueVector: NullableSmallIntVector) extends
201201
}
202202
}
203203

204-
private[sql] class IntegerWriter(val valueVector: NullableIntVector) extends ArrowFieldWriter {
204+
private[arrow] class IntegerWriter(val valueVector: NullableIntVector) extends ArrowFieldWriter {
205205

206206
override def valueMutator: NullableIntVector#Mutator = valueVector.getMutator()
207207

@@ -218,7 +218,7 @@ private[sql] class IntegerWriter(val valueVector: NullableIntVector) extends Arr
218218
}
219219
}
220220

221-
private[sql] class LongWriter(val valueVector: NullableBigIntVector) extends ArrowFieldWriter {
221+
private[arrow] class LongWriter(val valueVector: NullableBigIntVector) extends ArrowFieldWriter {
222222

223223
override def valueMutator: NullableBigIntVector#Mutator = valueVector.getMutator()
224224

@@ -235,7 +235,7 @@ private[sql] class LongWriter(val valueVector: NullableBigIntVector) extends Arr
235235
}
236236
}
237237

238-
private[sql] class FloatWriter(val valueVector: NullableFloat4Vector) extends ArrowFieldWriter {
238+
private[arrow] class FloatWriter(val valueVector: NullableFloat4Vector) extends ArrowFieldWriter {
239239

240240
override def valueMutator: NullableFloat4Vector#Mutator = valueVector.getMutator()
241241

@@ -252,7 +252,7 @@ private[sql] class FloatWriter(val valueVector: NullableFloat4Vector) extends Ar
252252
}
253253
}
254254

255-
private[sql] class DoubleWriter(val valueVector: NullableFloat8Vector) extends ArrowFieldWriter {
255+
private[arrow] class DoubleWriter(val valueVector: NullableFloat8Vector) extends ArrowFieldWriter {
256256

257257
override def valueMutator: NullableFloat8Vector#Mutator = valueVector.getMutator()
258258

@@ -269,7 +269,7 @@ private[sql] class DoubleWriter(val valueVector: NullableFloat8Vector) extends A
269269
}
270270
}
271271

272-
private[sql] class DecimalWriter(
272+
private[arrow] class DecimalWriter(
273273
val valueVector: NullableDecimalVector,
274274
precision: Int,
275275
scale: Int) extends ArrowFieldWriter {
@@ -292,7 +292,7 @@ private[sql] class DecimalWriter(
292292
}
293293
}
294294

295-
private[sql] class StringWriter(val valueVector: NullableVarCharVector) extends ArrowFieldWriter {
295+
private[arrow] class StringWriter(val valueVector: NullableVarCharVector) extends ArrowFieldWriter {
296296

297297
override def valueMutator: NullableVarCharVector#Mutator = valueVector.getMutator()
298298

@@ -311,7 +311,8 @@ private[sql] class StringWriter(val valueVector: NullableVarCharVector) extends
311311
}
312312
}
313313

314-
private[sql] class BinaryWriter(val valueVector: NullableVarBinaryVector) extends ArrowFieldWriter {
314+
private[arrow] class BinaryWriter(
315+
val valueVector: NullableVarBinaryVector) extends ArrowFieldWriter {
315316

316317
override def valueMutator: NullableVarBinaryVector#Mutator = valueVector.getMutator()
317318

@@ -329,7 +330,7 @@ private[sql] class BinaryWriter(val valueVector: NullableVarBinaryVector) extend
329330
}
330331
}
331332

332-
private[sql] class ArrayWriter(
333+
private[arrow] class ArrayWriter(
333334
val valueVector: ListVector,
334335
val elementWriter: ArrowFieldWriter) extends ArrowFieldWriter {
335336

@@ -364,7 +365,7 @@ private[sql] class ArrayWriter(
364365
}
365366
}
366367

367-
private[sql] class StructWriter(
368+
private[arrow] class StructWriter(
368369
val valueVector: NullableMapVector,
369370
children: Array[ArrowFieldWriter]) extends ArrowFieldWriter {
370371

sql/core/src/test/scala/org/apache/spark/sql/execution/arrow/ArrowConvertersSuite.scala

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,6 @@ import org.scalatest.BeforeAndAfterAll
3131

3232
import org.apache.spark.SparkException
3333
import org.apache.spark.sql.{DataFrame, Row}
34-
import org.apache.spark.sql.execution.vectorized.ArrowUtils
3534
import org.apache.spark.sql.test.SharedSQLContext
3635
import org.apache.spark.sql.types.{BinaryType, Decimal, IntegerType, StructField, StructType}
3736
import org.apache.spark.util.Utils

sql/core/src/test/scala/org/apache/spark/sql/execution/vectorized/ArrowUtilsSuite.scala renamed to sql/core/src/test/scala/org/apache/spark/sql/execution/arrow/ArrowUtilsSuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
* limitations under the License.
1616
*/
1717

18-
package org.apache.spark.sql.execution.vectorized
18+
package org.apache.spark.sql.execution.arrow
1919

2020
import org.apache.spark.SparkFunSuite
2121
import org.apache.spark.sql.types._

0 commit comments

Comments
 (0)