PMQueriesService -> ChatService

1 files changed, 150 insertions, 0 deletions

diff --git a/src/main/java/com/juick/service/ChatServiceImpl.java b/src/main/java/com/juick/service/ChatServiceImpl.java
new file mode 100644
index 00000000..2d3fe31a
--- /dev/null
+++ b/src/main/java/com/juick/service/ChatServiceImpl.java
@@ -0,0 +1,150 @@
+/*
+ * Copyright (C) 2008-2020, Juick
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.juick.service;
+
+import com.juick.model.Chat;
+import com.juick.model.User;
+import com.juick.model.Message;
+import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
+import org.springframework.jdbc.core.namedparam.SqlParameterSource;
+import org.springframework.stereotype.Repository;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.List;
+
+/**
+ * Created by aalexeev on 11/13/16.
+ */
+@Repository
+public class ChatServiceImpl extends BaseJdbcService implements ChatService {
+
+    @Transactional
+    @Override
+    public boolean createMessage(final int uidFrom, final int uid_to, final String body) {
+        return getJdbcTemplate().update(
+                "INSERT INTO pm(user_id, user_id_to, txt) VALUES (?, ?, ?)",
+                uidFrom, uid_to, body) > 0;
+    }
+
+    @Transactional
+    @Override
+    public boolean addPMinRoster(final int uid, final String jid) {
+        return getJdbcTemplate().update(
+                "INSERT INTO pm_inroster(user_id, jid) VALUES (?, ?) ON DUPLICATE KEY UPDATE user_id=user_id", uid, jid) > 0;
+    }
+
+    @Transactional
+    @Override
+    public boolean removePMinRoster(final int uid, final String jid) {
+        return getJdbcTemplate().update(
+                "DELETE FROM pm_inroster WHERE user_id = ? AND jid = ?", uid, jid) > 0;
+    }
+
+    @Transactional
+    @Override
+    public boolean havePMinRoster(final int uid, final String jid) {
+        List<Integer> res = getJdbcTemplate().queryForList(
+                "SELECT 1 FROM pm_inroster WHERE user_id = ? AND jid = ?",
+                Integer.class,
+                uid, jid);
+        return res.size() > 0;
+    }
+
+    @Transactional(readOnly = true)
+    @Override
+    public List<Chat> getLastChats(final User user) {
+        return getJdbcTemplate().query(
+                "SELECT l.user_id, users.nick, l.last, pm.txt FROM pm "
+                        + "INNER JOIN users ON users.id = pm.user_id "
+                        + ""
+                        + "INNER JOIN (SELECT user_id, MAX(ts) AS last FROM pm "
+                        + "WHERE user_id_to=? GROUP BY user_id) l ON l.last = pm.ts "
+                        + "WHERE pm.user_id_to=? "
+                        + "ORDER BY l.last DESC",
+                (rs, rowNum) -> {
+                    Chat u = new Chat();
+                    u.setUid(rs.getInt(1));
+                    u.setName(rs.getString(2));
+                    u.setLastMessageTimestamp(rs.getTimestamp(3).toInstant());
+                    u.setLastMessageText(rs.getString(4).trim());
+                    return u;
+                },
+                user.getUid(), user.getUid());
+    }
+
+    @Transactional
+    @Override
+    public List<Message> getChat(final int uid, final int uidTo) {
+        SqlParameterSource sqlParameterSource = new MapSqlParameterSource()
+                .addValue("uid", uid)
+                .addValue("uidTo", uidTo);
+
+        return getNamedParameterJdbcTemplate().query(
+                "SELECT pm.user_id, pm.txt, pm.ts, users.nick FROM pm INNER JOIN users ON users.id=pm.user_id WHERE (user_id = :uid AND user_id_to = :uidTo) "
+                        + "OR (user_id_to = :uid AND user_id = :uidTo) ORDER BY ts DESC LIMIT 20",
+                sqlParameterSource,
+                (rs, rowNum) -> {
+                    Message msg = new Message();
+                    int uuid = rs.getInt(1);
+                    User user = new User();
+                    user.setUid(uuid);
+                    user.setName(rs.getString(4));
+                    msg.setUser(user);
+                    msg.setText(rs.getString(2).trim());
+                    msg.setCreated(rs.getTimestamp(3).toInstant());
+                    return msg;
+                });
+    }
+
+    @Transactional(readOnly = true)
+    @Override
+    public List<Message> getInbox(final int uid) {
+        return getJdbcTemplate().query(
+                "SELECT pm.user_id, users.nick, pm.txt, pm.ts " +
+                        "FROM pm INNER JOIN users ON pm.user_id=users.id WHERE pm.user_id_to=? ORDER BY pm.ts DESC LIMIT 20",
+                (rs, num) -> {
+                    Message msg = new Message();
+                    msg.setUser(new User());
+                    msg.getUser().setUid(rs.getInt(1));
+                    msg.getUser().setName(rs.getString(2));
+                    msg.setText(rs.getString(3).trim());
+                    msg.setCreated(rs.getTimestamp(4).toInstant());
+                    return msg;
+                },
+                uid);
+    }
+
+    @Transactional(readOnly = true)
+    @Override
+    public List<Message> getOutbox(final int uid) {
+        return getJdbcTemplate().query(
+                "SELECT pm.user_id_to, users.nick, pm.txt, " +
+                        "pm.ts FROM pm INNER JOIN users ON pm.user_id_to=users.id " +
+                        "WHERE pm.user_id=? ORDER BY pm.ts DESC LIMIT 20",
+                (rs, num) -> {
+                    Message msg = new Message();
+                    msg.setUser(new User());
+                    msg.getUser().setUid(rs.getInt(1));
+                    msg.getUser().setName(rs.getString(2));
+                    msg.setText(rs.getString(3).trim());
+                    msg.setCreated(rs.getTimestamp(4).toInstant());
+                    return msg;
+                },
+                uid);
+    }
+}