You are viewing a plain text version of this content. The canonical link for it is here.
Posted to server-dev@james.apache.org by pg...@apache.org on 2002/10/04 06:55:19 UTC
cvs commit: jakarta-james/src/java/org/apache/james/transport/mailets JDBCVirtualUserTable.java
pgoldstein 2002/10/03 21:55:19
Added: src/java/org/apache/james/transport/mailets
JDBCVirtualUserTable.java
Log:
Adding JDBCVirtualUserTable mailet.
Thanks to Noel Bergman
Revision Changes Path
1.1 jakarta-james/src/java/org/apache/james/transport/mailets/JDBCVirtualUserTable.java
Index: JDBCVirtualUserTable.java
===================================================================
/*
* Copyright (C) The Apache Software Foundation. All rights reserved.
*
* This software is published under the terms of the Apache Software License
* version 1.1, a copy of which has been included with this distribution in
* the LICENSE file.
*/
package org.apache.james.transport.mailets;
import org.apache.avalon.cornerstone.services.datasource.DataSourceSelector;
import org.apache.avalon.excalibur.datasource.DataSourceComponent;
import org.apache.avalon.framework.component.ComponentManager;
import org.apache.james.Constants;
import org.apache.james.util.JDBCUtil;
import org.apache.mailet.GenericMailet;
import org.apache.mailet.Mail;
import org.apache.mailet.MailAddress;
import org.apache.mailet.MailetException;
import javax.mail.MessagingException;
import javax.mail.internet.ParseException;
import java.sql.*;
import java.util.Collection;
import java.util.Iterator;
import java.util.Locale;
import java.util.Vector;
/**
* Implements a Virtual User Table for JAMES. Derived from the
* JDBCAlias mailet, but whereas that mailet uses a simple map from a
* source address to a destination address, this handles simple
* wildcard selection, verifies that a catchall address is for a domain
* in the Virtual User Table, and handles forwarding.
*
* With JDBCAlias, if the destination address were remote it would be
* subject to relay testing, even though it should be treated as a local
* address. JDBCVirtualUserTable incorporates JDBCAlias processing for
* local destinations, and Forward processing for remote destinations.
*
* To prevent from breaking existing JDBCAlias applications, and to
* allow for evolution of this mailet, it is released as a new mailet,
* rather than as an update to JDBCAlias. However, anyone using
* JDBCAlias should be able to upgrade to JDBCVirtualUserTable.
*
* As with JDBCAlias, JDBCVirtualUserTable does not provide any
* administation tools. You'll have to create the VirtualUserTable
* yourself. The standard configuration is as follows:
*
* CREATE TABLE VirtualUserTable
* (
* user varchar(64) NOT NULL default '',
* domain varchar(255) NOT NULL default '',
* target_address varchar(255) NOT NULL default '',
* PRIMARY KEY (user,domain)
* );
*
* The standard query used with VirtualUserTable is:
*
* select VirtualUserTable.target_address from VirtualUserTable, VirtualUserTable as VUTDomains
* where (VirtualUserTable.user like ? or VirtualUserTable.user like "\%")
* and (VirtualUserTable.domain like ?
* or (VirtualUserTable.domain like "\%" and VUTDomains.domain like ?))
* order by concat(VirtualUserTable.user,'@',VirtualUserTable.domain) desc limit 1
*
* For a given [user, domain, domain] used with the query, this will
* match as follows (in precedence order):
*
* 1. user@domain - explicit mapping for user@domain
* 2. user@% - catchall mapping for user anywhere
* 3. %@domain - catchall mapping for anyone at domain
* 4. null - no valid mapping
*
* You need to set the connection. At the moment, there is a limit to
* what you can change regarding the SQL Query, because there isn't a
* means to specify where in the query to replace parameters. [TODO]
*
* <mailet match="All" class="JDBCVirtualUserTable">
* <table>db://maildb/VirtualUserTable</table>
* <sqlquery>sqlquery</sqlquery>
* </mailet>
*
* @author Noel J. Begman <no...@devtech.com>
*/
public class JDBCVirtualUserTable extends GenericMailet
{
protected DataSourceComponent datasource;
/**
* The query used by the mailet to get the alias mapping
*/
protected String query = null;
/**
* The JDBCUtil helper class
*/
private final JDBCUtil theJDBCUtil = new JDBCUtil() {
protected void delegatedLog(String logString) {
log("JDBCVirtualUserTable: " + logString);
}
};
/**
* Initialize the mailet
*/
public void init() throws MessagingException {
if (getInitParameter("table") == null) {
throw new MailetException("Table location not specified for JDBCVirtualUserTable");
}
String tableURL = getInitParameter("table");
String datasourceName = tableURL.substring(5);
int pos = datasourceName.indexOf("/");
String tableName = datasourceName.substring(pos + 1);
datasourceName = datasourceName.substring(0, pos);
Connection conn = null;
try {
ComponentManager componentManager = (ComponentManager)getMailetContext().getAttribute(Constants.AVALON_COMPONENT_MANAGER);
// Get the DataSourceSelector service
DataSourceSelector datasources = (DataSourceSelector)componentManager.lookup(DataSourceSelector.ROLE);
// Get the data-source required.
datasource = (DataSourceComponent)datasources.select(datasourceName);
conn = datasource.getConnection();
// Check if the required table exists. If not, complain.
DatabaseMetaData dbMetaData = conn.getMetaData();
// Need to ask in the case that identifiers are stored, ask the DatabaseMetaInfo.
// Try UPPER, lower, and MixedCase, to see if the table is there.
if (!(theJDBCUtil.tableExists(dbMetaData, tableName))) {
StringBuffer exceptionBuffer =
new StringBuffer(128)
.append("Could not find table '")
.append(tableName)
.append("' in datasource '")
.append(datasourceName)
.append("'");
throw new MailetException(exceptionBuffer.toString());
}
//Build the query
query = getInitParameter("sqlquery");
if (query == null) {
query = "select VirtualUserTable.target_address from VirtualUserTable, VirtualUserTable as VUTDomains where (VirtualUserTable.user like ? or VirtualUserTable.user like '\\%') and (VirtualUserTable.domain like ? or (VirtualUserTable.domain like '\\%' and VUTDomains.domain like ?)) order by concat(VirtualUserTable.user,'@',VirtualUserTable.domain) desc limit 1";
}
} catch (MailetException me) {
throw me;
} catch (Exception e) {
throw new MessagingException("Error initializing JDBCVirtualUserTable", e);
} finally {
theJDBCUtil.closeJDBCConnection(conn);
}
}
/**
* Checks the recipient list of the email for user mappings. Maps recipients as
* appropriate, modifying the recipient list of the mail and sends mail to any new
* non-local recipients.
*
* @param mail the mail to process
*/
public void service(Mail mail)
throws MessagingException {
Connection conn = null;
PreparedStatement mappingStmt = null;
Collection recipients = mail.getRecipients();
Collection recipientsToRemove = new Vector();
Collection recipientsToAddLocal = new Vector();
Collection recipientsToAddForward = new Vector();
try {
conn = datasource.getConnection();
mappingStmt = conn.prepareStatement(query);
for (Iterator i = recipients.iterator(); i.hasNext(); ) {
ResultSet mappingRS = null;
try {
MailAddress source = (MailAddress)i.next();
mappingStmt.setString(1, source.getUser());
mappingStmt.setString(2, source.getHost());
mappingStmt.setString(3, source.getHost());
mappingRS = mappingStmt.executeQuery();
if (mappingRS.next()) {
try {
String targetString = mappingRS.getString(1);
MailAddress target = (targetString.indexOf('@') < 0) ? new MailAddress(targetString, "localhost")
: new MailAddress(targetString);
//Mark this source address as an address to remove from the recipient list
recipientsToRemove.add(source);
//Need to separate local and remote recipients.
if (getMailetContext().isLocalServer(target.getHost())) {
recipientsToAddLocal.add(target);
} else {
recipientsToAddForward.add(target);
}
} catch (ParseException pe) {
//Don't map this address... there's an invalid address mapping here
StringBuffer exceptionBuffer =
new StringBuffer(128)
.append("There is an invalid map from ")
.append(source)
.append(" to ")
.append(mappingRS.getString(1));
log(exceptionBuffer.toString());
continue;
}
}
} finally {
theJDBCUtil.closeJDBCResultSet(mappingRS);
}
}
} catch (SQLException sqle) {
throw new MessagingException("Error accessing database", sqle);
} finally {
theJDBCUtil.closeJDBCStatement(mappingStmt);
theJDBCUtil.closeJDBCConnection(conn);
}
// Remove mapped recipients
recipients.removeAll(recipientsToRemove);
// Add mapped recipients that are local
recipients.addAll(recipientsToAddLocal);
// Forward to mapped recipients that are remote
if (recipientsToAddForward.size() != 0) {
getMailetContext().sendMail(mail.getSender(), recipientsToAddForward, mail.getMessage());
}
// If there are no recipients left, Ghost the message
if (recipients.size() == 0) {
mail.setState(Mail.GHOST);
}
}
public String getMailetInfo() {
return "JDBC Virtual User Table mailet";
}
}
--
To unsubscribe, e-mail: <ma...@jakarta.apache.org>
For additional commands, e-mail: <ma...@jakarta.apache.org>