Henrique Santos 10 лет назад
Родитель
Сommit
5d2fc89137

+ 4 - 0
services/filestore/app/coffee/AWSSDKPersistorManager.coffee

@@ -85,3 +85,7 @@ module.exports =
 				return callback err
 			callback null, data.ETag?
 
+	getProjectSize:(bucketName, key, callback)->
+		logger.log bucketName:bucketName, key:key, "get project size in s3"
+		callback null, 1024
+

+ 4 - 0
services/filestore/app/coffee/FSPersistorManager.coffee

@@ -87,3 +87,7 @@ module.exports =
     fs.exists "#{location}/#{filteredName}", (exists) ->
       logger.log location:location, name:filteredName, exists:exists, "checked if file exists"
       callback null, exists
+
+  getProjectSize:(location, name, callback)->
+    logger.log location:location, name:name, "get project size in file system"
+    callback null, 1024

+ 2 - 2
services/filestore/app/coffee/ProjectController.coffee

@@ -10,9 +10,9 @@ module.exports = projectController =
 		metrics.inc "projectSize"
 		{project_id, bucket} = req
 		logger.log project_id:project_id, bucket:bucket, "reciving request to project size"
-		FileHandler.getFile bucket, project_id, req, (err)->
+		ProjectHandler.getSize bucket, project_id, req, (err, size)->
 			if err?
 				logger.log err: err, project_id: project_id, bucket: bucket, "error inserting file"
 				res.send 500
 			else
-				res.send 200
+				res.json {'total bytes' : size}

+ 13 - 0
services/filestore/app/coffee/ProjectHandler.coffee

@@ -0,0 +1,13 @@
+settings = require("settings-sharelatex")
+PersistorManager = require("./PersistorManager")
+logger = require("logger-sharelatex")
+async = require("async")
+
+module.exports =
+
+	getSize: (bucket, project_id, opts = {}, callback)->
+		logger.log bucket:bucket, project_id:project_id, opts:opts, "getting project size"
+		PersistorManager.getProjectSize bucket, project_id, opts, (err, size)->
+			if err?
+				logger.err  bucket:bucket, project_id:project_id, opts:opts, "error getting size"
+			callback err, size

+ 4 - 0
services/filestore/app/coffee/S3PersistorManager.coffee

@@ -137,3 +137,7 @@ module.exports =
 			exists = res.statusCode == 200
 			logger.log bucketName:bucketName, key:key, exists:exists, "checked if file exsists in s3"
 			callback(err, exists)
+
+	getProjectSize:(bucketName, key, callback)->
+		logger.log bucketName:bucketName, key:key, "get project size in s3"
+		callback null, 1024