创建用户时检查用户的有效性

This commit is contained in:
heavyrain2012
2020-09-01 21:58:10 +08:00
parent e5d63a8d5f
commit bae0805e64
3 changed files with 32 additions and 0 deletions
@@ -2186,6 +2186,33 @@ public class DatabaseStore {
}
});
}
boolean isUidAndNameConflict(String uid, String name) {
Connection connection = null;
PreparedStatement statement = null;
ResultSet rs = null;
long start = System.currentTimeMillis();
try {
connection = DBUtil.getConnection();
String sql = "select _uid from t_user where _name = ? and _uid <> ? limit 1";
statement = connection.prepareStatement(sql);
statement.setString(1, name);
statement.setString(2, uid);
rs = statement.executeQuery();
if (rs.next()) {
String conflictId = rs.getString(1);
LOG.error("user {} already have name {} !!!", conflictId, name);
return true;
}
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
Utility.printExecption(LOG, e);
} finally {
DBUtil.closeDB(connection, statement, rs);
}
return false;
}
void updateUser(final WFCMessage.User user) throws Exception {
LOG.info("Database update user info {} {}", user.getUid(), user.getUpdateDt());
@@ -1783,6 +1783,9 @@ public class MemoryMessagesStore implements IMessagesStore {
public void addUserInfo(WFCMessage.User user, String password) throws Exception {
HazelcastInstance hzInstance = m_Server.getHazelcastInstance();
IMap<String, WFCMessage.User> mUserMap = hzInstance.getMap(USERS);
if (databaseStore.isUidAndNameConflict(user.getUid(), user.getName())) {
throw new Exception("用户名不能重复,必须唯一!!!");
}
databaseStore.updateUser(user);
mUserMap.put(user.getUid(), user);
databaseStore.updateUserPassword(user.getUid(), password);