Add possibility to add supplier for enum

BlackBeltTechnology · Aug 23, 2024 · f9a9c29 · f9a9c29
1 parent 59e6489
commit f9a9c29
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/i18n-api/src/main/java/hu/blackbelt/osgi/i18n/api/EnumI18nService.java b/i18n-api/src/main/java/hu/blackbelt/osgi/i18n/api/EnumI18nService.java
@@ -1,5 +1,8 @@
 package hu.blackbelt.osgi.i18n.api;
 
+import java.util.Locale;
+import java.util.function.Supplier;
+
 public interface EnumI18nService {
 
     /**
@@ -9,6 +12,15 @@ public interface EnumI18nService {
      */
     void register(Class<? extends Enum> clazz);
 
+    /**
+     * Register an Enum i18N message.
+     *
+     * @param clazz
+     * @param localeSupplier
+     */
+    void register(Class<? extends Enum> clazz, Supplier<Locale> localeSupplier);
+
+
     /**
      * Unregister an Enm i18n message.
      * @param clazz

diff --git a/...sourcebundle/src/main/java/hu/blackbelt/osgi/i18n/resourcebundle/EnumI18nServiceImpl.java b/...sourcebundle/src/main/java/hu/blackbelt/osgi/i18n/resourcebundle/EnumI18nServiceImpl.java
@@ -49,6 +49,11 @@ void activate(Config config) {
     }
 
     public void register(Class<? extends Enum> clazz) {
+        register(clazz, this.localeSupplier);
+    }
+
+    @Override
+    public void register(Class<? extends Enum> clazz, Supplier<Locale> localeSupplier) {
         MessageStreamLoader messageStreamLoader = new ClassLoaderBasedMessageStreamLoader(clazz.getClassLoader());
         ClassAndRequestContextLocaleBasedMessageResolver messageResolver =
                 new ClassAndRequestContextLocaleBasedMessageResolver(localeSupplier, clazz, messageStreamLoader, null);