/*
* Copyright (c) 2001-2004 Ant-Contrib project. All rights reserved.
*
* Licensed 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 net.sf.antcontrib.antserver.server;
import java.io.*;
import java.net.Socket;
import org.apache.tools.ant.Project;
import org.apache.xml.serialize.OutputFormat;
import org.apache.xml.serialize.XMLSerializer;
import net.sf.antcontrib.antserver.Command;
import net.sf.antcontrib.antserver.Response;
import net.sf.antcontrib.antserver.Util;
import net.sf.antcontrib.antserver.commands.DisconnectCommand;
import net.sf.antcontrib.antserver.commands.ShutdownCommand;
/****************************************************************************
* Place class description here.
*
* @author <a href='mailto:mattinger@yahoo.com'>Matthew Inger</a>
* @author <additional author>
*
* @since
*
****************************************************************************/
public class ConnectionHandler
implements Runnable
{
private static long nextGroupId = 0;
private ServerTask task;
private Socket socket;
private Thread thread;
private Throwable thrown;
public ConnectionHandler(ServerTask task, Socket socket)
{
super();
this.socket = socket;
this.task = task;
}
public void start()
{
long gid = nextGroupId;
if (nextGroupId == Long.MAX_VALUE)
nextGroupId = 0;
else
nextGroupId++;
ThreadGroup group = new ThreadGroup("server-tg-" + gid);
thread = new Thread(group, this);
thread.start();
}
public Throwable getThrown()
{
return thrown;
}
public void run()
{
InputStream is = null;
OutputStream os = null;
try
{
ConnectionBuildListener cbl = null;
is = socket.getInputStream();
os = socket.getOutputStream();
ObjectInputStream ois = new ObjectInputStream(is);
ObjectOutputStream oos = new ObjectOutputStream(os);
// Write the initial response object so that the
// object stream is initialized
oos.writeObject(new Response());
boolean disconnect = false;
Command inputCommand = null;
Response response = null;
while (! disconnect)
{
task.getProject().log("Reading command object.",
Project.MSG_DEBUG);
inputCommand = (Command) ois.readObject();
task.getProject().log("Executing command object: " + inputCommand,
Project.MSG_DEBUG);
response = new Response();
try
{
cbl = new ConnectionBuildListener();
task.getProject().addBuildListener(cbl);
inputCommand.execute(task.getProject(),
inputCommand.getContentLength(),
is);
response.setSucceeded(true);
}
catch (Throwable t)
{
response.setSucceeded(false);
response.setThrowable(t);
}
finally
{
if (cbl != null)
task.getProject().removeBuildListener(cbl);
}
ByteArrayOutputStream baos = new ByteArrayOutputStream();
XMLSerializer serial = new XMLSerializer();
OutputFormat fmt = new OutputFormat();
fmt.setOmitDocumentType(true);
fmt.setOmitXMLDeclaration(false);
serial.setOutputFormat(fmt);
serial.setOutputByteStream(baos);
serial.serialize(cbl.getDocument());
response.setResultsXml(baos.toString());
task.getProject().log("Executed command object: " + inputCommand,
Project.MSG_DEBUG);
task.getProject().log("Sending response: " + response,
Project.MSG_DEBUG);
response.setContentLength(inputCommand.getContentLength());
oos.writeObject(response);
if (inputCommand.getResponseContentLength() != 0)
{
Util.transferBytes(inputCommand.getReponseContentStream(),
inputCommand.getResponseContentLength(),
os,
true);
}
if (inputCommand instanceof DisconnectCommand)
{
disconnect = true;
task.getProject().log("Got disconnect command",
Project.MSG_DEBUG);
}
else if (inputCommand instanceof ShutdownCommand)
{
disconnect = true;
task.getProject().log("Got shutdown command",
Project.MSG_DEBUG);
task.shutdown();
}
}
}
catch (ClassNotFoundException e)
{
thrown = e;
}
catch (IOException e)
{
thrown = e;
}
catch (Throwable t)
{
thrown = t;
}
finally
{
if (is != null)
{
try
{
is.close();
}
catch (IOException e)
{
}
}
if (os != null)
{
try
{
os.close();
}
catch (IOException e)
{
}
}
if (socket != null)
{
try
{
socket.close();
}
catch (IOException e)
{
}
}
}
}
}