Remove unassigned keys when user type changes.

skavanagh · skavanagh · commit 04d2ad43d8d6 · 2015-11-28T09:15:08.000-05:00
Remove unassigned keys when user type changes. Fixes bastillion-io#118
diff --git a/src/main/java/com/keybox/manage/db/UserDB.java b/src/main/java/com/keybox/manage/db/UserDB.java
@@ -277,6 +277,9 @@ public static void updateUserNoCredentials(User user) {
             stmt.setLong(6, user.getId());
             stmt.execute();
             DBUtils.closeStmt(stmt);
+            if (User.ADMINISTRATOR.equals(user.getUserType())) {
+                PublicKeyDB.deleteUnassignedKeysByUser(con, user.getId());
+            }
 
         } catch (Exception e) {
             log.error(e.toString(), e);
@@ -307,6 +310,9 @@ public static void updateUserCredentials(User user) {
             stmt.setLong(8, user.getId());
             stmt.execute();
             DBUtils.closeStmt(stmt);
+            if(User.ADMINISTRATOR.equals(user.getUserType())) {
+                PublicKeyDB.deleteUnassignedKeysByUser(con, user.getId());
+            }
 
         } catch (Exception e) {
             log.error(e.toString(), e);
