Remove unassigned keys when user type changes.

Remove unassigned keys when user type changes. Fixes bastillion-io#118
webiis · Nov 28, 2015 · 04d2ad4 · 04d2ad4
1 parent e543f43
commit 04d2ad4
Showing 1 changed file with 6 additions and 0 deletions.
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);