feat: update template bean

Signed-off-by: Otavio Santana <otaviopolianasantana@gmail.com>
eclipse · Mar 2, 2024 · ec83aee · ec83aee
1 parent 728e2e6
commit ec83aee
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/...osql-mapping-column/src/main/java/org/eclipse/jnosql/mapping/column/spi/TemplateBean.java b/...osql-mapping-column/src/main/java/org/eclipse/jnosql/mapping/column/spi/TemplateBean.java
@@ -16,6 +16,8 @@
 
 
 import jakarta.enterprise.context.spi.CreationalContext;
+import jakarta.nosql.Template;
+import org.eclipse.jnosql.communication.semistructured.DatabaseManager;
 import org.eclipse.jnosql.mapping.DatabaseQualifier;
 import org.eclipse.jnosql.mapping.DatabaseType;
 import org.eclipse.jnosql.mapping.column.ColumnTemplate;
@@ -29,7 +31,7 @@
 
 class TemplateBean extends AbstractBean<ColumnTemplate> {
 
-    private static final Set<Type> TYPES = Set.of(ColumnTemplate.class, ColumnTemplate.class);
+    private static final Set<Type> TYPES = Set.of(ColumnTemplate.class, Template.class);
     private final String provider;
 
     private final Set<Annotation> qualifiers;
@@ -54,12 +56,12 @@ public Class<?> getBeanClass() {
     public ColumnTemplate create(CreationalContext<ColumnTemplate> context) {
 
         ColumnTemplateProducer producer = getInstance(ColumnTemplateProducer.class);
-        ColumnTemplate manager = getColumnManager();
+        DatabaseManager manager = getColumnManager();
         return producer.apply(manager);
     }
 
-    private ColumnTemplate getColumnManager() {
-        return getInstance(ColumnTemplate.class, DatabaseQualifier.ofColumn(provider));
+    private DatabaseManager getColumnManager() {
+        return getInstance(DatabaseManager.class, DatabaseQualifier.ofColumn(provider));
     }
 
     @Override