001: /* Copyright (c) 2001-2007, The HSQL Development Group
002: * All rights reserved.
003: *
004: * Redistribution and use in source and binary forms, with or without
005: * modification, are permitted provided that the following conditions are met:
006: *
007: * Redistributions of source code must retain the above copyright notice, this
008: * list of conditions and the following disclaimer.
009: *
010: * Redistributions in binary form must reproduce the above copyright notice,
011: * this list of conditions and the following disclaimer in the documentation
012: * and/or other materials provided with the distribution.
013: *
014: * Neither the name of the HSQL Development Group nor the names of its
015: * contributors may be used to endorse or promote products derived from this
016: * software without specific prior written permission.
017: *
018: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
019: * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
020: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
021: * ARE DISCLAIMED. IN NO EVENT SHALL HSQL DEVELOPMENT GROUP, HSQLDB.ORG,
022: * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
023: * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
024: * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
025: * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
026: * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
027: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
028: * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
029: */
030:
031: package org.hsqldb.util;
032:
033: import java.io.File;
034: import java.io.FileInputStream;
035: import java.util.ArrayList;
036: import java.util.Date;
037: import java.util.Properties;
038:
039: /* $Id: SqlToolSprayer.java,v 1.21 2007/06/12 02:37:57 unsaved Exp $ */
040:
041: /**
042: * Sql Tool Sprayer.
043: * Invokes SqlTool.objectMain() multiple times with the same SQL.
044: * Invokes for multiple urlids and/or retries.
045: *
046: * See JavaDocs for the main method for syntax of how to run.
047: *
048: * System properties used if set:
049: * <UL>
050: * <LI>sqltoolsprayer.period (in ms.)</LI>
051: * <LI>sqltoolsprayer.maxtime (in ms.)</LI>
052: * <LI>sqltoolsprayer.rcfile (filepath)</LI>
053: * </UL>
054: *
055: * @see @main()
056: * @version $Revision: 1.21 $
057: * @author Blaine Simpson unsaved@users
058: */
059: public class SqlToolSprayer {
060:
061: public static String LS = System.getProperty("line.separator");
062: private static String SYNTAX_MSG = "SYNTAX: java [-D...] SqlToolSprayer 'SQL;' [urlid1 urlid2...]\n"
063: + "System properties you may use [default values]:\n"
064: + " sqltoolsprayer.period (in ms.) [500]\n"
065: + " sqltoolsprayer.maxtime (in ms.) [0]\n"
066: + " sqltoolsprayer.monfile (filepath) [none]\n"
067: + " sqltoolsprayer.rcfile (filepath) [none. SqlTool default used.]"
068: + "\n sqltoolsprayer.propfile (filepath) [none]";
069: static {
070: if (!LS.equals("\n")) {
071: SYNTAX_MSG = SYNTAX_MSG.replaceAll("\n", LS);
072: }
073: }
074:
075: public static void main(String[] sa) {
076:
077: if (sa.length < 1) {
078: System.err.println(SYNTAX_MSG);
079: System.exit(4);
080: }
081:
082: long period = ((System.getProperty("sqltoolsprayer.period") == null) ? 500
083: : Integer.parseInt(System
084: .getProperty("sqltoolsprayer.period")));
085: long maxtime = ((System.getProperty("sqltoolsprayer.maxtime") == null) ? 0
086: : Integer.parseInt(System
087: .getProperty("sqltoolsprayer.maxtime")));
088: String rcFile = System.getProperty("sqltoolsprayer.rcfile");
089: String propfile = System.getProperty("sqltoolsprayer.propfile");
090: File monitorFile = (System
091: .getProperty("sqltoolsprayer.monfile") == null) ? null
092: : new File(System.getProperty("sqltoolsprayer.monfile"));
093: ArrayList urlids = new ArrayList();
094:
095: if (propfile != null) {
096: try {
097: getUrlsFromPropFile(propfile, urlids);
098: } catch (Exception e) {
099: System.err.println("Failed to load property file '"
100: + propfile + "': " + e);
101: System.exit(3);
102: }
103: }
104:
105: for (int i = 1; i < sa.length; i++) {
106: urlids.add(sa[i]);
107: }
108:
109: boolean[] status = new boolean[urlids.size()];
110:
111: for (int i = 0; i < status.length; i++) {
112: status[i] = false;
113: }
114:
115: String[] withRcArgs = { "--sql", sa[0], "--rcfile", rcFile,
116: null };
117: String[] withoutRcArgs = { "--sql", sa[0], null };
118: String[] sqlToolArgs = (rcFile == null) ? withoutRcArgs
119: : withRcArgs;
120: boolean onefailed = false;
121: long startTime = (new Date()).getTime();
122:
123: while (true) {
124: if (monitorFile != null && !monitorFile.exists()) {
125: System.err.println("Required file is gone: "
126: + monitorFile);
127: System.exit(2);
128: }
129:
130: onefailed = false;
131:
132: for (int i = 0; i < status.length; i++) {
133: if (status[i]) {
134: continue;
135: }
136:
137: sqlToolArgs[sqlToolArgs.length - 1] = (String) urlids
138: .get(i);
139:
140: try {
141: new SqlTool().objectMain(sqlToolArgs);
142:
143: status[i] = true;
144:
145: System.err.println("Success for instance '"
146: + urlids.get(i) + "'");
147: } catch (SqlTool.SqlToolException se) {
148: onefailed = true;
149: }
150: }
151:
152: if (!onefailed) {
153: break;
154: }
155:
156: if (maxtime == 0
157: || (new Date()).getTime() > startTime + maxtime) {
158: break;
159: }
160:
161: try {
162: Thread.sleep(period);
163: } catch (InterruptedException ie) {
164: }
165: }
166:
167: ArrayList failedUrlids = new ArrayList();
168:
169: // If all statuses true, then System.exit(0);
170: for (int i = 0; i < status.length; i++) {
171: if (status[i] != true) {
172: failedUrlids.add(urlids.get(i));
173: }
174: }
175:
176: if (failedUrlids.size() > 0) {
177: System.err.println("Failed instances: " + failedUrlids);
178: System.exit(1);
179: }
180:
181: System.exit(0);
182: }
183:
184: private static void getUrlsFromPropFile(String fileName,
185: ArrayList al) throws Exception {
186:
187: Properties p = new Properties();
188:
189: p.load(new FileInputStream(fileName));
190:
191: int i = -1;
192: String val;
193:
194: while (true) {
195: i++;
196:
197: val = p.getProperty("server.urlid." + i);
198:
199: if (val == null) {
200: return;
201: }
202:
203: al.add(val);
204: }
205: }
206: }
|