Update database client according latest database changes. Issue #14

gravidence · Nov 11, 2013 · 828b6e6 · 828b6e6
1 parent 99cf717
commit 828b6e6
Showing 1 changed file with 23 additions and 2 deletions.
diff --git a/src/main/java/org/gravidence/gravifon/db/ScrobblesDBClient.java b/src/main/java/org/gravidence/gravifon/db/ScrobblesDBClient.java
@@ -50,6 +50,21 @@ public class ScrobblesDBClient extends BasicDBClient<ScrobbleDocument> implement
      * JAX-RS client target associated with <code>main/all_scrobbles</code> view.
      */
     private WebTarget viewMainAllScrobblesTarget;
+
+    /**
+     * JAX-RS client target associated with <code>main/all_user_scrobbles</code> view.
+     */
+    private WebTarget viewMainAllUserScrobblesTarget;
+
+    /**
+     * JAX-RS client target associated with <code>main/all_track_scrobbles_amount</code> view.
+     */
+    private WebTarget viewMainAllTrackScrobblesAmountTarget;
+
+    /**
+     * JAX-RS client target associated with <code>main/all_track_scrobbles_duration</code> view.
+     */
+    private WebTarget viewMainAllTrackScrobblesDurationTarget;
 
     @Override
     public void afterPropertiesSet() throws Exception {
@@ -58,6 +73,12 @@ public void afterPropertiesSet() throws Exception {
 
         viewMainAllScrobblesTarget = ViewUtils.getViewTarget(
                 getDbTarget(), "main", "all_scrobbles");
+        viewMainAllUserScrobblesTarget = ViewUtils.getViewTarget(
+                getDbTarget(), "main", "all_user_scrobbles");
+        viewMainAllTrackScrobblesAmountTarget = ViewUtils.getViewTarget(
+                getDbTarget(), "main", "all_track_scrobbles_amount");
+        viewMainAllTrackScrobblesDurationTarget = ViewUtils.getViewTarget(
+                getDbTarget(), "main", "all_track_scrobbles_duration");
     }
 
     /**
@@ -95,7 +116,7 @@ public List<ScrobbleDocument> retrieveScrobblesByKey(String userId, DateTime scr
         key.add(userId);
         key.add(SharedInstanceHolder.OBJECT_MAPPER.valueToTree(DateTimeUtils.dateTimeToArray(scrobbleStartDatetime)));
 
-        return retrieveByKey(viewMainAllScrobblesTarget, key, ScrobbleDocument.class);
+        return retrieveByKey(viewMainAllUserScrobblesTarget, key, ScrobbleDocument.class);
     }
 
     /**
@@ -150,7 +171,7 @@ else if (end != null) {
             subKeyEnd = null;
         }
 
-        return retrievePage(viewMainAllScrobblesTarget, key, subKeyStart, subKeyEnd, ascending, limit,
+        return retrievePage(viewMainAllUserScrobblesTarget, key, subKeyStart, subKeyEnd, ascending, limit,
                 ScrobbleDocument.class);
     }