diff options
author | 2013-07-16 04:07:30 +0700 | |
---|---|---|
committer | 2013-07-16 04:07:30 +0700 | |
commit | 9fa3da4667af7a93da5826f3eaa00d5384971179 (patch) | |
tree | 5b05ba13271a95f769f58fa2df8ec18f094d3612 | |
parent | 378b014b82158d06a6ee986cb7522a04bc59f5bc (diff) |
GroupQueries
-rw-r--r-- | src/com/juick/server/GroupsQueries.java | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/src/com/juick/server/GroupsQueries.java b/src/com/juick/server/GroupsQueries.java new file mode 100644 index 00000000..45a4942d --- /dev/null +++ b/src/com/juick/server/GroupsQueries.java @@ -0,0 +1,149 @@ +/* + * Juick + * Copyright (C) 2008-2013, Ugnich Anton + * + * 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.server; + +import com.juick.Group; +import java.sql.Connection; +import java.sql.PreparedStatement; +import java.sql.ResultSet; +import java.sql.SQLException; +import java.sql.Statement; +import java.util.ArrayList; + +/** + * + * @author Ugnich Anton + */ +public class GroupsQueries { + + public static ArrayList<com.juick.Group> getGroups(Connection sql, int uid) { + ArrayList<com.juick.Group> groups = new ArrayList<Group>(); + + PreparedStatement stmt = null; + ResultSet rs = null; + try { + stmt = sql.prepareStatement("SELECT groups.group_id,groups.name,COUNT(groups_users.user_id) FROM groups LEFT JOIN groups_users USING(group_id) WHERE groups.user_id=? GROUP BY group_id"); + stmt.setInt(1, uid); + rs = stmt.executeQuery(); + rs.beforeFirst(); + while (rs.next()) { + com.juick.Group g = new com.juick.Group(); + g.GID = rs.getInt(1); + g.Name = rs.getString(2); + g.UsersCnt = rs.getInt(3); + groups.add(g); + } + } catch (SQLException e) { + System.err.println(e); + } finally { + Utils.finishSQL(rs, stmt); + } + + return groups; + } + + public static int getGroupID(Connection sql, int uid, String name) { + int gid = 0; + + PreparedStatement stmt = null; + ResultSet rs = null; + try { + stmt = sql.prepareStatement("SELECT group_id FROM groups WHERE user_id=? AND name=?"); + stmt.setInt(1, uid); + stmt.setString(2, name); + rs = stmt.executeQuery(); + if (rs.first()) { + gid = rs.getInt(1); + } + } catch (SQLException e) { + System.err.println(e); + } finally { + Utils.finishSQL(rs, stmt); + } + + if (gid == 0) { + try { + stmt = sql.prepareStatement("INSERT INTO groups(user_id,name) VALUES (?)", Statement.RETURN_GENERATED_KEYS); + stmt.setString(1, name); + stmt.executeUpdate(); + rs = stmt.getGeneratedKeys(); + if (rs.first()) { + gid = rs.getInt(1); + } + } catch (SQLException e) { + System.err.println(e); + } finally { + Utils.finishSQL(rs, stmt); + } + } + + return gid; + } + + public static boolean removeGroupUser(Connection sql, int gid, int uid) { + boolean ret = false; + PreparedStatement stmt = null; + try { + stmt = sql.prepareStatement("DELETE FROM groups_users WHERE group_id=? AND user_id=?"); + stmt.setInt(1, gid); + stmt.setInt(2, uid); + if (stmt.executeUpdate() > 0) { + ret = true; + } + } catch (SQLException e) { + System.err.println(e); + } finally { + Utils.finishSQL(null, stmt); + } + + if (ret) { + int cnt = SQLHelpers.getInt(sql, "SELECT COUNT(*) FROM groups_users WHERE group_id=?", gid, -1); + if (cnt == 0) { + try { + stmt = sql.prepareStatement("DELETE FROM groups WHERE group_id=?"); + stmt.setInt(1, gid); + stmt.executeUpdate(); + } catch (SQLException e) { + System.err.println(e); + } finally { + Utils.finishSQL(null, stmt); + } + } + } + + return ret; + } + + public static boolean addGroupUser(Connection sql, int gid, int uid) { + boolean ret = false; + PreparedStatement stmt = null; + try { + stmt = sql.prepareStatement("INSERT INTO groups_users(group_id,user_id) VALUES (?,?)"); + stmt.setInt(1, gid); + stmt.setInt(2, uid); + if (stmt.executeUpdate() > 0) { + ret = true; + } + } catch (SQLException e) { + System.err.println(e); + } finally { + Utils.finishSQL(null, stmt); + } + return ret; + } +} |