001 /* 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, 013 * software distributed under the License is distributed on an 014 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 015 * KIND, either express or implied. See the License for the 016 * specific language governing permissions and limitations 017 * under the License. 018 * 019 */ 020 package org.apache.directory.server.core.trigger; 021 022 023 import java.util.Map; 024 025 import org.apache.directory.server.core.interceptor.context.OperationContext; 026 import org.apache.directory.shared.ldap.entry.ServerEntry; 027 import org.apache.directory.shared.ldap.exception.LdapException; 028 import org.apache.directory.shared.ldap.exception.LdapInvalidDnException; 029 import org.apache.directory.shared.ldap.name.DN; 030 import org.apache.directory.shared.ldap.trigger.StoredProcedureParameter; 031 032 033 public class AddStoredProcedureParameterInjector extends AbstractStoredProcedureParameterInjector 034 { 035 private DN addedEntryName; 036 private ServerEntry addedEntry; 037 038 039 public AddStoredProcedureParameterInjector( OperationContext opContext, DN addedEntryName, 040 ServerEntry addedEntry ) 041 { 042 super( opContext ); 043 this.addedEntryName = addedEntryName; 044 this.addedEntry = addedEntry; 045 Map<Class<?>, MicroInjector> injectors = super.getInjectors(); 046 injectors.put( StoredProcedureParameter.Add_ENTRY.class, $entryInjector ); 047 injectors.put( StoredProcedureParameter.Add_ATTRIBUTES.class, $attributesInjector ); 048 } 049 050 051 MicroInjector $entryInjector = new MicroInjector() 052 { 053 public Object inject( OperationContext opContext, StoredProcedureParameter param ) throws LdapInvalidDnException 054 { 055 // Return a safe copy constructed with user provided name. 056 return new DN( addedEntryName.getName() ); 057 } 058 }; 059 060 061 MicroInjector $attributesInjector = new MicroInjector() 062 { 063 public Object inject( OperationContext opContext, StoredProcedureParameter param ) throws LdapException 064 { 065 return addedEntry.clone(); 066 } 067 }; 068 069 }