/*
|
* CDDL HEADER START
|
*
|
* The contents of this file are subject to the terms of the Common Development and
|
* Distribution License (the License). You may not use this file except in compliance with the
|
* License.
|
*
|
* You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
|
* specific language governing permission and limitations under the License.
|
*
|
* When distributing Covered Software, include this CDDL Header Notice in each file and include
|
* the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
|
* Header, with the fields enclosed by brackets [] replaced by your own identifying
|
* information: "Portions Copyright [year] [name of copyright owner]".
|
*
|
* CDDL HEADER END
|
*
|
* Copyright 2013 ForgeRock AS.
|
* Portions Copyright 2013 IS4U.
|
*/
|
|
package org.forgerock.opendj.virtual;
|
|
import java.io.IOException;
|
import java.sql.DriverManager;
|
import java.sql.SQLException;
|
import java.sql.Statement;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.Iterator;
|
import java.util.Map;
|
|
import org.forgerock.opendj.ldap.AbstractSynchronousConnection;
|
import org.forgerock.opendj.ldap.Attribute;
|
import org.forgerock.opendj.ldap.ByteString;
|
import org.forgerock.opendj.ldap.ConnectionEventListener;
|
import org.forgerock.opendj.ldap.ErrorResultException;
|
import org.forgerock.opendj.ldap.IntermediateResponseHandler;
|
import org.forgerock.opendj.ldap.ResultCode;
|
import org.forgerock.opendj.ldap.SearchResultHandler;
|
import org.forgerock.opendj.ldap.requests.AddRequest;
|
import org.forgerock.opendj.ldap.requests.BindRequest;
|
import org.forgerock.opendj.ldap.requests.CompareRequest;
|
import org.forgerock.opendj.ldap.requests.DeleteRequest;
|
import org.forgerock.opendj.ldap.requests.ExtendedRequest;
|
import org.forgerock.opendj.ldap.requests.ModifyDNRequest;
|
import org.forgerock.opendj.ldap.requests.ModifyRequest;
|
import org.forgerock.opendj.ldap.requests.SearchRequest;
|
import org.forgerock.opendj.ldap.requests.SimpleBindRequest;
|
import org.forgerock.opendj.ldap.requests.UnbindRequest;
|
import org.forgerock.opendj.ldap.responses.BindResult;
|
import org.forgerock.opendj.ldap.responses.CompareResult;
|
import org.forgerock.opendj.ldap.responses.ExtendedResult;
|
import org.forgerock.opendj.ldap.responses.Responses;
|
import org.forgerock.opendj.ldap.responses.Result;
|
|
public final class JDBCConnection extends AbstractSynchronousConnection {
|
private final String driverName = "com.mysql.jdbc.Driver";
|
private java.sql.Connection connection;
|
private String connectionUrl;
|
private String userName;
|
private String userPass;
|
private JDBCMapper jdbcm;
|
private MappingConfigurationManager mcm;
|
|
JDBCConnection(final String connectionURL) {
|
this.connectionUrl = connectionURL;
|
}
|
|
public void initializeMapper(JDBCMapper jdbcmapper) throws SQLException, ErrorResultException, IOException{
|
jdbcm = jdbcmapper;
|
jdbcm.fillMaps();
|
mcm = new MappingConfigurationManager(jdbcm);
|
jdbcm.loadMappingConfig(mcm.loadMapping());
|
}
|
|
private Map<String, Object> getValuesMap(AddRequest request, String tableName, String OUName){
|
final String baseDN = request.getName().toString();
|
Iterable<Attribute> attributesCollection = request.getAllAttributes();
|
Iterator<Attribute> attributeIter = attributesCollection.iterator();
|
Map<String, Object> map = new HashMap<String, Object>();
|
|
while(attributeIter.hasNext()){
|
Attribute att = attributeIter.next();
|
Iterator<ByteString> valueIter = att.iterator();
|
String attributeName = att.getAttributeDescriptionAsString();
|
String columnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, attributeName);
|
String columnValue = "";
|
|
if (columnName == null) continue;
|
|
while(valueIter.hasNext()){
|
columnValue = columnValue.concat(valueIter.next().toString());
|
}
|
map.put(columnName, columnValue);
|
}
|
return map;
|
}
|
|
private ArrayList<String> getSQLVariablesStrings(String tableName, Map<String, Object> columnValuesMap){
|
ArrayList<String>columnList = null;
|
try {
|
columnList = jdbcm.getTableColumns(tableName);
|
} catch (SQLException e1) {
|
e1.printStackTrace();
|
}
|
String columnNamesString = " (";
|
String columnValuesString = " (";
|
|
for(int i = 0; i < columnList.size(); i++){
|
if (i > 0){
|
columnNamesString = columnNamesString.concat(", ");
|
columnValuesString = columnValuesString.concat(", ");
|
}
|
String columnName = columnList.get(i);
|
Object columnValue = columnValuesMap.get(columnName);
|
Object dataType = jdbcm.getTableColumnDataType(tableName, columnName);
|
if(columnValue == null){
|
if(dataType.equals(Integer.class)) columnValue = "0";
|
else columnValue = "Default Value";
|
}
|
if(dataType.equals(Integer.class)) columnValue = Integer.parseInt(columnValue.toString());
|
|
columnNamesString = columnNamesString.concat(columnName);
|
columnValuesString = columnValuesString.concat("'" + columnValue + "'");
|
}
|
columnNamesString = columnNamesString.concat(")");
|
columnValuesString = columnValuesString.concat(")");
|
|
ArrayList<String> newlist = new ArrayList<String>();
|
newlist.add(columnNamesString);
|
newlist.add(columnValuesString);
|
|
return newlist;
|
}
|
|
public java.sql.Connection getSqlConnection(){
|
return connection;
|
}
|
|
@Override
|
public Result add(AddRequest request) throws ErrorResultException
|
{
|
Result r;
|
try {
|
final String baseDN = request.getName().toString();
|
String[] stringSplitter = baseDN.split("ou=");
|
stringSplitter = stringSplitter[1].split(",");
|
final String organizationalUnitName = stringSplitter[0];
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, organizationalUnitName);
|
final Map<String, Object> columnValuesMap = getValuesMap(request, tableName, organizationalUnitName);
|
final ArrayList<String> SQLStringList = getSQLVariablesStrings(tableName, columnValuesMap);
|
String columnNamesString = SQLStringList.get(0), columnValuesString = SQLStringList.get(1);
|
|
Statement st = connection.createStatement();
|
String sql = "INSERT INTO " + tableName + columnNamesString + " VALUES" + columnValuesString;
|
st.executeUpdate(sql);
|
r = Responses.newResult(ResultCode.SUCCESS);
|
} catch (SQLException e) {
|
System.out.println(e.toString());
|
r = Responses.newResult(ResultCode.OPERATIONS_ERROR);
|
}
|
return r;
|
}
|
|
@Override
|
public void addConnectionEventListener(ConnectionEventListener listener)
|
{
|
// TODO Auto-generated method stub
|
}
|
|
@Override
|
public BindResult bind(BindRequest request) throws ErrorResultException
|
{
|
BindResult r;
|
this.userName = request.getName();
|
byte[] password = null;
|
if (request instanceof SimpleBindRequest) {
|
password = ((SimpleBindRequest) request).getPassword();
|
} else {
|
r = Responses.newBindResult(ResultCode.PROTOCOL_ERROR);
|
return r;
|
}
|
this.userPass = new String(password);
|
|
try {
|
Class.forName(driverName);
|
this.connection = DriverManager
|
.getConnection(this.connectionUrl,this.userName,this.userPass);
|
} catch (ClassNotFoundException e) {
|
System.out.println(e.toString());
|
r = Responses.newBindResult(ResultCode.OTHER);
|
return r;
|
} catch (SQLException e) {
|
System.out.println(e.toString());
|
r = Responses.newBindResult(ResultCode.CLIENT_SIDE_CONNECT_ERROR);
|
return r;
|
}
|
r = Responses.newBindResult(ResultCode.SUCCESS);
|
return r;
|
}
|
|
@Override
|
public void close(UnbindRequest request, String reason)
|
{
|
try
|
{
|
this.connection.close();
|
}
|
catch (SQLException e)
|
{
|
e.printStackTrace();
|
}
|
}
|
|
@Override
|
public CompareResult compare(CompareRequest request)
|
throws ErrorResultException
|
{
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
@Override
|
public Result delete(DeleteRequest request) throws ErrorResultException
|
{
|
Result r = null;
|
try{
|
final String baseDN = request.getName().toString();
|
String[] stringSplitter = baseDN.split("ou=");
|
stringSplitter = stringSplitter[1].split(",");
|
final String organizationalUnitName = stringSplitter[0];
|
stringSplitter = baseDN.split("cn=");
|
stringSplitter = stringSplitter[1].split(",");
|
final String commonName = stringSplitter[0];
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, organizationalUnitName);
|
String columnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, organizationalUnitName, "cn");
|
|
Statement st = connection.createStatement();
|
String sql = "DELETE FROM " + tableName + " WHERE " + columnName + "='" + commonName + "'";
|
st.executeUpdate(sql);
|
r = Responses.newResult(ResultCode.SUCCESS);
|
} catch (SQLException e) {
|
System.out.println(e.toString());
|
r = Responses.newResult(ResultCode.OPERATIONS_ERROR);
|
}
|
return r;
|
}
|
|
@Override
|
public <R extends ExtendedResult> R extendedRequest(
|
ExtendedRequest<R> request, IntermediateResponseHandler handler)
|
throws ErrorResultException
|
{
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
@Override
|
public boolean isClosed()
|
{
|
// TODO Auto-generated method stub
|
return false;
|
}
|
|
@Override
|
public boolean isValid()
|
{
|
// TODO Auto-generated method stub
|
return false;
|
}
|
|
@Override
|
public Result modify(ModifyRequest request) throws ErrorResultException
|
{
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
@Override
|
public Result modifyDN(ModifyDNRequest request) throws ErrorResultException
|
{
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
@Override
|
public void removeConnectionEventListener(ConnectionEventListener listener)
|
{
|
// TODO Auto-generated method stub
|
|
}
|
|
@Override
|
public Result search(SearchRequest request, SearchResultHandler handler)
|
throws ErrorResultException
|
{
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
@Override
|
public String toString()
|
{
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
}
|
