116 lines
3.8 KiB
Java
116 lines
3.8 KiB
Java
package org.gcube.contentmanager.storageserver.store;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import com.mongodb.BasicDBObject;
|
|
import com.mongodb.BasicDBObjectBuilder;
|
|
import com.mongodb.DBCollection;
|
|
import com.mongodb.DBCursor;
|
|
import com.mongodb.DBObject;
|
|
|
|
public class StorageStatusOperationManager {
|
|
|
|
DBCollection ssCollection;
|
|
Logger logger=LoggerFactory.getLogger(StorageStatusOperationManager.class);
|
|
|
|
public StorageStatusOperationManager(DBCollection ssCollection){
|
|
setSsCollection(ssCollection);
|
|
}
|
|
|
|
public void putSSRecord(String consumer, long volume, int count){
|
|
BasicDBObject doc = new BasicDBObject("consumer", consumer)
|
|
.append("volume", volume)
|
|
.append("count", count);
|
|
getSsCollection().insert(doc);
|
|
}
|
|
|
|
public StorageStatusRecord updateUser(StorageStatusRecord ssRecord, String lastOperation){
|
|
StorageStatusRecord oldSsr=getSSRecord(ssRecord.getConsumer());
|
|
if(oldSsr != null){
|
|
|
|
int partialCount=oldSsr.getCount();
|
|
int count = countCalculation(ssRecord.getCount(), partialCount, lastOperation);
|
|
ssRecord.setCount(count);
|
|
|
|
long partialVolume=oldSsr.getVolume();
|
|
long volume = volumeCalculation(ssRecord.getVolume(), partialVolume, lastOperation);
|
|
ssRecord.setVolume(volume);
|
|
|
|
final BasicDBObject query = new BasicDBObject("consumer", ssRecord.getConsumer());
|
|
// Creating BasicDBObjectBuilder object without arguments
|
|
DBObject documentBuilder = BasicDBObjectBuilder.start()
|
|
.add("volume", volume).add("count", count).get();
|
|
// get the dbobject from builder and Inserting document
|
|
getSsCollection().update(query,new BasicDBObject("$set", documentBuilder), true, false);
|
|
// close();
|
|
}else{
|
|
putSSRecord(ssRecord.getConsumer(), ssRecord.getVolume(), ssRecord.getCount());
|
|
}
|
|
return ssRecord;
|
|
}
|
|
|
|
|
|
|
|
public StorageStatusRecord getSSRecord(String consumer){
|
|
BasicDBObject query = new BasicDBObject("consumer", consumer);
|
|
DBCursor cursor=getSsCollection().find(query);
|
|
DBObject obj=null;
|
|
try{
|
|
if(cursor.hasNext()){
|
|
obj=cursor.next();
|
|
|
|
}
|
|
}finally{
|
|
cursor.close();
|
|
}
|
|
if(obj!=null){
|
|
String cons=null;
|
|
if(obj.containsField("consumer")) cons=(String) obj.get("consumer");
|
|
else logger.error("incomplete record found. consumer field is missing");
|
|
long vol =0;
|
|
if(obj.containsField("volume")) vol=(long) obj.get("volume");
|
|
else logger.error("incomplete record found. volume field is missing");
|
|
int count=0;
|
|
if(obj.containsField("count")) count=(int) obj.get("count");
|
|
else logger.error("incomplete record found. count field is missing");
|
|
String id=(String)obj.get("id");
|
|
return new StorageStatusRecord(id, cons, vol, count, obj);
|
|
}else{
|
|
return null;
|
|
}
|
|
}
|
|
|
|
private long volumeCalculation(long currentVolume, long partialVolume, String operation) {
|
|
logger.info("accounting: operation "+operation+" total Volume "+partialVolume+" current volume "+currentVolume);
|
|
if(operation.equalsIgnoreCase("UPLOAD") || operation.equalsIgnoreCase("COPY")){
|
|
partialVolume=partialVolume+currentVolume;
|
|
}else if(operation.equalsIgnoreCase("DELETE")){
|
|
partialVolume=partialVolume-currentVolume;
|
|
}
|
|
logger.info("new volume "+partialVolume);
|
|
return partialVolume;
|
|
}
|
|
|
|
private int countCalculation(int currentCount, int partialCount, String operation) {
|
|
logger.info("accounting: operation "+operation+" total count "+partialCount+" current count"+currentCount);
|
|
if(operation.equalsIgnoreCase("UPLOAD")|| operation.equalsIgnoreCase("COPY"))
|
|
partialCount=partialCount+currentCount;
|
|
else if(operation.equalsIgnoreCase("DELETE"))
|
|
partialCount=partialCount-currentCount;
|
|
logger.info("new count: "+partialCount);
|
|
return partialCount;
|
|
}
|
|
|
|
public DBCollection getSsCollection() {
|
|
return ssCollection;
|
|
}
|
|
|
|
public void setSsCollection(DBCollection ssCollection) {
|
|
this.ssCollection = ssCollection;
|
|
}
|
|
|
|
|
|
|
|
}
|