import quincy 17.2.0

[ceph.git] / ceph / src / arrow / go / parquet / internal / testutils / utils.go

diff --git a/ceph/src/arrow/go/parquet/internal/testutils/utils.go b/ceph/src/arrow/go/parquet/internal/testutils/utils.go
new file mode 100644 (file)
index 0000000..b4415b1
--- /dev/null
+++ b/ceph/src/arrow/go/parquet/internal/testutils/utils.go
@@ -0,0 +1,42 @@
+// 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 testutils
+
+import (
+       "reflect"
+
+       "github.com/apache/arrow/go/v6/parquet"
+)
+
+var typeToParquetTypeMap = map[reflect.Type]parquet.Type{
+       reflect.TypeOf(true):                        parquet.Types.Boolean,
+       reflect.TypeOf(int32(0)):                    parquet.Types.Int32,
+       reflect.TypeOf(int64(0)):                    parquet.Types.Int64,
+       reflect.TypeOf(float32(0)):                  parquet.Types.Float,
+       reflect.TypeOf(float64(0)):                  parquet.Types.Double,
+       reflect.TypeOf(parquet.ByteArray{}):         parquet.Types.ByteArray,
+       reflect.TypeOf(parquet.Int96{}):             parquet.Types.Int96,
+       reflect.TypeOf(parquet.FixedLenByteArray{}): parquet.Types.FixedLenByteArray,
+}
+
+func TypeToParquetType(typ reflect.Type) parquet.Type {
+       ret, ok := typeToParquetTypeMap[typ]
+       if !ok {
+               panic("invalid type for parquet type")
+       }
+       return ret
+}