01: /*
02: * <copyright>
03: *
04: * Copyright 1997-2004 BBNT Solutions, LLC
05: * under sponsorship of the Defense Advanced Research Projects
06: * Agency (DARPA).
07: *
08: * You can redistribute this software and/or modify it under the
09: * terms of the Cougaar Open Source License as published on the
10: * Cougaar Open Source Website (www.cougaar.org).
11: *
12: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
13: * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
14: * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
15: * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
16: * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
17: * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
18: * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19: * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20: * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
22: * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23: *
24: * </copyright>
25: */
26:
27: package org.cougaar.mlm.plugin.ldm;
28:
29: import org.cougaar.glm.ldm.oplan.Oplan;
30:
31: /** Reads oplan info from a database table. Assumes it's being invoked on
32: * behalf of SQLOplanPlugin. Updates oplan maintained by SQLOplanPlugin.
33: */
34:
35: public class ActiveStagesQueryHandler extends SQLOplanQueryHandler {
36: private static final String QUERY_NAME = "ActiveStagesQuery";
37:
38: private int minRequiredStage = 0;
39:
40: /** this method is called before a query is started,
41: * before even getQuery.
42: **/
43: public void startQuery() {
44: minRequiredStage = 0; // In case it's not specified
45: }
46:
47: /** Construct and return an SQL query to be used by the Database engine.
48: * Subclasses are required to implement this method.
49: **/
50: public String getQuery() {
51: return (String) getParameter(QUERY_NAME);
52: }
53:
54: /** Process a single row in a result set,
55: * doing whatever is required.
56: **/
57: public void processRow(Object[] rowData) {
58: if (rowData.length != 1) {
59: System.err
60: .println("ActiveStagesQueryHandler.processRow()- expected 1 columns of data, "
61: + " got " + rowData.length);
62: }
63: try {
64: Number n = (Number) rowData[0];
65: minRequiredStage = n == null ? 0 : n.intValue();
66: } catch (Exception usee) {
67: System.err
68: .println("Caught exception while executing a query: "
69: + usee);
70: usee.printStackTrace();
71: }
72: }
73:
74: /** this method is called when a query is complete,
75: * afer the last call to processRow.
76: **/
77: public void endQuery() {
78: String oplanID = getParameter(OplanReaderPlugin.OPLAN_ID_PARAMETER);
79: //should already have this oplan
80: Oplan oplan = (Oplan) myPlugin.getOplan(oplanID).clone();
81: oplan.setMinRequiredStage(minRequiredStage);
82: myPlugin.updateOplanInfo(oplan);
83: }
84: }
|