/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.apache.avro.ipc.reflect; import java.io.IOException; import java.lang.reflect.Proxy; import org.apache.avro.Protocol; import org.apache.avro.Schema; import org.apache.avro.io.DatumReader; import org.apache.avro.io.DatumWriter; import org.apache.avro.reflect.ReflectData; import org.apache.avro.reflect.ReflectDatumReader; import org.apache.avro.reflect.ReflectDatumWriter; import org.apache.avro.ipc.Transceiver; import org.apache.avro.ipc.specific.SpecificRequestor; /** A {@link org.apache.avro.ipc.Requestor} for existing interfaces. */ public class ReflectRequestor extends SpecificRequestor { public ReflectRequestor(Class<?> iface, Transceiver transceiver) throws IOException { this(iface, transceiver, new ReflectData(iface.getClassLoader())); } protected ReflectRequestor(Protocol protocol, Transceiver transceiver) throws IOException { this(protocol, transceiver, ReflectData.get()); } public ReflectRequestor(Class<?> iface, Transceiver transceiver, ReflectData data) throws IOException { this(data.getProtocol(iface), transceiver, data); } public ReflectRequestor(Protocol protocol, Transceiver transceiver, ReflectData data) throws IOException { super(protocol, transceiver, data); } public ReflectData getReflectData() { return (ReflectData)getSpecificData(); } @Override protected DatumWriter<Object> getDatumWriter(Schema schema) { return new ReflectDatumWriter<Object>(schema, getReflectData()); } @Override protected DatumReader<Object> getDatumReader(Schema writer, Schema reader) { return new ReflectDatumReader<Object>(writer, reader, getReflectData()); } /** Create a proxy instance whose methods invoke RPCs. */ public static <T> T getClient(Class<T> iface, Transceiver transciever) throws IOException { return getClient(iface, transciever, new ReflectData(iface.getClassLoader())); } /** Create a proxy instance whose methods invoke RPCs. */ @SuppressWarnings("unchecked") public static <T> T getClient(Class<T> iface, Transceiver transciever, ReflectData reflectData) throws IOException { Protocol protocol = reflectData.getProtocol(iface); return (T)Proxy.newProxyInstance (reflectData.getClassLoader(), new Class[] { iface }, new ReflectRequestor(protocol, transciever, reflectData)); } /** Create a proxy instance whose methods invoke RPCs. */ @SuppressWarnings("unchecked") public static <T> T getClient(Class<T> iface, ReflectRequestor rreq) throws IOException { return (T)Proxy.newProxyInstance(rreq.getReflectData().getClassLoader(), new Class[] { iface }, rreq); } }