2019-12-30 17:18:03 +01:00
|
|
|
package envoy.client;
|
|
|
|
|
|
|
|
import java.io.InputStream;
|
|
|
|
import java.io.ObjectInputStream;
|
|
|
|
import java.util.HashMap;
|
|
|
|
import java.util.Map;
|
2019-12-31 10:57:11 +01:00
|
|
|
import java.util.function.Consumer;
|
2019-12-30 17:18:03 +01:00
|
|
|
import java.util.logging.Level;
|
|
|
|
import java.util.logging.Logger;
|
|
|
|
|
|
|
|
import envoy.client.util.EnvoyLog;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Project: <strong>envoy-client</strong><br>
|
|
|
|
* File: <strong>Receiver.java</strong><br>
|
|
|
|
* Created: <strong>30.12.2019</strong><br>
|
|
|
|
*
|
|
|
|
* @author Kai S. K. Engelbart
|
|
|
|
* @since Envoy v0.3-alpha
|
|
|
|
*/
|
|
|
|
public class Receiver implements Runnable {
|
|
|
|
|
2019-12-31 10:57:11 +01:00
|
|
|
private InputStream in;
|
|
|
|
private Map<Class<?>, Consumer<?>> processors = new HashMap<>();
|
2019-12-30 17:18:03 +01:00
|
|
|
|
|
|
|
private static final Logger logger = EnvoyLog.getLogger(Receiver.class.getSimpleName());
|
|
|
|
|
2019-12-31 10:57:11 +01:00
|
|
|
/**
|
|
|
|
* Creates an instance of {@link Receiver}.
|
|
|
|
*
|
|
|
|
* @param in the {@link InputStream} to parse objects from
|
|
|
|
*/
|
2019-12-30 17:18:03 +01:00
|
|
|
public Receiver(InputStream in) { this.in = in; }
|
|
|
|
|
|
|
|
@SuppressWarnings("unchecked")
|
|
|
|
@Override
|
|
|
|
public void run() {
|
|
|
|
try (ObjectInputStream oin = new ObjectInputStream(in)) {
|
|
|
|
while (true) {
|
|
|
|
Object obj = oin.readObject();
|
|
|
|
logger.finest("Received object " + obj);
|
|
|
|
|
|
|
|
// Get appropriate processor
|
2019-12-31 10:57:11 +01:00
|
|
|
@SuppressWarnings("rawtypes")
|
|
|
|
Consumer processor = processors.get(obj.getClass());
|
2019-12-30 17:18:03 +01:00
|
|
|
if (processor == null)
|
|
|
|
logger.severe(String.format("The received object has the class %s for which no processor is defined.", obj.getClass()));
|
2019-12-31 10:57:11 +01:00
|
|
|
else processor.accept(obj);
|
2019-12-30 17:18:03 +01:00
|
|
|
}
|
2019-12-31 10:57:11 +01:00
|
|
|
} catch (Exception e) {
|
2019-12-30 17:18:03 +01:00
|
|
|
logger.log(Level.SEVERE, "Error on receiver thread", e);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-12-31 10:57:11 +01:00
|
|
|
/**
|
|
|
|
* Adds an object processor to this {@link Receiver}. It will be called once an
|
|
|
|
* object of the accepted class has been received.
|
|
|
|
*
|
|
|
|
* @param processorClass the object class accepted by the processor
|
|
|
|
* @param processor the object processor
|
|
|
|
*/
|
|
|
|
public <T> void registerProcessor(Class<T> processorClass, Consumer<T> processor) { processors.put(processorClass, processor); }
|
2019-12-30 17:18:03 +01:00
|
|
|
}
|