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> clz, PersistenceBackendConfiguration configuration) throws Exception { + public void addRecordToDB(Class> 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 clz) throws Exception { + public synchronized RecordToDBConnection getRecordDBInfo(Class 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