ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/cms/source/server/uk/org/iscream/cms/server/filtermanager/HostInit.java
(Generate patch)

Comparing projects/cms/source/server/uk/org/iscream/cms/server/filtermanager/HostInit.java (file contents):
Revision 1.8 by ajm, Mon Nov 27 23:09:06 2000 UTC vs.
Revision 1.26 by tdb, Tue Mar 13 13:40:26 2001 UTC

# Line 1 | Line 1
1   //---PACKAGE DECLARATION---
2 + package uk.ac.ukc.iscream.filtermanager;
3  
4   //---IMPORTS---
5   import uk.ac.ukc.iscream.core.*;
6   import uk.ac.ukc.iscream.filter.*;
7 < import java.net.Socket;
8 < import java.io.InputStream;
9 < import java.io.OutputStream;
9 < import java.io.IOException;
7 > import uk.ac.ukc.iscream.util.*;
8 > import uk.ac.ukc.iscream.componentmanager.*;
9 > import java.net.*;
10   import java.io.*;
11 < import org.omg.CORBA.*;
12 < import org.omg.CosNaming.*;
11 > import java.util.*;
12  
13   /**
14 < * <ONE LINE DESCRIPTION>
15 < * <DETAILED DESCRIPTION>
14 > * Handles setting up a host.
15 > * This class provides a host with appropriate configuration
16 > * and a reference to a Filter to which it should pass data.
17   *
18   * @author  $Author$
19   * @version $Id$
# Line 31 | Line 31 | class HostInit extends Thread {
31  
32   //---CONSTRUCTORS---
33  
34 <    public HostInit(Socket socket, ConfigurationManager configManager, Logger logRef, NamingContextExt ncRef) throws IOException {
35 <        _configManager = configManager;
36 <        _logRef = logRef;
34 >    /**
35 >     * Construct a new HostInit.
36 >     *
37 >     * @param socket The socket to which the host is connected
38 >     */
39 >    public HostInit(Socket socket) throws IOException {
40 >        // set the Thread name
41 >        setName("filtermanager.HostInit");
42 >        
43          _socket = socket;
44 <        _ncRef = ncRef;
44 >        // setup reader & writer
45          _socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream()));
46 <        _socketOut = new PrintWriter(_socket.getOutputStream());
47 <        _logRef.write(toString(), Logger.SYSINIT, "created");
46 >        _socketOut = new PrintWriter(_socket.getOutputStream(), true);
47 >        _logger.write(toString(), Logger.SYSINIT, "created");
48      }
49      
50   //---PUBLIC METHODS---
51  
52 +    /**
53 +     * Main method in this class, which handles communicating with
54 +     * the host to determine it's setup.
55 +     */
56      public void run() {
57 +        // get a hook on the config proxy
58 +        ConfigurationProxy cp = ConfigurationProxy.getInstance();
59 +        // this is our config name
60 +        String configName = "Host." + _socket.getInetAddress().getHostName().toLowerCase();
61          try {
62 <            String inBound = _socketIn.readLine();
63 <            if (!inBound.equals("STARTCONFIG")) {
64 <                _socketOut.println("ERROR");
65 <                _socketOut.flush();
66 <                throw new IOException("protocol error - expected:STARTCONFIG got:" + inBound);
67 <            }
68 <
69 <            Configuration myConfig = _configManager.getConfiguration("Host." + _socket.getInetAddress().getHostName().toLowerCase());
70 <            if (myConfig == null) {
71 <                _socketOut.println("ERROR");
72 <                throw new IOException("error in host configuration");
73 <            } else {
74 <                _socketOut.println("OK");
75 <                _socketOut.flush();
76 <                
77 <                // get lastmodified
78 <                inBound = _socketIn.readLine();
79 <                if(!inBound.equals("LASTMODIFIED")) {
80 <                        // protocol error
62 >            // try for STARTCONFIG
63 >            getInBound("STARTCONFIG");
64 >            _socketOut.println("OK");
65 >            
66 >            // try for LASTMODIFIED
67 >            getInBound("LASTMODIFIED");
68 >            _socketOut.println(cp.getLastModified(configName));
69 >            
70 >            // try for FILELIST
71 >            getInBound("FILELIST");
72 >            _socketOut.println(cp.getFileList(configName));
73 >            
74 >            // try for FQDN
75 >            getInBound("FQDN");
76 >            _socketOut.println(_socket.getInetAddress().getHostName().toLowerCase());
77 >            
78 >            // get properties
79 >            String reqProperty = getInBound();
80 >            while(!reqProperty.equals("ENDCONFIG")) {
81 >                // get the property
82 >                try {
83 >                    String returnedProperty = cp.getProperty(configName, "Host."+reqProperty);
84 >                    _socketOut.println(returnedProperty);
85 >                } catch (PropertyNotFoundException e) {
86                      _socketOut.println("ERROR");
68                    _socketOut.flush();
69                    throw new IOException("protocol error - expected:LASTMODIFIED got:" + inBound);
87                  }
88 <                else {
89 <                        // send info
90 <                        _socketOut.println(myConfig.getLastModified());
91 <                        _socketOut.flush();
88 >                // get the next request
89 >                reqProperty = _socketIn.readLine();
90 >            }
91 >            _logger.write(toString(), Logger.SYSMSG, "configured host");
92 >            _socketOut.println("OK");
93 >            
94 >            // get filter reference
95 >            getInBound("FILTER");
96 >                // send info
97 >                String filterList = cp.getProperty(configName, "Host.filter");
98 >                Filter filterRef = null;
99 >                String filter = null;
100 >                StringTokenizer st = new StringTokenizer(filterList, ";");
101 >            while (filterRef==null && st.hasMoreTokens()) {
102 >                filter = st.nextToken();
103 >                _logger.write(toString(), Logger.DEBUG, " looking for filter- " + filter);
104 >                try {
105 >                    filterRef = FilterHelper.narrow(ReferenceManager.getInstance().getCORBARef("iscream.Filter." + filter));
106 >                } catch (ComponentCORBAException e) {
107 >                    _logger.write(toString(), Logger.DEBUG, " unable to find filter- " + filter);
108                  }
76                
77                // get config fileList
78                inBound = _socketIn.readLine();
79                if(!inBound.equals("FILELIST")) {
80                        // protocol error
81                    _socketOut.println("ERROR");
82                    _socketOut.flush();
83                    throw new IOException("protocol error - expected:FILELIST got:" + inBound);
84                }
85                else {
86                        // send info
87                        _socketOut.println(myConfig.getFileList());
88                        _socketOut.flush();
89                }
90
91                // get properties
92                inBound = _socketIn.readLine();
93                while(!inBound.equals("ENDCONFIG")) {
94                          
95                    // get the property
96                    try {
97                        String returnedProperty = myConfig.getProperty(inBound);    
98                        
99                        _socketOut.println(returnedProperty);
100                        _socketOut.flush();
101    
102                    } catch (org.omg.CORBA.MARSHAL e) {
103                        _socketOut.println("ERROR");
104                        _socketOut.flush();
105                    }
106                    inBound = _socketIn.readLine();
107                }
108                _logRef.write(toString(), Logger.SYSMSG, "configured host");
109                _socketOut.println("OK");
110                _socketOut.flush();
111                
112                // get filter reference
113                inBound = _socketIn.readLine();
114                if(!inBound.equals("FILTER")) {
115                        // protocol error
116                    _socketOut.println("ERROR");
117                    _socketOut.flush();
118                    throw new IOException("protocol error - expected:FILTER got:" + inBound);
119                }
120                else {
121                        // send info
122                        String parentFilter =  myConfig.getProperty("Host.filter");
123                        _logRef.write(toString(), Logger.DEBUG, " looking for parent - " + parentFilter);
124                        org.omg.CORBA.Object objRef = _ncRef.resolve(_ncRef.to_name("iscream.Filter." + parentFilter));
125                    Filter filter = FilterHelper.narrow(objRef);
126                        _socketOut.println(filter.getHostName() + ";"
127                                         + filter.getUDPPort() + ";"
128                                         + filter.getTCPPort());
129                        _socketOut.flush();
130                }
131                
132                // confirm that all is ok
133                inBound = _socketIn.readLine();
134                if(!inBound.equals("END")) {
135                        // protocol error
136                    _socketOut.println("ERROR");
137                    _socketOut.flush();
138                    throw new IOException("protocol error - expected:END got:" + inBound);
139                }
140                else {
141                        // send ok
142                        _socketOut.println("OK");
143                        _socketOut.flush();
144                }
145
109              }
110 +                
111 +                // hopefully we found a filter
112 +            if(filterRef != null) {
113 +                _logger.write(toString(), Logger.DEBUG, " found filter- " + filter);
114 +                // tell the host about it...
115 +                    _socketOut.println(filterRef.getHostName() + ";"
116 +                                     + filterRef.getUDPPort() + ";"
117 +                                     + filterRef.getTCPPort());
118 +                }
119 +                else {
120 +                    // ...or throw a wobbly (and tell the host!)
121 +                    _socketOut.println("ERROR");
122 +                    throw new IOException("unable to find filter for host");
123 +                }
124              
125 +            // confirm ok with END
126 +            getInBound("END");
127 +            _socketOut.println("OK");
128 +            
129          } catch (Exception e) {
130 <            _logRef.write(toString(), Logger.ERROR, "ERROR: " + e.getMessage());
130 >            _logger.write(toString(), Logger.ERROR, "ERROR - " + e);
131          }
132          
152        _socketOut.flush();
133          // Disconnect streams & socket
134          try {
135              _socketIn.close();
136              _socketOut.close();
137              _socket.close();
138          } catch (IOException e) {
139 <            _logRef.write(toString(), Logger.ERROR, "exception on socket close");
139 >            _logger.write(toString(), Logger.ERROR, "exception on socket close");
140          }
141 <        _logRef.write(toString(), Logger.SYSMSG, "finished");
141 >        _logger.write(toString(), Logger.SYSMSG, "finished");
142      }
143      
144      /**
145       * Overrides the {@link java.lang.Object#toString() Object.toString()}
146       * method to provide clean logging (every class should have this).
147       *
148 +     * This uses the uk.ac.ukc.iscream.util.NameFormat class
149 +     * to format the toString()
150 +     *
151       * @return the name of this class and its CVS revision
152       */
153      public String toString() {
154 <        return this.getClass().getName() + "{" + _socket.getInetAddress().getHostName()
155 <         + "}(" + REVISION.substring(11, REVISION.length() - 2) + ")";
156 <        
154 >        return FormatName.getName(
155 >            _name,
156 >            getClass().getName(),
157 >            REVISION);
158      }
159  
160   //---PRIVATE METHODS---
161  
162 +    private String getInBound(String expected) throws IOException {
163 +        // grab the input
164 +        String inBound = getInBound();
165 +        // check if it's what we're expecting
166 +        if(!inBound.equals(expected)) {
167 +            throw new IOException("protocol error - expected:"+expected+" got:" + inBound);
168 +        }
169 +        // it should be ok then
170 +        return inBound;
171 +    }
172 +    
173 +    private String getInBound() throws IOException {
174 +        // grab the input
175 +        String inBound = _socketIn.readLine();
176 +        // check for null's, likely disconnection
177 +        if(inBound == null) {
178 +            throw new IOException("got null from host, maybe it died");
179 +        }
180 +        // it's a valid message it seems
181 +        return inBound;
182 +    }
183 +
184   //---ACCESSOR/MUTATOR METHODS---
185  
186   //---ATTRIBUTES---
187  
188 <    ConfigurationManager _configManager;
189 <    Logger _logRef;
190 <    Socket _socket;
191 <    BufferedReader _socketIn;
192 <    PrintWriter _socketOut;
193 <    NamingContextExt _ncRef;
188 >    /**
189 >     * This holds a reference to the
190 >     * system logger that is being used.
191 >     */
192 >    private Logger _logger = ReferenceManager.getInstance().getLogger();
193 >    
194 >    /**
195 >     * A reference to the Configuration Manager the system is using
196 >     */
197 >    private ConfigurationManager _configManager = ReferenceManager.getInstance().getCM();
198 >    
199 >    /**
200 >     * This is the friendly identifier of the
201 >     * component this class is running in.
202 >     * eg, a Filter may be called "filter1",
203 >     * If this class does not have an owning
204 >     * component,  a name from the configuration
205 >     * can be placed here.  This name could also
206 >     * be changed to null for utility classes.
207 >     */
208 >    private String _name = FilterManager.NAME;
209 >    
210 >    /**
211 >     * The socket this class uses
212 >     */
213 >    private Socket _socket;
214 >
215 >    /**
216 >     * Used for the input stream of this socket
217 >     */
218 >    private BufferedReader _socketIn;
219 >    
220 >    /**
221 >     * Used for the output stream of this socket
222 >     */
223 >    private PrintWriter _socketOut;
224 >
225   //---STATIC ATTRIBUTES---
226  
227   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines