com.jtstand.TestStation Maven / Gradle / Ivy
/*
* Copyright (c) 2009 Albert Kurucz.
*
* This file, TestStation.java is part of JTStand.
*
* JTStand is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* JTStand is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with GTStand. If not, see properties = new ArrayList();
@OneToMany(cascade = CascadeType.ALL, mappedBy = "testStation", fetch = FetchType.LAZY)
@OrderBy("testLimitPosition ASC")
private List testLimits = new ArrayList();
// @OneToMany(cascade = CascadeType.ALL, mappedBy = "testStation")
// @OrderBy("testTypeReferencePosition ASC")
// private List testTypes = new ArrayList();
@ManyToOne
private FileRevision creator;
@ManyToOne
private TestProject testProject;
@OneToMany(cascade = CascadeType.ALL, mappedBy = "testStation")
@OrderBy("testFixturePosition ASC")
private List fixtures = new ArrayList();
private int testStationPosition;
private static EntityManagerFactory entityManagerFactory;
private transient final Object propertiesLock = new Object();
// private transient final Object testTypesLock = new Object();
private transient final Object testFixturesLock = new Object();
private transient final Object testLimitsLock = new Object();
private transient Map localVariablesMap = new HashMap();
// public void initializeProperties() throws ScriptException {
// for (TestProjectProperty tp : getTestProject().getProperties()) {
// if (tp.isEager() != null && tp.isEager()) {
// System.out.println("Evaluating eager project property: " + tp.getName());
// put(tp.getName(), tp.getPropertyObject(getBindings()));
// }
// }
// for (TestStationProperty tp : properties) {
// if (tp.isEager() != null && tp.isEager()) {
// System.out.println("Evaluating eager station property: " + tp.getName());
// put(tp.getName(), tp.getPropertyObject(getBindings()));
// }
// }
// }
@XmlElement(name = "limit")
public List getTestLimits() {
synchronized (testLimitsLock) {
if (testLimits == null) {
System.err.println("testLimits is null!");
}
return testLimits;
}
}
public void setTestLimits(List testLimits) {
this.testLimits = testLimits;
if (testLimits != null) {
for (ListIterator iterator = testLimits.listIterator(); iterator.hasNext();) {
int index = iterator.nextIndex();
TestStationLimit testLimit = iterator.next();
testLimit.setTestStation(this);
testLimit.setPosition(index);
}
}
}
// private Integer rmiPort;
//
// @XmlAttribute
// public Integer getRmiPort() {
// return rmiPort;
// }
//
// public void setRmiPort(Integer rmiPort) {
// this.rmiPort = rmiPort;
// }
public EntityManager createEntityManager() {
EntityManagerFactory theEmf = getEntityManagerFactory();
if (theEmf == null) {
return null;
}
return theEmf.createEntityManager();
}
public TestSequenceInstance getTestSequenceInstance(long id) {
EntityManagerFactory theEmf = getEntityManagerFactory();
if (theEmf == null) {
return null;
}
EntityManager em = createEntityManager();
if (em == null) {
return null;
}
TestSequenceInstance seq = em.find(TestSequenceInstance.class, id);
if (seq != null) {
seq.em = em;
}
return seq;
}
@XmlTransient
public EntityManagerFactory getEntityManagerFactory() {
if (entityManagerFactory == null && getTestProject() != null && getTestProject().getPun() != null) {
try {
entityManagerFactory = Persistence.createEntityManagerFactory(getTestProject().getPun(), getPeristencePropertiesFixedMap());
} catch (ScriptException ex) {
Logger.getLogger(TestStation.class.getName()).log(Level.SEVERE, null, ex);
}
}
return entityManagerFactory;
}
@XmlTransient
public Map getPeristencePropertiesFixedMap() throws ScriptException {
Map map = getPeristencePropertiesMap();
if (getPeristencePropertiesMap().get("hibernate.connection.driver_class") == null) {
map.put("hibernate.connection.driver_class", getDriver().driverClass);
}
if (getPeristencePropertiesMap().get("hibernate.dialect") == null) {
map.put("hibernate.dialect", getDriver().dialect);
}
if (getPeristencePropertiesMap().get("hibernate.connection.url") == null) {
String url = getPersistenceProtocol() + getPersistencePath();
System.out.println("hibernate.connection.url: " + url);
map.put("hibernate.connection.url", url);
}
if (Driver.derby.equals(getDriver()) && System.getProperty("derby.stream.error.file") == null) {
System.setProperty("derby.stream.error.file", getUserHome() + File.separator + ".jtstand" + File.separator + "derby.log");
}
return map;
}
@XmlTransient
private Map getPeristencePropertiesMap() throws ScriptException {
Map ppm = getTestProject() != null ? getTestProject().getPeristencePropertiesMap() : new HashMap();
Object sppm = getPropertyObject("persistenceProperties");
if (sppm != null) {
if (sppm instanceof Map) {
ppm.putAll((Map) sppm);
} else {
throw new IllegalArgumentException("Stations's persistenceProperties should be a Map, but it is: " + sppm.getClass().getCanonicalName());
}
}
return ppm;
}
@XmlTransient
public int getPosition() {
return testStationPosition;
}
public void setPosition(int position) {
this.testStationPosition = position;
}
@XmlTransient
public Long getId() {
return id;
}
// @XmlElement(name = "testType")
// public List getTestTypes() {
// synchronized (testTypesLock) {
// return testTypes;
// }
// }
//
// public void setTestTypes(List testTypes) {
// this.testTypes = testTypes;
// if (testTypes != null) {
// for (ListIterator iterator = testTypes.listIterator(); iterator.hasNext();) {
// int index = iterator.nextIndex();
// TestTypeReference testTypeReference = iterator.next();
// testTypeReference.setTestStation(this);
// testTypeReference.setPosition(index);
// }
// }
// }
@XmlAttribute
public String getHostName() {
return hostName;
}
public void setHostName(String hostName) {
this.hostName = hostName;
}
@XmlElement
public String getRemark() {
return remark;
}
public void setRemark(String remark) {
this.remark = remark;
}
@XmlElement(name = "property")
public List getProperties() {
synchronized (propertiesLock) {
return properties;
}
}
public void setProperties(List properties) {
this.properties = properties;
if (properties != null) {
for (ListIterator iterator = properties.listIterator(); iterator.hasNext();) {
int index = iterator.nextIndex();
TestStationProperty testStationProperty = iterator.next();
testStationProperty.setTestStation(this);
testStationProperty.setPosition(index);
}
}
}
@XmlElement(name = "fixture")
public List getFixtures() {
synchronized (testFixturesLock) {
return fixtures;
}
}
public void setFixtures(List fixtures) {
this.fixtures = fixtures;
if (fixtures != null) {
for (ListIterator iterator = fixtures.listIterator(); iterator.hasNext();) {
int index = iterator.nextIndex();
TestFixture testFixture = iterator.next();
testFixture.setTestStation(this);
testFixture.setPosition(index);
}
}
}
public TestFixture getTestFixture(String name) {
List testFixtures = getFixtures();
if (testFixtures != null) {
for (TestFixture tf : testFixtures) {
if (tf.getFixtureName().equals(name)) {
return tf;
}
}
}
return null;
}
@XmlTransient
public FileRevision getCreator() {
return creator;
}
public void setCreator(FileRevision creator) {
this.creator = creator;
setProperties(getProperties());
setTestLimits(getTestLimits());
setFixtures(getFixtures());
// setTestTypes(getTestTypes());
}
@XmlTransient
public TestProject getTestProject() {
return testProject;
}
public void setTestProject(TestProject testProject) {
this.testProject = testProject;
setFixtures(getFixtures());
if (testProject != null) {
setCreator(testProject.getCreator());
}
}
@Override
public int hashCode() {
int hash = 0;
hash += (creator != null ? creator.hashCode() : 0);
hash += (hostName != null ? hostName.hashCode() : 0);
return hash;
}
@Override
public boolean equals(Object object) {
if (!(object instanceof TestStation)) {
return false;
}
TestStation other = (TestStation) object;
if ((this.creator == null && other.getCreator() != null) || (this.creator != null && !this.creator.equals(other.getCreator()))) {
return false;
}
if ((this.hostName == null && other.getHostName() != null) || (this.hostName != null && !this.hostName.equals(other.getHostName()))) {
return false;
}
return true;
}
@Override
public String toString() {
return "[" + TestStation.class.getCanonicalName() + ":" + getCreator() + ":" + getHostName() + "]";
}
@Override
public Bindings getBindings() {
return this;
}
@Override
public Object getPropertyObjectUsingBindings(String keyString, Bindings bindings) throws ScriptException {
// if (bindings != null) {
// bindings.put("station", this);
// }
for (TestProperty tsp : getProperties()) {
if (tsp.getName().equals(keyString)) {
return tsp.getPropertyObject(bindings);
}
}
if (getTestProject() != null) {
return getTestProject().getPropertyObjectUsingBindings(keyString, bindings);
}
return null;
}
@XmlTransient
public List getHostNames() {
return queryStringList("select distinct ts.hostName from TestStation ts");
}
public List getTestFixtureNames(String hostName) {
return queryStringList((hostName != null && !hostName.equals("All")) ? "select distinct ts.fixtureName from TestFixture ts where ts.testStation.hostName = '" + hostName + "'" : "select distinct ts.fixtureName from TestFixture ts");
}
@XmlTransient
public List getPartNumbers() {
return queryStringList("select distinct ts.partNumber from Product ts");
}
public List getPartNumberRevs(String partNumber) {
return queryStringList((partNumber != null && !partNumber.equals("All")) ? "select distinct ts.partRevision from Product ts where ts.partNumber = '" + partNumber + "'" : "select distinct ts.partRevision from Product ts");
}
public List getTestTypes(String partNumber, String partRevision) {
if (partNumber == null || partNumber.equals("All")) {
if (partRevision == null || partRevision.equals("All")) {
return queryStringList("select distinct ts.name from TestType ts");
} else {
return queryStringList("select distinct ts.name from TestType ts where ts.product.partRevision = '" + partRevision + "'");
}
} else {
if (partRevision == null || partRevision.equals("All")) {
return queryStringList("select distinct ts.name from TestType ts where ts.product.partNumber = '" + partNumber + "'");
} else {
return queryStringList("select distinct ts.name from TestType ts where ts.product.partNumber = '" + partNumber + "' and ts.product.partRevision = '" + partRevision + "'");
}
}
}
@SuppressWarnings("unchecked")
public List queryStringList(String qstr) {
return (List) (new GeneralQuery(getEntityManagerFactory(), qstr)).query();
}
@XmlTransient
public String getPersistenceProtocol() throws ScriptException {
String pp = "jdbc:" + getDriver().toString() + ":";
System.out.println("Persistence protocol: " + pp);
return pp;
}
@Override
public void putAll(Map toMerge) {
for (Entry variableEntry : toMerge.entrySet()) {
put(variableEntry.getKey(), variableEntry.getValue());
}
}
@Override
public boolean containsKey(Object key) {
return super.containsKey(key.toString())
|| "station".equals(key)
|| localVariablesMap.containsKey(key.toString())
|| containsProperty(key.toString());
}
@Override
public Object get(Object key) {
if ("$type$".equals(key)) {
return getClass().getName();
}
if ("context".equals(key)) {
return ScriptContext.ENGINE_SCOPE;
}
if ("id".equals(key)) {
return id;
}
if ("testStation".equals(key)) {
return this;
}
if ("position".equals(key)) {
return testStationPosition;
}
if ("hostName".equals(key)) {
return hostName;
}
if ("remark".equals(key)) {
return remark;
}
if ("properties".equals(key)) {
return properties;
}
if ("testLimits".equals(key)) {
return testLimits;
}
if ("creator".equals(key)) {
return creator;
}
if ("testProject".equals(key)) {
return testProject;
}
if ("fixtures".equals(key)) {
return fixtures;
}
if (localVariablesMap.containsKey((String) key)) {
return localVariablesMap.get((String) key);
}
try {
return getVariable((String) key);
} catch (ScriptException ex) {
Logger.getLogger(TestFixture.class.getName()).log(Level.SEVERE, null, ex);
throw new IllegalStateException(ex.getMessage());
} catch (InterruptedException ex) {
throw new IllegalStateException(ex.getMessage());
}
}
@Override
public Object remove(Object key) {
return localVariablesMap.remove((String) key);
}
@Override
public int size() {
return keySet().size();
}
@Override
public boolean isEmpty() {
return false;
}
@Override
public void clear() {
super.clear();
localVariablesMap.clear();
}
@Override
public Set> entrySet() {
Set> entries = new HashSet>();
for (String key : keySet()) {
entries.add(new HashMap.SimpleEntry(key, get(key)));
}
return entries;
}
public static enum Driver {
h2("org.h2.Driver", "org.hibernate.dialect.H2Dialect"),
derby("org.apache.derby.jdbc.EmbeddedDriver", "org.hibernate.dialect.DerbyDialect"),
postgresql("org.postgresql.Driver", "org.hibernate.dialect.PostgreSQLDialect"),
mysql("com.mysql.jdbc.Driver", "org.hibernate.dialect.MySQLDialect");
public final String driverClass;
public final String dialect;
Driver(String driverClass, String dialect) {
this.driverClass = driverClass;
this.dialect = dialect;
}
}
@XmlTransient
public Driver getDriver() throws ScriptException {
String driverClass = getPeristencePropertiesMap().get("hibernate.connection.driver_class");
if (driverClass == null) {
return Driver.derby;
// return Driver.h2;
}
Driver[] drivers = Driver.values();
for (int i = 0; i < drivers.length; i++) {
if (drivers[i].driverClass.equals(driverClass)) {
return drivers[i];
}
}
throw new IllegalArgumentException("Unsupported driver: " + driverClass);
}
@XmlTransient
public String getPersistencePath() throws ScriptException {
String url = getPeristencePropertiesMap().get("hibernate.connection.url");
if (url == null) {
Driver driver = getDriver();
switch (driver) {
case h2:
return getUserHome() + File.separator + ".jtstand" + File.separator + "h2" + File.separator + "jtfw";
case derby:
return getUserHome() + File.separator + ".jtstand" + File.separator + "derby";
case postgresql:
return "//localhost/jtfw";
case mysql:
return "//localhost:3306/jtfw";
default:
throw new IllegalArgumentException("Database driver is not supported: " + driver);
}
}
if (!url.startsWith("jdbc:")) {
throw new IllegalArgumentException("url should start with 'jdbc:': " + url);
}
int start = url.indexOf(':', 5);
if (start < 0) {
throw new IllegalArgumentException("invalid url: " + url);
}
start++;
int end = url.indexOf(';');
String path = end > 0 ? url.substring(start, end) : url.substring(start);
System.out.println("Persistence path: " + path);
return path;
}
public boolean databaseReset(String username, String password) throws ClassNotFoundException, SQLException, ScriptException {
createDB(username, password);
return databaseUpdate();
}
private void createDB(String username, String password) throws ClassNotFoundException, SQLException, ScriptException {
switch (getDriver()) {
case h2:
case derby:
deleteDir(new File(getPersistencePath()));
break;
case postgresql:
createDB_PSQL(username, password);
break;
case mysql:
createDB_MYSQL(username, password);
break;
default:
throw new IllegalArgumentException("Unsupported database driver");
}
}
private static void createDB_MYSQL(String username, String password) throws ClassNotFoundException, SQLException {
String url = "jdbc:mysql://localhost:3306";
System.out.println("Loading driver:'" + TestStation.Driver.mysql.driverClass + "'...");
Class.forName(TestStation.Driver.mysql.driverClass);
System.out.println("Connecting...");
Connection c = DriverManager.getConnection(url, username, password);
Statement s = c.createStatement();
try {
System.out.println("Revoking user privileges...");
s.execute("REVOKE ALL PRIVILEGES, GRANT OPTION FROM 'dbuser'@'localhost'");
} catch (SQLException ex) {
System.out.println(ex.getMessage());
}
try {
System.out.println("Dropping user...");
s.execute("DROP USER 'dbuser'@'localhost'");
} catch (SQLException ex) {
System.out.println(ex.getMessage());
}
try {
System.out.println("Dropping database...");
s.execute("DROP DATABASE jtfw");
} catch (SQLException ex) {
System.out.println(ex.getMessage());
}
System.out.println("Creating database...");
s.execute("CREATE DATABASE jtfw");
System.out.println("Creating user...");
s.execute("CREATE USER 'dbuser'@'localhost' IDENTIFIED BY 'password'");
System.out.println("Grant...");
s.execute("GRANT ALL ON jtfw.* TO 'dbuser'@'localhost'");
System.out.println("Closing connection...");
s.close();
c.close();
System.out.println("New database is created.");
}
private static void createDB_PSQL(String username, String password) throws ClassNotFoundException, SQLException {
String url = "jdbc:postgresql://localhost/template1";
System.out.println("Loading driver:'" + TestStation.Driver.postgresql.driverClass + "'...");
Class.forName(TestStation.Driver.postgresql.driverClass);
System.out.println("Connecting...");
Connection c = DriverManager.getConnection(url, username, password);
Statement s = c.createStatement();
try {
System.out.println("Dropping database...");
s.execute("DROP DATABASE jtfw");
} catch (SQLException ex) {
System.out.println(ex.getMessage());
}
try {
System.out.println("Dropping user...");
s.execute("DROP USER dbuser");
} catch (SQLException ex) {
System.out.println(ex.getMessage());
}
System.out.println("Creating user...");
s.execute("CREATE USER dbuser WITH PASSWORD 'password'");
System.out.println("Creating database...");
s.execute("CREATE DATABASE jtfw OWNER dbuser TEMPLATE DEFAULT ENCODING 'UNICODE'");
System.out.println("Closing connection...");
s.close();
c.close();
System.out.println("New database is created.");
}
public static boolean deleteDir(File dir) {
if (dir.isDirectory()) {
String[] children = dir.list();
for (int i = 0; i < children.length; i++) {
boolean success = deleteDir(new File(dir, children[i]));
if (!success) {
return false;
}
}
}
// The directory is now empty so delete it
return dir.delete();
}
public boolean databaseValidate() throws ScriptException {
if (Driver.derby.equals(getDriver())) {
if (!(new File(getPersistencePath())).isDirectory()) {
System.out.println("Derby directory is missing: " + getPersistencePath());
return false;
}
}
if (Driver.h2.equals(getDriver())) {
String path = getPersistencePath();
String dir = path.substring(0, path.lastIndexOf(File.separatorChar));
System.out.println("H2 dir: " + dir);
if (!(new File(dir)).isDirectory()) {
System.out.println("H2 directory is missing: " + dir);
return false;
}
}
return auto("validate");
}
public boolean databaseUpdate() {
return auto("update");
}
private boolean auto(String operation) {
long startTime = System.currentTimeMillis();
System.out.println("Operation: " + operation + "...");
HibernateEntityManagerFactory emf = null;
EntityManager em = null;
try {
Map map = getPeristencePropertiesFixedMap();
if (operation != null) {
map.put("hibernate.hbm2ddl.auto", operation);
}
if (operation.equals("update")) {
if (getDriver().equals(TestStation.Driver.derby)) {
String url = map.get("hibernate.connection.url");
if (!url.contains("create=true")) {
url += ";create=true";
}
// if (!url.contains("logDevice=")) {
// url += ";logDevice=" + getUserHome() + File.separator + ".jtstand";
// }
map.put("hibernate.connection.url", url);
}
}
emf = (HibernateEntityManagerFactory) Persistence.createEntityManagerFactory(getTestProject().getPun(), map);
// emf.getSessionFactory().getAllClassMetadata();
// System.out.println(emf.getSessionFactory().getAllClassMetadata());
em = emf.createEntityManager();
em.getTransaction().begin();
em.getTransaction().commit();
// System.out.println("Closing entity manager");
em.close();
// System.out.println("Closing entity manager factory");
emf.close();
System.out.println("Database " + operation + " operation succeeded in " + Long.toString(System.currentTimeMillis() - startTime) + "ms");
return true;
} catch (Exception ex) {
ex.printStackTrace();
System.out.println(ex.getMessage());
if (em != null && em.isOpen()) {
em.close();
}
if (emf != null && emf.isOpen()) {
emf.close();
}
}
System.out.println("Database " + operation + " operation failed in " + Long.toString(System.currentTimeMillis() - startTime) + "ms");
return false;
}
public static String getUserHome() {
String userHome = System.getProperty("user.home");
// System.out.println("User home: " + userHome);
return userHome;
}
// public static File getUserHomeDirectory() {
// return new File(getUserHome());
// }
public static final String STR_SAVE_DIRECTORY = "jtstand.saveDirectory";
public static final String STR_SAVE_DIRECTORY_DEFAULT = "save";
public static final String STR_SAVED_DIRECTORY = "jtstand.savedDirectory";
public static final String STR_SAVED_DIRECTORY_DEFAULT = "saved";
public static final String STR_SAVED_ERROR_DIRECTORY = "jtstand.savedErrorDirectory";
public static final String STR_SAVED_ERROR_DIRECTORY_DEFAULT = "error";
public File getDirectoryParameter(String parameterName, String defaultDirectoryName) throws ScriptException {
File directory;
Object o = getPropertyObject(parameterName);
if (o != null) {
if (File.class.isAssignableFrom(o.getClass())) {
directory = (File) o;
} else {
directory = new File(o.toString());
}
} else {
directory = new File(getUserHome() + File.separator + ".jtstand" + File.separator + defaultDirectoryName);
}
if (directory.isFile()) {
throw new IllegalArgumentException("Directory is not a directory: a File is specified!");
}
if (!directory.isDirectory()) {
if (directory.mkdirs()) {
LOGGER.info("Directory is created: " + directory.getPath());
} else {
throw new IllegalArgumentException("Directory does not exist and cannot be created: " + directory.getPath());
}
}
if (!directory.canWrite()) {
throw new IllegalArgumentException("Directory does not exist and cannot be written: " + directory.getPath());
}
return directory;
}
@XmlTransient
public File getSaveDirectory() throws ScriptException {
return getDirectoryParameter(STR_SAVE_DIRECTORY, STR_SAVE_DIRECTORY_DEFAULT);
}
@XmlTransient
public File getSavedDirectory() throws ScriptException {
return getDirectoryParameter(STR_SAVED_DIRECTORY, STR_SAVED_DIRECTORY_DEFAULT);
}
@XmlTransient
public File getSavedErrorDirectory() throws ScriptException {
return getDirectoryParameter(STR_SAVED_ERROR_DIRECTORY, STR_SAVED_ERROR_DIRECTORY_DEFAULT);
}
public boolean containsProperty(String key) {
if ("station".equals(key)) {
return true;
}
for (TestProperty tsp : getProperties()) {
if (tsp.getName().equals(key)) {
return true;
}
}
return getTestProject().containsProperty(key);
}
public Object getVariable(String keyString) throws InterruptedException, ScriptException {
for (TestStationProperty tsp : getProperties()) {
if (tsp.getName().equals(keyString)) {
return getVariable(keyString, tsp, this);
}
}
if (getTestProject() != null) {
for (TestProjectProperty tsp : getTestProject().getProperties()) {
if (tsp.getName().equals(keyString)) {
return getVariable(keyString, tsp, this);
}
}
}
if ("out".equals(keyString)) {
return System.out;
}
throw new IllegalArgumentException("Undefined variable:" + keyString);
}
@Override
public Set keySet() {
Set keys = super.keySet();
keys.add("fixture");
keys.addAll(localVariablesMap.keySet());
return keys;
}
// @Override
// public Set getPropertyNames() {
// Set propertyNames = new HashSet();
// for (TestProperty tp : getProperties()) {
// propertyNames.add(tp.getName());
// }
// return propertyNames;
// }
@Override
public boolean containsValue(Object value) {
for (String key : keySet()) {
if (value.equals(get(key))) {
return true;
}
}
return false;
}
@Override
public Object put(String name, Object value) {
// System.out.println("putting to station variable: '" + name + "' value: " + value);
return super.put(name, value);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy