Skip to content

Commit

Permalink
streams: adds the ability to set the default encoding for writable st…
Browse files Browse the repository at this point in the history
…reams

Fixes nodejs#7159
  • Loading branch information
johnny-ray committed Sep 23, 2014
1 parent 6e08bb9 commit 398ac4f
Show file tree
Hide file tree
Showing 3 changed files with 67 additions and 0 deletions.
6 changes: 6 additions & 0 deletions doc/api/stream.markdown
Original file line number Diff line number Diff line change
Expand Up @@ -564,6 +564,12 @@ Buffered data will be flushed either at `.uncork()` or at `.end()` call.

Flush all data, buffered since `.cork()` call.

#### writable.setDefaultEncoding(encoding)

`encoding` {String} The new default encoding

Sets the default encoding for a writable stream

#### writable.end([chunk], [encoding], [callback])

* `chunk` {String | Buffer} Optional data to write
Expand Down
6 changes: 6 additions & 0 deletions lib/_stream_writable.js
Original file line number Diff line number Diff line change
Expand Up @@ -221,6 +221,12 @@ Writable.prototype.uncork = function() {
}
};

Writable.prototype.setDefaultEncoding = function(encoding) {
if (encoding) {
this._writableState.defaultEncoding = encoding;
}
};

function decodeChunk(state, chunk, encoding) {
if (!state.objectMode &&
state.decodeStrings !== false &&
Expand Down
55 changes: 55 additions & 0 deletions test/simple/test-stream-writable-change-default-encoding.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,55 @@
// Copyright Joyent, Inc. and other Node contributors.
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to permit
// persons to whom the Software is furnished to do so, subject to the
// following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
// USE OR OTHER DEALINGS IN THE SOFTWARE.

var common = require('../common');
var assert = require('assert');

var stream = require('stream');
var util = require('util');

function MyWritable(fn, options) {
stream.Writable.call(this, options);
this.fn = fn;
};

util.inherits(MyWritable, stream.Writable);

MyWritable.prototype._write = function (chunk, encoding, callback) {
this.fn(Buffer.isBuffer(chunk), typeof chunk, encoding);
callback();
};

;(function defaultCondingIsUtf8(){
var m = new MyWritable(function(isBuffer, type, enc) {
assert.equal(enc, 'utf8');
}, { decodeStrings: false });
m.write('foo');
m.end();
}());

;(function changeDefaultEncodingToAscii(){
var m = new MyWritable(function(isBuffer, type, enc) {
assert.equal(enc, 'ascii');
}, { decodeStrings: false });
m.setDefaultEncoding('ascii');
m.write('bar');
m.end();
}());

0 comments on commit 398ac4f

Please sign in to comment.