org.snapscript.studio.agent.local.LocalMessageConsumer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of snap-agent Show documentation

Dynamic scripting for the JVM

There is a newer version: 1.4.6

package org.snapscript.studio.agent.local;

import java.io.InputStream;
import java.io.ObjectInput;
import java.io.ObjectInputStream;
import java.io.ObjectOutput;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.Socket;

import org.snapscript.studio.agent.local.message.AttachRequest;
import org.snapscript.studio.agent.local.message.AttachResponse;
import org.snapscript.studio.agent.local.message.DetachRequest;
import org.snapscript.studio.agent.local.message.DetachResponse;

public class LocalMessageConsumer {

   private final LocalProcessController listener;
   
   public LocalMessageConsumer(LocalProcessController listener){
      this.listener = listener;
   }
   
   public void consume(Socket socket) {
      try {
         InputStream in = socket.getInputStream();
         OutputStream out = socket.getOutputStream();
         ObjectInput input = new ObjectInputStream(in);
         ObjectOutput output = new ObjectOutputStream(out);
         Object value = input.readObject();
         
         if(AttachRequest.class.isInstance(value)) {
            String process = listener.attachRequest((AttachRequest)value);
            AttachResponse response = new AttachResponse(process);
            
            try {
               output.writeObject(response);
            }finally {
               output.close();
            }
         }
         if(DetachRequest.class.isInstance(value)) {
            String process = listener.detachRequest((DetachRequest)value);
            DetachResponse response = new DetachResponse(process);
            
            try {
               output.writeObject(response);
            }finally {
               output.close();
            }
         }
      } catch(Exception e) {
         e.printStackTrace();
      }
   }
}