KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > objectstyle > cayenne > dba > sqlserver > SQLServerBatchAction


1 /* ====================================================================
2  *
3  * The ObjectStyle Group Software License, version 1.1
4  * ObjectStyle Group - http://objectstyle.org/
5  *
6  * Copyright (c) 2002-2005, Andrei (Andrus) Adamchik and individual authors
7  * of the software. All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  *
13  * 1. Redistributions of source code must retain the above copyright
14  * notice, this list of conditions and the following disclaimer.
15  *
16  * 2. Redistributions in binary form must reproduce the above copyright
17  * notice, this list of conditions and the following disclaimer in
18  * the documentation and/or other materials provided with the
19  * distribution.
20  *
21  * 3. The end-user documentation included with the redistribution, if any,
22  * must include the following acknowlegement:
23  * "This product includes software developed by independent contributors
24  * and hosted on ObjectStyle Group web site (http://objectstyle.org/)."
25  * Alternately, this acknowlegement may appear in the software itself,
26  * if and wherever such third-party acknowlegements normally appear.
27  *
28  * 4. The names "ObjectStyle Group" and "Cayenne" must not be used to endorse
29  * or promote products derived from this software without prior written
30  * permission. For written permission, email
31  * "andrus at objectstyle dot org".
32  *
33  * 5. Products derived from this software may not be called "ObjectStyle"
34  * or "Cayenne", nor may "ObjectStyle" or "Cayenne" appear in their
35  * names without prior written permission.
36  *
37  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
38  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
39  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
40  * DISCLAIMED. IN NO EVENT SHALL THE OBJECTSTYLE GROUP OR
41  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
42  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
43  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
44  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
45  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
46  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
47  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
48  * SUCH DAMAGE.
49  * ====================================================================
50  *
51  * This software consists of voluntary contributions made by many
52  * individuals and hosted on ObjectStyle Group web site. For more
53  * information on the ObjectStyle Group, please see
54  * <http://objectstyle.org/>.
55  */
56 package org.objectstyle.cayenne.dba.sqlserver;
57
58 import java.sql.Connection JavaDoc;
59 import java.sql.SQLException JavaDoc;
60 import java.sql.Statement JavaDoc;
61 import java.util.Collections JavaDoc;
62 import java.util.Iterator JavaDoc;
63
64 import org.objectstyle.cayenne.access.OperationObserver;
65 import org.objectstyle.cayenne.access.QueryLogger;
66 import org.objectstyle.cayenne.access.jdbc.BatchAction;
67 import org.objectstyle.cayenne.dba.DbAdapter;
68 import org.objectstyle.cayenne.map.DbAttribute;
69 import org.objectstyle.cayenne.map.EntityResolver;
70 import org.objectstyle.cayenne.query.BatchQuery;
71 import org.objectstyle.cayenne.query.InsertBatchQuery;
72
73 /**
74  * @since 1.2
75  * @author Andrei Adamchik
76  */
77 public class SQLServerBatchAction extends BatchAction {
78
79     public SQLServerBatchAction(BatchQuery batchQuery, DbAdapter adapter,
80             EntityResolver entityResolver) {
81         super(batchQuery, adapter, entityResolver);
82     }
83
84     public void performAction(Connection JavaDoc connection, OperationObserver observer)
85             throws SQLException JavaDoc, Exception JavaDoc {
86
87         // this condition checks if identity columns are present in the query and adapter
88 // is not ready to process them... e.g. if we are using a MS driver...
89 boolean identityOverride = expectsToOverrideIdentityColumns();
90         if (identityOverride) {
91             setIdentityInsert(connection, true);
92         }
93
94         try {
95             super.performAction(connection, observer);
96         }
97         finally {
98
99             // important: turn off IDENTITY_INSERT as SQL Server won't be able to process
100 // other identity columns in the same transaction
101
102             // TODO: if an error happens here this would mask the parent error
103 if (identityOverride) {
104                 setIdentityInsert(connection, false);
105             }
106         }
107     }
108     
109     protected void setIdentityInsert(Connection JavaDoc connection, boolean on)
110             throws SQLException JavaDoc {
111
112         String JavaDoc flag = on ? " ON" : " OFF";
113         String JavaDoc configSQL = "SET IDENTITY_INSERT "
114                 + query.getDbEntity().getFullyQualifiedName()
115                 + flag;
116
117         QueryLogger.logQuery(query.getLoggingLevel(), configSQL, Collections.EMPTY_LIST);
118
119         Statement JavaDoc statement = connection.createStatement();
120         try {
121             statement.execute(configSQL);
122         }
123         finally {
124             try {
125                 statement.close();
126             }
127             catch (Exception JavaDoc e) {
128             }
129         }
130     }
131  
132     /**
133      * Returns whether a table has identity columns.
134      */
135     protected boolean expectsToOverrideIdentityColumns() {
136         // jTDS driver supports identity columns, no need for tricks...
137 if (getAdapter().supportsGeneratedKeys()) {
138             return false;
139         }
140
141         if (!(query instanceof InsertBatchQuery) || query.getDbEntity() == null) {
142             return false;
143         }
144
145         // find identity attributes
146 Iterator JavaDoc it = query.getDbEntity().getAttributes().iterator();
147         while (it.hasNext()) {
148             DbAttribute attribute = (DbAttribute) it.next();
149             if (attribute.isGenerated()) {
150                 return true;
151             }
152         }
153
154         return false;
155     }
156 }
157
Popular Tags