| /** |
| * 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.oodt.xmlps.profile; |
| |
| //OODT imports |
| import org.apache.oodt.xmlps.mapping.DatabaseTable; |
| import org.apache.oodt.xmlps.mapping.MappingReader; |
| import org.apache.oodt.xmlps.product.XMLPSProductHandler; |
| import org.apache.oodt.xmlps.profile.DBMSExecutor; |
| import org.apache.oodt.xmlps.queryparser.Expression; |
| import org.apache.oodt.xmlps.queryparser.HandlerQueryParser; |
| |
| //JDK imports |
| import java.io.FileInputStream; |
| import java.io.FileNotFoundException; |
| import java.io.IOException; |
| import java.sql.SQLException; |
| import java.util.Iterator; |
| import java.util.List; |
| import java.util.Stack; |
| import java.util.logging.Level; |
| import java.util.logging.Logger; |
| |
| //OODT imports |
| import org.apache.oodt.profile.Profile; |
| import org.apache.oodt.profile.ProfileException; |
| import org.apache.oodt.profile.handlers.ProfileHandler; |
| import org.apache.oodt.xmlquery.QueryElement; |
| import org.apache.oodt.xmlquery.XMLQuery; |
| |
| /** |
| * |
| * <p> |
| * An implementation of a {@link ProfileHandler} that extends the capabilities |
| * of the {@link XMLPSProductHandler}, and uses the XML Specification defined |
| * by the mapping file to represent its field mapping information. |
| * </p>. |
| */ |
| public class XMLPSProfileHandler extends XMLPSProductHandler implements |
| ProfileHandler { |
| |
| private DBMSExecutor executor; |
| |
| /* our log stream */ |
| private static final Logger LOG = Logger.getLogger(XMLPSProfileHandler.class |
| .getName()); |
| |
| private String resLocationSpec; |
| |
| public XMLPSProfileHandler() throws InstantiationException { |
| super(null); |
| String mappingFilePath = System |
| .getProperty("org.apache.oodt.xmlps.profile.xml.mapFilePath"); |
| |
| if (mappingFilePath == null) { |
| throw new InstantiationException( |
| "Need to specify path to xml mapping file!"); |
| } |
| |
| try { |
| mapping = MappingReader.getMapping(mappingFilePath); |
| } catch (Exception e) { |
| throw new InstantiationException( |
| "Unable to parse profile mapping xml file: [" |
| + mappingFilePath + "]: reason: " |
| + e.getMessage()); |
| } |
| |
| // load the db properties file |
| // if one exists: otherwise, don't bother and just print out the SQL to |
| // the console. |
| // |
| String dbPropFilePath = System |
| .getProperty("org.apache.oodt.xmlps.profile.xml.dbPropFilePath"); |
| if (dbPropFilePath != null) { |
| try { |
| System.getProperties() |
| .load(new FileInputStream(dbPropFilePath)); |
| } catch (FileNotFoundException e) { |
| // TODO Auto-generated catch block |
| e.printStackTrace(); |
| } catch (IOException e) { |
| e.printStackTrace(); |
| throw new InstantiationException(e.getMessage()); |
| } |
| |
| executor = new DBMSExecutor(); |
| } |
| |
| this.resLocationSpec = System |
| .getProperty("org.apache.oodt.xmlps.profile.xml.resLocationSpec"); |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.apache.oodt.profile.handlers.ProfileHandler#findProfiles(org.apache.oodt.xmlquery.XMLQuery) |
| */ |
| public List<Profile> findProfiles(XMLQuery query) throws ProfileException { |
| List<QueryElement> whereSet = query.getWhereElementSet(); |
| List<QueryElement> selectSet = query.getSelectElementSet(); |
| try { |
| translateToDomain(selectSet, true); |
| translateToDomain(whereSet, false); |
| } catch (Exception e) { |
| e.printStackTrace(); |
| throw new ProfileException(e.getMessage()); |
| } |
| List<Profile> profs = queryAndPackageProfiles(query); |
| return profs; |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.apache.oodt.profile.handlers.ProfileHandler#get(java.lang.String) |
| */ |
| public Profile get(String id) throws ProfileException { |
| throw new ProfileException("Method not implemented!"); |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.apache.oodt.profile.handlers.ProfileHandler#getID() |
| */ |
| public String getID() { |
| return mapping.getId(); |
| } |
| |
| protected List<Profile> queryAndPackageProfiles(XMLQuery query) { |
| Stack<QueryElement> queryStack = HandlerQueryParser |
| .createQueryStack(query.getWhereElementSet()); |
| Expression parsedQuery = HandlerQueryParser.parse(queryStack, |
| this.mapping); |
| List<Profile> profs = null; |
| |
| StringBuffer sqlBuf = new StringBuffer("SELECT *"); |
| sqlBuf.append(" FROM "); |
| sqlBuf.append(mapping.getDefaultTable()); |
| sqlBuf.append(" "); |
| |
| if (mapping.getNumTables() > 0) { |
| for (Iterator<String> i = mapping.getTableNames().iterator(); i |
| .hasNext();) { |
| String tableName = i.next(); |
| if(tableName.equals(mapping.getDefaultTable())) continue; |
| DatabaseTable tbl = mapping.getTableByName(tableName); |
| sqlBuf.append("INNER JOIN "); |
| sqlBuf.append(tbl.getName()); |
| sqlBuf.append(" ON "); |
| sqlBuf.append(tbl.getName()); |
| sqlBuf.append("."); |
| sqlBuf.append(tbl.getJoinFieldName()); |
| sqlBuf.append(" = "); |
| sqlBuf.append(tbl.getDefaultTableJoin()); |
| sqlBuf.append("."); |
| sqlBuf.append(tbl.getDefaultTableJoinFieldName()); |
| sqlBuf.append(" "); |
| } |
| } |
| |
| if (parsedQuery != null) { |
| sqlBuf.append(" WHERE "); |
| sqlBuf.append(parsedQuery.evaluate()); |
| } |
| |
| LOG.log(Level.INFO, sqlBuf.toString()); |
| |
| if (executor != null) { |
| try { |
| profs = executor.executeLocalQuery(this.mapping, sqlBuf |
| .toString(), this.resLocationSpec); |
| |
| } catch (SQLException e) { |
| e.printStackTrace(); |
| LOG.log(Level.WARNING, "Error executing sql: [" |
| + sqlBuf.toString() + "]: Message: " + e.getMessage()); |
| } |
| } |
| |
| return profs; |
| } |
| |
| protected void translateToDomain(List<QueryElement> elemSet, |
| boolean selectSet) throws Exception { |
| super.translateToDomain(elemSet, selectSet); |
| } |
| |
| } |