Merge pull request #264 from art-community/feature/context-thread-safe

Feature/context thread safe

Author: Anton Bashirov

application-core/src/main/java/ru/art/core/context/Context.java

@@ -131,119 +131,154 @@ public <C extends ModuleConfiguration, S extends ModuleState> C getModule(String
         if (isNull(INSTANCE) || state != READY) {
             return toLoadIfNotExists.getDefaultConfiguration();
         }
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ModuleContainer<? extends ModuleConfiguration, ? extends ModuleState> moduleContainer = modules.get(moduleId);
         PreconfiguredModuleProvider preconfiguredModulesProvider;
         if (nonNull(moduleContainer)) {
+            lock.unlock();
             return cast(moduleContainer.getConfiguration());
         }
         C configuration;
         if (nonNull(preconfiguredModulesProvider = contextConfiguration().getPreconfiguredModulesProvider())) {
+            lock.unlock();
             return loadModule(toLoadIfNotExists, preconfiguredModulesProvider);
         }
+        lock.unlock();
         loadModule(toLoadIfNotExists, (configuration = toLoadIfNotExists.getDefaultConfiguration()));
         return cast(configuration);
     }
 
     public <C extends ModuleConfiguration, S extends ModuleState> S getModuleState(String moduleId, Module<C, S> toLoadIfNotExists) {
         if (isNull(moduleId)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         if (isNull(INSTANCE) || state == LOADING_MODULES) {
+            lock.unlock();
             return toLoadIfNotExists.getState();
         }
         ModuleContainer<? extends ModuleConfiguration, ? extends ModuleState> moduleContainer = modules.get(moduleId);
         PreconfiguredModuleProvider preconfiguredModulesProvider;
         if (nonNull(moduleContainer)) {
+            lock.unlock();
             return cast(moduleContainer.getModule().getState());
         }
         if (nonNull(preconfiguredModulesProvider = contextConfiguration().getPreconfiguredModulesProvider())) {
+            lock.unlock();
             loadModule(toLoadIfNotExists, preconfiguredModulesProvider);
             return cast(toLoadIfNotExists.getState());
         }
+        lock.unlock();
         loadModule(toLoadIfNotExists, toLoadIfNotExists.getDefaultConfiguration());
         return cast(toLoadIfNotExists.getState());
     }
 
     public <C extends ModuleConfiguration, S extends ModuleState> C getModule(String moduleId, Supplier<Module<C, S>> toLoadIfNotExists) {
         if (isNull(moduleId)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         if (isNull(INSTANCE) || state != READY) {
+            lock.unlock();
             return toLoadIfNotExists.get().getDefaultConfiguration();
         }
         ModuleContainer<? extends ModuleConfiguration, ? extends ModuleState> moduleContainer = modules.get(moduleId);
         PreconfiguredModuleProvider preconfiguredModulesProvider;
         if (nonNull(moduleContainer)) {
+            lock.unlock();
             return cast(moduleContainer.getConfiguration());
         }
         C configuration;
         Module<C, S> module = toLoadIfNotExists.get();
         if (nonNull(preconfiguredModulesProvider = contextConfiguration().getPreconfiguredModulesProvider())) {
+            lock.unlock();
             return loadModule(module, preconfiguredModulesProvider);
         }
+        lock.unlock();
         loadModule(module, (configuration = module.getDefaultConfiguration()));
         return cast(configuration);
     }
 
     public <C extends ModuleConfiguration, S extends ModuleState> S getModuleState(String moduleId, Supplier<Module<C, S>> toLoadIfNotExists) {
         if (isNull(moduleId)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         if (isNull(INSTANCE) || state == LOADING_MODULES) {
+            lock.unlock();
             return toLoadIfNotExists.get().getState();
         }
         ModuleContainer<? extends ModuleConfiguration, ? extends ModuleState> moduleContainer = modules.get(moduleId);
         PreconfiguredModuleProvider preconfiguredModulesProvider;
         if (nonNull(moduleContainer)) {
+            lock.unlock();
             return cast(moduleContainer.getModule().getState());
         }
         Module<C, S> module = toLoadIfNotExists.get();
         if (nonNull(preconfiguredModulesProvider = contextConfiguration().getPreconfiguredModulesProvider())) {
+            lock.unlock();
             loadModule(module, preconfiguredModulesProvider);
             return module.getState();
         }
+        lock.unlock();
         loadModule(module, module.getDefaultConfiguration());
         return cast(module.getState());
     }
 
     public <C extends ModuleConfiguration, S extends ModuleState> Context loadModule(Module<C, S> module) {
         if (isNull(module)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ContextState currentState = state;
         state = LOADING_MODULES;
         C configuration = module.getDefaultConfiguration();
         modules.put(module.getId(), new ModuleContainer<>(module, cast(configuration)));
         out.println(format(MODULE_LOADED_MESSAGE, module.getId(), currentTimeMillis() - lastActionTimestamp, configuration.getClass().getName()));
         state = currentState;
+        lock.unlock();
         module.onLoad();
         lastActionTimestamp = currentTimeMillis();
         return this;
     }
 
     public <C extends ModuleConfiguration, S extends ModuleState> Context loadModule(Module<C, S> module, ModuleConfigurator<C, S> moduleConfigurator) {
         if (isNull(module)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ContextState currentState = state;
         state = LOADING_MODULES;
         C configuration = cast(moduleConfigurator.configure(module));
         modules.put(module.getId(), new ModuleContainer<>(module, configuration));
         out.println(format(MODULE_LOADED_MESSAGE, module.getId(), currentTimeMillis() - lastActionTimestamp, configuration.getClass().getName()));
         state = currentState;
+        lock.unlock();
         module.onLoad();
         lastActionTimestamp = currentTimeMillis();
         return this;
     }
 
     public <C extends ModuleConfiguration, S extends ModuleState> Context loadModule(Module<C, S> module, C customModuleConfiguration) {
         if (isNull(module)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ContextState currentState = state;
         state = LOADING_MODULES;
         C configuration = cast(nonNull(customModuleConfiguration) ? customModuleConfiguration : module.getDefaultConfiguration());
         modules.put(module.getId(), new ModuleContainer<>(module, configuration));
         state = currentState;
+        lock.unlock();
         module.onLoad();
         out.println(format(MODULE_LOADED_MESSAGE, module.getId(), currentTimeMillis() - lastActionTimestamp, configuration.getClass().getName()));
         lastActionTimestamp = currentTimeMillis();
         return this;
     }
 
     private <C extends ModuleConfiguration, S extends ModuleState> C loadModule(Module<C, S> module, PreconfiguredModuleProvider preconfiguredModulesProvider) {
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ContextState currentState = state;
         state = LOADING_MODULES;
         C configuration = cast(preconfiguredModulesProvider.getModuleConfiguration(module.getId()).orElse(module.getDefaultConfiguration()));
         state = currentState;
+        lock.unlock();
         loadModule(module, configuration);
         return cast(configuration);
     }
@@ -254,56 +289,80 @@ public <C extends ModuleConfiguration> Context overrideModule(String moduleId, C
         if (isNull(customModuleConfiguration)) {
             throw new ContextInitializationException(CUSTOM_MODULE_CONFIGURATION_IS_NULL);
         }
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ModuleContainer<? extends ModuleConfiguration, ? extends ModuleState> moduleContainer = modules.get(moduleId);
-        if (isNull(moduleContainer)) return this;
+        if (isNull(moduleContainer)) {
+            lock.unlock();
+            return this;
+        }
         ContextState currentState = state;
         state = LOADING_MODULES;
         modules.put(moduleId, moduleContainer.overrideConfiguration(cast(customModuleConfiguration)));
         out.println(format(MODULE_OVERRIDDEN_MESSAGE, moduleId, currentTimeMillis() - lastActionTimestamp));
         lastActionTimestamp = currentTimeMillis();
         state = currentState;
+        lock.unlock();
         return this;
     }
 
     @SuppressWarnings("Duplicates")
     public Context reloadModule(String moduleId) {
         if (isNull(moduleId)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ModuleContainer<? extends ModuleConfiguration, ? extends ModuleState> moduleContainer = modules.get(moduleId);
-        if (isNull(moduleContainer)) return this;
+        if (isNull(moduleContainer)) {
+            lock.unlock();
+            return this;
+        }
         ContextState currentState = state;
         state = RELOADING_MODULES;
         moduleContainer.reloadModule();
         out.println(format(MODULE_RELOADED_MESSAGE, moduleId, currentTimeMillis() - lastActionTimestamp));
         lastActionTimestamp = currentTimeMillis();
         state = currentState;
+        lock.unlock();
         return this;
     }
 
     @SuppressWarnings("Duplicates")
     public Context refreshModule(String moduleId) {
         if (isNull(moduleId)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ModuleContainer<? extends ModuleConfiguration, ? extends ModuleState> moduleContainer = modules.get(moduleId);
-        if (isNull(moduleContainer)) return this;
+        if (isNull(moduleContainer)) {
+            lock.unlock();
+            return this;
+        }
         ContextState currentState = state;
         state = REFRESHING_MODULES;
         moduleContainer.refreshConfiguration();
         out.println(format(MODULE_REFRESHED_MESSAGE, moduleId, currentTimeMillis() - lastActionTimestamp));
         lastActionTimestamp = currentTimeMillis();
         state = currentState;
+        lock.unlock();
         return this;
     }
 
     public Context refreshAndReloadModule(String moduleId) {
         if (isNull(moduleId)) throw new ContextInitializationException(MODULE_ID_IS_NULL);
+        ReentrantLock lock = Context.LOCK;
+        lock.lock();
         ContextState currentState = state;
         state = REFRESHING_AND_RELOADING_MODULES;
         ModuleContainer<? extends ModuleConfiguration, ? extends ModuleState> moduleContainer = modules.get(moduleId);
-        if (isNull(moduleContainer)) return this;
+        if (isNull(moduleContainer)) {
+            lock.unlock();
+            return this;
+        }
         moduleContainer.reloadModule();
         moduleContainer.refreshConfiguration();
         out.println(format(MODULE_REFRESHED_AND_RELOADED_MESSAGE, moduleId, currentTimeMillis() - lastActionTimestamp));
         lastActionTimestamp = currentTimeMillis();
         state = currentState;
+        lock.unlock();
         return this;
     }