From dc8e1fafdd6eaaa4a4e5919a8005ba146046236c Mon Sep 17 00:00:00 2001
From: Luca Frosini <luca.frosini@isti.cnr.it>
Date: Fri, 19 Mar 2021 13:12:21 +0100
Subject: [PATCH] Fixed bug

---
 .../accounting/utility/postgresql/RecordToDBMapping.java  | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/gcube/accounting/utility/postgresql/RecordToDBMapping.java b/src/main/java/org/gcube/accounting/utility/postgresql/RecordToDBMapping.java
index 934524e..c568ebf 100644
--- a/src/main/java/org/gcube/accounting/utility/postgresql/RecordToDBMapping.java
+++ b/src/main/java/org/gcube/accounting/utility/postgresql/RecordToDBMapping.java
@@ -37,11 +37,17 @@ public class RecordToDBMapping {
 		recordToDBInfo.put(type, recordDBInfo);
 	}
 	
+	@SuppressWarnings("unchecked")
 	public static synchronized RecordToDBFields getRecordToDBFields(Class<? extends Record> clz) throws Exception {
 		String type = getRecordTypeByClass(clz);
 		RecordToDBFields recordToDBFields = classToRecordToDBMapper.get(type);
 		if(recordToDBFields == null) {
-			Class<? extends AggregatedRecord<?,?>> aggregatedRecordClass = RecordUtility.getAggregatedRecordClass(type);
+			Class<? extends AggregatedRecord<?,?>> aggregatedRecordClass;
+			if(AggregatedRecord.class.isAssignableFrom(clz)) {
+				aggregatedRecordClass = (Class<? extends AggregatedRecord<?, ?>>) clz;
+			}else {
+				aggregatedRecordClass = RecordUtility.getAggregatedRecordClass(type);
+			}
 			recordToDBFields = new RecordToDBFields(type, aggregatedRecordClass);
 			classToRecordToDBMapper.put(type, recordToDBFields);
 		}