1 files changed, 179 insertions, 0 deletions
diff --git a/das-java/tags/1.0-incubator-M2/rdb/src/main/java/org/apache/tuscany/das/rdb/impl/ChangeFactory.java b/das-java/tags/1.0-incubator-M2/rdb/src/main/java/org/apache/tuscany/das/rdb/impl/ChangeFactory.java
new file mode 100644
index 0000000000..c6628d3456
--- /dev/null
+++ b/das-java/tags/1.0-incubator-M2/rdb/src/main/java/org/apache/tuscany/das/rdb/impl/ChangeFactory.java
@@ -0,0 +1,179 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.tuscany.das.rdb.impl;
+
+import org.apache.log4j.Logger;
+import org.apache.tuscany.das.rdb.config.Create;
+import org.apache.tuscany.das.rdb.config.Delete;
+import org.apache.tuscany.das.rdb.config.Table;
+import org.apache.tuscany.das.rdb.config.Update;
+import org.apache.tuscany.das.rdb.config.wrapper.MappingWrapper;
+import org.apache.tuscany.das.rdb.config.wrapper.TableWrapper;
+import org.apache.tuscany.das.rdb.generator.impl.DeleteGenerator;
+import org.apache.tuscany.das.rdb.generator.impl.InsertGenerator;
+import org.apache.tuscany.das.rdb.generator.impl.UpdateGenerator;
+import org.apache.tuscany.das.rdb.util.LoggerFactory;
+
+import commonj.sdo.DataObject;
+
+public class ChangeFactory {
+    private final Logger logger = LoggerFactory.INSTANCE.getLogger(ChangeFactory.class);
+
+    private InsertCommandImpl createCommand;
+
+    private UpdateCommandImpl updateCommand;
+
+    private DeleteCommandImpl deleteCommand;
+
+    private final MappingWrapper mapping;
+
+    private final ConnectionImpl connection;
+
+    public ChangeFactory(MappingWrapper mapping, ConnectionImpl connection) {
+        this.mapping = mapping;
+        this.connection = connection;
+    }
+
+    public void setCreateCommand(InsertCommandImpl cmd) {
+        createCommand = cmd;
+    }
+
+    public void setUpdateCommand(UpdateCommandImpl cmd) {
+        if (this.logger.isDebugEnabled()) {
+            this.logger.debug("Setting Update Command to " + cmd);
+        }
+
+        updateCommand = cmd;
+    }
+
+    public void setDeleteCommand(DeleteCommandImpl cmd) {
+        deleteCommand = cmd;
+    }
+
+    ChangeOperation createUpdateOperation(DataObject changedObject, String propagatedID) {
+        return new UpdateOperation(getUpdateCommand(changedObject), changedObject, propagatedID);
+    }
+
+    ChangeOperation createUpdateOperation(DataObject changedObject) {
+        return createUpdateOperation(changedObject, null);
+    }
+
+    ChangeOperation createDeleteOperation(DataObject changedObject) {
+        return new DeleteOperation(getDeleteCommand(changedObject), changedObject);
+    }
+
+    ChangeOperation createInsertOperation(DataObject changedObject, String propagatedID) {
+        return new CreateOperation(getCreateCommand(changedObject), changedObject, propagatedID);
+    }
+
+    private InsertCommandImpl getCreateCommand(DataObject changedObject) {
+
+        if (createCommand == null) {
+            Table table = mapping.getTableByTypeName(changedObject.getType().getName());
+            if (table == null) {
+                if (changedObject.getType().getProperty("ID") != null) {
+                    // If the table is not defined in the config, assume it has a primary key of "ID"
+                    mapping.addPrimaryKey(changedObject.getType().getName() + ".ID");
+                    table = mapping.getTableByTypeName(changedObject.getType().getName());
+                } else {
+                    throw new RuntimeException("Table " + changedObject.getType().getName()
+                            + " was changed in the DataGraph but is not present in the Config");
+                }
+            }
+
+            Create create = table.getCreate();
+
+            if (create == null) {
+                createCommand = InsertGenerator.INSTANCE.getInsertCommand(mapping, changedObject, table);
+            } else {
+                createCommand = new InsertCommandImpl(create);
+            }
+            createCommand.setConnection(connection);
+            createCommand.configWrapper = mapping;
+        }
+        return createCommand;
+    }
+
+    private DeleteCommandImpl getDeleteCommand(DataObject changedObject) {
+
+        if (deleteCommand == null) {
+            Table table = mapping.getTableByTypeName(changedObject.getType().getName());
+            if (table == null) {
+                if (changedObject.getType().getProperty("ID") != null) {
+                    // If the table is not defined in the config, assume it has a primary key of "ID"
+                    mapping.addPrimaryKey(changedObject.getType().getName() + ".ID");
+                    table = mapping.getTableByTypeName(changedObject.getType().getName());
+                } else {
+                    throw new RuntimeException("Table " + changedObject.getType().getName()
+                            + " was changed in the DataGraph but is not present in the Config");
+                }
+            }
+
+            Delete delete = table.getDelete();
+
+            if (delete == null) {
+                deleteCommand = DeleteGenerator.INSTANCE.getDeleteCommand(table);
+            } else {
+                deleteCommand = new DeleteCommandImpl(delete);
+            }
+            deleteCommand.setConnection(connection);
+            deleteCommand.configWrapper = mapping;
+        }
+        return deleteCommand;
+    }
+
+    private UpdateCommandImpl getUpdateCommand(DataObject changedObject) {
+
+        if (updateCommand == null) {
+            Table table = mapping.getTableByTypeName(changedObject.getType().getName());
+            if (table == null) {
+                if (changedObject.getType().getProperty("ID") != null) {
+                    mapping.addPrimaryKey(changedObject.getType().getName() + ".ID");
+                    table = mapping.getTableByTypeName(changedObject.getType().getName());
+                } else {
+                    throw new RuntimeException("Table " + changedObject.getType().getName()
+                            + " was changed in the DataGraph but is not present in the Config");
+                }
+            }
+            Update update = table.getUpdate();
+            if (update == null) {
+                updateCommand = UpdateGenerator.INSTANCE.getUpdateCommand(mapping, changedObject, table);
+            } else {
+                TableWrapper t = new TableWrapper(table);
+                if (t.getCollisionColumn() != null) {
+                    updateCommand = new OptimisticWriteCommandImpl(update);
+                } else {
+                    updateCommand = new UpdateCommandImpl(update);
+                }
+            }
+            updateCommand.setConnection(connection);
+            updateCommand.configWrapper = mapping;
+        }
+        if (this.logger.isDebugEnabled()) {
+            this.logger.debug("Returning updateCommand: " + updateCommand);
+        }
+
+        return updateCommand;
+    }
+
+    public MappingWrapper getConfig() {
+        return this.mapping;
+    }
+
+}