You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@cayenne.apache.org by nt...@apache.org on 2021/02/24 10:26:31 UTC

[cayenne] 01/03: CAY-2677 Custom Class Generation Templates for Embeddables and DataMaps applying patch from Stefan S, see JIRA issue

This is an automated email from the ASF dual-hosted git repository.

ntimofeev pushed a commit to branch STABLE-4.1
in repository https://gitbox.apache.org/repos/asf/cayenne.git

commit 3ad81a8f0fe77b64b95029fe77942f8d50e9dc9c
Author: Nikita Timofeev <st...@gmail.com>
AuthorDate: Wed Feb 24 13:24:42 2021 +0300

    CAY-2677 Custom Class Generation Templates for Embeddables and DataMaps
    applying patch from Stefan S, see JIRA issue
---
 .../org/apache/cayenne/gen/CgenConfiguration.java  |  6 ++
 .../apache/cayenne/gen/xml/CgenConfigHandler.java  | 43 ++++++++++
 .../cayenne/modeler/CodeTemplateManager.java       | 92 +++++++++++++--------
 .../modeler/editor/cgen/CustomModeController.java  | 75 +++++++++++++++++
 .../modeler/editor/cgen/CustomModePanel.java       | 93 +++++++++++++++++++++-
 5 files changed, 274 insertions(+), 35 deletions(-)

diff --git a/cayenne-cgen/src/main/java/org/apache/cayenne/gen/CgenConfiguration.java b/cayenne-cgen/src/main/java/org/apache/cayenne/gen/CgenConfiguration.java
index 2f6e2e4..eb2c822 100644
--- a/cayenne-cgen/src/main/java/org/apache/cayenne/gen/CgenConfiguration.java
+++ b/cayenne-cgen/src/main/java/org/apache/cayenne/gen/CgenConfiguration.java
@@ -396,6 +396,12 @@ public class CgenConfiguration implements Serializable, XMLSerializable {
                 .simpleTag("mode", this.artifactsGenerationMode.getLabel())
                 .simpleTag("template", this.template)
                 .simpleTag("superTemplate", this.superTemplate)
+                .simpleTag("template", this.template)
+                .simpleTag("superTemplate", this.superTemplate)
+                .simpleTag("embeddableTemplate", this.embeddableTemplate)
+                .simpleTag("embeddableSuperTemplate", this.embeddableSuperTemplate)
+                .simpleTag("queryTemplate", this.queryTemplate)
+                .simpleTag("querySuperTemplate", this.querySuperTemplate)
                 .simpleTag("outputPattern", this.outputPattern)
                 .simpleTag("makePairs", Boolean.toString(this.makePairs))
                 .simpleTag("usePkgPath", Boolean.toString(this.usePkgPath))
diff --git a/cayenne-cgen/src/main/java/org/apache/cayenne/gen/xml/CgenConfigHandler.java b/cayenne-cgen/src/main/java/org/apache/cayenne/gen/xml/CgenConfigHandler.java
index 6a80f91..0a41a04 100644
--- a/cayenne-cgen/src/main/java/org/apache/cayenne/gen/xml/CgenConfigHandler.java
+++ b/cayenne-cgen/src/main/java/org/apache/cayenne/gen/xml/CgenConfigHandler.java
@@ -43,6 +43,10 @@ public class CgenConfigHandler extends NamespaceAwareNestedTagHandler{
     private static final String GENERATION_MODE_TAG = "mode";
     private static final String SUBCLASS_TEMPLATE_TAG = "template";
     private static final String SUPERCLASS_TEMPLATE_TAG = "superTemplate";
+    private static final String EMBEDDABLE_TEMPLATE_TAG = "embeddableTemplate";
+    private static final String EMBEDDABLE_SUPER_TEMPLATE_TAG = "embeddableSuperTemplate";
+    private static final String QUERY_TEMPLATE_TAG = "queryTemplate";
+    private static final String QUERY_SUPER_TEMPLATE_TAG = "querySuperTemplate";
     private static final String OUTPUT_PATTERN_TAG = "outputPattern";
     private static final String MAKE_PAIRS_TAG = "makePairs";
     private static final String USE_PKG_PATH_TAG = "usePkgPath";
@@ -97,6 +101,17 @@ public class CgenConfigHandler extends NamespaceAwareNestedTagHandler{
             case SUPERCLASS_TEMPLATE_TAG:
                 createSuperclassTemplate(data);
                 break;
+            case EMBEDDABLE_TEMPLATE_TAG:
+            	createEmbeddableTemplate(data);
+            	break;
+            case EMBEDDABLE_SUPER_TEMPLATE_TAG:
+            	createEmbeddableSuperTemplate(data);
+            case QUERY_TEMPLATE_TAG:
+            	createQueryTemplate(data);
+            	break;
+            case QUERY_SUPER_TEMPLATE_TAG:
+            	createQuerySuperTemplate(data);
+                break;
             case OUTPUT_PATTERN_TAG:
                 createOutputPattern(data);
                 break;
@@ -165,6 +180,34 @@ public class CgenConfigHandler extends NamespaceAwareNestedTagHandler{
         }
         configuration.setSuperTemplate(template);
     }
+    
+    private void createEmbeddableTemplate(String template) {
+    	if(template.trim().length() == 0) {
+    		return;
+    	}
+    	configuration.setEmbeddableTemplate(template);
+    }
+    
+    private void createEmbeddableSuperTemplate(String template) {
+    	if(template.trim().length() == 0) {
+    		return;
+    	}
+    	configuration.setEmbeddableSuperTemplate(template);
+    }
+    
+    private void createQueryTemplate(String template) {
+    	if(template.trim().length() == 0) {
+    		return;
+    	}
+    	configuration.setQueryTemplate(template);
+    }
+    
+    private void createQuerySuperTemplate(String template) {
+    	if(template.trim().length() == 0) {
+    		return;
+    	}
+    	configuration.setQuerySuperTemplate(template);
+    }
 
     private void createOutputPattern(String pattern) {
         if(pattern.trim().length() == 0) {
diff --git a/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/CodeTemplateManager.java b/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/CodeTemplateManager.java
index f9f6108..09ea39a 100644
--- a/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/CodeTemplateManager.java
+++ b/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/CodeTemplateManager.java
@@ -43,17 +43,19 @@ public class CodeTemplateManager {
 
 	public static final String STANDARD_SERVER_SUPERCLASS = "Standard Server Superclass";
 	public static final String STANDARD_SERVER_SUBCLASS = "Standard Server Subclass";
-	public static final String SINGLE_SERVER_CLASS = "Single Server class";
+	public static final String SINGLE_SERVER_CLASS = "Single Server Class";
 	static final String STANDARD_CLIENT_SUPERCLASS = "Standard Client Superclass";
 	static final String STANDARD_CLIENT_SUBCLASS = "Standard Client Subclass";
 
-	private static final String STANDART_EMBEDDABLE_SUPERCLASS = "Standart Embeddable Superclass";
-	private static final String STANDART_EMBEDDABLE_SUBCLASS = "Standart Embeddable Subclass";
-	private static final String SINGLE_EMBEDDABLE_CLASS = "Single Embeddable class";
+	private static final String STANDARD_EMBEDDABLE_SUPERCLASS = "Standard Embeddable Superclass";
+	private static final String STANDARD_EMBEDDABLE_SUBCLASS = "Standard Embeddable Subclass";
+	private static final String SINGLE_EMBEDDABLE_CLASS = "Single Embeddable Class";
 
-	private static final String STANDART_DATAMAP_SUPERCLASS = "Standart DataMap Superclass";
-	private static final String STANDART_DATAMAP_SUBCLASS = "Standart DataMap Subclass";
-	private static final String SINGLE_DATAMAP_CLASS = "Single DataMap class";
+	private static final String STANDARD_SERVER_DATAMAP_SUPERCLASS = "Standard Server DataMap Superclass";
+	private static final String STANDARD_SERVER_DATAMAP_SUBCLASS = "Standard Server DataMap Subclass";
+	private static final String STANDARD_CLIENT_DATAMAP_SUPERCLASS = "Standard Client DataMap Superclass";
+	private static final String STANDARD_CLIENT_DATAMAP_SUBCLASS = "Standard Client DataMap Subclass";
+	private static final String SINGLE_DATAMAP_CLASS = "Single DataMap Class";
 
 	public static final String NODE_NAME = "codeTemplateManager";
 
@@ -65,11 +67,13 @@ public class CodeTemplateManager {
 	private Map<String, String> reverseCustomTemplate;
 	private Map<String, String> standardTemplates;
 
-	private List<String> standartEmbeddableTemplates;
-	private List<String> standartEmbeddableSuperclassTemplates;
+	private List<String> standardEmbeddableTemplates;
+	private List<String> standardEmbeddableSuperclassTemplates;
 
-	private List<String> standartDataMapTemplates;
-	private List<String> standartDataMapSuperclassTemplates;
+	private List<String> standardServerDataMapTemplates;
+	private List<String> standardServerDataMapSuperclassTemplates;
+	private List<String> standardClientDataMapTemplates;
+	private List<String> standardClientDataMapSuperclassTemplates;
 
 	private Map<String, String> reverseStandartTemplates;
 
@@ -93,19 +97,25 @@ public class CodeTemplateManager {
 		standardClientSubclassTemplates = new ArrayList<>();
 		standardClientSubclassTemplates.add(STANDARD_CLIENT_SUBCLASS);
 
-		standartEmbeddableTemplates = new ArrayList<>();
-		standartEmbeddableTemplates.add(SINGLE_EMBEDDABLE_CLASS);
-		standartEmbeddableTemplates.add(STANDART_EMBEDDABLE_SUBCLASS);
+		standardEmbeddableTemplates = new ArrayList<>();
+		standardEmbeddableTemplates.add(STANDARD_EMBEDDABLE_SUBCLASS);
+		standardEmbeddableTemplates.add(SINGLE_EMBEDDABLE_CLASS);
 
-		standartEmbeddableSuperclassTemplates = new ArrayList<>();
-		standartEmbeddableSuperclassTemplates.add(STANDART_EMBEDDABLE_SUPERCLASS);
+		standardEmbeddableSuperclassTemplates = new ArrayList<>();
+		standardEmbeddableSuperclassTemplates.add(STANDARD_EMBEDDABLE_SUPERCLASS);
 
-		standartDataMapTemplates = new ArrayList<>();
-		standartDataMapTemplates.add(STANDART_DATAMAP_SUBCLASS);
-		standartDataMapTemplates.add(SINGLE_DATAMAP_CLASS);
+		standardServerDataMapTemplates = new ArrayList<>();
+		standardServerDataMapTemplates.add(STANDARD_SERVER_DATAMAP_SUBCLASS);
+		standardServerDataMapTemplates.add(SINGLE_DATAMAP_CLASS);
 
-		standartDataMapSuperclassTemplates = new ArrayList<>();
-		standartDataMapSuperclassTemplates.add(STANDART_DATAMAP_SUPERCLASS);
+		standardServerDataMapSuperclassTemplates = new ArrayList<>();
+		standardServerDataMapSuperclassTemplates.add(STANDARD_SERVER_DATAMAP_SUPERCLASS);
+		
+		standardClientDataMapTemplates = new ArrayList<>();
+		standardClientDataMapTemplates.add(STANDARD_CLIENT_DATAMAP_SUBCLASS);
+		
+		standardClientDataMapSuperclassTemplates = new ArrayList<>();
+		standardClientDataMapSuperclassTemplates.add(STANDARD_CLIENT_DATAMAP_SUPERCLASS);
 
 		updateCustomTemplates(getTemplatePreferences(application));
 		reverseCustomTemplate = new HashMap<>();
@@ -120,13 +130,15 @@ public class CodeTemplateManager {
 		standardTemplates.put(STANDARD_CLIENT_SUBCLASS, ClientClassGenerationAction.SUBCLASS_TEMPLATE);
 		standardTemplates.put(SINGLE_SERVER_CLASS, ClassGenerationAction.SINGLE_CLASS_TEMPLATE);
 
-		standardTemplates.put(STANDART_EMBEDDABLE_SUPERCLASS, ClassGenerationAction.EMBEDDABLE_SUPERCLASS_TEMPLATE);
-		standardTemplates.put(STANDART_EMBEDDABLE_SUBCLASS, ClassGenerationAction.EMBEDDABLE_SUBCLASS_TEMPLATE);
+		standardTemplates.put(STANDARD_EMBEDDABLE_SUPERCLASS, ClassGenerationAction.EMBEDDABLE_SUPERCLASS_TEMPLATE);
+		standardTemplates.put(STANDARD_EMBEDDABLE_SUBCLASS, ClassGenerationAction.EMBEDDABLE_SUBCLASS_TEMPLATE);
 		standardTemplates.put(SINGLE_EMBEDDABLE_CLASS, ClassGenerationAction.EMBEDDABLE_SINGLE_CLASS_TEMPLATE);
 
-		standardTemplates.put(STANDART_DATAMAP_SUBCLASS, ClassGenerationAction.DATAMAP_SUBCLASS_TEMPLATE);
+		standardTemplates.put(STANDARD_SERVER_DATAMAP_SUBCLASS, ClassGenerationAction.DATAMAP_SUBCLASS_TEMPLATE);
+		standardTemplates.put(STANDARD_CLIENT_DATAMAP_SUBCLASS, ClientClassGenerationAction.DMAP_SUBCLASS_TEMPLATE);
 		standardTemplates.put(SINGLE_DATAMAP_CLASS, ClassGenerationAction.DATAMAP_SINGLE_CLASS_TEMPLATE);
-		standardTemplates.put(STANDART_DATAMAP_SUPERCLASS, ClassGenerationAction.DATAMAP_SUPERCLASS_TEMPLATE);
+		standardTemplates.put(STANDARD_SERVER_DATAMAP_SUPERCLASS, ClassGenerationAction.DATAMAP_SUPERCLASS_TEMPLATE);
+		standardTemplates.put(STANDARD_CLIENT_DATAMAP_SUPERCLASS, ClientClassGenerationAction.DMAP_SUPERCLASS_TEMPLATE);
 
 		reverseStandartTemplates = new HashMap<>();
 		reverseStandartTemplates.put(ClassGenerationAction.SUBCLASS_TEMPLATE, STANDARD_SERVER_SUBCLASS);
@@ -135,13 +147,15 @@ public class CodeTemplateManager {
 		reverseStandartTemplates.put(ClientClassGenerationAction.SUPERCLASS_TEMPLATE, STANDARD_CLIENT_SUPERCLASS);
 		reverseStandartTemplates.put(ClassGenerationAction.SUPERCLASS_TEMPLATE, STANDARD_SERVER_SUPERCLASS);
 
-		reverseStandartTemplates.put(ClassGenerationAction.EMBEDDABLE_SUPERCLASS_TEMPLATE, STANDART_EMBEDDABLE_SUPERCLASS);
-		reverseStandartTemplates.put(ClassGenerationAction.EMBEDDABLE_SUBCLASS_TEMPLATE, STANDART_EMBEDDABLE_SUBCLASS);
+		reverseStandartTemplates.put(ClassGenerationAction.EMBEDDABLE_SUPERCLASS_TEMPLATE, STANDARD_EMBEDDABLE_SUPERCLASS);
+		reverseStandartTemplates.put(ClassGenerationAction.EMBEDDABLE_SUBCLASS_TEMPLATE, STANDARD_EMBEDDABLE_SUBCLASS);
 		reverseStandartTemplates.put(ClassGenerationAction.EMBEDDABLE_SINGLE_CLASS_TEMPLATE, SINGLE_EMBEDDABLE_CLASS);
 
-		reverseStandartTemplates.put(ClassGenerationAction.DATAMAP_SUBCLASS_TEMPLATE, STANDART_DATAMAP_SUBCLASS);
+		reverseStandartTemplates.put(ClassGenerationAction.DATAMAP_SUBCLASS_TEMPLATE, STANDARD_SERVER_DATAMAP_SUBCLASS);
+		reverseStandartTemplates.put(ClientClassGenerationAction.DMAP_SUBCLASS_TEMPLATE, STANDARD_CLIENT_DATAMAP_SUBCLASS);
 		reverseStandartTemplates.put(ClassGenerationAction.DATAMAP_SINGLE_CLASS_TEMPLATE, SINGLE_DATAMAP_CLASS);
-		reverseStandartTemplates.put(ClassGenerationAction.DATAMAP_SUPERCLASS_TEMPLATE, STANDART_DATAMAP_SUPERCLASS);
+		reverseStandartTemplates.put(ClassGenerationAction.DATAMAP_SUPERCLASS_TEMPLATE, STANDARD_SERVER_DATAMAP_SUPERCLASS);
+		reverseStandartTemplates.put(ClientClassGenerationAction.DMAP_SUPERCLASS_TEMPLATE, STANDARD_CLIENT_DATAMAP_SUPERCLASS);
 	}
 
 	/**
@@ -212,14 +226,26 @@ public class CodeTemplateManager {
 	}
 
 	public List<String> getStandartEmbeddableTemplates() {
-		return standartEmbeddableTemplates;
+		return standardEmbeddableTemplates;
 	}
 
 	public List<String> getStandartEmbeddableSuperclassTemplates() {
-		return standartEmbeddableSuperclassTemplates;
+		return standardEmbeddableSuperclassTemplates;
 	}
 
-	public List<String> getStandartDataMapTemplates() { return standartDataMapTemplates; }
+	public List<String> getStandartDataMapTemplates() {
+		return standardServerDataMapTemplates; 
+	}
 
-	public List<String> getStandartDataMapSuperclassTemplates() { return standartDataMapSuperclassTemplates; }
+	public List<String> getStandartDataMapSuperclassTemplates() {
+		return standardServerDataMapSuperclassTemplates;
+	}
+	
+	public List<String> getStandardClientDataMapTemplates() {
+		return standardClientDataMapTemplates; 
+	}
+	
+	public List<String> getStandardClientDataMapSuperclassTemplates() { 
+		return standardClientDataMapSuperclassTemplates; 
+	}
 }
diff --git a/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/editor/cgen/CustomModeController.java b/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/editor/cgen/CustomModeController.java
index 46f8f33..7dfc013 100644
--- a/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/editor/cgen/CustomModeController.java
+++ b/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/editor/cgen/CustomModeController.java
@@ -79,8 +79,32 @@ public class CustomModeController extends GeneratorController {
         Collections.sort(subTemplates);
         subTemplates.addAll(customTemplates);
 
+        List<String> querySuperTemplates = isClient ?
+                new ArrayList<>(templateManager.getStandardClientDataMapSuperclassTemplates()) :
+                new ArrayList<>(templateManager.getStandartDataMapSuperclassTemplates());
+        Collections.sort(querySuperTemplates);
+        querySuperTemplates.addAll(customTemplates);
+
+        List<String> queryTemplates = isClient ?
+                new ArrayList<>(templateManager.getStandardClientDataMapTemplates()) :
+                new ArrayList<>(templateManager.getStandartDataMapTemplates());
+        Collections.sort(queryTemplates);
+        queryTemplates.addAll(customTemplates);
+        
+        List<String> embeddableSuperTemplates = new ArrayList<>(templateManager.getStandartEmbeddableSuperclassTemplates());
+		Collections.sort(embeddableSuperTemplates);
+		embeddableSuperTemplates.addAll(customTemplates);
+        		
+		List<String> embeddableTemplates = new ArrayList<>(templateManager.getStandartEmbeddableTemplates());
+		Collections.sort(embeddableTemplates);
+		embeddableTemplates.addAll(customTemplates);
+
         this.view.getSubclassTemplate().getComboBox().setModel(new DefaultComboBoxModel<>(subTemplates.toArray(new String[0])));
         this.view.getSuperclassTemplate().getComboBox().setModel(new DefaultComboBoxModel<>(superTemplates.toArray(new String[0])));
+        this.view.getQueryTemplate().getComboBox().setModel(new DefaultComboBoxModel<>(queryTemplates.toArray(new String[0])));
+        this.view.getQuerySuperTemplate().getComboBox().setModel(new DefaultComboBoxModel<>(querySuperTemplates.toArray(new String[0])));
+        this.view.getEmbeddableTemplate().getComboBox().setModel(new DefaultComboBoxModel<>(embeddableTemplates.toArray(new String[0])));
+        this.view.getEmbeddableSuperTemplate().getComboBox().setModel(new DefaultComboBoxModel<>(embeddableSuperTemplates.toArray(new String[0])));
     }
 
     public Component getView() {
@@ -108,8 +132,22 @@ public class CustomModeController extends GeneratorController {
                 cgenConfiguration.getTemplate(), cgenConfiguration.getRootPath());
         String superTemplateName = getApplication().getCodeTemplateManager().getNameByPath(
                 cgenConfiguration.getSuperTemplate(), cgenConfiguration.getRootPath());
+        String embeddableTemplateName = getApplication().getCodeTemplateManager().getNameByPath(
+        		cgenConfiguration.getEmbeddableTemplate(), cgenConfiguration.getRootPath());
+        String embeddableSuperTemplateName = getApplication().getCodeTemplateManager().getNameByPath(
+        		cgenConfiguration.getEmbeddableSuperTemplate(), cgenConfiguration.getRootPath());
+        String queryTemplateName = getApplication().getCodeTemplateManager().getNameByPath(
+        		cgenConfiguration.getQueryTemplate(), cgenConfiguration.getRootPath());
+        String querySuperTemplateName = getApplication().getCodeTemplateManager().getNameByPath(
+        		cgenConfiguration.getQuerySuperTemplate(), cgenConfiguration.getRootPath());
+        
         String path = cgenConfiguration.getRootPath().resolve(Paths.get(cgenConfiguration.getTemplate())).normalize().toString();
         String superPath = cgenConfiguration.getRootPath().resolve(Paths.get(cgenConfiguration.getSuperTemplate())).normalize().toString();
+        String embeddableTemplatePath = cgenConfiguration.getRootPath().resolve(Paths.get(cgenConfiguration.getEmbeddableTemplate())).normalize().toString();
+        String embeddableSuperTemplatePath = cgenConfiguration.getRootPath().resolve(Paths.get(cgenConfiguration.getEmbeddableSuperTemplate())).normalize().toString();
+        String queryTemplatePath = cgenConfiguration.getRootPath().resolve(Paths.get(cgenConfiguration.getQueryTemplate())).normalize().toString();
+        String querySuperTemplatePath = cgenConfiguration.getRootPath().resolve(Paths.get(cgenConfiguration.getQuerySuperTemplate())).normalize().toString();
+        
         if(templateName == null && superTemplateName == null) {
             view.getSubclassTemplate().setItem(null);
             view.getSuperclassTemplate().setItem(null);
@@ -124,6 +162,27 @@ public class CustomModeController extends GeneratorController {
             view.getSubclassTemplate().setItem(templateName);
             view.getSuperclassTemplate().setItem(superTemplateName);
         }
+        
+        if(embeddableTemplateName == null && embeddableSuperTemplateName == null) {
+        	missTemplateDialog(cgenConfiguration, embeddableTemplatePath, embeddableSuperTemplatePath);
+        } else if(embeddableTemplateName == null) {
+        	missTemplateDialog(cgenConfiguration, embeddableTemplatePath, null);
+        } else if(embeddableSuperTemplateName == null) {
+        	missTemplateDialog(cgenConfiguration, null, embeddableSuperTemplatePath);
+        }
+    	view.getEmbeddableTemplate().setItem(embeddableTemplateName);
+    	view.getEmbeddableSuperTemplate().setItem(embeddableSuperTemplateName);
+    	
+    	if(queryTemplateName == null && querySuperTemplateName == null) {
+    		missTemplateDialog(cgenConfiguration, queryTemplatePath, querySuperTemplatePath);
+    	} else if(queryTemplateName == null) {
+    		missTemplateDialog(cgenConfiguration, queryTemplatePath, null);
+    	} else if(querySuperTemplateName == null) {
+    		missTemplateDialog(cgenConfiguration, null, querySuperTemplatePath);
+    	}
+    	view.getQueryTemplate().setItem(queryTemplateName);
+    	view.getQuerySuperTemplate().setItem(querySuperTemplateName);
+        
         view.setDisableSuperComboBoxes(view.getPairs().isSelected());
     }
 
@@ -179,9 +238,13 @@ public class CustomModeController extends GeneratorController {
             if(isSelected) {
                 cgenConfiguration.setTemplate(ClientClassGenerationAction.SUBCLASS_TEMPLATE);
                 cgenConfiguration.setSuperTemplate(ClientClassGenerationAction.SUPERCLASS_TEMPLATE);
+                cgenConfiguration.setQueryTemplate(ClientClassGenerationAction.DMAP_SUBCLASS_TEMPLATE);
+                cgenConfiguration.setQuerySuperTemplate(ClientClassGenerationAction.DMAP_SUPERCLASS_TEMPLATE);
             } else {
                 cgenConfiguration.setTemplate(ClassGenerationAction.SUBCLASS_TEMPLATE);
                 cgenConfiguration.setSuperTemplate(ClassGenerationAction.SUPERCLASS_TEMPLATE);
+                cgenConfiguration.setQueryTemplate(ClassGenerationAction.DATAMAP_SUBCLASS_TEMPLATE);
+                cgenConfiguration.setQuerySuperTemplate(ClassGenerationAction.DATAMAP_SUPERCLASS_TEMPLATE);
             }
             updateTemplates();
             String templateName = getApplication().getCodeTemplateManager().getNameByPath(
@@ -194,8 +257,20 @@ public class CustomModeController extends GeneratorController {
                             ClientClassGenerationAction.SUBCLASS_TEMPLATE :
                             ClassGenerationAction.SUBCLASS_TEMPLATE,
                     cgenConfiguration.getRootPath());
+            String queryTemplateName = getApplication().getCodeTemplateManager().getNameByPath(
+            		isSelected ?
+            				ClientClassGenerationAction.DMAP_SUBCLASS_TEMPLATE :
+            					ClassGenerationAction.DATAMAP_SUBCLASS_TEMPLATE,
+            					cgenConfiguration.getRootPath());
+            String querySuperTemplateName = getApplication().getCodeTemplateManager().getNameByPath(
+            		isSelected ?
+            				ClientClassGenerationAction.DMAP_SUPERCLASS_TEMPLATE :
+            					ClassGenerationAction.DATAMAP_SUPERCLASS_TEMPLATE,
+            					cgenConfiguration.getRootPath());
             view.getSubclassTemplate().setItem(templateName);
             view.getSuperclassTemplate().setItem(superTemplateName);
+            view.getQueryTemplate().setItem(queryTemplateName);
+            view.getQuerySuperTemplate().setItem(querySuperTemplateName);
             if(!getParentController().isInitFromModel()) {
                 getParentController().getProjectController().setDirty(true);
             }
diff --git a/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/editor/cgen/CustomModePanel.java b/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/editor/cgen/CustomModePanel.java
index 7bb7912..cada7f9 100644
--- a/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/editor/cgen/CustomModePanel.java
+++ b/modeler/cayenne-modeler/src/main/java/org/apache/cayenne/modeler/editor/cgen/CustomModePanel.java
@@ -44,6 +44,10 @@ public class CustomModePanel extends GeneratorControllerPanel {
 
     private ComboBoxAdapter<String> subclassTemplate;
     private ComboBoxAdapter<String> superclassTemplate;
+    private ComboBoxAdapter<String> embeddableTemplate;
+    private ComboBoxAdapter<String> embeddableSuperTemplate;
+    private ComboBoxAdapter<String> queryTemplate;
+    private ComboBoxAdapter<String> querySuperTemplate;
     private JCheckBox pairs;
     private JCheckBox overwrite;
     private JCheckBox usePackagePath;
@@ -57,6 +61,7 @@ public class CustomModePanel extends GeneratorControllerPanel {
 
     CustomModePanel(ProjectController projectController, CodeGeneratorControllerBase codeGeneratorControllerBase) {
         super(projectController, codeGeneratorControllerBase);
+        
         JComboBox<String> superclassField = new JComboBox<>();
         this.superclassTemplate = new ComboBoxAdapter<String>(superclassField) {
             @Override
@@ -82,6 +87,58 @@ public class CustomModePanel extends GeneratorControllerPanel {
                 }
             }
         };
+        
+        JComboBox<String> embeddableField = new JComboBox<>();
+        this.embeddableTemplate = new ComboBoxAdapter<String>(embeddableField) {
+        	@Override
+        	protected void updateModel(String item) throws ValidationException {
+        		CgenConfiguration cgenConfiguration = getCgenByDataMap();
+        		cgenConfiguration.setEmbeddableTemplate(Application.getInstance().getCodeTemplateManager().getTemplatePath(item,
+        				cgenConfiguration.getDataMap().getConfigurationSource()));
+        		if(!codeGeneratorControllerBase.isInitFromModel()) {
+        			projectController.setDirty(true);
+        		}
+        	}
+        };
+        
+        JComboBox<String> embeddableSuperField = new JComboBox<>();
+        this.embeddableSuperTemplate = new ComboBoxAdapter<String>(embeddableSuperField) {
+        	@Override
+        	protected void updateModel(String item) throws ValidationException {
+        		CgenConfiguration cgenConfiguration = getCgenByDataMap();
+        		cgenConfiguration.setEmbeddableSuperTemplate(Application.getInstance().getCodeTemplateManager().getTemplatePath(item,
+        				cgenConfiguration.getDataMap().getConfigurationSource()));
+        		if(!codeGeneratorControllerBase.isInitFromModel()) {
+        			projectController.setDirty(true);
+        		}
+        	}
+        };
+        
+        JComboBox<String> queryField = new JComboBox<>();
+        this.queryTemplate = new ComboBoxAdapter<String>(queryField) {
+        	@Override
+        	protected void updateModel(String item) throws ValidationException {
+        		CgenConfiguration cgenConfiguration = getCgenByDataMap();
+        		cgenConfiguration.setQueryTemplate(Application.getInstance().getCodeTemplateManager().getTemplatePath(item,
+        				cgenConfiguration.getDataMap().getConfigurationSource()));
+        		if(!codeGeneratorControllerBase.isInitFromModel()) {
+        			projectController.setDirty(true);
+        		}
+        	}
+        };
+        
+        JComboBox<String> querySuperField = new JComboBox<>();
+        this.querySuperTemplate = new ComboBoxAdapter<String>(querySuperField) {
+        	@Override
+        	protected void updateModel(String item) throws ValidationException {
+        		CgenConfiguration cgenConfiguration = getCgenByDataMap();
+        		cgenConfiguration.setQuerySuperTemplate(Application.getInstance().getCodeTemplateManager().getTemplatePath(item,
+        				cgenConfiguration.getDataMap().getConfigurationSource()));
+        		if(!codeGeneratorControllerBase.isInitFromModel()) {
+        			projectController.setDirty(true);
+        		}
+        	}
+        };
 
         this.pairs = new JCayenneCheckBox();
         this.overwrite = new JCayenneCheckBox();
@@ -121,7 +178,7 @@ public class CustomModePanel extends GeneratorControllerPanel {
 
         // assemble
         FormLayout layout = new FormLayout(
-                "right:74dlu, 1dlu, fill:240:grow, 1dlu, left:100dlu, 100dlu", "");
+                "right:105dlu, 1dlu, fill:240:grow, 1dlu, left:100dlu, 100dlu", "");
         DefaultFormBuilder builder = new DefaultFormBuilder(layout);
         builder.setDefaultDialogBorder();
 
@@ -133,6 +190,18 @@ public class CustomModePanel extends GeneratorControllerPanel {
 
         builder.append("Superclass Template:", superclassTemplate.getComboBox());
         builder.nextLine();
+        
+        builder.append("Embeddable Template:", embeddableTemplate.getComboBox());
+        builder.nextLine();
+        
+        builder.append("Embeddable Superclass Template:", embeddableSuperTemplate.getComboBox());
+        builder.nextLine();
+        
+        builder.append("DataMap Template:", queryTemplate.getComboBox());
+        builder.nextLine();
+        
+        builder.append("DataMap Superclass Template:", querySuperTemplate.getComboBox());
+        builder.nextLine();
 
         builder.append("Output Pattern:", outputPattern.getComponent());
         builder.nextLine();
@@ -167,17 +236,37 @@ public class CustomModePanel extends GeneratorControllerPanel {
 
     public void setDisableSuperComboBoxes(boolean val){
         superclassTemplate.getComboBox().setEnabled(val);
+        embeddableSuperTemplate.getComboBox().setEnabled(val);
+        querySuperTemplate.getComboBox().setEnabled(val);
     }
 
     public JButton getManageTemplatesLink() {
         return manageTemplatesLink;
     }
 
-    public ComboBoxAdapter<String> getSubclassTemplate() { return subclassTemplate; }
+    public ComboBoxAdapter<String> getSubclassTemplate() { 
+    	return subclassTemplate; 
+    }
 
     public ComboBoxAdapter<String> getSuperclassTemplate() {
         return superclassTemplate;
     }
+	
+    public ComboBoxAdapter<String> getEmbeddableTemplate() { 
+    	return embeddableTemplate; 
+    }
+    
+    public ComboBoxAdapter<String> getEmbeddableSuperTemplate() {
+    	return embeddableSuperTemplate;
+    }
+	
+    public ComboBoxAdapter<String> getQueryTemplate() { 
+    	return queryTemplate; 
+    }
+    
+    public ComboBoxAdapter<String> getQuerySuperTemplate() {
+    	return querySuperTemplate;
+    }
 
     public JCheckBox getOverwrite() {
         return overwrite;