ceph/src/arrow/ruby/red-arrow/lib/arrow/time32-array-builder.rb

   1 # Licensed to the Apache Software Foundation (ASF) under one
   2 # or more contributor license agreements.  See the NOTICE file
   3 # distributed with this work for additional information
   4 # regarding copyright ownership.  The ASF licenses this file
   5 # to you under the Apache License, Version 2.0 (the
   6 # "License"); you may not use this file except in compliance
   7 # with the License.  You may obtain a copy of the License at
   8 #
   9 #   http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing,
  12 # software distributed under the License is distributed on an
  13 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  14 # KIND, either express or implied.  See the License for the
  15 # specific language governing permissions and limitations
  16 # under the License.
  17
  18 module Arrow
  19   class Time32ArrayBuilder
  20     class << self
  21       def build(unit_or_data_type, values)
  22         builder = new(unit_or_data_type)
  23         builder.build(values)
  24       end
  25     end
  26
  27     alias_method :initialize_raw, :initialize
  28     def initialize(unit_or_data_type)
  29       case unit_or_data_type
  30       when DataType
  31         data_type = unit_or_data_type
  32       else
  33         unit = unit_or_data_type
  34         data_type = Time32DataType.new(unit)
  35       end
  36       initialize_raw(data_type)
  37     end
  38
  39     def unit
  40       @unit ||= value_data_type.unit
  41     end
  42
  43     private
  44     def convert_to_arrow_value(value)
  45       return value unless value.is_a?(Time)
  46       value.cast(unit).value
  47     end
  48   end
  49 end