Botanical/BotanSS
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

minor bug fixes

Browse Source
This commit is contained in:
斟酌 鵬兄 2015-07-28 12:25:10 +08:00
parent 8c146b6623
commit e10c108efe
2 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
net/WebFrame.js
View File

@ -140,6 +140,12 @@ Framework.prototype.plantResult = function()
this.HTTP.response.write( this.result ); this.HTTP.response.write( this.result );
this.HTTP.response.end(); this.HTTP.response.end();
} }
// Release resources
if( this.requestStr )
{
this.requestStr.discard();
}
}; };
// This won't handle path exists // This won't handle path exists

11
utils/CondStream.js
View File

@ -1,4 +1,6 @@
var Dragonfly = global.Dragonfly;
var fs = require( "fs" ); var fs = require( "fs" );
var path = require( "path" );
var crypto = require( "crypto" ); var crypto = require( "crypto" );
var util = require( "util" ); var util = require( "util" );
var ReadStream = require( "stream" ).Readable; var ReadStream = require( "stream" ).Readable;
@ -44,9 +46,9 @@ ConditionalStream.prototype.write = function( data )
// Trigger // Trigger
if( this.limit < this.size ) if( this.limit < this.size )
{ {
this.file = this.tmpPath + "ss_" + crypto.randomBytes( 8 ).toString( "hex" ); this.file = path.join( this.tmpPath, "ss_" + crypto.randomBytes( 8 ).toString( "hex" ) );
this.stream = fs.createWriteStream( this.file ); this.stream = fs.createWriteStream( this.file, { mode: 0600 } );
this.stream.addListener( "finish", this.__end.bind( this ) ); this.stream.addListener( "finish", this.__end.bind( this ) );
this.stream.write( this.hexData, "hex" ); this.stream.write( this.hexData, "hex" );
} }
@ -78,10 +80,11 @@ ConditionalStream.prototype.discard = function()
var _self = this; var _self = this;
this.__discard = true; this.__discard = true;
if( this.__finished ) if( this.__finished && this.file )
{ {
fs.unlink( this.file, function() fs.unlink( this.file, function( e )
{ {
Dragonfly.Debug( "Client Data Closed: " + _self.file );
if( _self.__error ) throw new Error( _self.__error ); if( _self.__error ) throw new Error( _self.__error );
} ); } );
} }