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.20 2001/03/04 05:24:08 tdb1 Exp $
 */
class TCPReaderInit extends Thread {

//---FINAL ATTRIBUTES---

    /**
     * The current CVS revision of this class
     */
    public final String REVISION = "$Revision: 1.20 $";
    
//---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 {
        // set the Thread name
        setName("filter.TCPReaderInit");
        
        _socket = socket;
        _queue = queue;
        // setup the reader & writer
        _socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream()));
        _socketOut = new PrintWriter(_socket.getOutputStream(), true);
        _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");
                throw new IOException("protocol error - expecting:HEARTBEAT got:" + inBound);
            } else {
                _socketOut.println("OK");
            }
            
            inBound = _socketIn.readLine();
            if(!inBound.equals("CONFIG")) {
                _socketOut.println("ERROR");
                throw new IOException("protocol error - expecting:CONFIG got:" + inBound);
            } else {
                _socketOut.println("OK");
            }
            
            inBound = _socketIn.readLine();
            filelist = inBound;
            _socketOut.println("OK");
            
            inBound = _socketIn.readLine();
            lastModified = inBound;
            
            boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
            
            if(newConfig) {
                _socketOut.println("ERROR");
            }
            else {
                _socketOut.println("OK");
            }
            
            inBound = _socketIn.readLine();
            if(!inBound.equals("ENDHEARTBEAT")) {
                _socketOut.println("ERROR");
                throw new IOException("protocol error - expecting:ENDHEARTBEAT got:" + inBound);
            } else {
                _socketOut.println("OK");
            }

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