--- /dev/null
+/*******************************************************************************\r
+ * ============LICENSE_START==================================================\r
+ * * org.onap.dmaap\r
+ * * ===========================================================================\r
+ * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
+ * * ===========================================================================\r
+ * * Licensed under the Apache License, Version 2.0 (the "License");\r
+ * * you may not use this file except in compliance with the License.\r
+ * * You may obtain a copy of the License at\r
+ * * \r
+ * * http://www.apache.org/licenses/LICENSE-2.0\r
+ * * \r
+ * * Unless required by applicable law or agreed to in writing, software\r
+ * * distributed under the License is distributed on an "AS IS" BASIS,\r
+ * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * * See the License for the specific language governing permissions and\r
+ * * limitations under the License.\r
+ * * ============LICENSE_END====================================================\r
+ * *\r
+ * * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
+ * *\r
+ ******************************************************************************/\r
+\r
+package com.att.research.datarouter.provisioning.beans;\r
+\r
+import java.io.InvalidObjectException;\r
+import java.sql.Connection;\r
+import java.sql.PreparedStatement;\r
+import java.sql.ResultSet;\r
+import java.sql.SQLException;\r
+import java.sql.Statement;\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.Date;\r
+import java.util.List;\r
+\r
+import org.apache.log4j.Logger;\r
+import org.json.JSONObject;\r
+\r
+import com.att.research.datarouter.provisioning.utils.DB;\r
+import com.att.research.datarouter.provisioning.utils.URLUtilities;\r
+\r
+/**\r
+ * The representation of a Subscription. Subscriptions can be retrieved from the DB, or stored/updated in the DB.\r
+ * @author vikram\r
+ * @version $Id: Group.java,v 1.0 2016/07/19 \r
+ */\r
+public class Group extends Syncable {\r
+ private static Logger intlogger = Logger.getLogger("com.att.research.datarouter.provisioning.internal");\r
+ private static int next_groupid = getMaxGroupID() + 1;\r
+\r
+ private int groupid;\r
+ private String authid;\r
+ private String name;\r
+ private String description;\r
+ private String classification;\r
+ private String members;\r
+ private Date last_mod;\r
+ \r
+ \r
+ public static Group getGroupMatching(Group gup) {\r
+ String sql = String.format(\r
+ "select * from GROUPS where NAME = \"%s\"",\r
+ gup.getName()\r
+ );\r
+ List<Group> list = getGroupsForSQL(sql);\r
+ return list.size() > 0 ? list.get(0) : null;\r
+ }\r
+ \r
+ public static Group getGroupMatching(Group gup, int groupid) {\r
+ String sql = String.format(\r
+ "select * from GROUPS where NAME = \"%s\" and GROUPID != %d ",\r
+ gup.getName(),\r
+ gup.getGroupid()\r
+ );\r
+ List<Group> list = getGroupsForSQL(sql);\r
+ return list.size() > 0 ? list.get(0) : null;\r
+ }\r
+ \r
+ public static Group getGroupById(int id) {\r
+ String sql = "select * from GROUPS where GROUPID = " + id;\r
+ List<Group> list = getGroupsForSQL(sql);\r
+ return list.size() > 0 ? list.get(0) : null;\r
+ }\r
+ \r
+ public static Group getGroupByAuthId(String id) {\r
+ String sql = "select * from GROUPS where AUTHID = '" + id +"'";\r
+ List<Group> list = getGroupsForSQL(sql);\r
+ return list.size() > 0 ? list.get(0) : null;\r
+ }\r
+ \r
+ public static Collection<Group> getAllgroups() {\r
+ return getGroupsForSQL("select * from GROUPS");\r
+ }\r
+ private static List<Group> getGroupsForSQL(String sql) {\r
+ List<Group> list = new ArrayList<Group>();\r
+ try {\r
+ DB db = new DB();\r
+ @SuppressWarnings("resource")\r
+ Connection conn = db.getConnection();\r
+ Statement stmt = conn.createStatement();\r
+ ResultSet rs = stmt.executeQuery(sql);\r
+ while (rs.next()) {\r
+ Group group = new Group(rs);\r
+ list.add(group);\r
+ }\r
+ rs.close();\r
+ stmt.close();\r
+ db.release(conn);\r
+ } catch (SQLException e) {\r
+ e.printStackTrace();\r
+ }\r
+ return list;\r
+ }\r
+ public static int getMaxGroupID() {\r
+ int max = 0;\r
+ try {\r
+ DB db = new DB();\r
+ @SuppressWarnings("resource")\r
+ Connection conn = db.getConnection();\r
+ Statement stmt = conn.createStatement();\r
+ ResultSet rs = stmt.executeQuery("select MAX(groupid) from GROUPS");\r
+ if (rs.next()) {\r
+ max = rs.getInt(1);\r
+ }\r
+ rs.close();\r
+ stmt.close();\r
+ db.release(conn);\r
+ } catch (SQLException e) {\r
+ intlogger.info("getMaxSubID: "+e.getMessage());\r
+ e.printStackTrace();\r
+ }\r
+ return max;\r
+ }\r
+ public static Collection<String> getGroupsByClassfication(String classfication) {\r
+ List<String> list = new ArrayList<String>();\r
+ String sql = "select * from GROUPS where classification = '"+classfication+"'";\r
+ try {\r
+ DB db = new DB();\r
+ @SuppressWarnings("resource")\r
+ Connection conn = db.getConnection();\r
+ Statement stmt = conn.createStatement();\r
+ ResultSet rs = stmt.executeQuery(sql);\r
+ while (rs.next()) {\r
+ int groupid = rs.getInt("groupid");\r
+ //list.add(URLUtilities.generateSubscriptionURL(groupid));\r
+ }\r
+ rs.close();\r
+ stmt.close();\r
+ db.release(conn);\r
+ } catch (SQLException e) {\r
+ e.printStackTrace();\r
+ }\r
+ return list;\r
+ }\r
+ /**\r
+ * Return a count of the number of active subscriptions in the DB.\r
+ * @return the count\r
+ */\r
+ public static int countActiveSubscriptions() {\r
+ int count = 0;\r
+ try {\r
+ DB db = new DB();\r
+ @SuppressWarnings("resource")\r
+ Connection conn = db.getConnection();\r
+ Statement stmt = conn.createStatement();\r
+ ResultSet rs = stmt.executeQuery("select count(*) from SUBSCRIPTIONS");\r
+ if (rs.next()) {\r
+ count = rs.getInt(1);\r
+ }\r
+ rs.close();\r
+ stmt.close();\r
+ db.release(conn);\r
+ } catch (SQLException e) {\r
+ intlogger.warn("PROV0008 countActiveSubscriptions: "+e.getMessage());\r
+ e.printStackTrace();\r
+ }\r
+ return count;\r
+ }\r
+\r
+ public Group() {\r
+ this("", "", "");\r
+ }\r
+ public Group(String name, String desc, String members) {\r
+ this.groupid = -1;\r
+ this.authid = "";\r
+ this.name = name;\r
+ this.description = desc;\r
+ this.members = members;\r
+ this.classification = "";\r
+ this.last_mod = new Date();\r
+ }\r
+ \r
+ \r
+ public Group(ResultSet rs) throws SQLException {\r
+ this.groupid = rs.getInt("GROUPID");\r
+ this.authid = rs.getString("AUTHID");\r
+ this.name = rs.getString("NAME");\r
+ this.description = rs.getString("DESCRIPTION");\r
+ this.classification = rs.getString("CLASSIFICATION");\r
+ this.members = rs.getString("MEMBERS");\r
+ this.last_mod = rs.getDate("LAST_MOD");\r
+ }\r
+ \r
+\r
+ \r
+ public Group(JSONObject jo) throws InvalidObjectException {\r
+ this("", "", "");\r
+ try {\r
+ // The JSONObject is assumed to contain a vnd.att-dr.group representation\r
+ this.groupid = jo.optInt("groupid", -1);\r
+ String gname = jo.getString("name");\r
+ String gdescription = jo.getString("description");\r
+ \r
+ this.authid = jo.getString("authid");\r
+ this.name = gname;\r
+ this.description = gdescription;\r
+ this.classification = jo.getString("classification");\r
+ this.members = jo.getString("members");\r
+ \r
+ if (gname.length() > 50)\r
+ throw new InvalidObjectException("Group name is too long");\r
+ if (gdescription.length() > 256)\r
+ throw new InvalidObjectException("Group Description is too long");\r
+ } catch (InvalidObjectException e) {\r
+ throw e;\r
+ } catch (Exception e) {\r
+ throw new InvalidObjectException("invalid JSON: "+e.getMessage());\r
+ }\r
+ }\r
+ public int getGroupid() {\r
+ return groupid;\r
+ }\r
+ \r
+ public static Logger getIntlogger() {\r
+ return intlogger;\r
+ }\r
+ public void setGroupid(int groupid) {\r
+ this.groupid = groupid;\r
+ }\r
+ \r
+ public static void setIntlogger(Logger intlogger) {\r
+ Group.intlogger = intlogger;\r
+ }\r
+ public static int getNext_groupid() {\r
+ return next_groupid;\r
+ }\r
+ public static void setNext_groupid(int next_groupid) {\r
+ Group.next_groupid = next_groupid;\r
+ }\r
+ public String getAuthid() {\r
+ return authid;\r
+ }\r
+ public void setAuthid(String authid) {\r
+ this.authid = authid;\r
+ }\r
+ public String getName() {\r
+ return name;\r
+ }\r
+ public void setName(String name) {\r
+ this.name = name;\r
+ }\r
+ public String getDescription() {\r
+ return description;\r
+ }\r
+ public void setDescription(String description) {\r
+ this.description = description;\r
+ }\r
+ public String getClassification() {\r
+ return classification;\r
+ }\r
+ public void setClassification(String classification) {\r
+ this.classification = classification;\r
+ }\r
+ public String getMembers() {\r
+ return members;\r
+ }\r
+ public void setMembers(String members) {\r
+ this.members = members;\r
+ }\r
+ public Date getLast_mod() {\r
+ return last_mod;\r
+ }\r
+ public void setLast_mod(Date last_mod) {\r
+ this.last_mod = last_mod;\r
+ }\r
+ \r
+\r
+ @Override\r
+ public JSONObject asJSONObject() {\r
+ JSONObject jo = new JSONObject();\r
+ jo.put("groupid", groupid);\r
+ jo.put("authid", authid);\r
+ jo.put("name", name);\r
+ jo.put("description", description);\r
+ jo.put("classification", classification);\r
+ jo.put("members", members);\r
+ jo.put("last_mod", last_mod.getTime());\r
+ return jo;\r
+ }\r
+ @Override\r
+ public boolean doInsert(Connection c) {\r
+ boolean rv = true;\r
+ PreparedStatement ps = null;\r
+ try {\r
+ if (groupid == -1) {\r
+ // No feed ID assigned yet, so assign the next available one\r
+ setGroupid(next_groupid++);\r
+ }\r
+ // In case we insert a gropup from synchronization\r
+ if (groupid > next_groupid)\r
+ next_groupid = groupid+1;\r
+\r
+ \r
+ // Create the GROUPS row\r
+ String sql = "insert into GROUPS (GROUPID, AUTHID, NAME, DESCRIPTION, CLASSIFICATION, MEMBERS) values (?, ?, ?, ?, ?, ?)";\r
+ ps = c.prepareStatement(sql, new String[] { "GROUPID" });\r
+ ps.setInt(1, groupid);\r
+ ps.setString(2, authid);\r
+ ps.setString(3, name);\r
+ ps.setString(4, description);\r
+ ps.setString(5, classification);\r
+ ps.setString(6, members);\r
+ ps.execute();\r
+ ps.close();\r
+ } catch (SQLException e) {\r
+ rv = false;\r
+ intlogger.warn("PROV0005 doInsert: "+e.getMessage());\r
+ e.printStackTrace();\r
+ } finally {\r
+ try {\r
+ ps.close();\r
+ } catch (SQLException e) {\r
+ e.printStackTrace();\r
+ }\r
+ }\r
+ return rv;\r
+ }\r
+ @Override\r
+ public boolean doUpdate(Connection c) {\r
+ boolean rv = true;\r
+ PreparedStatement ps = null;\r
+ try {\r
+ String sql = "update GROUPS set AUTHID = ?, NAME = ?, DESCRIPTION = ?, CLASSIFICATION = ? , MEMBERS = ? where GROUPID = ?";\r
+ ps = c.prepareStatement(sql);\r
+ ps.setString(1, authid);\r
+ ps.setString(2, name);\r
+ ps.setString(3, description);\r
+ ps.setString(4, classification);\r
+ ps.setString(5, members);\r
+ ps.setInt(6, groupid);\r
+ ps.executeUpdate();\r
+ } catch (SQLException e) {\r
+ rv = false;\r
+ intlogger.warn("PROV0006 doUpdate: "+e.getMessage());\r
+ e.printStackTrace();\r
+ } finally {\r
+ try {\r
+ ps.close();\r
+ } catch (SQLException e) {\r
+ e.printStackTrace();\r
+ }\r
+ }\r
+ return rv;\r
+ }\r
+ @Override\r
+ public boolean doDelete(Connection c) {\r
+ boolean rv = true;\r
+ PreparedStatement ps = null;\r
+ try {\r
+ String sql = "delete from GROUPS where GROUPID = ?";\r
+ ps = c.prepareStatement(sql);\r
+ ps.setInt(1, groupid);\r
+ ps.execute();\r
+ } catch (SQLException e) {\r
+ rv = false;\r
+ intlogger.warn("PROV0007 doDelete: "+e.getMessage());\r
+ e.printStackTrace();\r
+ } finally {\r
+ try {\r
+ ps.close();\r
+ } catch (SQLException e) {\r
+ e.printStackTrace();\r
+ }\r
+ }\r
+ return rv;\r
+ }\r
+ @Override\r
+ public String getKey() {\r
+ return ""+getGroupid();\r
+ }\r
+ @Override\r
+ public boolean equals(Object obj) {\r
+ if (!(obj instanceof Group))\r
+ return false;\r
+ Group os = (Group) obj;\r
+ if (groupid != os.groupid)\r
+ return false;\r
+ if (authid != os.authid)\r
+ return false;\r
+ if (!name.equals(os.name))\r
+ return false;\r
+ if (description != os.description)\r
+ return false;\r
+ if (!classification.equals(os.classification))\r
+ return false;\r
+ if (!members.equals(os.members))\r
+ return false;\r
+ \r
+ return true;\r
+ }\r
+\r
+ @Override\r
+ public String toString() {\r
+ return "GROUP: groupid=" + groupid;\r
+ }\r
+}\r
Code Review / dmaap / datarouter.git / blobdiff