ceph/src/arrow/ruby/red-arrow-dataset/lib/arrow-dataset/file-system-dataset-factory.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 ArrowDataset
  19   class FileSystemDatasetFactory
  20     alias_method :set_file_system_uri_raw, :set_file_system_uri
  21     def set_file_system_uri(uri)
  22       if uri.is_a?(URI)
  23         if uri.scheme.nil?
  24           uri = uri.dup
  25           absolute_path = File.expand_path(uri.path)
  26           if absolute_path.start_with?("/")
  27             uri.path = absolute_path
  28           else
  29             uri.path = "/#{absolute_path}"
  30           end
  31           uri.scheme = "file"
  32         end
  33         uri = uri.to_s
  34       end
  35       set_file_system_uri_raw(uri)
  36     end
  37     alias_method :file_system_uri=, :set_file_system_uri
  38   end
  39 end