Mercurial > dive4elements > river
diff flys-artifacts/src/main/java/de/intevation/flys/artifacts/datacage/Recommendations.java @ 1015:9a1a3080ad98
Bring user specific meta data service to life.
flys-artifacts/trunk@2461 c6561f87-3c4e-4783-a992-168aeb5c3f6f
author | Sascha L. Teichmann <sascha.teichmann@intevation.de> |
---|---|
date | Wed, 03 Aug 2011 16:40:04 +0000 |
parents | |
children | d42fa223be48 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/flys-artifacts/src/main/java/de/intevation/flys/artifacts/datacage/Recommendations.java Wed Aug 03 16:40:04 2011 +0000 @@ -0,0 +1,252 @@ +package de.intevation.flys.artifacts.datacage; + +import java.util.Map; +import java.util.HashMap; +import java.util.List; +import java.util.ArrayList; + +import java.io.InputStream; +import java.io.IOException; +import java.io.File; + +import java.io.FileInputStream; + +import java.sql.Connection; +import java.sql.SQLException; + +import javax.sql.DataSource; + +import org.apache.log4j.Logger; + +import org.w3c.dom.Document; +import org.w3c.dom.Node; + +import org.hibernate.Session; + +import org.hibernate.jdbc.Work; + +import de.intevation.artifacts.common.utils.Config; +import de.intevation.artifacts.common.utils.XMLUtils; + +import de.intevation.flys.artifacts.FLYSArtifact; + +import de.intevation.flys.backend.SessionHolder; + +import de.intevation.artifactdatabase.data.StateData; + +import de.intevation.flys.artifacts.datacage.templating.Builder; + +public class Recommendations +{ + private static Logger log = Logger.getLogger(Recommendations.class); + + public static final String XPATH_SYSTEM_TEMPLATE = + "/artifact-database/metadata/system/@template"; + + public static final String XPATH_USER_TEMPLATE = + "/artifact-database/metadata/user/@template"; + + private static Recommendations INSTANCE; + + protected Builder systemBuilder; + protected Builder userBuilder; + + public Recommendations() { + } + + public Recommendations(Builder systemBuilder, Builder userBuilder) { + this.systemBuilder = systemBuilder; + this.userBuilder = userBuilder; + } + + public Builder getUserBuilder() { + return userBuilder; + } + + public Builder getSystemBuilder() { + return systemBuilder; + } + + protected static void artifactToParameters( + FLYSArtifact artifact, + Map<String, Object> parameters + ) { + parameters.put("current-state-id", artifact.getCurrentStateId()); + + for (StateData sd: artifact.getAllData()) { + Object value = sd.getValue(); + if (value == null) { + continue; + } + String key = sd.getName().replace('.', '-'); + parameters.put(key, value); + } + } + + public void recommend( + FLYSArtifact artifact, + String userId, + String [] outs, + Map<String, Object> extraParameters, + Node result + ) { + Map<String, Object> parameters = new HashMap<String, Object>(); + + if (extraParameters != null) { + parameters.putAll(extraParameters); + } + + if (userId != null) { + parameters.put("user-id", userId); + } + + if (artifact != null) { + artifactToParameters(artifact, parameters); + } + + parameters.put("artifact-outs", outs); + + parameters.put("parameters", parameters); + + recommend(parameters, userId, result); + } + + public void recommend( + Map<String, Object> parameters, + String userId, + Node result + ) { + recommend(parameters, userId, result, SessionHolder.HOLDER.get()); + } + + public void recommend( + final Map<String, Object> parameters, + final String userId, + final Node result, + Session session + ) { + if (systemBuilder == null || userBuilder == null) { + log.error("builder not configured properly."); + return; + } + + session.doWork(new Work() { + @Override + public void execute(Connection systemConnection) + throws SQLException + { + List<Builder.NamedConnection> connections = + new ArrayList<Builder.NamedConnection>(2); + + if (userId != null) { // system and user templates + connections.add(new Builder.NamedConnection( + Builder.CONNECTION_USER, systemConnection)); + + // get connection to datacage db + DataSource dataSource = DBConfig + .getInstance() + .getDBConnection() + .getDataSource(); + + Connection userConnection = dataSource.getConnection(); + try { + connections.add(new Builder.NamedConnection( + Builder.CONNECTION_SYSTEM, userConnection, false)); + + userBuilder.build(connections, result, parameters); + } + finally { + userConnection.close(); + } + } + else { // system template only + connections.add(new Builder.NamedConnection( + Builder.CONNECTION_SYSTEM, systemConnection)); + + systemBuilder.build(connections, result, parameters); + } + } + }); + } + + public static synchronized Recommendations getInstance() { + if (INSTANCE == null) { + INSTANCE = createRecommendations(); + } + return INSTANCE; + } + + protected static Document loadTemplate(File file) throws IOException { + InputStream in = null; + + try { + in = new FileInputStream(file); + + Document template = XMLUtils.parseDocument(in); + + if (template == null) { + throw new IOException("cannot load template"); + } + return template; + } + finally { + if (in != null) { + try { + in.close(); + } + catch (IOException ioe) { + log.error(ioe); + } + } + } + } + + public static Recommendations createRecommendations( + File systemFile, + File userFile + ) { + log.debug("Recommendations.createBuilder"); + + if (!systemFile.isFile() || !systemFile.canRead()) { + log.error("Cannot open template file '" + systemFile + "'"); + return null; + } + + if (!userFile.isFile() || !userFile.canRead()) { + log.error("Cannot open template file '" + userFile + "'"); + return null; + } + + try { + Document systemTemplate = loadTemplate(systemFile); + Document userTemplate = loadTemplate(userFile); + return new Recommendations( + new Builder(systemTemplate), + new Builder(userTemplate)); + } + catch (IOException ioe) { + log.error(ioe); + return null; + } + } + + protected static Recommendations createRecommendations() { + log.debug("Recommendations.createRecommendations"); + + String systemPath = Config.getStringXPath(XPATH_SYSTEM_TEMPLATE); + String userPath = Config.getStringXPath(XPATH_USER_TEMPLATE); + + if (systemPath == null || userPath == null) { + log.error("no path to template file given"); + return null; + } + + systemPath = Config.replaceConfigDir(systemPath); + userPath = Config.replaceConfigDir(userPath); + + return createRecommendations( + new File(systemPath), + new File(userPath)); + } +} +// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :