001: /* ====================================================================
002: * The Apache Software License, Version 1.1
003: *
004: * Copyright (c) 1997-2003 The Apache Software Foundation. All rights
005: * reserved.
006: *
007: * Redistribution and use in source and binary forms, with or without
008: * modification, are permitted provided that the following conditions
009: * are met:
010: *
011: * 1. Redistributions of source code must retain the above copyright
012: * notice, this list of conditions and the following disclaimer.
013: *
014: * 2. Redistributions in binary form must reproduce the above copyright
015: * notice, this list of conditions and the following disclaimer in
016: * the documentation and/or other materials provided with the
017: * distribution.
018: *
019: * 3. The end-user documentation included with the redistribution,
020: * if any, must include the following acknowledgment:
021: * "This product includes software developed by the
022: * Apache Software Foundation (http://www.apache.org/)."
023: * Alternately, this acknowledgment may appear in the software
024: * itself, if and wherever such third-party acknowledgments
025: * normally appear.
026: *
027: * 4. The names "Jakarta", "Avalon", and "Apache Software Foundation"
028: * must not be used to endorse or promote products derived from this
029: * software without prior written permission. For written
030: * permission, please contact apache@apache.org.
031: *
032: * 5. Products derived from this software may not be called "Apache",
033: * nor may "Apache" appear in their name, without prior written
034: * permission of the Apache Software Foundation.
035: *
036: * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
037: * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
038: * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
039: * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
040: * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
041: * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
042: * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
043: * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
044: * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
045: * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
046: * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
047: * SUCH DAMAGE.
048: * ====================================================================
049: *
050: * This software consists of voluntary contributions made by many
051: * individuals on behalf of the Apache Software Foundation. For more
052: * information on the Apache Software Foundation, please see
053: * <http://www.apache.org/>.
054: */
055: package org.apache.log.output.db;
056:
057: import java.sql.PreparedStatement;
058: import java.sql.ResultSet;
059: import java.sql.SQLException;
060: import java.sql.Statement;
061: import java.util.HashMap;
062: import javax.sql.DataSource;
063: import org.apache.log.LogEvent;
064:
065: /**
066: * JDBC target that writes to normalized tables.
067: * This reduces overhead and cost of querying/storing logs.
068: *
069: * <p>Parts based on JDBC logger from prottomatter by
070: * <a href="mailto:nate@protomatter.com">Nate Sammons</a></p>
071: *
072: * @author <a href="mailto:peter@apache.org">Peter Donald</a>
073: */
074: public class NormalizedJDBCTarget extends DefaultJDBCTarget {
075: private HashMap m_categoryIDs = new HashMap();
076: private HashMap m_priorityIDs = new HashMap();
077:
078: public NormalizedJDBCTarget(final DataSource dataSource,
079: final String table, final ColumnInfo[] columns) {
080: super (dataSource, table, columns);
081: }
082:
083: /**
084: * Adds a single object into statement.
085: */
086: protected void specifyColumn(final PreparedStatement statement,
087: final int index, final LogEvent event) throws SQLException {
088: final ColumnInfo info = getColumn(index);
089: int id = 0;
090: String tableName = null;
091:
092: switch (info.getType()) {
093: case ColumnType.CATEGORY:
094: tableName = getTable() + "_" + ColumnType.CATEGORY_STR
095: + "_SET";
096: id = getID(tableName, m_categoryIDs, event.getCategory());
097: statement.setInt(index + 1, id);
098: break;
099:
100: case ColumnType.PRIORITY:
101: tableName = getTable() + "_" + ColumnType.PRIORITY_STR
102: + "_SET";
103: id = getID(tableName, m_priorityIDs, event.getPriority()
104: .getName());
105: statement.setInt(index + 1, id);
106: break;
107:
108: default:
109: super .specifyColumn(statement, index, event);
110: }
111: }
112:
113: protected synchronized int getID(final String tableName,
114: final HashMap idMap, final String instance)
115: throws SQLException {
116: final Integer id = (Integer) idMap.get(instance);
117: if (null != id)
118: return id.intValue();
119:
120: // see if it's been put in before.
121: Statement statement = null;
122: ResultSet resultSet = null;
123:
124: try {
125: statement = getConnection().createStatement();
126:
127: final String querySql = "SELECT ID FROM " + tableName
128: + " WHERE NAME='" + instance + "'";
129: resultSet = statement.executeQuery(querySql);
130:
131: if (resultSet.next()) {
132: final Integer newID = new Integer(resultSet.getInt(1));
133: idMap.put(instance, newID);
134: return newID.intValue();
135: }
136:
137: resultSet.close();
138:
139: //Note that the next part should be a transaction but
140: //it is not mega vital so ...
141:
142: //Find the max id in table and set
143: //max to it's value if any items are present in table
144: final String maxQuerySql = "SELECT MAX(ID) FROM "
145: + tableName;
146: resultSet = statement.executeQuery(maxQuerySql);
147: int max = 0;
148: if (resultSet.next())
149: max = resultSet.getInt(1);
150: resultSet.close();
151:
152: final int newID = max + 1;
153: final String insertSQL = "INSERT INTO " + tableName
154: + " (ID, NAME) VALUES ( " + newID + ", '"
155: + instance + "')";
156: statement.executeUpdate(insertSQL);
157:
158: idMap.put(instance, new Integer(newID));
159: return newID;
160: } finally {
161: // close up shop
162: if (null != resultSet) {
163: try {
164: resultSet.close();
165: } catch (final Exception e) {
166: }
167: }
168: if (null != statement) {
169: try {
170: statement.close();
171: } catch (final Exception e) {
172: }
173: }
174: }
175: }
176: }
|