TrainManagementSystem.java

import java.util.*;
import javax.jms.*;
import java.io.*;
import java.lang.*;

public class TrainManagementSystem{
 
 public static String serverUrl;
 public static TrainMsgSyncProducer breakdownMsger;
 public static TrainMsgProducer resumeServiceMsger;
 
 public static void main(String[] args){
 Scanner sc = new Scanner(System.in);
 boolean run = true;
 serverUrl = "localhost"; //if this is done locally 
 initializeMsgProducers();
 
 
 while(run){
 System.out.println();
 System.out.println();
 

 System.out.println("====================================");
 System.out.println(" Train Management System ");
 System.out.println("====================================");

 System.out.println();
 System.out.println();

 System.out.println("Server URL: " + serverUrl);


 System.out.println("1. Update Server URL");
 System.out.println("2. Send Breakdown Message");
 System.out.println("3. Send System Resumed Message");
 System.out.println("4. Quit Program");
 System.out.println();
 //cls();
 
 System.out.print("Option: ");
 int input = sc.nextInt();
 run = routeInput(input,sc);
 
 
 
 }

 }
 
 //Working
 public static boolean initializeMsgProducers(){
 breakdownMsger = new TrainMsgSyncProducer(serverUrl,"q.breakdown","q.deployed",TrainMsgProducer.QUEUE);
 resumeServiceMsger = new TrainMsgProducer(serverUrl,"q.resumed",TrainMsgProducer.QUEUE);
 return true;
 }
 
 //Route the input from main
 public static boolean routeInput(int input, Scanner sc){
 
 if(input == 1){
 updateURL(sc);
 cls();
 return true;
 }
 if(input == 2){
 messageScreen();
 cls();
 return true;
 }
 if (input == 3){
 systemResumedScreen();
 cls();
 return true;
 }

 cls();
 return false;
 
 }
 
 public static void updateURL(Scanner sc){
 System.out.println(); 
 System.out.println("Current serverURL: " + serverUrl);
 System.out.print("New serverURL: ");
 System.out.flush();
 
 serverUrl = sc.next(); //Get the new serverURL and update it
 System.out.flush();
 
 //if you update reinitialize the message producers
 initializeMsgProducers();
 }
 
 public static void messageScreen(){
 processAllFiles(); //This process the XML files and send it 
 }
 
 public static void systemResumedScreen(){
 System.out.println("Informing q.resumed that systems has resumed");
 resumeServiceMsger.sendMessage("SYSTEM RESUMED");
 System.out.println("MESSAGE SENT");
 }
 
 
 public static void processAllFiles(){
 String address = "C://EI//Project//breakdownReports";
 visitAllFiles(new File(address));
 }
 
 //Visit all the files in the directory and process them
 public static void visitAllFiles(File dir) {
 if (dir.isDirectory()) {
 String[] children = dir.list();
 if (children.length == 0) {
 System.out.println("There is no order to process.");
 }
 for (int i = 0; i < children.length; i++) {
 visitAllFiles(new File(dir, children[i]));
 }
 } else {
 readXML(dir);
 }
 }

 /*
 * This method reads an XML document and put the content into Vector data
 */
 public static void readXML(File filename) {
 
 BufferedReader bufferedReader = null;
 try {
 //Construct the BufferedReader object
 bufferedReader = new BufferedReader(new FileReader(filename));
 String line = null;
 Vector data = new Vector();
 while ((line = bufferedReader.readLine()) != null) {
 //Add the line from XML file to the message to be sent as JMS msg
 data.addElement(line);
 }
 sendMessage(data);
 } catch (FileNotFoundException ex) {
 ex.printStackTrace();
 } catch (IOException ex) {
 ex.printStackTrace();
 } finally {
 //Close the BufferedReader
 try {
 if (bufferedReader != null)
 bufferedReader.close();
 } catch (IOException ex) {
 ex.printStackTrace();
 }
 }
 }
 
 private static void sendMessage(Vector data){
 
 StringBuffer finalMessage = new StringBuffer();
 int i;
 if(data.size() == 0){
 System.err.println("***Error: Order file is empty\n");
 }else{
 finalMessage.append((String)data.elementAt(0));
 for(i = 1; i < data.size();i++){
 finalMessage.append(' ');
 finalMessage.append((String)data.elementAt(i));
 }
 }
 
 breakdownMsger.sendMessage(finalMessage.toString());
 MessageConsumer replyConsumer;
 
 }
 
 public static void cls(){
 for (int i = 0 ; i < 10; i ++){
 System.out.println();
 }
 
 }
}