forked from zhurui/management
108 lines
2.5 KiB
JavaScript
108 lines
2.5 KiB
JavaScript
|
module.exports = FileWriter
|
||
|
|
|
||
|
|
var fs = require('graceful-fs')
|
||
|
|
var Writer = require('./writer.js')
|
||
|
|
var inherits = require('inherits')
|
||
|
|
var EOF = {}
|
||
|
|
|
||
|
|
inherits(FileWriter, Writer)
|
||
|
|
|
||
|
|
function FileWriter (props) {
|
||
|
|
var self = this
|
||
|
|
if (!(self instanceof FileWriter)) {
|
||
|
|
throw new Error('FileWriter must be called as constructor.')
|
||
|
|
}
|
||
|
|
|
||
|
|
// should already be established as a File type
|
||
|
|
if (props.type !== 'File' || !props.File) {
|
||
|
|
throw new Error('Non-file type ' + props.type)
|
||
|
|
}
|
||
|
|
|
||
|
|
self._buffer = []
|
||
|
|
self._bytesWritten = 0
|
||
|
|
|
||
|
|
Writer.call(this, props)
|
||
|
|
}
|
||
|
|
|
||
|
|
FileWriter.prototype._create = function () {
|
||
|
|
var self = this
|
||
|
|
if (self._stream) return
|
||
|
|
|
||
|
|
var so = {}
|
||
|
|
if (self.props.flags) so.flags = self.props.flags
|
||
|
|
so.mode = Writer.filemode
|
||
|
|
if (self._old && self._old.blksize) so.bufferSize = self._old.blksize
|
||
|
|
|
||
|
|
self._stream = fs.createWriteStream(self._path, so)
|
||
|
|
|
||
|
|
self._stream.on('open', function () {
|
||
|
|
// console.error("FW open", self._buffer, self._path)
|
||
|
|
self.ready = true
|
||
|
|
self._buffer.forEach(function (c) {
|
||
|
|
if (c === EOF) self._stream.end()
|
||
|
|
else self._stream.write(c)
|
||
|
|
})
|
||
|
|
self.emit('ready')
|
||
|
|
// give this a kick just in case it needs it.
|
||
|
|
self.emit('drain')
|
||
|
|
})
|
||
|
|
|
||
|
|
self._stream.on('error', function (er) { self.emit('error', er) })
|
||
|
|
|
||
|
|
self._stream.on('drain', function () { self.emit('drain') })
|
||
|
|
|
||
|
|
self._stream.on('close', function () {
|
||
|
|
// console.error('\n\nFW Stream Close', self._path, self.size)
|
||
|
|
self._finish()
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
FileWriter.prototype.write = function (c) {
|
||
|
|
var self = this
|
||
|
|
|
||
|
|
self._bytesWritten += c.length
|
||
|
|
|
||
|
|
if (!self.ready) {
|
||
|
|
if (!Buffer.isBuffer(c) && typeof c !== 'string') {
|
||
|
|
throw new Error('invalid write data')
|
||
|
|
}
|
||
|
|
self._buffer.push(c)
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
var ret = self._stream.write(c)
|
||
|
|
// console.error('\t-- fw wrote, _stream says', ret, self._stream._queue.length)
|
||
|
|
|
||
|
|
// allow 2 buffered writes, because otherwise there's just too
|
||
|
|
// much stop and go bs.
|
||
|
|
if (ret === false && self._stream._queue) {
|
||
|
|
return self._stream._queue.length <= 2
|
||
|
|
} else {
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
FileWriter.prototype.end = function (c) {
|
||
|
|
var self = this
|
||
|
|
|
||
|
|
if (c) self.write(c)
|
||
|
|
|
||
|
|
if (!self.ready) {
|
||
|
|
self._buffer.push(EOF)
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
return self._stream.end()
|
||
|
|
}
|
||
|
|
|
||
|
|
FileWriter.prototype._finish = function () {
|
||
|
|
var self = this
|
||
|
|
if (typeof self.size === 'number' && self._bytesWritten !== self.size) {
|
||
|
|
self.error(
|
||
|
|
'Did not get expected byte count.\n' +
|
||
|
|
'expect: ' + self.size + '\n' +
|
||
|
|
'actual: ' + self._bytesWritten)
|
||
|
|
}
|
||
|
|
Writer.prototype._finish.call(self)
|
||
|
|
}
|