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
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
|
package org.openecomp.core.zusammen.plugin.dao.impl.cassandra;
import com.amdocs.zusammen.datatypes.Id;
import com.amdocs.zusammen.datatypes.SessionContext;
import com.amdocs.zusammen.plugin.statestore.cassandra.dao.types.ElementEntityContext;
import com.datastax.driver.core.ResultSet;
import com.datastax.driver.core.Row;
import com.datastax.driver.mapping.annotations.Accessor;
import com.datastax.driver.mapping.annotations.Query;
import org.openecomp.core.zusammen.plugin.dao.ElementSynchronizationStateRepository;
import org.openecomp.core.zusammen.plugin.dao.types.SynchronizationStateEntity;
import java.util.Collection;
import java.util.Collections;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;
public class ElementSynchronizationStateRepositoryImpl
implements ElementSynchronizationStateRepository {
@Override
public Collection<SynchronizationStateEntity> list(SessionContext context,
ElementEntityContext elementContext) {
List<Row> rows = getAccessor(context)
.list(elementContext.getSpace(),
elementContext.getItemId().toString(),
elementContext.getVersionId().toString()).all();
return rows == null ? new HashSet<>()
: rows.stream().map(this::getSynchronizationStateEntity).collect(Collectors.toSet());
}
@Override
public void create(SessionContext context, ElementEntityContext elementContext,
SynchronizationStateEntity elementSyncState) {
update(context, elementContext.getSpace(),
elementContext.getItemId(),
elementContext.getVersionId(),
elementContext.getRevisionId(),
elementSyncState.getRevisionId(),
elementSyncState.getId(),
elementSyncState.getPublishTime(),
elementSyncState.isDirty()
);
}
@Override
public void update(SessionContext context, ElementEntityContext elementContext,
SynchronizationStateEntity elementSyncState) {
update(context, elementContext.getSpace(),
elementContext.getItemId(),
elementContext.getVersionId(),
elementContext.getRevisionId(),
elementSyncState.getRevisionId(),
elementSyncState.getId(),
elementSyncState.getPublishTime(),
elementSyncState.isDirty()
);
}
@Override
public void markAsDirty(SessionContext context, ElementEntityContext elementContext,
SynchronizationStateEntity elementSyncState) {
getAccessor(context).updateDirty(true,
elementContext.getSpace(),
elementContext.getItemId().toString(),
elementContext.getVersionId().toString(),
elementSyncState.getId().toString(),
elementContext.getRevisionId().getValue());
getVersionElementsAccessor(context).addDirtyElements(
Collections.singleton(elementSyncState.getId().toString()), elementContext.getSpace(),
elementContext.getItemId().toString(),
elementContext.getVersionId().toString(),
elementContext.getRevisionId().getValue());
}
@Override
public void delete(SessionContext context, ElementEntityContext elementContext,
SynchronizationStateEntity elementSyncState) {
getAccessor(context).delete(elementContext.getSpace(),
elementContext.getItemId().toString(),
elementContext.getVersionId().toString(),
elementSyncState.getId().toString(),
elementContext.getRevisionId().getValue());
getVersionElementsAccessor(context).removeDirtyElements(
Collections.singleton(elementSyncState.getId().toString()), elementContext.getSpace(),
elementContext.getItemId().toString(),
elementContext.getVersionId().toString(),
elementContext.getRevisionId().getValue());
}
@Override
public Optional<SynchronizationStateEntity> get(SessionContext context,
ElementEntityContext elementContext,
SynchronizationStateEntity elementSyncState) {
Row row = getAccessor(context)
.get(elementContext.getSpace(),
elementContext.getItemId().toString(),
elementContext.getVersionId().toString(),
elementSyncState.getId().toString(),
elementSyncState.getRevisionId().getValue()).one();
return row == null ? Optional.empty() : Optional.of(getSynchronizationStateEntity(row));
}
private void update(SessionContext context, String space, Id itemId, Id versionId, Id
versionRevisionId,Id elementRevisionId,Id elementId, Date publishTime, boolean isDirty) {
getAccessor(context).update(publishTime,
isDirty,
space,
itemId.toString(),
versionId.toString(),
elementId.toString(),
elementRevisionId.getValue());
if (isDirty) {
getVersionElementsAccessor(context).addDirtyElements(
Collections.singleton(elementId.toString()), space,
itemId.toString(),
versionId.toString(),
versionRevisionId.getValue());
} else {
getVersionElementsAccessor(context).removeDirtyElements(
Collections.singleton(elementId.toString()), space,
itemId.toString(),
versionId.toString(),
versionRevisionId.getValue());
}
}
private SynchronizationStateEntity getSynchronizationStateEntity(Row row) {
SynchronizationStateEntity entity =
new SynchronizationStateEntity(new Id(row.getString(SynchronizationStateField.ID)),
new Id(row.getString(SynchronizationStateField.REVISION_ID)),
row.getDate(SynchronizationStateField.PUBLISH_TIME),
row.getBool(SynchronizationStateField.DIRTY));
entity.setRevisionId(new Id(row.getString(SynchronizationStateField.REVISION_ID)));
return entity;
}
private ElementSynchronizationStateAccessor getAccessor(SessionContext context) {
return CassandraDaoUtils.getAccessor(context, ElementSynchronizationStateAccessor.class);
}
private VersionElementsAccessor getVersionElementsAccessor(SessionContext context) {
return CassandraDaoUtils.getAccessor(context, VersionElementsAccessor.class);
}
@Accessor
interface ElementSynchronizationStateAccessor {
@Query("UPDATE element_synchronization_state SET publish_time=?, dirty=? " +
"WHERE space=? AND item_id=? AND version_id=? AND element_id=? AND revision_id = ? ")
void update(Date publishTime, boolean dirty, String space, String itemId, String versionId,
String elementId, String revisionId);
@Query("UPDATE element_synchronization_state SET dirty=? " +
"WHERE space=? AND item_id=? AND version_id=? AND element_id=? AND revision_id = ? ")
void updateDirty(boolean dirty, String space, String itemId, String versionId,
String elementId, String revisionId);
@Query("DELETE FROM element_synchronization_state " +
"WHERE space=? AND item_id=? AND version_id=? AND element_id=? AND revision_id = ? ")
void delete(String space, String itemId, String versionId, String elementId, String revisionId);
@Query("SELECT element_id,revision_id, publish_time, dirty FROM element_synchronization_state" +
" WHERE space=? AND item_id=? AND version_id=?")
ResultSet list(String space, String itemId, String versionId);
@Query("SELECT element_id,revision_id, publish_time, dirty FROM element_synchronization_state" +
" WHERE space=? AND item_id=? AND version_id=? AND element_id=? AND revision_id = ? ")
ResultSet get(String space, String itemId, String versionId, String elementId, String
revisionId);
}
private static final class SynchronizationStateField {
private static final String ID = "element_id";
private static final String PUBLISH_TIME = "publish_time";
private static final String DIRTY = "dirty";
private static final String REVISION_ID = "revision_id";
}
@Accessor
interface VersionElementsAccessor {
@Query("UPDATE version_elements SET dirty_element_ids=dirty_element_ids+? " +
"WHERE space=? AND item_id=? AND version_id=? AND revision_id=?")
void addDirtyElements(Set<String> elementIds, String space, String itemId, String versionId,
String revisionId);
@Query("UPDATE version_elements SET dirty_element_ids=dirty_element_ids-? " +
"WHERE space=? AND item_id=? AND version_id=? AND revision_id=? ")
void removeDirtyElements(Set<String> elementIds, String space, String itemId, String
versionId, String revisionId);
}
}
|
