Browse Source

Don't use stat in fs.readFile[Sync]

Original patch c/o Evan Larkin <evan.larkin.iit@gmail.com>
v0.7.4-release
isaacs 14 years ago
committed by Ryan Dahl
parent
commit
4d0456f827
  1. 106
      lib/fs.js
  2. 9
      test/simple/test-fs-read-file-sync-hostname.js

106
lib/fs.js

@ -43,71 +43,83 @@ fs.Stats.prototype.isSocket = function () {
return this._checkModeProperty(constants.S_IFSOCK);
};
fs.readFile = function (path, encoding_, callback) {
var encoding = typeof(encoding_) == 'string' ? encoding_ : null;
var callback_ = arguments[arguments.length - 1];
var callback = (typeof(callback_) == 'function' ? callback_ : noop);
binding.stat(path, function (err, stat) {
if (err) { callback(err); return; }
binding.open(path, constants.O_RDONLY, 0666, function (err, fd) {
if (err) { callback(err); return; }
var size = stat.size;
var buffer = new Buffer(size);
var offset = 0;
function doRead() {
if (size < 1) {
binding.close(fd);
callback(null, encoding ? '' : buffer);
return;
}
// position is offset or null so we can read files on unseekable mediums
binding.read(fd, buffer, offset, size - offset, offset || null, function (err, amount) {
if (err) {
callback(err);
binding.close(fd);
return;
}
if (amount + offset < size) {
offset += amount;
doRead();
return;
fs.readFile = function (path, encoding_) {
var encoding = typeof(encoding_) === 'string' ? encoding_ : null;
var callback = arguments[arguments.length-1];
if (typeof(callback) !== 'function') callback = noop;
var readStream = fs.createReadStream(path);
var buffers = [];
var nread = 0;
readStream.on("data", function (chunk) {
buffers.push(chunk);
nread += chunk.length;
});
readStream.on("error", function (er) {
callback(er);
readStream.destroy();
});
readStream.on("end", function () {
// copy all the buffers into one
var buffer;
switch (buffers.length) {
case 0: buffer = new Buffer(0); break;
case 1: buffer = buffers[0]; break;
default: // concat together
buffer = new Buffer(nread);
var n = 0;
buffers.forEach(function (b) {
var l = b.length;
b.copy(buffer, n, 0, l);
n += l;
});
break;
}
binding.close(fd);
if (encoding) {
try {
var str = buffer.toString(encoding);
} catch (err) {
callback(err);
return;
}
callback(null, str);
} else {
callback(null, buffer);
buffer = buffer.toString(encoding);
} catch (er) {
return callback(er);
}
});
}
doRead();
});
callback(null, buffer);
});
};
fs.readFileSync = function (path, encoding) {
var fd = fs.openSync(path, constants.O_RDONLY, 0666);
var stat = fs.statSync(path);
var buffer = new Buffer(stat.size);
var buffer = new Buffer(4048);
var buffers = [];
var nread = 0;
var lastRead = 0;
while (nread < buffer.length) {
nread += fs.readSync(fd, buffer, nread, buffer.length - nread, null);
do {
if (lastRead) {
buffer._bytesRead = lastRead;
nread += lastRead;
buffers.push(buffer);
}
var buffer = new Buffer(4048);
lastRead = fs.readSync(fd, buffer, 0, buffer.length, null);
} while (lastRead > 0);
fs.closeSync(fd);
if (encoding) {
return buffer.toString(encoding);
if (buffers.length > 1) {
var offset = 0;
var i;
buffer = new Buffer(nread);
buffers.forEach(function (i) {
if (!i._bytesRead) return;
i.copy(buffer,offset,0,i._bytesRead);
offset += i._bytesRead;
})
} else {
return buffer;
//buffers has exactly 1 (possibly zero length) buffer, so this should be a shortcut
buffer = buffers[0].slice(0, buffers[0]._bytesRead);
}
if (encoding) buffer = buffer.toString(encoding);
return buffer;
};

9
test/simple/test-fs-read-file-sync-hostname.js

@ -0,0 +1,9 @@
common = require("../common");
assert = common.assert
var fs = require('fs');
// test reading from hostname
if (process.platform === "linux2") {
var hostname = fs.readFileSync('/proc/sys/kernel/hostname');
assert.ok(hostname.length > 0);
}
Loading…
Cancel
Save