Skip to content

Commit 6d01d4f

Browse files
lloydwatkinlloydwatkin
committed
Add additional parameters so we can start back paging
1 parent b7bf8f1 commit 6d01d4f

File tree

10 files changed

+33
-19
lines changed

10 files changed

+33
-19
lines changed

‎src/main/java/org/buddycloud/channelserver/channel/ChannelManagerImpl.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,8 @@ public CloseableIterator<NodeItem> getNodeItems(String nodeId) throws NodeStoreE
125125

126126
@Override
127127
public ClosableIteratorImpl<NodeItem> getNodeItemReplies(String nodeId, String itemId,
128-
String afterItemId, int limit) throws NodeStoreException {
129-
return nodeStore.getNodeItemReplies(nodeId, itemId, afterItemId, limit);
128+
String afterItemId, boolean after, int limit) throws NodeStoreException {
129+
return nodeStore.getNodeItemReplies(nodeId, itemId, afterItemId, after, limit);
130130
}
131131

132132
@Override

‎src/main/java/org/buddycloud/channelserver/db/NodeStore.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ ResultSet<NodeMembershipWithConfiguration> getUserMembershipsWithConfiguration(J
268268
* @return
269269
* @throws NodeStoreException
270270
*/
271-
ClosableIteratorImpl<NodeItem> getNodeItemReplies(String nodeId, String itemId, String afterItemId, int limit) throws NodeStoreException;
271+
ClosableIteratorImpl<NodeItem> getNodeItemReplies(String nodeId, String itemId, String rsmItemId, boolean after, int limit) throws NodeStoreException;
272272

273273
/**
274274
* Get a count of the number of replies to an item

‎src/main/java/org/buddycloud/channelserver/db/jdbc/JDBCNodeStore.java‎

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1167,16 +1167,24 @@ public CloseableIterator<NodeItem> getNodeItems(String nodeId) throws NodeStoreE
11671167
}
11681168

11691169
@Override
1170-
public ClosableIteratorImpl<NodeItem> getNodeItemReplies(String nodeId, String itemId, String afterItemId, int limit) throws NodeStoreException {
1170+
public ClosableIteratorImpl<NodeItem> getNodeItemReplies(String nodeId, String itemId, String rsmItem, boolean after, int limit) throws NodeStoreException {
11711171
PreparedStatement stmt = null;
11721172

11731173
try {
1174-
Date since = new Date(0);
1175-
if (null != afterItemId) {
1176-
since = getNodeItem(nodeId, afterItemId).getUpdated();
1174+
Date since = new Date();
1175+
String afterReplacement = "<";
1176+
if (after) {
1177+
since = new Date(0);
1178+
afterReplacement = ">";
1179+
}
1180+
1181+
if (null != rsmItem) {
1182+
since = getNodeItem(nodeId, rsmItem).getUpdated();
11771183
}
11781184

1179-
String query = dialect.selectItemReplies();
1185+
String query = dialect.selectItemReplies()
1186+
.replace("%beforeAfter%", afterReplacement);
1187+
11801188
if (-1 != limit) {
11811189
query += " LIMIT ?";
11821190
}

‎src/main/java/org/buddycloud/channelserver/db/jdbc/dialect/Sql92NodeStoreDialect.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ public class Sql92NodeStoreDialect implements NodeStoreSQLDialect {
133133
private static final String COUNT_ITEMS_FOR_NODE = "SELECT COUNT(*)" + " FROM \"items\" WHERE \"node\" = ? %parentOnly%;";
134134

135135
private static final String SELECT_ITEM_REPLIES = "" + "SELECT \"id\", \"node\", \"xml\", \"updated\", \"in_reply_to\", \"created\" "
136-
+ "FROM \"items\" WHERE \"node\" = ? AND \"in_reply_to\" LIKE ? " + "AND \"updated\" > ? ORDER BY \"updated\" DESC";
136+
+ "FROM \"items\" WHERE \"node\" = ? AND \"in_reply_to\" LIKE ? " + "AND \"updated\" %beforeAfter% ? ORDER BY \"updated\" DESC";
137137

138138
private static final String SELECT_ITEM_THREAD = "" + "SELECT \"id\", \"node\", \"xml\", \"updated\", \"in_reply_to\", \"created\" "
139139
+ "FROM \"items\" WHERE \"node\" = ? " + "AND (\"in_reply_to\" LIKE ? OR \"id\" = ?) " + "AND \"updated\" > ? ORDER BY \"updated\" DESC";

‎src/main/java/org/buddycloud/channelserver/packetprocessor/iq/namespace/pubsub/PubSubElementProcessorAbstract.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ public abstract class PubSubElementProcessorAbstract implements PubSubElementPro
4747
// RSM details
4848
protected int maxResults = -1;
4949
protected String afterItemId = null;
50+
protected String beforeItemId;
5051

5152
protected Element resultSetManagement;
5253
protected String firstItem;
@@ -267,9 +268,13 @@ protected boolean parseRsmElement() throws NodeStoreException {
267268
maxResults = Integer.parseInt(max.getTextTrim());
268269
}
269270
Element after;
271+
Element before;
270272
if (null != (after = rsmElement.element("after"))) {
271273
afterItemId = after.getTextTrim();
272274
}
275+
if (null != (before = rsmElement.element("before"))) {
276+
beforeItemId = before.getTextTrim();
277+
}
273278

274279
return true;
275280
};

‎src/main/java/org/buddycloud/channelserver/packetprocessor/iq/namespace/pubsub/get/RepliesGet.java‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,8 @@ protected void addRsmElement() throws NodeStoreException {
9999

100100
private void addReplies() throws NodeStoreException {
101101

102-
CloseableIterator<NodeItem> items = channelManager.getNodeItemReplies(node, parentId, afterItemId, maxResults);
102+
boolean after = true;
103+
CloseableIterator<NodeItem> items = channelManager.getNodeItemReplies(node, parentId, afterItemId, after, maxResults);
103104
NodeItem item;
104105
Element entry;
105106
Element itemElement;

‎src/main/java/org/buddycloud/channelserver/packetprocessor/iq/namespace/pubsub/set/ItemDelete.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ private void deleteReplies() throws NodeStoreException {
9595
if (null != nodeItem.getInReplyTo()) {
9696
return;
9797
}
98-
ClosableIteratorImpl<NodeItem> replies = channelManager.getNodeItemReplies(node, itemId.getItemID(), null, -1);
98+
ClosableIteratorImpl<NodeItem> replies = channelManager.getNodeItemReplies(node, itemId.getItemID(), null, true, -1);
9999
NodeItem reply = null;
100100
while (replies.hasNext()) {
101101
reply = replies.next();

‎src/test/java/org/buddycloud/channelserver/db/jdbc/JDBCNodeStoreTest.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -391,7 +391,7 @@ public void canGetItemReplies() throws Exception {
391391
store.addNodeItem(testItem);
392392

393393
ClosableIteratorImpl<NodeItem> items = store.getNodeItemReplies(
394-
TEST_SERVER1_NODE1_ID, "a5", null, -1);
394+
TEST_SERVER1_NODE1_ID, "a5", null, true, -1);
395395

396396
int count = 0;
397397
NodeItem item = null;
@@ -421,7 +421,7 @@ public void canGetItemRepliesWithResultSetManagement() throws Exception {
421421
store.addNodeItem(testItem4);
422422

423423
ClosableIteratorImpl<NodeItem> items = store.getNodeItemReplies(
424-
TEST_SERVER1_NODE1_ID, "a5", "a7", 2);
424+
TEST_SERVER1_NODE1_ID, "a5", "a7", true, 2);
425425

426426
int count = 0;
427427
ArrayList<NodeItem> itemsResult = new ArrayList<NodeItem>();

‎src/test/java/org/buddycloud/channelserver/packetprocessor/iq/namespace/pubsub/get/RepliesGetTest.java‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ public void testIfItemDoesNotExistErrorStanzaIsReturned() throws Exception {
160160
@Test
161161
public void testNoRepliesReturnsEmptyStanza() throws Exception {
162162

163-
Mockito.when(channelManager.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.anyInt())).thenReturn(
163+
Mockito.when(channelManager.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.eq(true), Mockito.anyInt())).thenReturn(
164164
new ClosableIteratorImpl<NodeItem>(new ArrayList<NodeItem>().iterator()));
165165

166166
repliesGet.process(element, jid, request, null);
@@ -178,7 +178,7 @@ public void testOutgoingStanzaFormattedAsExpected() throws Exception {
178178
expectedResults.add(new NodeItemImpl(TEST_NODE, "1", new Date(), "<entry>value1</entry>"));
179179
expectedResults.add(new NodeItemImpl(TEST_NODE, "2", new Date(), "<entry>value2</entry>"));
180180

181-
Mockito.when(channelManager.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.anyInt())).thenReturn(
181+
Mockito.when(channelManager.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.eq(true), Mockito.anyInt())).thenReturn(
182182
new ClosableIteratorImpl<NodeItem>(expectedResults.iterator()));
183183

184184
repliesGet.process(element, jid, request, null);
@@ -198,7 +198,7 @@ public void testUnparsableItemEntriesAreSimplyIgnored() throws Exception {
198198
expectedResults.add(new NodeItemImpl(TEST_NODE, "1", new Date(), "<entry>value1</entry>"));
199199
expectedResults.add(new NodeItemImpl(TEST_NODE, "2", new Date(), "<entry>value2"));
200200

201-
Mockito.when(channelManager.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.anyInt())).thenReturn(
201+
Mockito.when(channelManager.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.eq(true), Mockito.anyInt())).thenReturn(
202202
new ClosableIteratorImpl<NodeItem>(expectedResults.iterator()));
203203

204204
repliesGet.process(element, jid, request, null);
@@ -225,7 +225,7 @@ public void testRsmElementIsAddedCorrectly() throws Exception {
225225
expectedResults.add(new NodeItemImpl(TEST_NODE, "3", new Date(), "<entry>value3</entry>"));
226226
expectedResults.add(new NodeItemImpl(TEST_NODE, "4", new Date(), "<entry>value4</entry>"));
227227

228-
Mockito.when(channelManager.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(), Mockito.eq("1"), Mockito.eq(4))).thenReturn(
228+
Mockito.when(channelManager.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(), Mockito.eq("1"), Mockito.eq(true), Mockito.eq(4))).thenReturn(
229229
new ClosableIteratorImpl<NodeItem>(expectedResults.iterator()));
230230

231231
repliesGet.process(element, jid, request, null);

‎src/test/java/org/buddycloud/channelserver/packetprocessor/iq/namespace/pubsub/set/ItemDeleteTest.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -396,7 +396,7 @@ public void doesNotRequestThreadWhenDealingWithReply() throws Exception {
396396
Mockito.doThrow(Exception.class)
397397
.when(channelManager)
398398
.getNodeItemReplies(Mockito.anyString(), Mockito.anyString(),
399-
Mockito.anyString(), Mockito.anyInt());
399+
Mockito.anyString(), Mockito.anyBoolean(), Mockito.anyInt());
400400

401401
Mockito.doReturn(new ResultSetImpl<NodeSubscription>(subscriptions))
402402
.when(channelManager).getNodeSubscriptionListeners(node);
@@ -431,7 +431,7 @@ public void requestsThreadWhenDeletingParentPost() throws Exception {
431431
Mockito.when(channelManager.getNodeItem(node, "item-id")).thenReturn(
432432
nodeItem);
433433
Mockito.when(channelManager.getNodeItemReplies(Mockito.eq(node), Mockito.eq("item-id"),
434-
Mockito.anyString(), Mockito.eq(-1))).thenReturn(new ClosableIteratorImpl<NodeItem>(replies
434+
Mockito.anyString(), Mockito.anyBoolean(), Mockito.eq(-1))).thenReturn(new ClosableIteratorImpl<NodeItem>(replies
435435
.iterator()));
436436

437437
Mockito.doReturn(new ResultSetImpl<NodeSubscription>(subscriptions))

0 commit comments

Comments
 (0)