1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
package main;
import com.google.gson.JsonIOException;
import com.google.gson.JsonSyntaxException;
import data.Tweet;
import database.DBConnection;
import io.FileTweetReader;
import io.ITweetReader;
import io.TweetReader;
import java.io.IOException;
/**
* The main class.
*/
public class Main {
/**
* The main method of the application.
*
* @param args the global arguments to pass to the program.
*/
public static void main(String[] args) {
try {
Main main = new Main(args);
} catch (IllegalArgumentException ex) {
System.err.println(ex.getMessage());
System.exit(1);
}
}
private String m_hostaddress;
private String m_filename;
public Main(String[] args) {
/* parse the global options. */
parseGlobalOptions(args);
if (m_hostaddress == null) {
throw new IllegalArgumentException("Missing --dbhost to specify the hostaddress.");
}
DBConnection connection = new DBConnection(m_hostaddress, "5432", "Twitter", "postgres", "2IOC02");
/* create the object that fills the database */
DataFiller filler = new DataFiller(connection);
ITweetReader reader = null;
try {
if (m_filename == null) {
reader = new TweetReader(System.in);
} else {
reader = new FileTweetReader(m_filename);
}
Tweet tweet;
while ((tweet = reader.getTweet()) != null) {
filler.processTweet(tweet);
}
} catch (JsonSyntaxException ex) {
System.err.println("Got an invalid tweet: " + ex);
} catch (IOException ex) {
System.err.println("Cannot open tweets: " + ex);
} finally {
if (reader != null) {
reader.close();
}
}
connection.close();
System.out.print("exit succesfull.");
}
private void parseGlobalOptions(String[] args) {
/* parse global options */
for (int i = 0; i < args.length; i++) {
if ("--help".equals(args[i])) {
printHelp();
} else if ("--dbhost".equals(args[i])) {
m_hostaddress = getParam(args, ++i);
} else if (args[i].startsWith("-")) {
throw new IllegalArgumentException("Invalid option: " + args[i]);
} else {
/* This should be the filename */
m_filename = getParam(args, i);
}
}
if (args.length == 0) {
throw new IllegalArgumentException("No parameters specified, see --help for usage info.");
}
}
/**
* Read an extra option for a command.
*/
private String getParam(String[] args, Integer index) {
if (index + 1 <= args.length) {
index++;
return args[index - 1];
} else {
throw new IllegalArgumentException("An extra option was missing.");
}
}
/**
* Print some useful help messages.
*/
private void printHelp() {
for (String line : HELP) {
System.out.println(line);
}
}
private final static String[] HELP = {
"Global options:",
" --help Print this help text.",
" --ip <ip> Specify the database ip address.",
" --file <tweet> <profile> Specify the tweet and profile filenames ",
" ",
" if no --file was specified read from standard input."
};
}
|
