server/filter/TCPReaderInit.java

//---PACKAGE DECLARATION---
package uk.ac.ukc.iscream.filter;

//---IMPORTS---
import uk.ac.ukc.iscream.core.*;
import uk.ac.ukc.iscream.filter.*;
import uk.ac.ukc.iscream.componentmanager.*;
import java.net.Socket;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.IOException;
import java.io.*;
import uk.ac.ukc.iscream.util.*;

/**
 * This provides Host heartbeat functionality
 *
 * @author  $Author: tdb1 $
 * @version $Id: TCPReaderInit.java,v 1.12 2001/01/18 23:13:36 tdb1 Exp $
 */
class TCPReaderInit extends Thread {

//---FINAL ATTRIBUTES---

    /**
     * The current CVS revision of this class
     */
    public final String REVISION = "$Revision: 1.12 $";
    
//---STATIC METHODS---

//---CONSTRUCTORS---
    
    /**
     * Construct a new TCPReaderInit.
     *
     * @param socket the Socket to which the host is connected
     * @param queue the Queue to which we'll add data
     * @throws IOException if something goes badly wrong
     */
    public TCPReaderInit(Socket socket, Queue queue) throws IOException {
        _socket = socket;
        _queue = queue;
        // setup the reader & writer
        _socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream()));
        _socketOut = new PrintWriter(_socket.getOutputStream());
        _logger.write(toString(), Logger.SYSINIT, "created");
    }
    
//---PUBLIC METHODS---
    
    /**
     * Main run method. Will communicate with the host, inform it
     * if any updates to it's configuration are needed, and send
     * a heartbeat packet into the system.
     */
    public void run() {
        try {
            //variables
            String filelist = "";
            String lastModified = "";
            String inBound = "";
            
            inBound = _socketIn.readLine();
            if(!inBound.equals("HEARTBEAT")) {
                _socketOut.println("ERROR");
                _socketOut.flush();
                throw new IOException("protocol error - expecting:HEARTBEAT got:" + inBound);
            } else {
                _socketOut.println("OK");
                _socketOut.flush();
            }
            
            inBound = _socketIn.readLine();
            if(!inBound.equals("CONFIG")) {
                _socketOut.println("ERROR");
                _socketOut.flush();
                throw new IOException("protocol error - expecting:CONFIG got:" + inBound);
            } else {
                _socketOut.println("OK");
                _socketOut.flush();
            }
            
            inBound = _socketIn.readLine();
            filelist = inBound;
            _socketOut.println("OK");
            _socketOut.flush();
            
            inBound = _socketIn.readLine();
            lastModified = inBound;
            
            boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
            
            if(newConfig) {
                _socketOut.println("ERROR");
            }
            else {
                _socketOut.println("OK");
            }
            _socketOut.flush();
            
            inBound = _socketIn.readLine();
            if(!inBound.equals("ENDHEARTBEAT")) {
                _socketOut.println("ERROR");
                _socketOut.flush();
                throw new IOException("protocol error - expecting:ENDHEARTBEAT got:" + inBound);
            } else {
                _socketOut.println("OK");
                _socketOut.flush();
            }

            String date = new Long(System.currentTimeMillis()).toString();
            String hostname = _socket.getInetAddress().getHostName();
            String ipadd = _socket.getInetAddress().getHostAddress();
            String xml = "<packet type=\"heartbeat\" machine_name=\""+hostname+"\" date=\""+date+"\" ip=\""+ipadd+"\"></packet>";
            
            _queue.add(xml);
            
        } catch (Exception e) {
            _logger.write(toString(), Logger.ERROR, "ERROR: " + e.getMessage());
        }
        
        _socketOut.flush();
        // Disconnect streams & socket
        try {
            _socketIn.close();
            _socketOut.close();
            _socket.close();
        } catch (IOException e) {
            _logger.write(toString(), Logger.ERROR, "exception on socket close");
        }
        _logger.write(toString(), Logger.SYSMSG, "finished");
    }
    
    /**
     * Overrides the {@link java.lang.Object#toString() Object.toString()}
     * method to provide clean logging (every class should have this).
     *
     * This uses the uk.ac.ukc.iscream.util.NameFormat class
     * to format the toString()
     *
     * @return the name of this class and its CVS revision
     */
    public String toString() {
        return FormatName.getName(
            _name,
            getClass().getName(),
            REVISION);
    }

//---PRIVATE METHODS---

//---ACCESSOR/MUTATOR METHODS---

//---ATTRIBUTES---

    /**
     * A reference to the configuration manager
     */
    ConfigurationManager _configManager = ReferenceManager.getInstance().getCM();
    
    /**
     * This is the friendly identifier of the
     * component this class is running in.
     * eg, a Filter may be called "filter1",
     * If this class does not have an owning
     * component,  a name from the configuration
     * can be placed here.  This name could also
     * be changed to null for utility classes.
     */
    private String _name = FilterMain.NAME;

    /**
     * This holds a reference to the
     * system logger that is being used.
     */
    private Logger _logger = ReferenceManager.getInstance().getLogger();
    
    /**
     * The socket we are talking on
     */
    Socket _socket;
    
    /** 
     * The input from the socket
     */
    BufferedReader _socketIn;
    
    /**
     * The output from the socket
     */
    PrintWriter _socketOut;
    
    /**
     * A reference to our Queue
     */
    Queue _queue;
//---STATIC ATTRIBUTES---

}
Revision:	1.13
Committed:	Sun Jan 28 05:34:38 2001 UTC (23 years, 3 months ago) by tdb
Branch:	MAIN
Changes since 1.12:	+17 -4 lines
Log Message:	Some tidying up.
#	User	Rev	Content
1	tdb	1.1	//---PACKAGE DECLARATION---
2	tdb	1.5	package uk.ac.ukc.iscream.filter;
3	tdb	1.1
4			//---IMPORTS---
5			import uk.ac.ukc.iscream.core.*;
6			import uk.ac.ukc.iscream.filter.*;
7	tdb	1.12	import uk.ac.ukc.iscream.componentmanager.*;
8	tdb	1.1	import java.net.Socket;
9			import java.io.InputStream;
10			import java.io.OutputStream;
11			import java.io.IOException;
12			import java.io.*;
13	ajm	1.9	import uk.ac.ukc.iscream.util.*;
14	tdb	1.1
15			/**
16	ajm	1.10	* This provides Host heartbeat functionality
17	tdb	1.1	*
18	tdb	1.12	* @author $Author: tdb1 $
19	tdb	1.13	* @version $Id: TCPReaderInit.java,v 1.12 2001/01/18 23:13:36 tdb1 Exp $
20	tdb	1.1	*/
21			class TCPReaderInit extends Thread {
22
23			//---FINAL ATTRIBUTES---
24
25			/**
26			* The current CVS revision of this class
27			*/
28	tdb	1.13	public final String REVISION = "$Revision: 1.12 $";
29	tdb	1.1
30			//---STATIC METHODS---
31
32			//---CONSTRUCTORS---
33	tdb	1.13
34			/**
35			* Construct a new TCPReaderInit.
36			*
37			* @param socket the Socket to which the host is connected
38			* @param queue the Queue to which we'll add data
39			* @throws IOException if something goes badly wrong
40			*/
41	tdb	1.11	public TCPReaderInit(Socket socket, Queue queue) throws IOException {
42	tdb	1.1	_socket = socket;
43	tdb	1.11	_queue = queue;
44	tdb	1.13	// setup the reader & writer
45	tdb	1.1	_socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream()));
46			_socketOut = new PrintWriter(_socket.getOutputStream());
47	tdb	1.8	_logger.write(toString(), Logger.SYSINIT, "created");
48	tdb	1.1	}
49
50			//---PUBLIC METHODS---
51	tdb	1.13
52			/**
53			* Main run method. Will communicate with the host, inform it
54			* if any updates to it's configuration are needed, and send
55			* a heartbeat packet into the system.
56			*/
57	tdb	1.1	public void run() {
58			try {
59	tdb	1.3	//variables
60			String filelist = "";
61			String lastModified = "";
62			String inBound = "";
63	tdb	1.2
64	tdb	1.3	inBound = _socketIn.readLine();
65			if(!inBound.equals("HEARTBEAT")) {
66			_socketOut.println("ERROR");
67			_socketOut.flush();
68			throw new IOException("protocol error - expecting:HEARTBEAT got:" + inBound);
69			} else {
70			_socketOut.println("OK");
71			_socketOut.flush();
72			}
73
74			inBound = _socketIn.readLine();
75			if(!inBound.equals("CONFIG")) {
76			_socketOut.println("ERROR");
77			_socketOut.flush();
78			throw new IOException("protocol error - expecting:CONFIG got:" + inBound);
79			} else {
80			_socketOut.println("OK");
81			_socketOut.flush();
82			}
83
84			inBound = _socketIn.readLine();
85			filelist = inBound;
86			_socketOut.println("OK");
87			_socketOut.flush();
88
89			inBound = _socketIn.readLine();
90			lastModified = inBound;
91
92			boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
93
94			if(newConfig) {
95			_socketOut.println("ERROR");
96			}
97			else {
98			_socketOut.println("OK");
99			}
100			_socketOut.flush();
101
102			inBound = _socketIn.readLine();
103			if(!inBound.equals("ENDHEARTBEAT")) {
104			_socketOut.println("ERROR");
105			_socketOut.flush();
106			throw new IOException("protocol error - expecting:ENDHEARTBEAT got:" + inBound);
107			} else {
108			_socketOut.println("OK");
109			_socketOut.flush();
110			}
111	tdb	1.4
112	tdb	1.6	String date = new Long(System.currentTimeMillis()).toString();
113			String hostname = _socket.getInetAddress().getHostName();
114			String ipadd = _socket.getInetAddress().getHostAddress();
115			String xml = "<packet type=\"heartbeat\" machine_name=\""+hostname+"\" date=\""+date+"\" ip=\""+ipadd+"\"></packet>";
116
117	tdb	1.11	_queue.add(xml);
118	tdb	1.1
119			} catch (Exception e) {
120	tdb	1.8	_logger.write(toString(), Logger.ERROR, "ERROR: " + e.getMessage());
121	tdb	1.1	}
122
123			_socketOut.flush();
124			// Disconnect streams & socket
125			try {
126			_socketIn.close();
127			_socketOut.close();
128			_socket.close();
129			} catch (IOException e) {
130	tdb	1.8	_logger.write(toString(), Logger.ERROR, "exception on socket close");
131	tdb	1.1	}
132	tdb	1.8	_logger.write(toString(), Logger.SYSMSG, "finished");
133	tdb	1.1	}
134
135			/**
136			* Overrides the {@link java.lang.Object#toString() Object.toString()}
137			* method to provide clean logging (every class should have this).
138			*
139	ajm	1.10	* This uses the uk.ac.ukc.iscream.util.NameFormat class
140			* to format the toString()
141			*
142	tdb	1.1	* @return the name of this class and its CVS revision
143			*/
144			public String toString() {
145	ajm	1.10	return FormatName.getName(
146			_name,
147			getClass().getName(),
148			REVISION);
149	tdb	1.1	}
150
151			//---PRIVATE METHODS---
152
153			//---ACCESSOR/MUTATOR METHODS---
154
155			//---ATTRIBUTES---
156
157	ajm	1.10	/**
158			* A reference to the configuration manager
159			*/
160	tdb	1.8	ConfigurationManager _configManager = ReferenceManager.getInstance().getCM();
161	ajm	1.10
162			/**
163			* This is the friendly identifier of the
164			* component this class is running in.
165			* eg, a Filter may be called "filter1",
166			* If this class does not have an owning
167			* component, a name from the configuration
168			* can be placed here. This name could also
169			* be changed to null for utility classes.
170			*/
171			private String _name = FilterMain.NAME;
172
173			/**
174			* This holds a reference to the
175			* system logger that is being used.
176			*/
177			private Logger _logger = ReferenceManager.getInstance().getLogger();
178
179			/**
180			* The socket we are talking on
181			*/
182	tdb	1.1	Socket _socket;
183	ajm	1.10
184			/**
185			* The input from the socket
186			*/
187	tdb	1.1	BufferedReader _socketIn;
188	ajm	1.10
189			/**
190			* The output from the socket
191			*/
192	tdb	1.1	PrintWriter _socketOut;
193	ajm	1.10
194			/**
195	tdb	1.11	* A reference to our Queue
196	ajm	1.10	*/
197	tdb	1.11	Queue _queue;
198	tdb	1.1	//---STATIC ATTRIBUTES---
199
200			}