mirror of https://github.com/OpenIdentityPlatform/OpenDJ.git
opendj3/opendj-virtual/src/main/java/org/forgerock/opendj/virtual/JDBCConnection.java
@@ -1,30 +1,43 @@ /* * 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.Collection; 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.Connection; import org.forgerock.opendj.ldap.ConnectionEventListener; import org.forgerock.opendj.ldap.DN; import org.forgerock.opendj.ldap.Entry; import org.forgerock.opendj.ldap.ErrorResultException; import org.forgerock.opendj.ldap.ErrorResultIOException; import org.forgerock.opendj.ldap.FutureResult; import org.forgerock.opendj.ldap.IntermediateResponseHandler; import org.forgerock.opendj.ldap.ResultCode; import org.forgerock.opendj.ldap.ResultHandler; import org.forgerock.opendj.ldap.SearchResultHandler; import org.forgerock.opendj.ldap.SearchResultReferenceIOException; import org.forgerock.opendj.ldap.SearchScope; import org.forgerock.opendj.ldap.requests.AbandonRequest; import org.forgerock.opendj.ldap.requests.AddRequest; import org.forgerock.opendj.ldap.requests.BindRequest; import org.forgerock.opendj.ldap.requests.CompareRequest; @@ -32,454 +45,265 @@ 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.Requests; 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.GenericExtendedResult; import org.forgerock.opendj.ldap.responses.Responses; import org.forgerock.opendj.ldap.responses.Result; import org.forgerock.opendj.ldap.responses.SearchResultEntry; import org.forgerock.opendj.ldap.responses.SearchResultReference; import org.forgerock.opendj.ldif.ChangeRecord; import org.forgerock.opendj.ldif.ConnectionEntryReader; public final class JDBCConnection implements Connection { 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; /** * Creates a new JDBC connection. * * @param connection * The SQL Connection using JDBC Driver. */ JDBCConnection(final String connectionURL) { this.connectionUrl = connectionURL; } public void initializeMapper(JDBCMapper jdbcmapper) throws SQLException, ErrorResultException, ErrorResultIOException, SearchResultReferenceIOException{ jdbcm = jdbcmapper; jdbcm.fillMaps(); mcm = new MappingConfigurationManager(jdbcm); jdbcm.loadMappingConfig(mcm.loadMapping()); 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); } public java.sql.Connection getSqlConnection(){ return connection; 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 = " ("; @Override public FutureResult<Void> abandonAsync(AbandonRequest request) { // TODO Auto-generated method stub return null; } private Map<String, Object> getValuesMap(AddRequest request, String tableName, String OUName){ 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, 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 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; } private Result addOperation(AddRequest request){ Result r; try { final String DN = request.getName().toString(); String[] stringSplitter = DN.split("ou="); stringSplitter = stringSplitter[1].split(","); final String organizationalUnitName = stringSplitter[0]; final String tableName = jdbcm.getTableNameFromMapping(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.UNWILLING_TO_PERFORM); } return r; } @Override public Result add(AddRequest request) throws ErrorResultException { return addOperation(request); } 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()); @Override public Result add(Entry entry) throws ErrorResultException { AddRequest addRequest = Requests.newAddRequest(entry); return addOperation(addRequest); } columnNamesString = columnNamesString.concat(columnName); columnValuesString = columnValuesString.concat("'" + columnValue + "'"); } columnNamesString = columnNamesString.concat(")"); columnValuesString = columnValuesString.concat(")"); @Override public Result add(String... ldifLines) throws ErrorResultException { AddRequest addRequest = Requests.newAddRequest(ldifLines); return addOperation(addRequest); } ArrayList<String> newlist = new ArrayList<String>(); newlist.add(columnNamesString); newlist.add(columnValuesString); @Override public FutureResult<Result> addAsync(AddRequest request, IntermediateResponseHandler intermediateResponseHandler, ResultHandler<? super Result> resultHandler) { // TODO Auto-generated method stub return null; } return newlist; } @Override public void addConnectionEventListener(ConnectionEventListener listener) { // TODO Auto-generated method stub } @Override public Result applyChange(ChangeRecord request) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public FutureResult<Result> applyChangeAsync(ChangeRecord request, IntermediateResponseHandler intermediateResponseHandler, ResultHandler<? super Result> resultHandler) { // TODO Auto-generated method stub return null; } @Override public BindResult bind(BindRequest request) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public BindResult bind(String name, char[] password) throws ErrorResultException { BindResult r; this.userName = name; 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 FutureResult<BindResult> bindAsync(BindRequest request, IntermediateResponseHandler intermediateResponseHandler, ResultHandler<? super BindResult> resultHandler) { // TODO Auto-generated method stub return null; } @Override public void close() { // TODO Auto-generated method stub } @Override public void close(UnbindRequest request, String reason) { // TODO Auto-generated method stub } @Override public CompareResult compare(CompareRequest request) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public CompareResult compare(String name, String attributeDescription, String assertionValue) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public FutureResult<CompareResult> compareAsync(CompareRequest request, IntermediateResponseHandler intermediateResponseHandler, ResultHandler<? super CompareResult> resultHandler) { // TODO Auto-generated method stub return null; } @Override public Result delete(DeleteRequest request) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public Result delete(String name) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public FutureResult<Result> deleteAsync(DeleteRequest request, IntermediateResponseHandler intermediateResponseHandler, ResultHandler<? super Result> resultHandler) { // TODO Auto-generated method stub return null; } @Override public <R extends ExtendedResult> R extendedRequest( ExtendedRequest<R> request) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public <R extends ExtendedResult> R extendedRequest( ExtendedRequest<R> request, IntermediateResponseHandler handler) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public GenericExtendedResult extendedRequest(String requestName, ByteString requestValue) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public <R extends ExtendedResult> FutureResult<R> extendedRequestAsync( ExtendedRequest<R> request, IntermediateResponseHandler intermediateResponseHandler, ResultHandler<? super R> resultHandler) { // 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 modify(String... ldifLines) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public FutureResult<Result> modifyAsync(ModifyRequest request, IntermediateResponseHandler intermediateResponseHandler, ResultHandler<? super Result> resultHandler) { // TODO Auto-generated method stub return null; } @Override public Result modifyDN(ModifyDNRequest request) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public Result modifyDN(String name, String newRDN) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public FutureResult<Result> modifyDNAsync(ModifyDNRequest request, IntermediateResponseHandler intermediateResponseHandler, ResultHandler<? super Result> resultHandler) { // TODO Auto-generated method stub return null; } @Override public SearchResultEntry readEntry(DN name, String... attributeDescriptions) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public SearchResultEntry readEntry(String name, String... attributeDescriptions) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public FutureResult<SearchResultEntry> readEntryAsync(DN name, Collection<String> attributeDescriptions, ResultHandler<? super SearchResultEntry> handler) { // TODO Auto-generated method stub return null; } @Override public void removeConnectionEventListener(ConnectionEventListener listener) { // TODO Auto-generated method stub } @Override public ConnectionEntryReader search(SearchRequest request) { // TODO Auto-generated method stub return null; } @Override public Result search(SearchRequest request, Collection<? super SearchResultEntry> entries) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public Result search(SearchRequest request, Collection<? super SearchResultEntry> entries, Collection<? super SearchResultReference> references) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public Result search(SearchRequest request, SearchResultHandler handler) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public ConnectionEntryReader search(String baseObject, SearchScope scope, String filter, String... attributeDescriptions) { // TODO Auto-generated method stub return null; } @Override public FutureResult<Result> searchAsync(SearchRequest request, IntermediateResponseHandler intermediateResponseHandler, SearchResultHandler resultHandler) { // TODO Auto-generated method stub return null; } @Override public SearchResultEntry searchSingleEntry(SearchRequest request) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public SearchResultEntry searchSingleEntry(String baseObject, SearchScope scope, String filter, String... attributeDescriptions) throws ErrorResultException { // TODO Auto-generated method stub return null; } @Override public FutureResult<SearchResultEntry> searchSingleEntryAsync( SearchRequest request, ResultHandler<? super SearchResultEntry> handler) { // TODO Auto-generated method stub return null; } public java.sql.Connection getSqlConnection(){ return connection; } @Override public Result deleteSubtree(String name) throws ErrorResultException 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; } }
