KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > objectstyle > cayenne > access > DataDomainCommitObserver


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.access;
57
58 import java.util.Iterator JavaDoc;
59 import java.util.List JavaDoc;
60
61 import org.apache.log4j.Level;
62 import org.objectstyle.cayenne.CayenneException;
63 import org.objectstyle.cayenne.CayenneRuntimeException;
64 import org.objectstyle.cayenne.DataRow;
65 import org.objectstyle.cayenne.ObjectId;
66 import org.objectstyle.cayenne.map.DbAttribute;
67 import org.objectstyle.cayenne.query.InsertBatchQuery;
68 import org.objectstyle.cayenne.query.Query;
69 import org.objectstyle.cayenne.util.Util;
70
71 /**
72  * Commit operation observer that assigns generated keys to the corresponding objects.
73  *
74  * @since 1.2
75  * @author Andrus Adamchik
76  */
77 class DataDomainCommitObserver implements OperationObserver {
78
79     public void nextBatchCount(Query query, int[] resultCount) {
80         // noop
81 }
82
83     public void nextCount(Query query, int resultCount) {
84         // noop
85 }
86
87     public void nextDataRows(Query query, List JavaDoc dataRows) {
88         // noop
89 }
90
91     public void nextDataRows(Query q, ResultIterator it) {
92         // noop
93 }
94
95     /**
96      * Assigns generated keys.
97      */
98     public void nextGeneratedDataRows(Query query, ResultIterator keysIterator) {
99         // read and close the iterator before doing anything else
100 List JavaDoc keys;
101         try {
102             keys = keysIterator.dataRows(true);
103         }
104         catch (CayenneException ex) {
105             throw new CayenneRuntimeException("Error reading primary key", Util
106                     .unwindException(ex));
107         }
108
109         if (!(query instanceof InsertBatchQuery)) {
110             throw new CayenneRuntimeException(
111                     "Generated keys only supported for InsertBatchQuery, instead got "
112                             + query);
113         }
114
115         InsertBatchQuery batch = (InsertBatchQuery) query;
116
117         ObjectId id = batch.getObjectId();
118         if (id == null || !id.isTemporary()) {
119             // why would this happen?
120 return;
121         }
122
123         if (keys.size() != 1) {
124             throw new CayenneRuntimeException(
125                     "One and only one PK row is expected, instead got " + keys.size());
126         }
127
128         DataRow key = (DataRow) keys.get(0);
129
130         // empty key?
131 if (key.size() == 0) {
132             throw new CayenneRuntimeException("Empty key generated.");
133         }
134
135         // determine DbAttribute name...
136
137         // As of now (01/2005) all tested drivers don't provide decent descriptors of
138 // identity result sets, so a data row will contain garbage labels. Also most
139 // DBs only support one autogenerated key per table... So here we will have to
140 // infer the key name and currently will only support a single column...
141 if (key.size() > 1) {
142             throw new CayenneRuntimeException(
143                     "Only a single column autogenerated PK is supported. "
144                             + "Generated key: "
145                             + key);
146         }
147
148         Iterator JavaDoc it = batch.getDbEntity().getGeneratedAttributes().iterator();
149         while (it.hasNext()) {
150             DbAttribute attribute = (DbAttribute) it.next();
151
152             // batch can have generated attributes that are not PKs, e.g. columns with
153 // DB DEFAULT values. Ignore those.
154 if (attribute.isPrimaryKey()) {
155                 Object JavaDoc value = key.values().iterator().next();
156
157                 // I guess we should override any existing value,
158 // as generated key is the latest thing that exists in the DB.
159 id.getReplacementIdMap().put(attribute.getName(), value);
160                 break;
161             }
162         }
163     }
164
165     public void nextGlobalException(Exception JavaDoc ex) {
166         throw new CayenneRuntimeException("Commit error.", Util.unwindException(ex));
167     }
168
169     public void nextQueryException(Query query, Exception JavaDoc ex) {
170         throw new CayenneRuntimeException("Commit query error.", Util.unwindException(ex));
171     }
172
173     public Level getLoggingLevel() {
174         return Query.DEFAULT_LOG_LEVEL;
175     }
176
177     public boolean isIteratedResult() {
178         return false;
179     }
180 }
181
Popular Tags