datarouter-prov/src/main/java/org/onap/dmaap/datarouter/provisioning/beans/Group.java

   1 /*******************************************************************************\r
   2  * ============LICENSE_START==================================================\r
   3  * * org.onap.dmaap\r
   4  * * ===========================================================================\r
   5  * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
   6  * * ===========================================================================\r
   7  * * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * * you may not use this file except in compliance with the License.\r
   9  * * You may obtain a copy of the License at\r
  10  * *\r
  11  *  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  * *\r
  13  *  * Unless required by applicable law or agreed to in writing, software\r
  14  * * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * * See the License for the specific language governing permissions and\r
  17  * * limitations under the License.\r
  18  * * ============LICENSE_END====================================================\r
  19  * *\r
  20  * * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
  21  * *\r
  22  ******************************************************************************/\r
  23 \r
  24 package org.onap.dmaap.datarouter.provisioning.beans;\r
  25 \r
  26 import java.io.InvalidObjectException;\r
  27 import java.sql.Connection;\r
  28 import java.sql.PreparedStatement;\r
  29 import java.sql.ResultSet;\r
  30 import java.sql.SQLException;\r
  31 import java.sql.Statement;\r
  32 import java.util.*;\r
  33 \r
  34 import org.apache.log4j.Logger;\r
  35 import org.json.JSONObject;\r
  36 import org.onap.dmaap.datarouter.provisioning.utils.DB;\r
  37 import org.onap.dmaap.datarouter.provisioning.utils.URLUtilities;\r
  38 \r
  39 /**\r
  40  * The representation of a Subscription.  Subscriptions can be retrieved from the DB, or stored/updated in the DB.\r
  41  *\r
  42  * @author vikram\r
  43  * @version $Id: Group.java,v 1.0 2016/07/19\r
  44  */\r
  45 public class Group extends Syncable {\r
  46 \r
  47     private static Logger intlogger = Logger.getLogger("org.onap.dmaap.datarouter.provisioning.internal");\r
  48     private static int next_groupid = getMaxGroupID() + 1;\r
  49 \r
  50     private int groupid;\r
  51     private String authid;\r
  52     private String name;\r
  53     private String description;\r
  54     private String classification;\r
  55     private String members;\r
  56     private Date last_mod;\r
  57 \r
  58 \r
  59     public static Group getGroupMatching(Group gup) {\r
  60         String sql = String.format(\r
  61                 "select * from GROUPS where NAME='%s'",\r
  62                 gup.getName()\r
  63         );\r
  64         List<Group> list = getGroupsForSQL(sql);\r
  65         return list.size() > 0 ? list.get(0) : null;\r
  66     }\r
  67 \r
  68     public static Group getGroupMatching(Group gup, int groupid) {\r
  69         String sql = String.format(\r
  70                 "select * from GROUPS where  NAME = '%s' and GROUPID != %d ",\r
  71                 gup.getName(),\r
  72                 gup.getGroupid()\r
  73         );\r
  74         List<Group> list = getGroupsForSQL(sql);\r
  75         return list.size() > 0 ? list.get(0) : null;\r
  76     }\r
  77 \r
  78     public static Group getGroupById(int id) {\r
  79         String sql = "select * from GROUPS where GROUPID = " + id;\r
  80         List<Group> list = getGroupsForSQL(sql);\r
  81         return list.size() > 0 ? list.get(0) : null;\r
  82     }\r
  83 \r
  84     public static Group getGroupByAuthId(String id) {\r
  85         String sql = "select * from GROUPS where AUTHID = '" + id + "'";\r
  86         List<Group> list = getGroupsForSQL(sql);\r
  87         return list.size() > 0 ? list.get(0) : null;\r
  88     }\r
  89 \r
  90     public static Collection<Group> getAllgroups() {\r
  91         return getGroupsForSQL("select * from GROUPS");\r
  92     }\r
  93 \r
  94     private static List<Group> getGroupsForSQL(String sql) {\r
  95         List<Group> list = new ArrayList<Group>();\r
  96         try {\r
  97             DB db = new DB();\r
  98             @SuppressWarnings("resource")\r
  99             Connection conn = db.getConnection();\r
 100             try (Statement stmt = conn.createStatement()) {\r
 101                 try (ResultSet rs = stmt.executeQuery(sql)) {\r
 102                     while (rs.next()) {\r
 103                         Group group = new Group(rs);\r
 104                         list.add(group);\r
 105                     }\r
 106                 }\r
 107             }\r
 108             db.release(conn);\r
 109         } catch (SQLException e) {\r
 110             intlogger.error("SQLException " + e.getMessage());\r
 111         }\r
 112         return list;\r
 113     }\r
 114 \r
 115     public static int getMaxGroupID() {\r
 116         int max = 0;\r
 117         try {\r
 118             DB db = new DB();\r
 119             @SuppressWarnings("resource")\r
 120             Connection conn = db.getConnection();\r
 121             try (Statement stmt = conn.createStatement()) {\r
 122                 try (ResultSet rs = stmt.executeQuery("select MAX(groupid) from GROUPS")) {\r
 123                     if (rs.next()) {\r
 124                         max = rs.getInt(1);\r
 125                     }\r
 126                 }\r
 127             }\r
 128             db.release(conn);\r
 129         } catch (SQLException e) {\r
 130             intlogger.info("getMaxSubID: " + e.getMessage());\r
 131         }\r
 132         return max;\r
 133     }\r
 134 \r
 135     public static Collection<String> getGroupsByClassfication(String classfication) {\r
 136         List<String> list = new ArrayList<>();\r
 137         String sql = "select * from GROUPS where classification = ?";\r
 138         try {\r
 139             DB db = new DB();\r
 140             @SuppressWarnings("resource")\r
 141             Connection conn = db.getConnection();\r
 142             try (PreparedStatement stmt = conn.prepareStatement(sql)) {\r
 143                 stmt.setString(1, classfication);\r
 144                 try (ResultSet rs = stmt.executeQuery()) {\r
 145                     while (rs.next()) {\r
 146                         int groupid = rs.getInt("groupid");\r
 147 \r
 148                     }\r
 149                 }\r
 150             }\r
 151             db.release(conn);\r
 152         } catch (SQLException e) {\r
 153             intlogger.error("SQLException " + e.getMessage());\r
 154         }\r
 155         return list;\r
 156     }\r
 157 \r
 158     /**\r
 159      * Return a count of the number of active subscriptions in the DB.\r
 160      *\r
 161      * @return the count\r
 162      */\r
 163     public static int countActiveSubscriptions() {\r
 164         int count = 0;\r
 165         try {\r
 166             DB db = new DB();\r
 167             @SuppressWarnings("resource")\r
 168             Connection conn = db.getConnection();\r
 169             try (Statement stmt = conn.createStatement()) {\r
 170                 try (ResultSet rs = stmt.executeQuery("select count(*) from SUBSCRIPTIONS")) {\r
 171                     if (rs.next()) {\r
 172                         count = rs.getInt(1);\r
 173                     }\r
 174                 }\r
 175             }\r
 176             db.release(conn);\r
 177         } catch (SQLException e) {\r
 178             intlogger.warn("PROV0008 countActiveSubscriptions: " + e.getMessage());\r
 179         }\r
 180         return count;\r
 181     }\r
 182 \r
 183     public Group() {\r
 184         this("", "", "");\r
 185     }\r
 186 \r
 187     public Group(String name, String desc, String members) {\r
 188         this.groupid = -1;\r
 189         this.authid = "";\r
 190         this.name = name;\r
 191         this.description = desc;\r
 192         this.members = members;\r
 193         this.classification = "";\r
 194         this.last_mod = new Date();\r
 195     }\r
 196 \r
 197 \r
 198     public Group(ResultSet rs) throws SQLException {\r
 199         this.groupid = rs.getInt("GROUPID");\r
 200         this.authid = rs.getString("AUTHID");\r
 201         this.name = rs.getString("NAME");\r
 202         this.description = rs.getString("DESCRIPTION");\r
 203         this.classification = rs.getString("CLASSIFICATION");\r
 204         this.members = rs.getString("MEMBERS");\r
 205         this.last_mod = rs.getDate("LAST_MOD");\r
 206     }\r
 207 \r
 208 \r
 209     public Group(JSONObject jo) throws InvalidObjectException {\r
 210         this("", "", "");\r
 211         try {\r
 212             // The JSONObject is assumed to contain a vnd.dmaap-dr.group representation\r
 213             this.groupid = jo.optInt("groupid", -1);\r
 214             String gname = jo.getString("name");\r
 215             String gdescription = jo.getString("description");\r
 216 \r
 217             this.authid = jo.getString("authid");\r
 218             this.name = gname;\r
 219             this.description = gdescription;\r
 220             this.classification = jo.getString("classification");\r
 221             this.members = jo.getString("members");\r
 222 \r
 223             if (gname.length() > 50) {\r
 224                 throw new InvalidObjectException("Group name is too long");\r
 225             }\r
 226             if (gdescription.length() > 256) {\r
 227                 throw new InvalidObjectException("Group Description is too long");\r
 228             }\r
 229         } catch (InvalidObjectException e) {\r
 230             throw e;\r
 231         } catch (Exception e) {\r
 232             throw new InvalidObjectException("invalid JSON: " + e.getMessage());\r
 233         }\r
 234     }\r
 235 \r
 236     public int getGroupid() {\r
 237         return groupid;\r
 238     }\r
 239 \r
 240     public static Logger getIntlogger() {\r
 241         return intlogger;\r
 242     }\r
 243 \r
 244     public void setGroupid(int groupid) {\r
 245         this.groupid = groupid;\r
 246     }\r
 247 \r
 248     public static void setIntlogger(Logger intlogger) {\r
 249         Group.intlogger = intlogger;\r
 250     }\r
 251 \r
 252     public static int getNext_groupid() {\r
 253         return next_groupid;\r
 254     }\r
 255 \r
 256     public static void setNext_groupid(int next_groupid) {\r
 257         Group.next_groupid = next_groupid;\r
 258     }\r
 259 \r
 260     public String getAuthid() {\r
 261         return authid;\r
 262     }\r
 263 \r
 264     public void setAuthid(String authid) {\r
 265         this.authid = authid;\r
 266     }\r
 267 \r
 268     public String getName() {\r
 269         return name;\r
 270     }\r
 271 \r
 272     public void setName(String name) {\r
 273         this.name = name;\r
 274     }\r
 275 \r
 276     public String getDescription() {\r
 277         return description;\r
 278     }\r
 279 \r
 280     public void setDescription(String description) {\r
 281         this.description = description;\r
 282     }\r
 283 \r
 284     public String getClassification() {\r
 285         return classification;\r
 286     }\r
 287 \r
 288     public void setClassification(String classification) {\r
 289         this.classification = classification;\r
 290     }\r
 291 \r
 292     public String getMembers() {\r
 293         return members;\r
 294     }\r
 295 \r
 296     public void setMembers(String members) {\r
 297         this.members = members;\r
 298     }\r
 299 \r
 300     public Date getLast_mod() {\r
 301         return last_mod;\r
 302     }\r
 303 \r
 304     public void setLast_mod(Date last_mod) {\r
 305         this.last_mod = last_mod;\r
 306     }\r
 307 \r
 308 \r
 309     @Override\r
 310     public JSONObject asJSONObject() {\r
 311         JSONObject jo = new JSONObject();\r
 312         jo.put("groupid", groupid);\r
 313         jo.put("authid", authid);\r
 314         jo.put("name", name);\r
 315         jo.put("description", description);\r
 316         jo.put("classification", classification);\r
 317         jo.put("members", members);\r
 318         jo.put("last_mod", last_mod.getTime());\r
 319         return jo;\r
 320     }\r
 321 \r
 322     @Override\r
 323     public boolean doInsert(Connection c) {\r
 324         boolean rv = true;\r
 325         PreparedStatement ps = null;\r
 326         try {\r
 327             if (groupid == -1) {\r
 328                 // No feed ID assigned yet, so assign the next available one\r
 329                 setGroupid(next_groupid++);\r
 330             }\r
 331             // In case we insert a gropup from synchronization\r
 332             if (groupid > next_groupid) {\r
 333                 next_groupid = groupid + 1;\r
 334             }\r
 335 \r
 336             // Create the GROUPS row\r
 337             String sql = "insert into GROUPS (GROUPID, AUTHID, NAME, DESCRIPTION, CLASSIFICATION, MEMBERS) values (?, ?, ?, ?, ?, ?)";\r
 338             ps = c.prepareStatement(sql, new String[]{"GROUPID"});\r
 339             ps.setInt(1, groupid);\r
 340             ps.setString(2, authid);\r
 341             ps.setString(3, name);\r
 342             ps.setString(4, description);\r
 343             ps.setString(5, classification);\r
 344             ps.setString(6, members);\r
 345             ps.execute();\r
 346             ps.close();\r
 347         } catch (SQLException e) {\r
 348             rv = false;\r
 349             intlogger.warn("PROV0005 doInsert: " + e.getMessage());\r
 350         } finally {\r
 351             try {\r
 352                 if (ps != null) {\r
 353                     ps.close();\r
 354                 }\r
 355             } catch (SQLException e) {\r
 356                 intlogger.error("SQLException " + e.getMessage());\r
 357             }\r
 358         }\r
 359         return rv;\r
 360     }\r
 361 \r
 362     @Override\r
 363     public boolean doUpdate(Connection c) {\r
 364         boolean rv = true;\r
 365         PreparedStatement ps = null;\r
 366         try {\r
 367             String sql = "update GROUPS set AUTHID = ?, NAME = ?, DESCRIPTION = ?, CLASSIFICATION = ? ,  MEMBERS = ? where GROUPID = ?";\r
 368             ps = c.prepareStatement(sql);\r
 369             ps.setString(1, authid);\r
 370             ps.setString(2, name);\r
 371             ps.setString(3, description);\r
 372             ps.setString(4, classification);\r
 373             ps.setString(5, members);\r
 374             ps.setInt(6, groupid);\r
 375             ps.executeUpdate();\r
 376         } catch (SQLException e) {\r
 377             rv = false;\r
 378             intlogger.warn("PROV0006 doUpdate: " + e.getMessage());\r
 379         } finally {\r
 380             try {\r
 381                 if (ps != null) {\r
 382                     ps.close();\r
 383                 }\r
 384             } catch (SQLException e) {\r
 385                 intlogger.error("SQLException " + e.getMessage());\r
 386             }\r
 387         }\r
 388         return rv;\r
 389     }\r
 390 \r
 391     @Override\r
 392     public boolean doDelete(Connection c) {\r
 393         boolean rv = true;\r
 394         PreparedStatement ps = null;\r
 395         try {\r
 396             String sql = "delete from GROUPS where GROUPID = ?";\r
 397             ps = c.prepareStatement(sql);\r
 398             ps.setInt(1, groupid);\r
 399             ps.execute();\r
 400         } catch (SQLException e) {\r
 401             rv = false;\r
 402             intlogger.warn("PROV0007 doDelete: " + e.getMessage());\r
 403         } finally {\r
 404             try {\r
 405                 if (ps != null) {\r
 406                     ps.close();\r
 407                 }\r
 408             } catch (SQLException e) {\r
 409                 intlogger.error("SQLException " + e.getMessage());\r
 410             }\r
 411         }\r
 412         return rv;\r
 413     }\r
 414 \r
 415     @Override\r
 416     public String getKey() {\r
 417         return "" + getGroupid();\r
 418     }\r
 419 \r
 420     @Override\r
 421     public boolean equals(Object obj) {\r
 422         if (!(obj instanceof Group)) {\r
 423             return false;\r
 424         }\r
 425         Group os = (Group) obj;\r
 426         if (groupid != os.groupid) {\r
 427             return false;\r
 428         }\r
 429         if (authid != os.authid) {\r
 430             return false;\r
 431         }\r
 432         if (!name.equals(os.name)) {\r
 433             return false;\r
 434         }\r
 435         if (description != os.description) {\r
 436             return false;\r
 437         }\r
 438         if (!classification.equals(os.classification)) {\r
 439             return false;\r
 440         }\r
 441         if (!members.equals(os.members)) {\r
 442             return false;\r
 443         }\r
 444 \r
 445         return true;\r
 446     }\r
 447 \r
 448     @Override\r
 449     public String toString() {\r
 450         return "GROUP: groupid=" + groupid;\r
 451     }\r
 452 \r
 453     @Override\r
 454     public int hashCode() {\r
 455         return Objects.hash(groupid, authid, name, description, classification, members, last_mod);\r
 456     }\r
 457 }\r