Webionite/haste-server
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added support for haste maximum size

Browse Source
This commit is contained in:
John Crepezzi
2011-11-21 10:17:23 -05:00
parent 29a600fe87
commit a9e29c2986
4 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/document_handler.js
View File

@@ -5,6 +5,7 @@ var winston = require('winston');
var DocumentHandler = function(options) {
if (options) {
this.keyLength = options.keyLength || 10;
this.maxLength = options.maxLength; // none by default
this.store = options.store;
}
};
@@ -35,8 +36,15 @@ DocumentHandler.prototype.handlePost = function(request, response) {
response.writeHead(200, { 'content-type': 'application/json' });
}
buffer += data.toString();
if (_this.maxLength && buffer.length > _this.maxLength) {
_this.cancelled = true;
winston.warn('attempted to upload a document >maxLength', { maxLength: _this.maxLength });
response.writeHead(400, { 'content-type': 'application/json' });
response.end(JSON.stringify({ message: 'document exceeds maximum length' }));
}
});
request.on('end', function(end) {
if (_this.cancelled) return;
_this.store.set(key, buffer, function(res) {
if (res) {
winston.verbose('added document', { key: key });
@@ -44,6 +52,7 @@ DocumentHandler.prototype.handlePost = function(request, response) {
}
else {
winston.verbose('error adding document');
response.writeHead(500, { 'content-type': 'application/json' });
response.end(JSON.stringify({ message: 'error adding document' }));
}
});