]>
git.proxmox.com Git - ceph.git/blob - ceph/src/arrow/c_glib/test/helper/flight-server.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
9 # http://www.apache.org/licenses/LICENSE-2.0
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
18 require_relative
"flight-info-generator"
21 class FlightServer
< ArrowFlight
::Server
25 def virtual_do_list_flights(context
, criteria
)
26 generator
= FlightInfoGenerator
.new
30 def virtual_do_do_get(context
, ticket
)
31 generator
= FlightInfoGenerator
.new
32 unless ticket
== generator
.page_view_ticket
33 raise Arrow
::Error::Invalid.new("invalid ticket")
35 table
= generator
.page_view_table
36 reader
= Arrow
::TableBatchReader.new(table
)
37 ArrowFlight
::RecordBatchStream.new(reader
)