From e8f0cfb4bd051a5e80585164a13d040e5eec3efe Mon Sep 17 00:00:00 2001 From: Athou Date: Mon, 11 Aug 2014 05:33:00 +0200 Subject: [PATCH] better detection of empty database --- src/main/java/com/commafeed/backend/dao/UserDAO.java | 4 ++++ .../java/com/commafeed/backend/service/StartupService.java | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/commafeed/backend/dao/UserDAO.java b/src/main/java/com/commafeed/backend/dao/UserDAO.java index 14bdbf01..96618f36 100644 --- a/src/main/java/com/commafeed/backend/dao/UserDAO.java +++ b/src/main/java/com/commafeed/backend/dao/UserDAO.java @@ -28,4 +28,8 @@ public class UserDAO extends GenericDAO { return newQuery().from(user).where(user.email.equalsIgnoreCase(email)).leftJoin(user.roles, QUserRole.userRole).fetch() .uniqueResult(user); } + + public long count() { + return newQuery().from(user).count(); + } } diff --git a/src/main/java/com/commafeed/backend/service/StartupService.java b/src/main/java/com/commafeed/backend/service/StartupService.java index 56e9d087..4f13a09d 100644 --- a/src/main/java/com/commafeed/backend/service/StartupService.java +++ b/src/main/java/com/commafeed/backend/service/StartupService.java @@ -48,7 +48,7 @@ public class StartupService implements Managed { new UnitOfWork(sessionFactory) { @Override protected Void runInSession() throws Exception { - if (userDAO.findByName(CommaFeedApplication.USERNAME_ADMIN) == null) { + if (userDAO.count() == 0) { initialData(); } return null;