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.15 2001/02/07 13:47:11 tdb1 Exp $
 */
class TCPReaderInit extends Thread {

//---FINAL ATTRIBUTES---

    /**
     * The current CVS revision of this class
     */
    public final String REVISION = "$Revision: 1.15 $";
    
//---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(), 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()).toString();
            String hostname = _socket.getInetAddress().getHostName();
            String ipadd = _socket.getInetAddress().getHostAddress();
            
            // run the service checks for this host
            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.getMessage());
        }
        
        // 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.16
Committed:	Sun Feb 11 18:01:43 2001 UTC (23 years, 3 months ago) by ajm
Branch:	MAIN
Changes since 1.15:	+9 -3 lines
Log Message:	initial checkin of the service check functionality
#	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	ajm	1.16	* @version $Id: TCPReaderInit.java,v 1.15 2001/02/07 13:47:11 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	ajm	1.16	public final String REVISION = "$Revision: 1.15 $";
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	tdb	1.15	_socketOut = new PrintWriter(_socket.getOutputStream(), true);
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			throw new IOException("protocol error - expecting:HEARTBEAT got:" + inBound);
68			} else {
69			_socketOut.println("OK");
70			}
71
72			inBound = _socketIn.readLine();
73			if(!inBound.equals("CONFIG")) {
74			_socketOut.println("ERROR");
75			throw new IOException("protocol error - expecting:CONFIG got:" + inBound);
76			} else {
77			_socketOut.println("OK");
78			}
79
80			inBound = _socketIn.readLine();
81			filelist = inBound;
82			_socketOut.println("OK");
83
84			inBound = _socketIn.readLine();
85			lastModified = inBound;
86
87			boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
88
89			if(newConfig) {
90			_socketOut.println("ERROR");
91			}
92			else {
93			_socketOut.println("OK");
94			}
95
96			inBound = _socketIn.readLine();
97			if(!inBound.equals("ENDHEARTBEAT")) {
98			_socketOut.println("ERROR");
99			throw new IOException("protocol error - expecting:ENDHEARTBEAT got:" + inBound);
100			} else {
101			_socketOut.println("OK");
102			}
103	tdb	1.4
104	tdb	1.6	String date = new Long(System.currentTimeMillis()).toString();
105			String hostname = _socket.getInetAddress().getHostName();
106			String ipadd = _socket.getInetAddress().getHostAddress();
107
108	ajm	1.16	// run the service checks for this host
109			String checks = PluginServiceCheckManager.getInstance().runServiceChecks(hostname);
110
111			// build the heartbeat packet
112			String xml = "<packet type=\"heartbeat\" machine_name=\""+hostname+"\" date=\""+date+"\" ip=\""+ipadd+"\">" + checks + "</packet>";
113
114			// get it to be sent on
115	tdb	1.11	_queue.add(xml);
116	tdb	1.1
117			} catch (Exception e) {
118	tdb	1.8	_logger.write(toString(), Logger.ERROR, "ERROR: " + e.getMessage());
119	tdb	1.1	}
120
121			// Disconnect streams & socket
122			try {
123			_socketIn.close();
124			_socketOut.close();
125			_socket.close();
126			} catch (IOException e) {
127	tdb	1.8	_logger.write(toString(), Logger.ERROR, "exception on socket close");
128	tdb	1.1	}
129	tdb	1.14	_logger.write(toString(), Logger.DEBUG, "finished");
130	tdb	1.1	}
131
132			/**
133			* Overrides the {@link java.lang.Object#toString() Object.toString()}
134			* method to provide clean logging (every class should have this).
135			*
136	ajm	1.10	* This uses the uk.ac.ukc.iscream.util.NameFormat class
137			* to format the toString()
138			*
139	tdb	1.1	* @return the name of this class and its CVS revision
140			*/
141			public String toString() {
142	ajm	1.10	return FormatName.getName(
143			_name,
144			getClass().getName(),
145			REVISION);
146	tdb	1.1	}
147
148			//---PRIVATE METHODS---
149
150			//---ACCESSOR/MUTATOR METHODS---
151
152			//---ATTRIBUTES---
153
154	ajm	1.10	/**
155			* A reference to the configuration manager
156			*/
157	tdb	1.8	ConfigurationManager _configManager = ReferenceManager.getInstance().getCM();
158	ajm	1.10
159			/**
160			* This is the friendly identifier of the
161			* component this class is running in.
162			* eg, a Filter may be called "filter1",
163			* If this class does not have an owning
164			* component, a name from the configuration
165			* can be placed here. This name could also
166			* be changed to null for utility classes.
167			*/
168			private String _name = FilterMain.NAME;
169
170			/**
171			* This holds a reference to the
172			* system logger that is being used.
173			*/
174			private Logger _logger = ReferenceManager.getInstance().getLogger();
175
176			/**
177			* The socket we are talking on
178			*/
179	tdb	1.1	Socket _socket;
180	ajm	1.10
181			/**
182			* The input from the socket
183			*/
184	tdb	1.1	BufferedReader _socketIn;
185	ajm	1.10
186			/**
187			* The output from the socket
188			*/
189	tdb	1.1	PrintWriter _socketOut;
190	ajm	1.10
191			/**
192	tdb	1.11	* A reference to our Queue
193	ajm	1.10	*/
194	tdb	1.11	Queue _queue;
195	tdb	1.1	//---STATIC ATTRIBUTES---
196
197			}