/* * This file is part of the Wildfire Chat package. * (c) Heavyrain2012 * * For the full copyright and license information, please view the LICENSE * file that was distributed with this source code. */ package io.moquette.persistence; import cn.wildfirechat.pojos.SystemSettingPojo; import cn.wildfirechat.proto.ProtoConstants; import cn.wildfirechat.proto.WFCMessage; import cn.wildfirechat.server.ThreadPoolExecutorWrapper; import com.hazelcast.core.HazelcastInstance; import com.hazelcast.core.MultiMap; import com.hazelcast.util.StringUtil; import com.xiaoleilu.loServer.model.FriendData; import io.moquette.server.Server; import io.moquette.spi.ClientSession; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import win.liyufan.im.DBUtil; import win.liyufan.im.MessageBundle; import win.liyufan.im.MessageShardingUtil; import win.liyufan.im.Utility; import java.io.IOException; import java.security.MessageDigest; import java.sql.*; import java.util.*; import java.util.function.Function; import static cn.wildfirechat.proto.ProtoConstants.PersistFlag.Transparent; import static io.moquette.persistence.MemoryMessagesStore.USER_STATUS; import static io.moquette.server.Constants.MAX_MESSAGE_QUEUE; public class DatabaseStore { private static final Logger LOG = LoggerFactory.getLogger(DatabaseStore.class); private final ThreadPoolExecutorWrapper mScheduler; public DatabaseStore(ThreadPoolExecutorWrapper scheduler) { this.mScheduler = scheduler; } TreeMap reloadUserMessageMaps(String userId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; TreeMap out = new TreeMap<>(); try { connection = DBUtil.getConnection(); String sql = "select `_seq`, `_mid` from " + getUserMessageTable(userId) + " where `_uid` = ? order by `_seq` DESC limit " + MAX_MESSAGE_QUEUE; statement = connection.prepareStatement(sql); statement.setString(1, userId); rs = statement.executeQuery(); while (rs.next()) { int index = 1; long msgSeq = rs.getLong(index++); long msgId = rs.getLong(index); out.put(msgSeq, msgId); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return out; } boolean updateSystemSetting(int id, String value, String desc) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_settings " + " (`id`, `value`, `desc`) values(?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`value` = ?," + "`desc` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setLong(index++, id); statement.setString(index++, value); statement.setString(index++, desc); statement.setString(index++, value); statement.setString(index++, desc); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); return true; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } return false; } SystemSettingPojo getSystemSetting(int id) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `value`, `desc` from t_settings where `id` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setLong(index++, id); rs = statement.executeQuery(); while (rs.next()) { SystemSettingPojo out = new SystemSettingPojo(); index = 1; out.id = id; String value = rs.getString(index++); value = (value == null ? "" : value); out.value = value; value = rs.getString(index++); value = (value == null ? "" : value); out.desc = value; return out; } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } List searchUserFromDB(String keyword, boolean buzzy, int page) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; ArrayList out = new ArrayList<>(); try { connection = DBUtil.getConnection(); String sql = "select `_uid`, `_name`" + ", `_display_name`" + ", `_portrait`" + ", `_mobile`" + ", `_gender`" + ", `_email`" + ", `_address`" + ", `_company`" + ", `_social`" + ", `_extra`" + ", `_dt` from t_user"; if (buzzy) { sql += " where (`_display_name` like ? or `_name` = ? or `_mobile` = ?) "; } else { sql += " where (`_name` = ? or `_mobile` = ?) "; } sql += " and _type <> 2"; //can search normal user(0) and robot(1), can not search things if (buzzy) { sql += " limit 20"; } else { sql += " limit 1"; } if (page > 0) { sql += "offset = '" + page * 20 + "'"; } statement = connection.prepareStatement(sql); int index = 1; if (buzzy) { statement.setString(index++, "%" + keyword + "%"); } statement.setString(index++, keyword); statement.setString(index++, keyword); rs = statement.executeQuery(); while (rs.next()) { WFCMessage.User.Builder builder = WFCMessage.User.newBuilder(); index = 1; String value = rs.getString(index++); value = (value == null ? "" : value); builder.setUid(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setName(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setDisplayName(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setPortrait(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setMobile(value); int gender = rs.getInt(index++); builder.setGender(gender); value = rs.getString(index++); value = (value == null ? "" : value); builder.setEmail(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setAddress(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setCompany(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setSocial(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setExtra(value); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); WFCMessage.User user = builder.build(); out.add(user); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return out; } Integer getUserStatus(String userId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_status` from t_user_status where `_uid` = ?"; statement = connection.prepareStatement(sql); statement.setString(1, userId); rs = statement.executeQuery(); if (rs.next()) { return rs.getInt(1); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return 0; } synchronized Collection reloadGroupMemberFromDB(HazelcastInstance hzInstance, String groupId) { MultiMap groupMembers = hzInstance.getMultiMap(MemoryMessagesStore.GROUP_MEMBERS); if (groupMembers.get(groupId).size() > 0) { return groupMembers.get(groupId); } Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_mid`" + ", `_alias`" + ", `_type`" + ", `_dt` from t_group_member where _gid = ?"; statement = connection.prepareStatement(sql); statement.setString(1, groupId); int index; rs = statement.executeQuery(); while (rs.next()) { WFCMessage.GroupMember.Builder builder = WFCMessage.GroupMember.newBuilder(); index = 1; String value = rs.getString(index++); value = (value == null ? "" : value); builder.setMemberId(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setAlias(value); int intvalue = rs.getInt(index++); builder.setType(intvalue); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); WFCMessage.GroupMember member = builder.build(); groupMembers.put(groupId, member); } return groupMembers.get(groupId); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return new ArrayList<>(); } void reloadFriendsFromDB(HazelcastInstance hzInstance) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; MultiMap friendsMap = hzInstance.getMultiMap(MemoryMessagesStore.USER_FRIENDS); if (friendsMap.size() > 0) { return; } try { connection = DBUtil.getConnection(); String sql = "select `_uid`, `_friend_uid`, `_alias`, `_state`, `_blacked`, `_dt` from t_friend"; statement = connection.prepareStatement(sql); int index; rs = statement.executeQuery(); while (rs.next()) { FriendData builder = new FriendData(); index = 1; String value = rs.getString(index++); value = (value == null ? "" : value); builder.setUserId(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setFriendUid(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setAlias(value); int intvalue = rs.getInt(index++); builder.setState(intvalue); intvalue = rs.getInt(index++); builder.setBlacked(intvalue); long longvalue = rs.getLong(index++); builder.setTimestamp(longvalue); friendsMap.put(builder.getUserId(), builder); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } } void reloadFriendRequestsFromDB(HazelcastInstance hzInstance) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; MultiMap requestMap = hzInstance.getMultiMap(MemoryMessagesStore.USER_FRIENDS_REQUEST); if (requestMap.size() > 0) { return; } try { connection = DBUtil.getConnection(); String sql = "select `_uid`, `_friend_uid`, `_reason`, `_status`, `_dt`, `_from_read_status`, `_to_read_status` from t_friend_request"; statement = connection.prepareStatement(sql); int index; rs = statement.executeQuery(); while (rs.next()) { WFCMessage.FriendRequest.Builder builder = WFCMessage.FriendRequest.newBuilder(); index = 1; String value = rs.getString(index++); value = (value == null ? "" : value); builder.setFromUid(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setToUid(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setReason(value); int intvalue = rs.getInt(index++); builder.setStatus(intvalue); long longvalue = rs.getLong(index++); builder.setUpdateDt(longvalue); intvalue = rs.getInt(index++); builder.setFromReadStatus(intvalue > 0); intvalue = rs.getInt(index++); builder.setToReadStatus(intvalue > 0); WFCMessage.FriendRequest request = builder.build(); requestMap.put(request.getFromUid(), request); requestMap.put(request.getToUid(), request); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } } void persistMessage(final WFCMessage.Message message, boolean update) { if(message.getContent().getPersistFlag() == Transparent) { return; } mScheduler.execute(()-> { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into " + MessageShardingUtil.getMessageTable(message.getMessageId()) + " (`_mid`, `_from`, `_type`, `_target`, `_line`, `_data`, `_searchable_key`, `_dt`, `_content_type`) values(?, ?, ?, ?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_data` = ?," + "`_searchable_key` = ?," + "`_dt` = ?," + "`_content_type` = ?"; String searchableContent = message.getContent().getSearchableContent() == null ? "" : message.getContent().getSearchableContent(); statement = connection.prepareStatement(sql); int index = 1; statement.setLong(index++, message.getMessageId()); statement.setString(index++, message.getFromUser()); statement.setInt(index++, message.getConversation().getType()); statement.setString(index++, message.getConversation().getTarget()); statement.setInt(index++, message.getConversation().getLine()); Blob blob = connection.createBlob(); blob.setBytes(1, message.getContent().toByteArray()); statement.setBlob(index++, blob); statement.setString(index++, searchableContent); statement.setTimestamp(index++, new Timestamp(message.getServerTimestamp())); statement.setInt(index++, message.getContent().getType()); statement.setBlob(index++, blob); statement.setString(index++, searchableContent); statement.setTimestamp(index++, new Timestamp(message.getServerTimestamp())); statement.setInt(index++, message.getContent().getType()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void persistSensitiveMessage(final WFCMessage.Message message) { if(message.getContent().getPersistFlag() == Transparent) { return; } mScheduler.execute(()-> { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into " + "t_sensitive_messages" + " (`_mid`, `_from`, `_type`, `_target`, `_line`, `_data`, `_searchable_key`, `_dt`, `_content_type`) values(?, ?, ?, ?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_data` = ?," + "`_searchable_key` = ?," + "`_dt` = ?," + "`_content_type` = ?"; String searchableContent = message.getContent().getSearchableContent() == null ? "" : message.getContent().getSearchableContent(); statement = connection.prepareStatement(sql); int index = 1; statement.setLong(index++, message.getMessageId()); statement.setString(index++, message.getFromUser()); statement.setInt(index++, message.getConversation().getType()); statement.setString(index++, message.getConversation().getTarget()); statement.setInt(index++, message.getConversation().getLine()); Blob blob = connection.createBlob(); blob.setBytes(1, message.getContent().toByteArray()); statement.setBlob(index++, blob); statement.setString(index++, searchableContent); statement.setTimestamp(index++, new Timestamp(message.getServerTimestamp())); statement.setInt(index++, message.getContent().getType()); statement.setBlob(index++, blob); statement.setString(index++, searchableContent); statement.setTimestamp(index++, new Timestamp(message.getServerTimestamp())); statement.setInt(index++, message.getContent().getType()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } Map getMessages(Collection keys) { Map> messageTableMap = new HashMap<>(); for (Long key : keys) { String messageTableId = MessageShardingUtil.getMessageTable(key); messageTableMap.computeIfAbsent(messageTableId, new Function>() { @Override public List apply(String s) { return new ArrayList<>(); } }); messageTableMap.get(messageTableId).add(key); } Map out = null; Connection connection = null; try { connection = DBUtil.getConnection(); out = new HashMap<>(); for (Map.Entry> entry : messageTableMap.entrySet()) { String sql = "select `_mid`, `_from`, `_type`, `_target`, `_line`, `_data`, `_dt` from " + entry.getKey() +" where _mid in ("; for (int i = 0; i < entry.getValue().size(); i++) { sql += entry.getValue().get(i); if (i != entry.getValue().size() - 1) { sql += ","; } } sql += ")"; ResultSet resultSet = null; try { PreparedStatement statement = connection.prepareStatement(sql); resultSet = statement.executeQuery(); while (resultSet.next()) { WFCMessage.Message.Builder builder = WFCMessage.Message.newBuilder(); int index = 1; builder.setMessageId(resultSet.getLong(index++)); builder.setFromUser(resultSet.getString(index++)); WFCMessage.Conversation.Builder cb = WFCMessage.Conversation.newBuilder(); cb.setType(resultSet.getInt(index++)); cb.setTarget(resultSet.getString(index++)); cb.setLine(resultSet.getInt(index++)); builder.setConversation(cb.build()); Blob blob = resultSet.getBlob(index++); WFCMessage.MessageContent messageContent = WFCMessage.MessageContent.parseFrom(blob.getBinaryStream()); builder.setContent(messageContent); builder.setServerTimestamp(resultSet.getTimestamp(index++).getTime()); WFCMessage.Message message = builder.build(); out.put(message.getMessageId(),new MessageBundle(message.getMessageId(), message.getFromUser(), null, message)); } } catch (SQLException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } catch (IOException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } finally { try { if (resultSet!=null) { resultSet.close(); } } catch (SQLException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } } } } catch (SQLException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, null); } return out; } MessageBundle getMessage(long messageId) { String sql = "select `_from`, `_type`, `_target`, `_line`, `_data`, `_dt` from " + MessageShardingUtil.getMessageTable(messageId) +" where _mid = ? limit 1"; Connection connection = null; PreparedStatement statement = null; ResultSet resultSet = null; try { connection = DBUtil.getConnection(); statement = connection.prepareStatement(sql); statement.setLong(1, messageId); resultSet = statement.executeQuery(); if (resultSet.next()) { WFCMessage.Message.Builder builder = WFCMessage.Message.newBuilder(); builder.setMessageId(messageId); int index = 1; builder.setFromUser(resultSet.getString(index++)); WFCMessage.Conversation.Builder cb = WFCMessage.Conversation.newBuilder(); cb.setType(resultSet.getInt(index++)); cb.setTarget(resultSet.getString(index++)); cb.setLine(resultSet.getInt(index++)); builder.setConversation(cb.build()); Blob blob = resultSet.getBlob(index++); WFCMessage.MessageContent messageContent = WFCMessage.MessageContent.parseFrom(blob.getBinaryStream()); builder.setContent(messageContent); builder.setServerTimestamp(resultSet.getTimestamp(index++).getTime()); WFCMessage.Message message = builder.build(); return new MessageBundle(messageId, message.getFromUser(), null, message); } } catch (SQLException | IOException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, resultSet); } return null; } void deleteMessage(long messageId) { String sql = "delete from " + MessageShardingUtil.getMessageTable(messageId) + " where _mid = ?"; Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); statement = connection.prepareStatement(sql); statement.setLong(1, messageId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } List loadRemoteMessages(String user, WFCMessage.Conversation conversation, long beforeUid, int count) { List messages = loadRemoteMessagesFromTable(user, conversation, beforeUid, count, MessageShardingUtil.getMessageTable(beforeUid)); if (messages != null && messages.size() < count) { String nexTable = MessageShardingUtil.getPreviousMessageTable(beforeUid); if (!StringUtil.isNullOrEmpty(nexTable)) { List nextMessages = loadRemoteMessagesFromTable(user, conversation, beforeUid, count - messages.size(), nexTable); if (nextMessages != null) { messages.addAll(nextMessages); } } } return messages; } List loadRemoteMessagesFromTable(String user, WFCMessage.Conversation conversation, long beforeUid, int count, String table) { String sql = "select `_mid`, `_from`, `_type`, `_target`, `_line`, `_data`, `_dt` from " + table +" where"; if (conversation.getType() == ProtoConstants.ConversationType.ConversationType_Private) { sql += " _type = ? and _line = ? and _mid < ? and ((_target = ? and _from = ?) or (_target = ? and _from = ?))"; } else { sql += " _type = ? and _line = ? and _mid < ? and _target = ?"; } sql += "order by `_mid` DESC limit ?"; Connection connection = null; PreparedStatement statement = null; ResultSet resultSet = null; List out = new ArrayList<>(); try { connection = DBUtil.getConnection(); statement = connection.prepareStatement(sql); int index = 1; statement.setInt(index++, conversation.getType()); statement.setInt(index++, conversation.getLine()); statement.setLong(index++, beforeUid); statement.setString(index++, conversation.getTarget()); if (conversation.getType() == ProtoConstants.ConversationType.ConversationType_Private) { statement.setString(index++, user); statement.setString(index++, user); statement.setString(index++, conversation.getTarget()); } statement.setInt(index++, count); resultSet = statement.executeQuery(); while (resultSet.next()) { WFCMessage.Message.Builder builder = WFCMessage.Message.newBuilder(); index = 1; builder.setMessageId(resultSet.getLong(index++)); builder.setFromUser(resultSet.getString(index++)); WFCMessage.Conversation.Builder cb = WFCMessage.Conversation.newBuilder(); cb.setType(resultSet.getInt(index++)); cb.setTarget(resultSet.getString(index++)); cb.setLine(resultSet.getInt(index++)); builder.setConversation(cb.build()); Blob blob = resultSet.getBlob(index++); WFCMessage.MessageContent messageContent = WFCMessage.MessageContent.parseFrom(blob.getBinaryStream()); builder.setContent(messageContent); builder.setServerTimestamp(resultSet.getTimestamp(index++).getTime()); WFCMessage.Message message = builder.build(); out.add(message); } } catch (SQLException | IOException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, resultSet); } return out; } void persistUserMessage(final String userId, final long messageId, final long messageSeq) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into " + getUserMessageTable(userId) + " (`_mid`, `_uid`, `_seq`) values(?, ?, ?)"; statement = connection.prepareStatement(sql); int index = 1; statement.setLong(index++, messageId); statement.setString(index++, userId); statement.setLong(index++, messageSeq); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void removeFavGroup(final String groupId, final List memberIds) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); StringBuilder sb = new StringBuilder("update t_user_setting set _value = ?, _dt = ? where _uid in ("); for (int i = 0; i < memberIds.size(); i++) { sb.append("?"); if (i != memberIds.size() - 1) { sb.append(","); } } sb.append(")"); sb.append(" and _scope = 6 and _key = ?"); statement = connection.prepareStatement(sb.toString()); int index = 1; statement.setString(index++, "0"); statement.setLong(index++, System.currentTimeMillis()); for (int i = 0; i < memberIds.size(); i++) { statement.setString(index++, memberIds.get(i)); } statement.setString(index++, groupId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } void persistUserSetting(final String userId, WFCMessage.UserSettingEntry entry) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_user_setting (`_uid`" + ", `_scope`" + ", `_key`" + ", `_value`" + ", `_dt`) values(?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_value` = ?," + "`_dt` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, userId); statement.setInt(index++, entry.getScope()); statement.setString(index++, entry.getKey()); statement.setString(index++, entry.getValue()); statement.setLong(index++, entry.getUpdateDt()); statement.setString(index++, entry.getValue()); statement.setLong(index++, entry.getUpdateDt()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } List getPersistUserSetting(final String userId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select " + "`_scope`" + ", `_key`" + ", `_value`" + ", `_dt`" + " from t_user_setting where `_uid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, userId); rs = statement.executeQuery(); List out = new ArrayList<>(); while (rs.next()) { WFCMessage.UserSettingEntry.Builder builder = WFCMessage.UserSettingEntry.newBuilder(); index = 1; int intvalue = rs.getInt(index++); builder.setScope(intvalue); String value = rs.getString(index++); value = (value == null ? "" : value); builder.setKey(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setValue(value); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); out.add(builder.build()); } return out; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } void removeGroupUserSettings(String groupId, List users) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); StringBuilder sb = new StringBuilder("delete from t_user_setting where _scope in (1,3,5,7) and _uid in ("); for (int i = 0; i < users.size(); i++) { sb.append("?"); if (i != users.size() - 1) { sb.append(","); } } sb.append(") and _key like '1-_-"); sb.append(groupId); sb.append("'"); statement = connection.prepareStatement(sb.toString()); int index = 1; for (String userId:users) { statement.setString(index++, userId); } int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } void persistGroupInfo(final WFCMessage.GroupInfo groupInfo) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_group (`_gid`" + ", `_name`" + ", `_portrait`" + ", `_owner`" + ", `_type`" + ", `_extra`" + ", `_dt`" + ", `_member_count`" + ", `_mute`" + ", `_join_type`" + ", `_private_chat`" + ", `_searchable`" + ", `_member_dt`) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_name` = ?," + "`_portrait` = ?," + "`_owner` = ?," + "`_type` = ?," + "`_extra` = ?," + "`_dt` = ?," + "`_member_count` = ?," + "`_mute` = ?" + ", `_join_type` = ?" + ", `_private_chat` = ?" + ", `_searchable` = ?, " + "`_member_dt` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, groupInfo.getTargetId()); statement.setString(index++, groupInfo.getName()); statement.setString(index++, groupInfo.getPortrait()); statement.setString(index++, groupInfo.getOwner()); statement.setInt(index++, groupInfo.getType()); statement.setString(index++, groupInfo.getExtra()); statement.setLong(index++, groupInfo.getUpdateDt() == 0 ? System.currentTimeMillis() : groupInfo.getUpdateDt()); statement.setInt(index++, groupInfo.getMemberCount()); statement.setInt(index++, groupInfo.getMute()); statement.setInt(index++, groupInfo.getJoinType()); statement.setInt(index++, groupInfo.getPrivateChat()); statement.setInt(index++, groupInfo.getSearchable()); statement.setLong(index++, groupInfo.getMemberUpdateDt() == 0 ? System.currentTimeMillis() : groupInfo.getUpdateDt()); statement.setString(index++, groupInfo.getName()); statement.setString(index++, groupInfo.getPortrait()); statement.setString(index++, groupInfo.getOwner()); statement.setInt(index++, groupInfo.getType()); statement.setString(index++, groupInfo.getExtra()); statement.setLong(index++, groupInfo.getUpdateDt() == 0 ? System.currentTimeMillis() : groupInfo.getUpdateDt()); statement.setInt(index++, groupInfo.getMemberCount()); statement.setInt(index++, groupInfo.getMute()); statement.setInt(index++, groupInfo.getJoinType()); statement.setInt(index++, groupInfo.getPrivateChat()); statement.setInt(index++, groupInfo.getSearchable()); statement.setLong(index++, groupInfo.getMemberUpdateDt() == 0 ? System.currentTimeMillis() : groupInfo.getUpdateDt()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } List getUserActivedSessions(String uid) { String sql = "select `_cid`, `_package_name`,`_token`,`_voip_token`,`_secret`,`_db_secret`,`_platform`,`_push_type`,`_device_name`,`_device_version`,`_phone_name`,`_language`,`_carrier_name`, `_dt` from t_user_session where `_uid` = ? and `_deleted` = 0"; Connection connection = null; PreparedStatement statement = null; List result = new ArrayList<>(); ResultSet resultSet = null; try { connection = DBUtil.getConnection(); statement = connection.prepareStatement(sql); statement.setString(1, uid); resultSet = statement.executeQuery(); while (resultSet.next()) { int index = 1; String cid = resultSet.getString(index++); ClientSession clientSession = new ClientSession(cid, Server.getServer().getStore().sessionsStore()); MemorySessionStore.Session session = new MemorySessionStore.Session(uid, cid, clientSession); session.setAppName(resultSet.getString(index++)); session.setDeviceToken(resultSet.getString(index++)); session.setVoipDeviceToken(resultSet.getString(index++)); session.setSecret(resultSet.getString(index++)); session.setDbSecret(resultSet.getString(index++)); session.setPlatform(resultSet.getInt(index++)); session.setPushType(resultSet.getInt(index++)); session.setDeviceName(resultSet.getString(index++)); session.setDeviceVersion(resultSet.getString(index++)); session.setPhoneName(resultSet.getString(index++)); session.setLanguage(resultSet.getString(index++)); session.setCarrierName(resultSet.getString(index++)); session.setUpdateDt(resultSet.getLong(index++)); result.add(session); } } catch (SQLException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, resultSet); } return result; } MemorySessionStore.Session getSession(String uid, String clientId, ClientSession clientSession) { String sql = "select `_package_name`,`_token`,`_voip_token`,`_secret`,`_db_secret`,`_platform`,`_push_type`,`_device_name`,`_device_version`,`_phone_name`,`_language`,`_carrier_name`, `_dt`, `_deleted` from t_user_session where `_uid` = ? and `_cid` = ? limit 1"; Connection connection = null; PreparedStatement statement = null; ResultSet resultSet = null; try { connection = DBUtil.getConnection(); statement = connection.prepareStatement(sql); statement.setString(1, uid); statement.setString(2, clientId); resultSet = statement.executeQuery(); if (resultSet.next()) { MemorySessionStore.Session session = new MemorySessionStore.Session(uid, clientId, clientSession); int index = 1; session.setAppName(resultSet.getString(index++)); session.setDeviceToken(resultSet.getString(index++)); session.setVoipDeviceToken(resultSet.getString(index++)); session.setSecret(resultSet.getString(index++)); session.setDbSecret(resultSet.getString(index++)); session.setPlatform(resultSet.getInt(index++)); session.setPushType(resultSet.getInt(index++)); session.setDeviceName(resultSet.getString(index++)); session.setDeviceVersion(resultSet.getString(index++)); session.setPhoneName(resultSet.getString(index++)); session.setLanguage(resultSet.getString(index++)); session.setCarrierName(resultSet.getString(index++)); session.setUpdateDt(resultSet.getLong(index++)); session.setDeleted(resultSet.getInt(index)); return session; } } catch (SQLException e) { e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, resultSet); } return null; } void clearOtherSessionToken(String cid, String token, int pushType, boolean voipToken) { if (voipToken) { return; } Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "update t_user_session set `_token` = ?, `_dt` = ? where `_token` = ? and `_push_type` = ? and `_cid` <> ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, ""); statement.setLong(index++, System.currentTimeMillis()); statement.setString(index++, token); statement.setInt(index++, pushType); statement.setString(index++, cid); int c = statement.executeUpdate(); LOG.info("Update rows {}", c); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } void updateSessionToken(String uid, String cid, String token, int pushType, boolean voipToken) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; clearOtherSessionToken(cid, token, pushType, voipToken); try { connection = DBUtil.getConnection(); String sql ; if (voipToken) { sql = "update t_user_session set `_voip_token` = ?, `_dt` = ? where `_uid` = ? and `_cid` = ?"; } else { sql = "update t_user_session set `_token` = ?, `_push_type` = ?, `_dt` = ? where `_uid` = ? and `_cid` = ?"; } statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, token); if (!voipToken) { statement.setInt(index++, pushType); } statement.setLong(index++, System.currentTimeMillis()); statement.setString(index++, uid); statement.setString(index++, cid); int c = statement.executeUpdate(); LOG.info("Update rows {}", c); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void updateSessionDeleted(String uid, String cid, int deleted) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "update t_user_session set `_deleted` = ? where `_uid` = ? and `_cid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setInt(index++, deleted); statement.setString(index++, uid); statement.setString(index++, cid); int c = statement.executeUpdate(); LOG.info("Update rows {}", c); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void updateSessionPlatform(String uid, String cid, int platform) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "update t_user_session set `_platform` = ? where `_uid` = ? and `_cid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setInt(index++, platform); statement.setString(index++, uid); statement.setString(index++, cid); int c = statement.executeUpdate(); LOG.info("Update rows {}", c); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } private boolean strEqual(String left, String right) { if (left == right) return true; if (left == null) return false; return left.equals(right); } void updateSession(String uid, String cid, MemorySessionStore.Session session, WFCMessage.RouteRequest request) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "update t_user_session set"; if (!strEqual(session.getAppName(), request.getApp())) { sql += " `_package_name` = ?,"; } if (session.getPlatform() != request.getPlatform()) { sql += " `_platform` = ?,"; } if (session.getPushType() != request.getPushType()) { sql += " `_push_type` = ?,"; } if (!strEqual(session.getDeviceName(), request.getDeviceName())) { sql += " `_device_name` = ?,"; } if (!strEqual(session.getDeviceVersion(), request.getDeviceVersion())) { sql += " `_device_version` = ?,"; } if (!strEqual(session.getPhoneName(), request.getPhoneName())) { sql += " `_phone_name` = ?,"; } if (!strEqual(session.getLanguage(), request.getLanguage())) { sql += " `_language` = ?,"; } if (!strEqual(session.getCarrierName(), request.getCarrierName())) { sql += " `_carrier_name` = ?,"; } sql += " `_dt` = ?"; sql += " where `_uid` = ? and `_cid` = ?"; statement = connection.prepareStatement(sql); int index = 1; if (!strEqual(session.getAppName(), request.getApp())) { statement.setString(index++, request.getApp()); } if (session.getPlatform() != request.getPlatform()) { statement.setInt(index++, request.getPlatform()); } if (session.getPushType() != request.getPushType()) { statement.setInt(index++, request.getPushType() >= 32 ? 0 : request.getPushType()); } if (!strEqual(session.getDeviceName(), request.getDeviceName())) { statement.setString(index++, request.getDeviceName()); } if (!strEqual(session.getDeviceVersion(), request.getDeviceVersion())) { statement.setString(index++, request.getDeviceVersion()); } if (!strEqual(session.getPhoneName(), request.getPhoneName())) { statement.setString(index++, request.getPhoneName()); } if (!strEqual(session.getLanguage(), request.getLanguage())) { statement.setString(index++, request.getLanguage()); } if (!strEqual(session.getCarrierName(), request.getCarrierName())) { statement.setString(index++, request.getCarrierName()); } statement.setLong(index++, System.currentTimeMillis()); statement.setString(index++, uid); statement.setString(index++, cid); int c = statement.executeUpdate(); LOG.info("Update rows {}", c); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void clearMultiEndpoint(String uid, String clientId, int platform) { LOG.info("clearMultiEndpoint {}, {}", uid, clientId); Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql; if (platform == ProtoConstants.Platform.Platform_Windows || platform == ProtoConstants.Platform.Platform_OSX || platform == ProtoConstants.Platform.Platform_iOS || platform == ProtoConstants.Platform.Platform_Android) { sql = "update t_user_session set `_deleted` = ?, `_token` = ?, `_voip_token` = ?, `_dt` = ? where `_uid`=? and (`_platform` = ? or `_platform` = ?) and `_cid` <> ? and `_deleted` = 0"; } else { sql = "update t_user_session set `_deleted` = ?, `_token` = ?, `_voip_token` = ?, `_dt` = ? where `_uid`=? and `_platform` = ? and `_cid` <> ? and `_deleted` = 0"; } statement = connection.prepareStatement(sql); int index = 1; statement.setInt(index++, 1); statement.setString(index++, ""); statement.setString(index++, ""); statement.setLong(index++, System.currentTimeMillis()); statement.setString(index++, uid); if (platform == ProtoConstants.Platform.Platform_Windows || platform == ProtoConstants.Platform.Platform_OSX) { statement.setInt(index++, ProtoConstants.Platform.Platform_Windows); statement.setInt(index++, ProtoConstants.Platform.Platform_OSX); } else if(platform == ProtoConstants.Platform.Platform_iOS || platform == ProtoConstants.Platform.Platform_Android) { statement.setInt(index++, ProtoConstants.Platform.Platform_iOS); statement.setInt(index++, ProtoConstants.Platform.Platform_Android); } else { statement.setInt(index++, platform); } statement.setString(index++, clientId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } MemorySessionStore.Session createSession(String uid, String clientId, ClientSession clientSession, int platform) { Connection connection = null; PreparedStatement statement = null; LOG.info("Database create session {},{}", uid, clientId); try { connection = DBUtil.getConnection(); String sql = "insert into t_user_session (`_uid`,`_cid`,`_platform`,`_secret`,`_db_secret`, `_dt`) values (?,?,?,?,?,?)"; statement = connection.prepareStatement(sql); int index = 1; MemorySessionStore.Session session = new MemorySessionStore.Session(uid, clientId, clientSession); session.setPlatform(platform); statement.setString(index++, uid); statement.setString(index++, clientId); statement.setInt(index++, platform); session.setSecret(UUID.randomUUID().toString()); statement.setString(index++, session.getSecret()); session.setDbSecret(UUID.randomUUID().toString()); statement.setString(index++, session.getDbSecret()); statement.setLong(index++, System.currentTimeMillis()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); return session; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } return null; } void updateGroupMemberCountDt(final String groupId, final int count, final long dt) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql ; if (count >= 0) { sql = "update t_group set `_member_count` = ?, `_member_dt` = ? , `_dt` = ? where `_gid` = ?"; } else { sql = "update t_group set `_member_dt` = ?, `_dt` = ? where `_gid` = ?"; } statement = connection.prepareStatement(sql); int index = 1; if (count >=0) { statement.setInt(index++, count); } statement.setLong(index++, dt); statement.setLong(index++, dt); statement.setString(index++, groupId); int c = statement.executeUpdate(); LOG.info("Update rows {}", c); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void persistGroupMember(final String groupId, final List memberList) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); connection.setAutoCommit(false); String sql = "insert into t_group_member (`_gid`" + ", `_mid`" + ", `_alias`" + ", `_type`" + ", `_dt`) values(?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_alias` = ?," + "`_type` = ?," + "`_dt` = ?"; statement = connection.prepareStatement(sql); for (WFCMessage.GroupMember member : memberList ) { int index = 1; long dt = System.currentTimeMillis(); if (member.getUpdateDt() > 0) { dt = member.getUpdateDt(); } statement.setString(index++, groupId); statement.setString(index++, member.getMemberId()); statement.setString(index++, member.getAlias()); statement.setInt(index++, member.getType()); statement.setLong(index++, dt); statement.setString(index++, member.getAlias()); statement.setInt(index++, member.getType()); statement.setLong(index++, dt); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } connection.commit(); connection.setAutoCommit(true); } catch (SQLException e) { // TODO Auto-generated catch block if (connection != null) { try { connection.commit(); connection.setAutoCommit(true); } catch (SQLException e1) { e1.printStackTrace(); } } e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } int removeGroupMember(String groupId, List groupMembers) { Connection connection = null; PreparedStatement statement = null; PreparedStatement statement2 = null; try { connection = DBUtil.getConnection(); connection.setAutoCommit(false); StringBuilder sqlBuilder = new StringBuilder("update t_group_member set `_type` = ?, `_dt` = ? where `_mid` in ("); for (int i = 0; i < groupMembers.size(); i++) { sqlBuilder.append("?"); if (i != groupMembers.size()-1) { sqlBuilder.append(","); } } sqlBuilder.append(")"); sqlBuilder.append(" and _gid = ?"); statement = connection.prepareStatement(sqlBuilder.toString()); int index = 1; long current = System.currentTimeMillis(); statement.setInt(index++, ProtoConstants.GroupMemberType.GroupMemberType_Removed); statement.setLong(index++, current); for (String memberId:groupMembers) { statement.setString(index++, memberId); } statement.setString(index++, groupId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); String sql = "update t_group set `_member_count` = `_member_count` - ?, `_member_dt` = ? , `_dt` = ? where `_gid` = ?"; statement2 = connection.prepareStatement(sql); index = 1; statement2.setInt(index++, count); statement2.setLong(index++, current); statement2.setLong(index++, current); statement2.setString(index++, groupId); statement2.executeUpdate(); connection.commit(); connection.setAutoCommit(true); } catch (SQLException e) { // TODO Auto-generated catch block if (connection != null) { try { connection.commit(); connection.setAutoCommit(true); } catch (SQLException e1) { e1.printStackTrace(); } } e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); DBUtil.closeDB(connection, statement2); } return 0; } WFCMessage.GroupInfo getPersistGroupInfo(String groupId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_name`" + ", `_portrait`" + ", `_owner`" + ", `_type`" + ", `_extra`" + ", `_dt`" + ", `_member_count`" + ", `_member_dt`" + ", `_mute`" + ", `_join_type`" + ", `_private_chat`" + ", `_searchable`" + " from t_group where `_gid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, groupId); rs = statement.executeQuery(); if (rs.next()) { String strValue; int intValue; WFCMessage.GroupInfo.Builder builder = WFCMessage.GroupInfo.newBuilder(); index = 1; builder.setTargetId(groupId); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setName(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setPortrait(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setOwner(strValue); intValue = rs.getInt(index++); builder.setType(intValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setExtra(strValue); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); intValue = rs.getInt(index++); builder.setMemberCount(intValue); longValue = rs.getLong(index++); builder.setMemberUpdateDt(longValue); intValue = rs.getInt(index++); builder.setMute(intValue); intValue = rs.getInt(index++); builder.setJoinType(intValue); intValue = rs.getInt(index++); builder.setPrivateChat(intValue); intValue = rs.getInt(index++); builder.setSearchable(intValue); return builder.build(); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } void updateChatroomInfo(String chatroomId, WFCMessage.ChatroomInfo chatroomInfo) { LOG.info("Database update chatroom info {}", chatroomId); mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_chatroom (`_cid`" + ", `_title`" + ", `_portrait`" + ", `_state`" + ", `_desc`" + ", `_extra`" + ", `_dt`) values(?, ?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE `_title`=?" + ", `_portrait`=?" + ", `_state`=?" + ", `_desc`=?" + ", `_extra`=?" + ", `_dt`=?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, chatroomId); statement.setString(index++, chatroomInfo.getTitle()); statement.setString(index++, chatroomInfo.getPortrait()); statement.setInt(index++, chatroomInfo.getState()); statement.setString(index++, chatroomInfo.getDesc()); statement.setString(index++, chatroomInfo.getExtra()); statement.setLong(index++, chatroomInfo.getUpdateDt() == 0 ? System.currentTimeMillis() : chatroomInfo.getUpdateDt()); statement.setString(index++, chatroomInfo.getTitle()); statement.setString(index++, chatroomInfo.getPortrait()); statement.setInt(index++, chatroomInfo.getState()); statement.setString(index++, chatroomInfo.getDesc()); statement.setString(index++, chatroomInfo.getExtra()); statement.setLong(index++, chatroomInfo.getUpdateDt() == 0 ? System.currentTimeMillis() : chatroomInfo.getUpdateDt()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void removeChatroomInfo(String chatroomId) { LOG.info("Database remove chatroom {}", chatroomId); mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "delete from t_chatroom where `_cid`=?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, chatroomId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } WFCMessage.ChatroomInfo getPersistChatroomInfo(String chatroomId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_title`" + ", `_portrait`" + ", `_state`" + ", `_desc`" + ", `_extra`" + ", `_dt`" + " from t_chatroom where `_cid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, chatroomId); rs = statement.executeQuery(); if (rs.next()) { String strValue; int intValue; WFCMessage.ChatroomInfo.Builder builder = WFCMessage.ChatroomInfo.newBuilder(); index = 1; strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setTitle(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setPortrait(strValue); intValue = rs.getInt(index++); builder.setState(intValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setDesc(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setExtra(strValue); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); builder.setCreateDt(longValue); return builder.build(); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } void updateUserPassword(final String userId, final String password) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "update t_user set `_passwd_md5` = ? where `_uid` = ?"; statement = connection.prepareStatement(sql); int index = 1; try { MessageDigest md5 = MessageDigest.getInstance("MD5"); String passwdMd5 = Base64.getEncoder().encodeToString(md5.digest(password.getBytes("utf-8"))); statement.setString(index, passwdMd5); } catch (Exception e) { statement.setString(index, ""); } index++; statement.setString(index++, userId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void deleteUser(final String userId) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "delete from t_user where `_uid`=?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, userId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void deleteRobot(String robotId) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "delete from t_robot where `_uid`=?"; statement = connection.prepareStatement(sql); statement.setString(1, robotId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } WFCMessage.Robot getRobot(String robotId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_owner`" + ", `_secret`" + ", `_callback`" + ", `_state`" + ", `_extra`" + ", `_dt` from t_robot where `_uid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(1, robotId); rs = statement.executeQuery(); if (rs.next()) { WFCMessage.Robot.Builder builder = WFCMessage.Robot.newBuilder(); index = 1; String value = rs.getString(index++); value = (value == null ? "" : value); builder.setUid(robotId); builder.setOwner(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setSecret(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setCallback(value); int state = rs.getInt(index++); builder.setState(state); value = rs.getString(index++); value = (value == null ? "" : value); builder.setExtra(value); long longValue = rs.getLong(index++); return builder.build(); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } void updateRobot(final WFCMessage.Robot robot) { LOG.info("Database update user info {}", robot.getUid()); mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_robot (`_uid`" + ", `_owner`" + ", `_secret`" + ", `_callback`" + ", `_state`" + ", `_extra`" + ", `_dt`) values(?, ?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE `_owner`=?" + ", `_secret`=?" + ", `_callback`=?" + ", `_state`=?" + ", `_extra`=?" + ", `_dt`=?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, robot.getUid()); statement.setString(index++, robot.getOwner()); statement.setString(index++, robot.getSecret()); statement.setString(index++, robot.getCallback()); statement.setInt(index++, robot.getState()); statement.setString(index++, robot.getExtra()); statement.setLong(index++, System.currentTimeMillis()); statement.setString(index++, robot.getOwner()); statement.setString(index++, robot.getSecret()); statement.setString(index++, robot.getCallback()); statement.setInt(index++, robot.getState()); statement.setString(index++, robot.getExtra()); statement.setLong(index++, System.currentTimeMillis()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void updateUser(final WFCMessage.User user) { LOG.info("Database update user info {} {}", user.getUid(), user.getUpdateDt()); Connection connection = null; PreparedStatement statement = null; LOG.info("Database update user info {}", user.getDisplayName()); try { connection = DBUtil.getConnection(); String sql = "insert into t_user (`_uid`" + ", `_name`" + ", `_display_name`" + ", `_portrait`" + ", `_mobile`" + ", `_gender`" + ", `_email`" + ", `_address`" + ", `_company`" + ", `_social`" + ", `_extra`" + ", `_type`" + ", `_dt`) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE `_name`=?" + ", `_display_name`=?" + ", `_portrait`=?" + ", `_mobile`=?" + ", `_gender`=?" + ", `_email`=?" + ", `_address`=?" + ", `_company`=?" + ", `_social`=?" + ", `_extra`=?" + ", `_type`=?" + ", `_dt`=?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, user.getUid()); statement.setString(index++, user.getName()); statement.setString(index++, user.getDisplayName()); statement.setString(index++, user.getPortrait()); statement.setString(index++, user.getMobile()); statement.setInt(index++, user.getGender()); statement.setString(index++, user.getEmail()); statement.setString(index++, user.getAddress()); statement.setString(index++, user.getCompany()); statement.setString(index++, user.getSocial()); statement.setString(index++, user.getExtra()); statement.setInt(index++, user.getType()); statement.setLong(index++, user.getUpdateDt() == 0 ? System.currentTimeMillis() : user.getUpdateDt()); statement.setString(index++, user.getName()); statement.setString(index++, user.getDisplayName()); statement.setString(index++, user.getPortrait()); statement.setString(index++, user.getMobile()); statement.setInt(index++, user.getGender()); statement.setString(index++, user.getEmail()); statement.setString(index++, user.getAddress()); statement.setString(index++, user.getCompany()); statement.setString(index++, user.getSocial()); statement.setString(index++, user.getExtra()); statement.setInt(index++, user.getType()); statement.setLong(index++, user.getUpdateDt() == 0 ? System.currentTimeMillis() : user.getUpdateDt()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } void deleteUserStatus(String userId) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "delete from t_user_status where _uid = ?"; statement = connection.prepareStatement(sql); statement.setString(1, userId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void updateUserStatus(String userId, int status) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); if (status == 0) { String sql = "delete from t_user_status where _uid = ?"; statement = connection.prepareStatement(sql); statement.setString(1, userId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } else { String sql = "insert into t_user_status (`_uid`, `_status`, `_dt`) values(?,?,?) ON DUPLICATE KEY UPDATE `_status` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, userId); statement.setInt(index++, status); statement.setLong(index++, System.currentTimeMillis()); statement.setInt(index++, status); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } int getGeneratedId() { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "INSERT INTO `t_id_generator` (`id`) VALUES (NULL);"; statement = connection.prepareStatement(sql); if(statement.executeUpdate()> 0) { sql = "SELECT LAST_INSERT_ID()"; try { if (statement!=null) { statement.close(); } } catch (Exception e) { e.printStackTrace(); Utility.printExecption(LOG, e); } statement = connection.prepareStatement(sql); rs = statement.executeQuery(); if (rs.next()) { return rs.getInt(1); } } return -1; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); return -1; } finally { DBUtil.closeDB(connection, statement, rs); } } WFCMessage.User getPersistUser(String userId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_name`" + ", `_display_name`" + ", `_portrait`" + ", `_mobile`" + ", `_gender`" + ", `_email`" + ", `_address`" + ", `_company`" + ", `_social`" + ", `_extra`" + ", `_type`" + ", `_dt` from t_user where `_uid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(1, userId); rs = statement.executeQuery(); if (rs.next()) { WFCMessage.User.Builder builder = WFCMessage.User.newBuilder(); index = 1; String value = rs.getString(index++); value = (value == null ? "" : value); builder.setUid(userId); builder.setName(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setDisplayName(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setPortrait(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setMobile(value); int gender = rs.getInt(index++); builder.setGender(gender); value = rs.getString(index++); value = (value == null ? "" : value); builder.setEmail(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setAddress(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setCompany(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setSocial(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setExtra(value); int type = rs.getInt(index++); builder.setType(type); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); return builder.build(); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } String getUserIdByName(String name) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_uid` from t_user where `_name` = ? limit 1"; statement = connection.prepareStatement(sql); statement.setString(1, name); rs = statement.executeQuery(); if (rs.next()) { return rs.getString(1); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } String getUserIdByMobile(String mobile) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_uid` from t_user where `_mobile` = ? limit 1"; statement = connection.prepareStatement(sql); statement.setString(1, mobile); rs = statement.executeQuery(); if (rs.next()) { return rs.getString(1); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } Set getAllEnds() { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select distinct(`_uid`) from t_user_session where `_deleted` = 0"; statement = connection.prepareStatement(sql); rs = statement.executeQuery(); Set out = new HashSet<>(); while (rs.next()) { String uid = rs.getString(1); out.add(uid); } return out; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } List getPersistFriends(String userId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_friend_uid`, `_alias`, `_state`, `_blacked`, `_dt` from t_friend where `_uid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, userId); rs = statement.executeQuery(); List out = new ArrayList<>(); while (rs.next()) { String uid = rs.getString(1); String alias = rs.getString(2); int state = rs.getInt(3); int blacked = rs.getInt(4); long timestamp = rs.getLong(5); FriendData data = new FriendData(userId, uid, alias, state, blacked, timestamp); out.add(data); } return out; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } List getPersistFriendRequests(String userId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_uid`" + ", `_friend_uid`" + ", `_reason`" + ", `_status`" + ", `_dt`" + ", `_from_read_status`" + ", `_to_read_status` from t_friend_request where `_uid` = ? UNION ALL " + "select `_uid`" + ", `_friend_uid`" + ", `_reason`" + ", `_status`" + ", `_dt`" + ", `_from_read_status`" + ", `_to_read_status` from t_friend_request where `_friend_uid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, userId); statement.setString(index++, userId); rs = statement.executeQuery(); List out = new ArrayList<>(); while (rs.next()) { WFCMessage.FriendRequest.Builder builder = WFCMessage.FriendRequest.newBuilder(); index = 1; String value = rs.getString(index++); value = (value == null ? "" : value); builder.setFromUid(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setToUid(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setReason(value); int intValue = rs.getInt(index++); builder.setStatus(intValue); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); boolean b = rs.getBoolean(index++); builder.setFromReadStatus(b); b = rs.getBoolean(index++); builder.setToReadStatus(b); out.add(builder.build()); } return out; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } void persistFriendRequestUnreadStatus(String userId, long readDt, long updateDt) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "update t_friend_request set `_dt`=? , `_to_read_status`=?" + " where " + "`_friend_uid` = ? and" + "`_dt` <= ? and" + "`_to_read_status` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setLong(index++, updateDt); statement.setBoolean(index++, true); statement.setString(index++, userId); statement.setLong(index++, readDt); statement.setBoolean(index++, false); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } // void persistOrUpdateFriendRequest(final WFCMessage.FriendRequest request) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_friend_request (`_uid`, `_friend_uid`, `_reason`, `_status`, `_dt`, `_from_read_status`, `_to_read_status`) values(?, ?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_reason` = ?," + "`_status` = ?," + "`_dt` = ?," + "`_from_read_status` = ?," + "`_to_read_status` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, request.getFromUid()); statement.setString(index++, request.getToUid()); statement.setString(index++, request.getReason()); statement.setInt(index++, request.getStatus()); statement.setLong(index++, request.getUpdateDt()); statement.setInt(index++, request.getFromReadStatus() ? 1 : 0); statement.setInt(index++, request.getToReadStatus() ? 1 : 0); statement.setString(index++, request.getReason()); statement.setInt(index++, request.getStatus()); statement.setLong(index++, request.getUpdateDt()); statement.setInt(index++, request.getFromReadStatus() ? 1 : 0); statement.setInt(index++, request.getToReadStatus() ? 1 : 0); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } void persistOrUpdateFriendData(final FriendData request) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_friend (`_uid`, `_friend_uid`, `_alias`, `_state`, `_blacked`, `_dt`) values(?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_alias` = ?," + "`_state` = ?," + "`_blacked` = ?," + "`_dt` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, request.getUserId()); statement.setString(index++, request.getFriendUid()); statement.setString(index++, request.getAlias()); statement.setInt(index++, request.getState()); statement.setInt(index++, request.getBlacked()); statement.setLong(index++, request.getTimestamp()); statement.setString(index++, request.getAlias()); statement.setInt(index++, request.getState()); statement.setInt(index++, request.getBlacked()); statement.setLong(index++, request.getTimestamp()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } boolean removeGroupInfoFromDB(String groupId) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "delete from t_group where _gid = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, groupId); return statement.executeUpdate() > 0; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } return false; } boolean removeGroupMemberFromDB(String groupId) { Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "delete from t_group_member where _gid = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, groupId); return statement.executeUpdate() > 0; } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } return false; } void updateChannelInfo(final WFCMessage.ChannelInfo channelInfo) { LOG.info("Database update channel info {} {}", channelInfo.getTargetId(), channelInfo.getUpdateDt()); Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_channel (`_cid`" + ", `_name`" + ", `_portrait`" + ", `_owner`" + ", `_status`" + ", `_desc`" + ", `_extra`" + ", `_secret`" + ", `_callback`" + ", `_automatic`" + ", `_dt`) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)" + " ON DUPLICATE KEY UPDATE `_name`=?" + ", `_portrait`=?" + ", `_owner`=?" + ", `_status`=?" + ", `_desc`=?" + ", `_extra`=?" + ", `_secret`=?" + ", `_callback`=?" + ", `_automatic`=?" + ", `_dt`=?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, channelInfo.getTargetId()); statement.setString(index++, channelInfo.getName()); statement.setString(index++, channelInfo.getPortrait()); statement.setString(index++, channelInfo.getOwner()); statement.setInt(index++, channelInfo.getStatus()); statement.setString(index++, channelInfo.getDesc()); statement.setString(index++, channelInfo.getExtra()); statement.setString(index++, channelInfo.getSecret()); statement.setString(index++, channelInfo.getCallback()); statement.setInt(index++, channelInfo.getAutomatic()); statement.setLong(index++, channelInfo.getUpdateDt() == 0 ? System.currentTimeMillis() : channelInfo.getUpdateDt()); statement.setString(index++, channelInfo.getName()); statement.setString(index++, channelInfo.getPortrait()); statement.setString(index++, channelInfo.getOwner()); statement.setInt(index++, channelInfo.getStatus()); statement.setString(index++, channelInfo.getDesc()); statement.setString(index++, channelInfo.getExtra()); statement.setString(index++, channelInfo.getSecret()); statement.setString(index++, channelInfo.getCallback()); statement.setInt(index++, channelInfo.getAutomatic()); statement.setLong(index++, channelInfo.getUpdateDt() == 0 ? System.currentTimeMillis() : channelInfo.getUpdateDt()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } } void removeChannelInfo(final String channelId) { LOG.info("Database remove channel {}", channelId); mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "delete from t_channel where `_cid`=?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, channelId); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } WFCMessage.ChannelInfo getPersistChannelInfo(String channelId) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_name`" + ", `_portrait`" + ", `_owner`" + ", `_status`" + ", `_desc`" + ", `_extra`" + ", `_secret`" + ", `_callback`" + ", `_automatic`" + ", `_dt` from t_channel where `_cid` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, channelId); rs = statement.executeQuery(); if (rs.next()) { String strValue; int intValue; WFCMessage.ChannelInfo.Builder builder = WFCMessage.ChannelInfo.newBuilder(); index = 1; builder.setTargetId(channelId); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setName(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setPortrait(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setOwner(strValue); intValue = rs.getInt(index++); builder.setStatus(intValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setDesc(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setExtra(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setSecret(strValue); strValue = rs.getString(index++); strValue = (strValue == null ? "" : strValue); builder.setCallback(strValue); intValue = rs.getInt(index++); builder.setAutomatic(intValue); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); return builder.build(); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return null; } void persistChannelListener(final String groupId, final List memberList) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); connection.setAutoCommit(false); String sql = "insert into t_channel_listener (`_cid`" + ", `_mid`" + ", `_dt`) values(?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_dt` = ?"; statement = connection.prepareStatement(sql); long dt = System.currentTimeMillis(); for (String member : memberList) { int index = 1; statement.setString(index++, groupId); statement.setString(index++, member); statement.setLong(index++, dt); statement.setLong(index++, dt); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } connection.commit(); connection.setAutoCommit(true); } catch (SQLException e) { // TODO Auto-generated catch block if (connection != null) { try { connection.commit(); connection.setAutoCommit(true); } catch (SQLException e1) { e1.printStackTrace(); } } e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } List getChannelListener(String channelId) { List out = new ArrayList<>(); Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; try { connection = DBUtil.getConnection(); String sql = "select `_mid` from t_channel_listener where _cid = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, channelId); rs = statement.executeQuery(); while (rs.next()) { String value = rs.getString(index++); if (!StringUtil.isNullOrEmpty(value)) { out.add(value); } } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return out; } List searchChannelFromDB(String keyword, boolean buzzy, int page) { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; ArrayList out = new ArrayList<>(); try { connection = DBUtil.getConnection(); String sql = "select `_cid`, `_name`" + ", `_portrait`" + ", `_owner`" + ", `_status`" + ", `_desc`" + ", `_extra`" + ", `_dt` from t_channel"; if (buzzy) { sql += " where `_name` like ?"; } else { sql += " where `_name` = ?"; } sql += " and _status = 0"; sql += " limit 20"; if (page > 0) { sql += "offset = '" + page * 20 + "'"; } statement = connection.prepareStatement(sql); int index = 1; if (buzzy) { statement.setString(index++, "%" + keyword + "%"); } else { statement.setString(index++, keyword); } rs = statement.executeQuery(); while (rs.next()) { WFCMessage.ChannelInfo.Builder builder = WFCMessage.ChannelInfo.newBuilder(); index = 1; String value = rs.getString(index++); value = (value == null ? "" : value); builder.setTargetId(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setName(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setPortrait(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setOwner(value); int status = rs.getInt(index++); builder.setStatus(status); value = rs.getString(index++); value = (value == null ? "" : value); builder.setDesc(value); value = rs.getString(index++); value = (value == null ? "" : value); builder.setExtra(value); long longValue = rs.getLong(index++); builder.setUpdateDt(longValue); out.add(builder.build()); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return out; } void updateChannelListener(final String channelId, final String listener, final boolean listen) { LOG.info("Database remove channel {}", channelId); mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql; if (listen) { sql = "insert into t_channel_listener (`_cid`" + ", `_mid`" + ", `_dt`) values(?, ?, ?)" + " ON DUPLICATE KEY UPDATE " + "`_dt` = ?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, channelId); statement.setString(index++, listener); statement.setLong(index++, System.currentTimeMillis()); statement.setLong(index++, System.currentTimeMillis()); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } else { sql = "delete from t_channel_listener where `_cid`=? and `_mid`=?"; statement = connection.prepareStatement(sql); int index = 1; statement.setString(index++, channelId); statement.setString(index++, listener); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } public Set getSensitiveWord() { Connection connection = null; PreparedStatement statement = null; ResultSet rs = null; Set out = new HashSet<>(); try { connection = DBUtil.getConnection(); String sql = "select `_word` from t_sensitiveword order by `id` desc"; statement = connection.prepareStatement(sql); rs = statement.executeQuery(); while (rs.next()) { String value = rs.getString(1); out.add(value); } } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement, rs); } return out; } void deleteSensitiveWord(final String word) { LOG.info("delete sensitive word {}", word); mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "delete from t_sensitiveword where `_word` = ?"; statement = connection.prepareStatement(sql); statement.setString(1, word); int c = statement.executeUpdate(); LOG.info("Update rows {}", c); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } void persistSensitiveWord(final String word) { mScheduler.execute(()->{ Connection connection = null; PreparedStatement statement = null; try { connection = DBUtil.getConnection(); String sql = "insert into t_sensitiveword (`_word`) values(?)"; statement = connection.prepareStatement(sql); statement.setString(1, word); int count = statement.executeUpdate(); LOG.info("Update rows {}", count); } catch (SQLException e) { // TODO Auto-generated catch block e.printStackTrace(); Utility.printExecption(LOG, e); } finally { DBUtil.closeDB(connection, statement); } }); } private String getUserMessageTable(String uid) { if (DBUtil.IsEmbedDB) { return "t_user_messages"; } int hashId = Math.abs(uid.hashCode())%128; return "t_user_messages_" + hashId; } }