From 347589a78ba6db05bb4fe62ad92a0b73e4a57c3e Mon Sep 17 00:00:00 2001
From: Jesse Morgan <jesse@jesterpm.net>
Date: Fri, 23 May 2014 22:40:39 -0700
Subject: Adding dynamo support.

---
 src/com/p4square/grow/frontend/SurveyPageResource.java | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'src/com/p4square/grow/frontend/SurveyPageResource.java')

diff --git a/src/com/p4square/grow/frontend/SurveyPageResource.java b/src/com/p4square/grow/frontend/SurveyPageResource.java
index 313fb7b..cfc4eed 100644
--- a/src/com/p4square/grow/frontend/SurveyPageResource.java
+++ b/src/com/p4square/grow/frontend/SurveyPageResource.java
@@ -249,9 +249,12 @@ public class SurveyPageResource extends FreeMarkerPageResource {
         if (nextQuestionId == null) {
             // Just finished the last question. Update the user's account
             try {
-                UserRecord account = mUserRecordProvider.get(mUserId);
-                if (account == null) {
-                    account = new UserRecord();
+                UserRecord account = null;
+                try {
+                    account = mUserRecordProvider.get(mUserId);
+                } catch (NotFoundException e) {
+                    // User record doesn't exist, so create a new one.
+                    account = new UserRecord(getRequest().getClientInfo().getUser());
                 }
                 account.setLanding("training");
                 mUserRecordProvider.put(mUserId, account);
-- 
cgit v1.2.3