Wrong partition length raises exception
[phpeclipse.git] / archive / net.sourceforge.phpeclipse.wiki / src / net / sourceforge / phpeclipse / wiki / sql / WikipediaDB.java
1 package net.sourceforge.phpeclipse.wiki.sql;
2
3 import java.sql.Connection;
4 import java.sql.DriverManager;
5 import java.sql.PreparedStatement;
6 import java.sql.ResultSet;
7 import java.sql.ResultSetMetaData;
8 import java.sql.SQLException;
9 import java.sql.Statement;
10 import java.util.ArrayList;
11 import java.util.List;
12
13 import net.sourceforge.phpeclipse.wiki.editor.WikiEditorPlugin;
14 import net.sourceforge.phpeclipse.wiki.internal.ConfigurationManager;
15 import net.sourceforge.phpeclipse.wiki.internal.IConfiguration;
16 import net.sourceforge.phpeclipse.wiki.preferences.Util;
17
18 public class WikipediaDB {
19
20   public static void dump(ResultSet rs) throws SQLException {
21
22     // the order of the rows in a cursor
23     // are implementation dependent unless you use the SQL ORDER statement
24     ResultSetMetaData meta = rs.getMetaData();
25     int colmax = meta.getColumnCount();
26     int i;
27     Object o = null;
28
29     // the result set is a cursor into the data. You can only
30     // point to one row at a time
31     // assume we are pointing to BEFORE the first row
32     // rs.next() points to next row and returns true
33     // or false if there is no next row, which breaks the loop
34     for (; rs.next();) {
35       for (i = 0; i < colmax; ++i) {
36         o = rs.getObject(i + 1); // Is SQL the first column is indexed
37         // with 1 not 0
38         System.out.print(o.toString() + " ");
39       }
40
41       System.out.println(" ");
42     }
43   }
44
45   public static ArrayList getResultAsString(ResultSet rs) throws SQLException {
46     ArrayList list = new ArrayList();
47     int maxProposals = 500;
48     // the order of the rows in a cursor
49     // are implementation dependent unless you use the SQL ORDER statement
50     ResultSetMetaData meta = rs.getMetaData();
51     int colmax = meta.getColumnCount();
52     int i;
53     Object o = null;
54
55     // the result set is a cursor into the data. You can only
56     // point to one row at a time
57     // assume we are pointing to BEFORE the first row
58     // rs.next() points to next row and returns true
59     // or false if there is no next row, which breaks the loop
60     for (; rs.next();) {
61       for (i = 0; i < colmax; ++i) {
62         o = rs.getObject(i + 1); // Is SQL the first column is indexed
63         // with 1 not 0
64         list.add(o.toString());
65         maxProposals--;
66         if (maxProposals <= 0) {
67           return list;
68         }
69       }
70     }
71     return list;
72   }
73
74   public static String getFirstRow(ResultSet rs) throws SQLException {
75     // the order of the rows in a cursor
76     // are implementation dependent unless you use the SQL ORDER statement
77     ResultSetMetaData meta = rs.getMetaData();
78     int colmax = meta.getColumnCount();
79     int i;
80     Object o = null;
81
82     // the result set is a cursor into the data. You can only
83     // point to one row at a time
84     // assume we are pointing to BEFORE the first row
85     // rs.next() points to next row and returns true
86     // or false if there is no next row, which breaks the loop
87     for (; rs.next();) {
88       for (i = 0; i < colmax; ++i) {
89         o = rs.getObject(i + 1); // Is SQL the first column is indexed
90         // with 1 not 0
91         return o.toString();
92       }
93     }
94     return null;
95   }
96
97   public static void main(String[] args) {
98     WikipediaDB db = null;
99
100     try {
101       db = new WikipediaDB();
102     } catch (Exception ex1) {
103       ex1.printStackTrace(); // could not start db
104       return; // bye bye
105     }
106
107     try {
108       // do a query
109       ArrayList list = db.queryPrefixTexts("Programming:PHP");
110       //      db.query("SELECT * FROM cur WHERE cur_title like 'Programming:PHP%'"); // WHERE num_col < 250");
111       for (int i = 0; i < list.size(); i++) {
112         System.out.println(list.get(i).toString());
113       }
114       // at end of program
115       db.shutdown();
116     } catch (SQLException ex3) {
117       ex3.printStackTrace();
118     }
119   }
120
121   //  private static void readFile(WikipediaDB db, String filename) {
122   //    FileReader fileReader;
123   //    try {
124   //      BufferedReader bufferedReader = new BufferedReader(new FileReader(filename));
125   //      // String line;
126   //      LineTokenizer lineTokenizer = new LineTokenizer();
127   //      StringBuffer line = new StringBuffer(1024);
128   //      while (lineTokenizer.getToken(line, bufferedReader)) {
129   //        if (line.length() == 0) {
130   //          // this should not happen
131   //        } else {
132   //          // try {
133   //          // db.update("INSERT INTO wp_titles(title) VALUES('" + line + "')");
134   //          System.out.println(line);
135   //          line.delete(0, line.length());
136   //          // addLine(line);
137   //          // } catch (SQLException ex3) {
138   //          //// ex3.printStackTrace();
139   //          // }
140   //        }
141   //      }
142   //      bufferedReader.close();
143   //    } catch (FileNotFoundException e) {
144   //      // ignore this
145   //      // TODO DialogBox which asks the user if she/he likes to build new index?
146   //    } catch (IOException e) {
147   //      // TODO Auto-generated catch block
148   //      e.printStackTrace();
149   //    }
150   //  }
151
152   private final Connection conn;
153
154   private final PreparedStatement fGetPrefixTitles;
155
156   private final PreparedStatement fGetPrefixTexts;
157
158   private final PreparedStatement fGetExactText;
159
160   public WikipediaDB() throws Exception // note more general exception
161   {
162
163     // Load the Database Engine JDBC driver
164     // mysql-connector.jar should be in the class path or made part of the current jar
165     Class.forName("com.mysql.jdbc.Driver");
166
167     // determine the first SQL configuration
168     IConfiguration configuration = null;
169     try {
170       List allConfigsList = ConfigurationManager.getInstance().getConfigurations();
171       ArrayList configsList = new ArrayList();
172       for (int i = 0; i < allConfigsList.size(); i++) {
173         configuration = (IConfiguration) allConfigsList.get(i);
174         if (configuration.getType().equals(WikiEditorPlugin.WIKIPEDIA_SQL)) {
175           break;
176         }
177         configuration = null;
178       }
179     } catch (Throwable th) {
180       // 
181     }
182
183     // connect to the database. This will load the db files and start the
184     // database if it is not alread running.
185     // db_file_name_prefix is used to open or create files that hold the state
186     // of the db.
187     // It can contain directory names relative to the
188     // current working directory
189     if (configuration != null) {
190       conn = DriverManager.getConnection(configuration.getURL(), configuration.getUser(), configuration.getPassword());
191     } else {
192       // default configuration for XAMPP distribution
193       conn = DriverManager.getConnection("jdbc:mysql://localhost/wikidb", // filenames
194           "root", // category
195           ""); // password
196     }
197     fGetPrefixTitles = conn.prepareStatement("SELECT cur_title FROM cur WHERE LOWER( cur_title ) like ?");
198     fGetPrefixTexts = conn.prepareStatement("SELECT cur_text FROM cur WHERE LOWER( cur_title ) like ?");
199     fGetExactText = conn.prepareStatement("SELECT cur_text FROM cur WHERE cur_title = ?");
200   }
201
202   //use for SQL commands CREATE and SELECT
203   public synchronized void query(String expression) throws SQLException {
204
205     Statement st = null;
206     ResultSet rs = null;
207
208     st = conn.createStatement(); // statement objects can be reused with
209     // repeated calls to execute but we
210     // choose to make a new one each time
211     rs = st.executeQuery(expression); // run the query
212
213     // do something with the result set.
214     dump(rs);
215     st.close(); // NOTE!! if you close a statement the associated ResultSet is
216     // closed too
217     // so you should copy the contents to some other object.
218     // the result set is invalidated also if you recycle an Statement
219     // and try to execute some other query before the result set has been
220     // completely examined.
221   }
222
223   public synchronized ArrayList queryPrefixTitle(String prefix) throws SQLException {
224     fGetPrefixTitles.setString(1, prefix.toLowerCase() + '%');
225     ResultSet rs = null;
226     rs = fGetPrefixTitles.executeQuery(); // run the query
227     // do something with the result set.
228     ArrayList list = getResultAsString(rs);
229     if (list != null) {
230       // convert to editor format
231       for (int i = 0; i < list.size(); i++) {
232         list.set(i, Util.db2TitleLink((String) list.get(i)));
233       }
234     }
235     return list;
236     //    st.close(); // NOTE!! if you close a statement the associated ResultSet is
237   }
238
239   public synchronized ArrayList queryPrefixTexts(String prefix) throws SQLException {
240     fGetPrefixTexts.setString(1, prefix.toLowerCase() + '%');
241     ResultSet rs = null;
242     rs = fGetPrefixTexts.executeQuery(); // run the query
243     // do something with the result set.
244     return getResultAsString(rs);
245     //    st.close(); // NOTE!! if you close a statement the associated ResultSet is
246   }
247
248   public synchronized String queryExactText(String prefix) throws SQLException {
249     fGetExactText.setString(1, prefix);
250     ResultSet rs = null;
251     rs = fGetExactText.executeQuery(); // run the query
252     // do something with the result set.
253     return getFirstRow(rs);
254     //    st.close(); // NOTE!! if you close a statement the associated ResultSet is
255   }
256
257   public static String getExactText(String prefix) {
258     WikipediaDB db = null;
259
260     try {
261       db = new WikipediaDB();
262     } catch (Exception ex1) {
263       ex1.printStackTrace(); // could not start db
264       return null; // bye bye
265     }
266
267     try {
268       String text = db.queryExactText(prefix);
269       db.shutdown();
270       return text;
271     } catch (SQLException ex3) {
272       ex3.printStackTrace();
273     }
274     return null;
275   }
276
277   public void shutdown() throws SQLException {
278
279     conn.close(); // if there are no other open connection
280     // db writes out to files and shuts down
281     // this happens anyway at garbage collection
282     // when program ends
283   }
284
285   //use for SQL commands DROP and INSERT and UPDATE
286   public synchronized void update(String expression) throws SQLException {
287
288     Statement st = null;
289
290     st = conn.createStatement(); // statements
291
292     int i = st.executeUpdate(expression); // run the query
293
294     if (i == -1) {
295       System.out.println("db error : " + expression);
296     }
297
298     st.close();
299   }
300 }