elimu-ai#1684 addressed review comment to change hardcoded package na…

…me ai.elimu.model by dynaically pulling package name using BaseEntity class
diptechexpo · Jul 8, 2024 · 35bc4bd · 35bc4bd
1 parent 3221f77
commit 35bc4bd
Show file tree

Hide file tree

Showing 2 changed files with 1,133 additions and 1 deletion.
diff --git a/src/main/java/ai/elimu/web/servlet/CustomDispatcherServlet.java b/src/main/java/ai/elimu/web/servlet/CustomDispatcherServlet.java
@@ -1,5 +1,6 @@
 package ai.elimu.web.servlet;
 
+import ai.elimu.entity.BaseEntity;
 import ai.elimu.model.v2.enums.Environment;
 import ai.elimu.model.v2.enums.Language;
 import ai.elimu.util.ConfigHelper;
@@ -81,7 +82,7 @@ private void createJpaSchemaExport() {
         // Scan for classes annotated as JPA @Entity
         ClassPathScanningCandidateComponentProvider entityScanner = new ClassPathScanningCandidateComponentProvider(true);
         entityScanner.addIncludeFilter(new AnnotationTypeFilter(Entity.class));
-        for (BeanDefinition beanDefinition : entityScanner.findCandidateComponents("ai.elimu.model")) {
+        for (BeanDefinition beanDefinition : entityScanner.findCandidateComponents(BaseEntity.class.getPackageName())) {
             logger.info("beanDefinition.getBeanClassName(): " + beanDefinition.getBeanClassName());
             try {
                 Class<?> annotatedClass = Class.forName(beanDefinition.getBeanClassName());