Class: IOStreams::Encode::Writer

Inherits:
Writer
  • Object
show all
Defined in:
lib/io_streams/encode/writer.rb

Instance Attribute Summary collapse

Attributes inherited from Writer

#output_stream

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from Writer

file, open

Constructor Details

#initialize(output_stream, encoding: "UTF-8", cleaner: nil, replace: nil) ⇒ Writer

A delimited stream writer that will write to the supplied output stream Written data is encoded prior to writing.

Parameters

output_stream
  The output stream that implements #write

encoding: [String|Encoding]
  Encode returned data with this encoding.
  'US-ASCII':   Original 7 bit ASCII Format
  'ASCII-8BIT': 8-bit ASCII Format
  'UTF-8':      UTF-8 Format
  Etc.
  Default: 'UTF-8'

replace: [String]
  The character to replace with when a character cannot be converted to the target encoding.
  nil: Don't replace any invalid characters. Encoding::UndefinedConversionError is raised.
  Default: nil

cleaner: [nil|symbol|Proc]
  Cleanse data read from the input stream.
  nil:           No cleansing
  :printable Cleanse all non-printable characters except \r and \n
  Proc/lambda    Proc to call after every read to cleanse the data
  Default: nil



37
38
39
40
41
42
43
44
 
# File 'lib/io_streams/encode/writer.rb', line 37

def initialize(output_stream, encoding: "UTF-8", cleaner: nil, replace: nil)
  super(output_stream)

  @cleaner          = ::IOStreams::Encode::Reader.send(:extract_cleaner, cleaner)
  @encoding         = encoding.nil? || encoding.is_a?(Encoding) ? encoding : Encoding.find(encoding)
  @encoding_options = replace.nil? ? {} : {invalid: :replace, undef: :replace, replace: replace}
  @replace          = replace
end

Instance Attribute Details

#cleaner ⇒ Object (readonly)

Returns the value of attribute cleaner.



4
5
6
 
# File 'lib/io_streams/encode/writer.rb', line 4

def cleaner
  @cleaner
end

#encoding ⇒ Object (readonly)

Returns the value of attribute encoding.



4
5
6
 
# File 'lib/io_streams/encode/writer.rb', line 4

def encoding
  @encoding
end

Class Method Details

.stream(input_stream, original_file_name: nil, **args) {|new(input_stream, **args)| ... } ⇒ Object

Write a line at a time to a file or stream

Yields:

  • (new(input_stream, **args)) 


7
8
9
 
# File 'lib/io_streams/encode/writer.rb', line 7

def self.stream(input_stream, original_file_name: nil, **args)
  yield new(input_stream, **args)
end

Instance Method Details

#<<(record) ⇒ Object

Write a line to the output stream

Example:

IOStreams.writer('a.txt', encoding: 'UTF-8') do |stream|
  stream << 'first line' << 'second line'
end



52
53
54
55
 
# File 'lib/io_streams/encode/writer.rb', line 52

def <<(record)
  write(record)
  self
end

#write(data) ⇒ Object

Write a line to the output stream followed by the delimiter. Returns [Integer] the number of bytes written.

Example:

IOStreams.writer('a.txt', encoding: 'UTF-8') do |stream|
  count = stream.write('first line')
  puts "Wrote #{count} bytes to the output file, including the delimiter"
end



65
66
67
68
69
70
71
72
 
# File 'lib/io_streams/encode/writer.rb', line 65

def write(data)
  return 0 if data.nil?

  data  = data.to_s
  block = data.encoding == @encoding ? data : data.encode(@encoding, **@encoding_options)
  block = @cleaner.call(block, @replace) if @cleaner
  @output_stream.write(block)
end