2 * ============LICENSE_START====================================================
4 * ===========================================================================
5 * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
6 * ===========================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ============LICENSE_END====================================================
22 package org.onap.aaf.auth.dao.cass;
24 import java.nio.ByteBuffer;
25 import java.text.SimpleDateFormat;
26 import java.util.Date;
27 import java.util.List;
28 import java.util.UUID;
30 import org.onap.aaf.auth.dao.AbsCassDAO;
31 import org.onap.aaf.auth.dao.CassDAOImpl;
32 import org.onap.aaf.auth.dao.Loader;
33 import org.onap.aaf.auth.env.AuthzTrans;
34 import org.onap.aaf.auth.layer.Result;
36 import com.datastax.driver.core.Cluster;
37 import com.datastax.driver.core.ConsistencyLevel;
38 import com.datastax.driver.core.ResultSet;
39 import com.datastax.driver.core.Row;
44 * Originally written PE3617
47 * History is a special case, because we don't want Updates or Deletes... Too likely to mess up history.
49 * Jonathan 9-9-2013 - Found a problem with using "Prepare". You cannot prepare anything with a "now()" in it, as
50 * it is evaluated once during the prepare, and kept. That renders any use of "now()" pointless. Therefore
51 * the Create function needs to be run fresh everytime.
53 * Fixed in Cassandra 1.2.6 https://issues.apache.org/jira/browse/CASSANDRA-5616
56 public class HistoryDAO extends CassDAOImpl<AuthzTrans, HistoryDAO.Data> {
57 private static final String TABLE = "history";
59 private String[] helpers;
61 private HistLoader defLoader;
63 private AbsCassDAO<AuthzTrans, Data>.PSInfo readByUser, readBySubject, readByYRMN;
65 public HistoryDAO(AuthzTrans trans, Cluster cluster, String keyspace) {
66 super(trans, HistoryDAO.class.getSimpleName(),cluster,keyspace,Data.class,TABLE,ConsistencyLevel.LOCAL_ONE,ConsistencyLevel.ANY);
70 public HistoryDAO(AuthzTrans trans, AbsCassDAO<AuthzTrans,?> aDao) {
71 super(trans, HistoryDAO.class.getSimpleName(),aDao,Data.class,TABLE,ConsistencyLevel.LOCAL_ONE,ConsistencyLevel.ANY);
76 private static final int KEYLIMIT = 1;
77 public static class Data {
83 public String subject;
85 public ByteBuffer reconstruct;
88 public static class HistLoader extends Loader<Data> {
89 public HistLoader(int keylimit) {
94 public Data load(Data data, Row row) {
95 data.id = row.getUUID(0);
96 data.yr_mon = row.getInt(1);
97 data.user = row.getString(2);
98 data.action = row.getString(3);
99 data.target = row.getString(4);
100 data.subject = row.getString(5);
101 data.memo = row.getString(6);
102 data.reconstruct = row.getBytes(7);
107 protected void key(Data data, int idx, Object[] obj) {
112 protected void body(Data data, int _idx, Object[] obj) {
114 obj[idx]=data.yr_mon;
115 obj[++idx]=data.user;
116 obj[++idx]=data.action;
117 obj[++idx]=data.target;
118 obj[++idx]=data.subject;
119 obj[++idx]=data.memo;
120 obj[++idx]=data.reconstruct;
124 private void init(AuthzTrans trans) {
125 // Loader must match fields order
126 defLoader = new HistLoader(KEYLIMIT);
127 helpers = setCRUD(trans, TABLE, Data.class, defLoader);
129 // Need a specialty Creator to handle the "now()"
130 // 9/9/2013 - Jonathan - Just great... now() is evaluated once on Client side, invalidating usage (what point is a now() from a long time in the past?
131 // Unless this is fixed, we're putting in non-prepared statement
132 // Solved in Cassandra. Make sure you are running 1.2.6 Cassandra or later. https://issues.apache.org/jira/browse/CASSANDRA-5616
133 replace(CRUD.create, new PSInfo(trans, "INSERT INTO history (" + helpers[FIELD_COMMAS] +
134 ") VALUES(now(),?,?,?,?,?,?,?)",
137 protected void key(Data data, int idx, Object[] obj) {
142 replace(CRUD.read, new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] +
143 " FROM history WHERE id = ?", defLoader,readConsistency)
144 // new HistLoader(2) {
146 // protected void key(Data data, int idx, Object[] obj) {
147 // obj[idx]=data.yr_mon;
148 // obj[++idx]=data.id;
152 disable(CRUD.update);
153 disable(CRUD.delete);
155 readByUser = new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] +
156 " FROM history WHERE user = ?", defLoader,readConsistency);
157 readBySubject = new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] +
158 " FROM history WHERE subject = ? and target = ? ALLOW FILTERING", defLoader,readConsistency);
159 readByYRMN = new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] +
160 " FROM history WHERE yr_mon = ?", defLoader,readConsistency);
161 async(true); //TODO dropping messages with Async
164 public static Data newInitedData() {
165 Data data = new Data();
166 Date now = new Date();
167 // Sonar claims that SimpleDateFormat is not thread safe, so we can't be static
168 data.yr_mon = Integer.parseInt(new SimpleDateFormat("yyyyMM").format(now));
169 // data.day_time = Integer.parseInt(dayTimeFormat.format(now));
173 public void createBatch(StringBuilder sb, Data data) {
174 sb.append("INSERT INTO history (");
175 sb.append(helpers[FIELD_COMMAS]);
176 sb.append(") VALUES(now(),");
177 sb.append(data.yr_mon);
179 sb.append(data.user);
181 sb.append(data.action);
183 sb.append(data.target);
185 sb.append(data.subject);
187 sb.append(data.memo);
188 sb.append("',null);\n");
191 public Result<List<Data>> readByYYYYMM(AuthzTrans trans, int yyyymm) {
192 Result<ResultSet> rs = readByYRMN.exec(trans, "yr_mon", yyyymm);
194 return Result.err(rs);
196 return extract(defLoader,rs.value,null,dflt);
200 * Gets the history for a user in the specified year and month
201 * year - the year in yyyy format
202 * month - the month in a year ...values 1 - 12
204 public Result<List<Data>> readByUser(AuthzTrans trans, String user, int ... yyyymm) {
205 if (yyyymm.length==0) {
206 return Result.err(Status.ERR_BadData, "No or invalid yyyymm specified");
208 Result<ResultSet> rs = readByUser.exec(trans, "user", user);
210 return Result.err(rs);
212 return extract(defLoader,rs.value,null,yyyymm.length>0?new YYYYMM(yyyymm):dflt);
215 public Result<List<Data>> readBySubject(AuthzTrans trans, String subject, String target, int ... yyyymm) {
216 if (yyyymm.length==0) {
217 return Result.err(Status.ERR_BadData, "No or invalid yyyymm specified");
219 Result<ResultSet> rs = readBySubject.exec(trans, "subject", subject, target);
221 return Result.err(rs);
223 return extract(defLoader,rs.value,null,yyyymm.length>0?new YYYYMM(yyyymm):dflt);
226 private class YYYYMM implements Accept<Data> {
227 private int[] yyyymm;
228 public YYYYMM(int[] yyyymm) {
229 this.yyyymm = yyyymm;
232 public boolean ok(Data data) {
233 int dym = data.yr_mon;
234 for (int ym:yyyymm) {