diff --git a/CHANGELOG.md b/CHANGELOG.md
index 1283df1..9bf441c 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -2,6 +2,10 @@ This project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.htm
# Changelog for Accounting Postgresql Utilities
+## [v2.0.0-SNAPSHOT]
+
+- RecordToDBMapping is no more singleton
+
## [v1.0.0]
diff --git a/pom.xml b/pom.xml
index 31d7322..74ee6b0 100644
--- a/pom.xml
+++ b/pom.xml
@@ -7,7 +7,7 @@
org.gcube.accounting
accounting-postgresql-utilities
- 1.0.0
+ 2.0.0-SNAPSHOT
Accounting Postgresql Utilities
Utility classes to persist/query Record in PostgreSQL database
@@ -45,6 +45,12 @@
42.2.19
+
+ org.gcube.common
+ authorization-utils
+ [2.2.0, 3.0.0-SNAPSHOT)
+ test
+
org.gcube.accounting
accounting-lib
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 c568ebf..03953f2 100644
--- a/src/main/java/org/gcube/accounting/utility/postgresql/RecordToDBMapping.java
+++ b/src/main/java/org/gcube/accounting/utility/postgresql/RecordToDBMapping.java
@@ -22,15 +22,19 @@ public class RecordToDBMapping {
}
}
- protected final static Map classToRecordToDBMapper;
- protected final static Map recordToDBInfo;
+ protected static final Map classToRecordToDBMapper;
static {
classToRecordToDBMapper = new HashMap<>();
+ }
+
+ protected final Map recordToDBInfo;
+
+ public RecordToDBMapping() {
recordToDBInfo= new HashMap<>();
}
- public static void addRecordToDB(Class extends AggregatedRecord, ?>> clz, PersistenceBackendConfiguration configuration) throws Exception {
+ public void addRecordToDB(Class extends AggregatedRecord, ?>> clz, PersistenceBackendConfiguration configuration) throws Exception {
String type = getRecordTypeByClass(clz);
getRecordToDBFields(clz);
RecordToDBConnection recordDBInfo = new RecordToDBConnection(type, clz, configuration);
@@ -54,8 +58,9 @@ public class RecordToDBMapping {
return recordToDBFields;
}
- public static synchronized RecordToDBConnection getRecordDBInfo(Class extends Record> clz) throws Exception {
+ public synchronized RecordToDBConnection getRecordDBInfo(Class extends Record> clz) throws Exception {
String type = getRecordTypeByClass(clz);
return recordToDBInfo.get(type);
}
+
}
diff --git a/src/main/resources/AccountingManagerTables.sql b/src/main/resources/AccountingManagerTables.sql
index fc8077a..d6016ac 100644
--- a/src/main/resources/AccountingManagerTables.sql
+++ b/src/main/resources/AccountingManagerTables.sql
@@ -16,10 +16,11 @@ CREATE TABLE "aggregation_status"(
recovered_records_number NUMERIC NOT NULL,
malformed_records_number NUMERIC NOT NULL DEFAULT 0,
percentage NUMERIC NOT NULL,
+ restart_from NUMERIC NOT NULL DEFAULT 0,
context TEXT,
current_aggregation_state aggregation_state NOT NULL,
last_update_time TIMESTAMP WITH TIME ZONE NOT NULL,
- previous UUID REFERENCES aggregation_status (id)
+ previous UUID REFERENCES aggregation_status (id) DEFAULT NULL ON DELETE SET NULL
);
CREATE TABLE "aggregation_status_event"(
diff --git a/src/test/java/org/gcube/accounting/utility/postgresql/ContextTest.java b/src/test/java/org/gcube/accounting/utility/postgresql/ContextTest.java
index b8b0b44..820c23b 100644
--- a/src/test/java/org/gcube/accounting/utility/postgresql/ContextTest.java
+++ b/src/test/java/org/gcube/accounting/utility/postgresql/ContextTest.java
@@ -7,14 +7,14 @@ import java.io.IOException;
import java.io.InputStream;
import java.util.Properties;
-import org.gcube.common.authorization.client.Constants;
-import org.gcube.common.authorization.client.exceptions.ObjectNotFound;
-import org.gcube.common.authorization.library.AuthorizationEntry;
-import org.gcube.common.authorization.library.provider.AuthorizationProvider;
-import org.gcube.common.authorization.library.provider.ClientInfo;
-import org.gcube.common.authorization.library.provider.SecurityTokenProvider;
-import org.gcube.common.authorization.library.utils.Caller;
-import org.gcube.common.scope.api.ScopeProvider;
+import org.gcube.common.authorization.utils.manager.SecretManager;
+import org.gcube.common.authorization.utils.manager.SecretManagerProvider;
+import org.gcube.common.authorization.utils.secret.JWTSecret;
+import org.gcube.common.authorization.utils.secret.Secret;
+import org.gcube.common.authorization.utils.secret.SecretUtility;
+import org.gcube.common.keycloak.KeycloakClientFactory;
+import org.gcube.common.keycloak.KeycloakClientHelper;
+import org.gcube.common.keycloak.model.TokenResponse;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.slf4j.Logger;
@@ -27,80 +27,147 @@ public class ContextTest {
private static final Logger logger = LoggerFactory.getLogger(ContextTest.class);
- protected static Properties properties;
- protected static final String PROPERTIES_FILENAME = "token.properties";
+ protected static final String CONFIG_INI_FILENAME = "config.ini";
- public static final String PARENT_DEFAULT_TEST_SCOPE;
public static final String DEFAULT_TEST_SCOPE;
- public static final String ALTERNATIVE_TEST_SCOPE;
- public static final String DEFAULT_TEST_SCOPE_ANOTHER_USER;
+ public static final String GCUBE;
+ public static final String DEVNEXT;
+ public static final String NEXTNEXT;
+ public static final String DEVSEC;
+ public static final String DEVVRE;
+
+ public static final String ROOT_PROD;
+
+ protected static final Properties properties;
+
+ public static final String TYPE_PROPERTY_KEY = "type";
+ public static final String USERNAME_PROPERTY_KEY = "username";
+ public static final String PASSWORD_PROPERTY_KEY = "password";
+ public static final String CLIENT_ID_PROPERTY_KEY = "clientId";
static {
- properties = new Properties();
- InputStream input = ContextTest.class.getClassLoader().getResourceAsStream(PROPERTIES_FILENAME);
+ GCUBE = "/gcube";
+ DEVNEXT = GCUBE + "/devNext";
+ NEXTNEXT = DEVNEXT + "/NextNext";
+ DEVSEC = GCUBE + "/devsec";
+ DEVVRE = DEVSEC + "/devVRE";
+ ROOT_PROD = "/d4science.research-infrastructures.eu";
+
+ DEFAULT_TEST_SCOPE = GCUBE;
+
+
+ properties = new Properties();
+ InputStream input = ContextTest.class.getClassLoader().getResourceAsStream(CONFIG_INI_FILENAME);
try {
// load the properties file
properties.load(input);
- } catch(IOException e) {
+ } catch (IOException e) {
throw new RuntimeException(e);
}
- // PARENT_DEFAULT_TEST_SCOPE = "/pred4s"
- // DEFAULT_TEST_SCOPE_NAME = PARENT_DEFAULT_TEST_SCOPE + "/preprod";
- // ALTERNATIVE_TEST_SCOPE = DEFAULT_TEST_SCOPE_NAME + "/preVRE";
+ }
+
+ private enum Type{
+ USER, CLIENT_ID
+ };
+
+ public static void set(Secret secret) throws Exception {
+ SecretManagerProvider.instance.reset();
+ SecretManager secretManager = new SecretManager();
+ secretManager.addSecret(secret);
+ SecretManagerProvider.instance.set(secretManager);
+ SecretManagerProvider.instance.get().set();
+ }
+
+ public static void setContextByName(String fullContextName) throws Exception {
+ logger.debug("Going to set credentials for context {}", fullContextName);
+ Secret secret = getSecretByContextName(fullContextName);
+ set(secret);
+ }
+
+
+ private static TokenResponse getJWTAccessToken(String context) throws Exception {
+ Type type = Type.valueOf(properties.get(TYPE_PROPERTY_KEY).toString());
+ TokenResponse tr = null;
- PARENT_DEFAULT_TEST_SCOPE = "/gcube";
- DEFAULT_TEST_SCOPE = PARENT_DEFAULT_TEST_SCOPE + "/devNext";
- ALTERNATIVE_TEST_SCOPE = DEFAULT_TEST_SCOPE + "/NextNext";
+ int index = context.indexOf('/', 1);
+ String root = context.substring(0, index == -1 ? context.length() : index);
- DEFAULT_TEST_SCOPE_ANOTHER_USER = "lucio.lelii_" + DEFAULT_TEST_SCOPE;
+ switch (type) {
+ case CLIENT_ID:
+ String clientId = properties.getProperty(CLIENT_ID_PROPERTY_KEY);
+ String clientSecret = properties.getProperty(root);
+
+ tr = KeycloakClientFactory.newInstance().queryUMAToken(context, clientId, clientSecret, context, null);
+ break;
+ case USER:
+ default:
+ String username = properties.getProperty(USERNAME_PROPERTY_KEY);
+ String password = properties.getProperty(PASSWORD_PROPERTY_KEY);
+
+ switch (root) {
+ case "/gcube":
+ default:
+ clientId = "next.d4science.org";
+ break;
+
+ case "/pred4s":
+ clientId = "pre.d4science.org";
+ break;
+
+ case "/d4science.research-infrastructures.eu":
+ clientId = "services.d4science.org";
+ break;
+ }
+ clientSecret = null;
+
+ tr = KeycloakClientHelper.getTokenForUser(context, username, password);
+ break;
+
+ }
+
+ return tr;
+
+ }
+
+ public static Secret getSecretByContextName(String context) throws Exception {
+ TokenResponse tr = getJWTAccessToken(context);
+ Secret secret = new JWTSecret(tr.getAccessToken());
+ return secret;
+ }
+
+ public static void setContext(String token) throws Exception {
+ Secret secret = getSecret(token);
+ set(secret);
+ }
+
+ private static Secret getSecret(String token) throws Exception {
+ Secret secret = SecretUtility.getSecretByTokenString(token);
+ return secret;
+ }
+
+ public static String getUser() {
+ String user = "UNKNOWN";
try {
- setContextByName(DEFAULT_TEST_SCOPE);
+ user = SecretManagerProvider.instance.get().getUser().getUsername();
} catch(Exception e) {
- throw new RuntimeException(e);
+ logger.error("Unable to retrieve user. {} will be used", user);
}
- }
-
- public static String getCurrentContextFullName() throws ObjectNotFound, Exception {
- return getContextFullNameByToken(SecurityTokenProvider.instance.get());
- }
-
- public static String getContextFullNameByToken(String token) throws ObjectNotFound, Exception {
- AuthorizationEntry authorizationEntry = Constants.authorizationService().get(token);
- String context = authorizationEntry.getContext();
- logger.info("Context of token {} is {}", token, context);
- return context;
- }
-
- public static void setContextByName(String fullContextName) throws ObjectNotFound, Exception {
- String token = ContextTest.properties.getProperty(fullContextName);
- setContext(token);
- }
-
- private static void setContext(String token) throws ObjectNotFound, Exception {
- SecurityTokenProvider.instance.set(token);
- AuthorizationEntry authorizationEntry = Constants.authorizationService().get(token);
- ClientInfo clientInfo = authorizationEntry.getClientInfo();
- logger.debug("User : {} - Type : {}", clientInfo.getId(), clientInfo.getType().name());
- String qualifier = authorizationEntry.getQualifier();
- Caller caller = new Caller(clientInfo, qualifier);
- AuthorizationProvider.instance.set(caller);
- ScopeProvider.instance.set(getContextFullNameByToken(token));
+ return user;
}
@BeforeClass
public static void beforeClass() throws Exception {
- setContextByName(DEFAULT_TEST_SCOPE);
+ setContextByName(ROOT_PROD);
}
@AfterClass
public static void afterClass() throws Exception {
- SecurityTokenProvider.instance.reset();
- ScopeProvider.instance.reset();
+ SecretManagerProvider.instance.reset();
}
}
diff --git a/src/test/resources/.gitignore b/src/test/resources/.gitignore
index 8f712ff..b91c9ba 100644
--- a/src/test/resources/.gitignore
+++ b/src/test/resources/.gitignore
@@ -2,3 +2,4 @@
/devNext.gcubekey
/devsec.gcubekey
/gcube.gcubekey
+/config.ini