DatabaseClasses.java

package com.nortal.jroad.typegen.database;

import com.nortal.jroad.enums.XRoadProtocolVersion;
import com.nortal.jroad.model.XmlBeansXRoadMetadata;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

/**
 * @author Roman Tekhov
 */
public class DatabaseClasses {

  private String baseDirectory;
  private String packageName;
  private XRoadProtocolVersion version;

  private Map<String, DatabaseClass> classes = new HashMap<String, DatabaseClass>();


  public DatabaseClasses(String baseDirectory, String packageName, XRoadProtocolVersion version) {
    this.baseDirectory = baseDirectory;
    this.packageName = packageName;
    this.version = version;
  }


  public void add(String database, XmlBeansXRoadMetadata metadata) throws IOException {
    DatabaseClass databaseClass = classes.get(database);
    if (databaseClass == null) {
      databaseClass = new DatabaseClass(database, packageName, version);
      classes.put(database, databaseClass);
    }

    try {
      databaseClass.add(new DatabaseServiceMethod(metadata, baseDirectory));
    } catch (NoDescriptionFoundException e) {
      System.err.println("No XSB found for " + metadata.getOperationName());
    }
  }

  public Map<String, DatabaseClass> getClasses() {
    return classes;
  }
}