ceph/src/arrow/c_glib/test/helper/flight-info-generator.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 require_relative "buildable"
  19 require_relative "data-type"
  20 require_relative "writable"
  21
  22 module Helper
  23   class FlightInfoGenerator
  24     include Buildable
  25     include DataType
  26     include Writable
  27
  28     def page_view_table
  29       build_table("count" => build_uint64_array([1, 2, 3]),
  30                   "private" => build_boolean_array([true, false, true]))
  31     end
  32
  33     def page_view_descriptor
  34       ArrowFlight::PathDescriptor.new(["page-view"])
  35     end
  36
  37     def page_view_ticket
  38       ArrowFlight::Ticket.new("page-view")
  39     end
  40
  41     def page_view_endpoints
  42       locations = [
  43         ArrowFlight::Location.new("grpc+tcp://127.0.0.1:10000"),
  44         ArrowFlight::Location.new("grpc+tcp://127.0.0.1:10001"),
  45       ]
  46       [
  47         ArrowFlight::Endpoint.new(page_view_ticket, locations),
  48       ]
  49     end
  50
  51     def page_view
  52       table = page_view_table
  53       descriptor = page_view_descriptor
  54       endpoints = page_view_endpoints
  55       output = Arrow::ResizableBuffer.new(0)
  56       write_table(table, output, type: :stream)
  57       ArrowFlight::Info.new(table.schema,
  58                             descriptor,
  59                             endpoints,
  60                             table.n_rows,
  61                             output.size)
  62     end
  63   end
  64 end