summaryrefslogtreecommitdiff
path: root/src/main/java/com/amazon/carbonado/spi/LobEngineTrigger.java
blob: d2785564128a551abd37b0bcf33728dee7ed93bb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
/*
 * Copyright 2006 Amazon Technologies, Inc. or its affiliates.
 * Amazon, Amazon.com and Carbonado are trademarks or registered trademarks
 * of Amazon Technologies, Inc. or its affiliates.  All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.amazon.carbonado.spi;

import java.util.List;

import com.amazon.carbonado.FetchException;
import com.amazon.carbonado.PersistException;
import com.amazon.carbonado.Storable;
import com.amazon.carbonado.Trigger;

import com.amazon.carbonado.lob.Lob;

/**
 *
 *
 * @author Brian S O'Neill
 * @see LobEngine
 */
class LobEngineTrigger<S extends Storable> extends Trigger<S> {
    final LobEngine mEngine;
    private final int mBlockSize;
    private final LobProperty<Lob>[] mLobProperties;

    LobEngineTrigger(LobEngine engine, Class<S> type, int blockSize,
                     List<LobProperty<?>> lobProperties)
    {
        mEngine = engine;
        mBlockSize = blockSize;

        mLobProperties = new LobProperty[lobProperties.size()];
        lobProperties.toArray(mLobProperties);
    }

    // Returns user specified Lob values
    public Object beforeInsert(S storable) throws PersistException {
        // Capture user lob values for later and replace with new locators.
        int length = mLobProperties.length;
        Object[] userLobs = new Object[length];
        for (int i=0; i<length; i++) {
            LobProperty<Lob> prop = mLobProperties[i];
            Object userLob = storable.getPropertyValue(prop.mName);
            userLobs[i] = userLob;
            if (userLob != null) {
                Object lob = prop.createNewLob(mBlockSize);
                storable.setPropertyValue(prop.mName, lob);
            }
        }
        return userLobs;
    }

    public void afterInsert(S storable, Object state) throws PersistException {
        // Save user lob value contents into new lobs. This is done after the
        // insert of the enclosing record to avoid an expensive rollback if a
        // constraint violation is detected.
        Object[] userLobs = (Object[]) state;
        int length = mLobProperties.length;
        for (int i=0; i<length; i++) {
            Object userLob = userLobs[i];
            if (userLob != null) {
                LobProperty<Lob> prop = mLobProperties[i];
                Lob lob = (Lob) storable.getPropertyValue(prop.mName);
                prop.setLobValue(mEngine.getLocator(lob), (Lob) userLob);
            }
        }
    }

    public void failedInsert(S storable, Object state) {
        unreplaceLobs(storable, state);
    }

    public Object beforeUpdate(S storable) throws PersistException {
        // For each dirty lob property, capture it in case update fails. All
        // lob updates are made in this method.

        int length = mLobProperties.length;
        Object[] userLobs = new Object[length];
        S existing = null;

        for (int i=0; i<length; i++) {
            LobProperty<Lob> prop = mLobProperties[i];
            if (!storable.isPropertyDirty(prop.mName)) {
                continue;
            }

            try {
                if (existing == null && (existing = loadExisting(storable)) == null) {
                    // Update will fail so don't touch lobs.
                    return null;
                }
            } catch (FetchException e) {
                throw e.toPersistException();
            }

            Object userLob = storable.getPropertyValue(prop.mName);
            userLobs[i] = userLob;
            Lob existingLob = (Lob) existing.getPropertyValue(prop.mName);
            if (userLob == null) {
                if (existingLob != null) {
                    // User is setting existing lob to null, so delete it.
                    mEngine.deleteLob(existingLob);
                }
            } else {
                if (existingLob == null) {
                    // User is setting a lob that has no locator yet, so make one.
                    existingLob = prop.createNewLob(mBlockSize);
                }
                prop.setLobValue(mEngine.getLocator(existingLob), (Lob) userLob);
                storable.setPropertyValue(prop.mName, existingLob);
            }
        }

        return userLobs;
    }

    public void failedUpdate(S storable, Object state) {
        unreplaceLobs(storable, state);
    }

    // Returns existing Storable or null
    public Object beforeDelete(S storable) throws PersistException {
        S existing = (S) storable.copy();
        try {
            if (!existing.tryLoad()) {
                existing = null;
            }
            return existing;
        } catch (FetchException e) {
            throw e.toPersistException();
        }
    }

    public void afterDelete(S storable, Object existing) throws PersistException {
        if (existing != null) {
            // After successful delete of master storable, delete all the lobs.
            for (LobProperty<Lob> prop : mLobProperties) {
                Lob lob = (Lob) ((S) existing).getPropertyValue(prop.mName);
                mEngine.deleteLob(lob);
            }
        }
    }

    private S loadExisting(S storable) throws FetchException {
        S existing = (S) storable.copy();
        if (!existing.tryLoad()) {
            return null;
        }
        return existing;
    }

    private void unreplaceLobs(S storable, Object state) {
        if (state != null) {
            Object[] userLobs = (Object[]) state;
            int length = mLobProperties.length;
            for (int i=0; i<length; i++) {
                Object userLob = userLobs[i];
                if (userLob != null) {
                    storable.setPropertyValue(mLobProperties[i].mName, userLob);
                }
            }
        }
    }
}