File: composite_io.rb

package info (click to toggle)
ruby-google-apis-core 0.11.3-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 260 kB
  • sloc: ruby: 1,964; makefile: 4
file content (97 lines) | stat: -rw-r--r-- 2,728 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# Copyright 2020 Google LLC
#
# Licensed 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.
# Copyright 2015 Google Inc.
#
# Licensed 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.

require 'google/apis/core/http_command'
require 'google/apis/core/upload'
require 'google/apis/core/download'
require 'addressable/uri'
require 'securerandom'
module Google
  module Apis
    module Core
      class CompositeIO
        def initialize(*ios)
          @ios = ios.flatten
          @pos = 0
          @index = 0
          @sizes = @ios.map(&:size)
        end

        def read(length = nil, buf = nil)
          buf = buf ? buf.replace('') : ''

          begin
            io = @ios[@index]
            break if io.nil?
            result = io.read(length)
            if result
              buf << result
              if length
                length -= result.length
                break if length == 0
              end
            end
            @index += 1
          end while @index < @ios.length
          buf.length > 0 ? buf : nil
        end

        def size
          @sizes.reduce(:+)
        end

        alias_method :length, :size

        def pos
          @pos
        end

        def pos=(pos)
          fail ArgumentError, "Position can not be negative" if pos < 0
          @pos = pos
          new_index = nil
          @ios.each_with_index do |io,idx|
            size = io.size
            if pos <= size
              new_index ||= idx
              io.pos = pos
              pos = 0
            else
              io.pos = size
              pos -= size
            end
          end
          @index = new_index unless new_index.nil?
        end

        def rewind
          self.pos = 0
        end
      end
    end
  end
end