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 private 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.detail;
121 obj[++idx]=data.reconstruct;
125 private void init(AuthzTrans trans) {
126 // Loader must match fields order
127 defLoader = new HistLoader(KEYLIMIT);
128 helpers = setCRUD(trans, TABLE, Data.class, defLoader);
130 // Need a specialty Creator to handle the "now()"
131 // 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?
132 // Unless this is fixed, we're putting in non-prepared statement
133 // Solved in Cassandra. Make sure you are running 1.2.6 Cassandra or later. https://issues.apache.org/jira/browse/CASSANDRA-5616
134 replace(CRUD.create, new PSInfo(trans, "INSERT INTO history (" + helpers[FIELD_COMMAS] +
135 ") VALUES(now(),?,?,?,?,?,?,?)",
138 protected void key(Data data, int idx, Object[] obj) {
142 // disable(CRUD.Create);
144 replace(CRUD.read, new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] +
145 " FROM history WHERE id = ?", defLoader,readConsistency)
146 // new HistLoader(2) {
148 // protected void key(Data data, int idx, Object[] obj) {
149 // obj[idx]=data.yr_mon;
150 // obj[++idx]=data.id;
154 disable(CRUD.update);
155 disable(CRUD.delete);
157 readByUser = new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] +
158 " FROM history WHERE user = ?", defLoader,readConsistency);
159 readBySubject = new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] +
160 " FROM history WHERE subject = ? and target = ? ALLOW FILTERING", defLoader,readConsistency);
161 readByYRMN = new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] +
162 " FROM history WHERE yr_mon = ?", defLoader,readConsistency);
163 async(true); //TODO dropping messages with Async
166 public static Data newInitedData() {
167 Data data = new Data();
168 Date now = new Date();
169 // Sonar claims that SimpleDateFormat is not thread safe, so we can't be static
170 data.yr_mon = Integer.parseInt(new SimpleDateFormat("yyyyMM").format(now));
171 // data.day_time = Integer.parseInt(dayTimeFormat.format(now));
175 public Result<List<Data>> readByYYYYMM(AuthzTrans trans, int yyyymm) {
176 Result<ResultSet> rs = readByYRMN.exec(trans, "yr_mon", yyyymm);
178 return Result.err(rs);
180 return extract(defLoader,rs.value,null,dflt);
184 * Gets the history for a user in the specified year and month
185 * year - the year in yyyy format
186 * month - the month in a year ...values 1 - 12
188 public Result<List<Data>> readByUser(AuthzTrans trans, String user, int ... yyyymm) {
189 if (yyyymm.length==0) {
190 return Result.err(Status.ERR_BadData, "No or invalid yyyymm specified");
192 Result<ResultSet> rs = readByUser.exec(trans, "user", user);
194 return Result.err(rs);
196 return extract(defLoader,rs.value,null,yyyymm.length>0?new YYYYMM(yyyymm):dflt);
199 public Result<List<Data>> readBySubject(AuthzTrans trans, String subject, String target, int ... yyyymm) {
200 if (yyyymm.length==0) {
201 return Result.err(Status.ERR_BadData, "No or invalid yyyymm specified");
203 Result<ResultSet> rs = readBySubject.exec(trans, "subject", subject, target);
205 return Result.err(rs);
207 return extract(defLoader,rs.value,null,yyyymm.length>0?new YYYYMM(yyyymm):dflt);
210 private class YYYYMM implements Accept<Data> {
211 private int[] yyyymm;
212 public YYYYMM(int yyyymm[]) {
213 this.yyyymm = yyyymm;
216 public boolean ok(Data data) {
217 int dym = data.yr_mon;
218 for (int ym:yyyymm) {