/*

  * Copyright 2006 Outsource Cafe, Inc.

  *

  * Licensed 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.javagen.revgen.visitor;

 import org.javagen.agile.core.Generator;

 import org.javagen.agile.core.model.Model;

 import org.javagen.agile.core.util.ModelUtil;

 import org.javagen.agile.db.visitor.IdAssignerDatabaseVisitor;

 import org.javagen.agile.oo.visitor.IdAssignerOOVisitor;

 /**

  * Generates IDs for both Java and DB model instances.

  *

  * @author Richard Easterling

  */

 public class IdAssignerDb2OO implements Generator {

     protected IdAssignerOOVisitor ooVisitor = new IdAssignerOOVisitor();

     protected IdAssignerDatabaseVisitor dbVisitor = new IdAssignerDatabaseVisitor();

     public Model gen(Model input) {

         Model root = ModelUtil.getRootNode(input);

         dbVisitor.gen(root);

         ooVisitor.gen(root);

         return input;

     }

 }