auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/BatchDataView.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   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
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  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====================================================
  19  *
  20  */
  21 package org.onap.aaf.auth.batch.helpers;
  22
  23 import java.io.IOException;
  24 import java.util.ArrayList;
  25 import java.util.List;
  26
  27 import org.onap.aaf.auth.batch.approvalsets.DataView;
  28 import org.onap.aaf.auth.dao.cass.ApprovalDAO;
  29 import org.onap.aaf.auth.dao.cass.FutureDAO;
  30 import org.onap.aaf.auth.dao.cass.NsDAO;
  31 import org.onap.aaf.auth.dao.cass.RoleDAO;
  32 import org.onap.aaf.auth.dao.cass.UserRoleDAO;
  33 import org.onap.aaf.auth.dao.cass.UserRoleDAO.Data;
  34 import org.onap.aaf.auth.env.AuthzTrans;
  35 import org.onap.aaf.auth.layer.Result;
  36 import org.onap.aaf.cadi.Hash;
  37 import org.onap.aaf.misc.env.APIException;
  38 import org.onap.aaf.misc.env.TimeTaken;
  39 import org.onap.aaf.misc.env.Trans;
  40 import org.onap.aaf.misc.env.util.Chrono;
  41
  42 import com.datastax.driver.core.Session;
  43
  44 public class BatchDataView implements DataView {
  45     private static final String QUOTE_PAREN_SEMI = "');\n";
  46     private static final String QUOTE_COMMA = "',";
  47     private static final String QUOTE_COMMA_QUOTE = "','";
  48     private static final String COMMA_QUOTE = ",'";
  49     private final CQLBatchLoop cqlBatch;
  50     private final Session session;
  51
  52     public BatchDataView(final AuthzTrans trans, final Session session, final boolean dryRun ){
  53         this.session = session;
  54         cqlBatch = new CQLBatchLoop(new CQLBatch(trans.info(),session),50,dryRun);
  55     }
  56
  57     public Session getSession(AuthzTrans trans){
  58         return session;
  59     }
  60
  61     public Result<NsDAO.Data> ns(AuthzTrans trans, String id) {
  62         NS n;
  63         TimeTaken tt = trans.start("Get NS by ID %s", Trans.SUB, id);
  64         try {
  65             n=NS.data.get(id);
  66         } finally {
  67             tt.done();
  68         }
  69
  70         if(n==null || n.ndd==null) {
  71             return Result.err(Result.ERR_Backend,"Namespace '%s' does not exist", id);
  72         }
  73         return Result.ok(n.ndd);
  74     }
  75
  76
  77     @Override
  78     public Result<RoleDAO.Data> roleByName(AuthzTrans trans, String name) {
  79         Role r = Role.byName.get(name);
  80         if(r==null || r.rdd==null) {
  81             return Result.err(Result.ERR_Backend,"Role '%s' does not exist", name);
  82         }
  83         return Result.ok(r.rdd);
  84     }
  85
  86     @Override
  87     public Result<List<UserRoleDAO.Data>> ursByRole(AuthzTrans trans, String role) {
  88         List<UserRole> urs = UserRole.getByRole().get(role);
  89         if(urs==null) {
  90             return Result.err(Result.ERR_Backend, "UserRoles for Role '%s' does not exist", role);
  91         }
  92         return toLURDD(urs);
  93     }
  94
  95     private Result<List<Data>> toLURDD(List<UserRole> urs) {
  96         List<UserRoleDAO.Data> rv = new ArrayList<>();
  97         if(urs!=null) {
  98             for(UserRole ur : urs) {
  99                 rv.add(ur.urdd());
 100             }
 101         }
 102         return Result.ok(rv);
 103     }
 104
 105     @Override
 106     public Result<List<UserRoleDAO.Data>> ursByUser(AuthzTrans trans, String user) {
 107         List<UserRole> urs = UserRole.getByUser().get(user);
 108         if(urs==null) {
 109             return Result.err(Result.ERR_Backend, "UserRoles for User '%s' does not exist", user);
 110         }
 111         return toLURDD(urs);
 112     }
 113
 114     @Override
 115     public Result<FutureDAO.Data> delete(AuthzTrans trans, FutureDAO.Data fdd) {
 116         StringBuilder sb = cqlBatch.inc();
 117         sb.append("DELETE from authz.future WHERE id = ");
 118         sb.append(fdd.id.toString());
 119         return Result.ok(fdd);
 120     }
 121
 122     @Override
 123     public Result<ApprovalDAO.Data> delete(AuthzTrans trans, ApprovalDAO.Data add) {
 124         StringBuilder sb = cqlBatch.inc();
 125         sb.append("DELETE from authz.approval WHERE id = ");
 126         sb.append(add.id.toString());
 127         return Result.ok(add);
 128     }
 129
 130
 131     @Override
 132     public Result<ApprovalDAO.Data> insert(AuthzTrans trans, ApprovalDAO.Data add) {
 133         StringBuilder sb = cqlBatch.inc();
 134         sb.append("INSERT INTO authz.approval (id,approver,memo,operation,status,ticket,type,user) VALUES (");
 135         sb.append(add.id.toString());
 136         sb.append(COMMA_QUOTE);
 137         sb.append(add.approver);
 138 //        sb.append(QUOTE_COMMA_QUOTE);
 139 //        sb.append(Chrono.utcStamp(add.last_notified));
 140         sb.append(QUOTE_COMMA_QUOTE);
 141         sb.append(add.memo.replace("'", "''"));
 142         sb.append(QUOTE_COMMA_QUOTE);
 143         sb.append(add.operation);
 144         sb.append(QUOTE_COMMA_QUOTE);
 145         sb.append(add.status);
 146         sb.append(QUOTE_COMMA);
 147         sb.append(add.ticket.toString());
 148         sb.append(COMMA_QUOTE);
 149         sb.append(add.type);
 150         sb.append(QUOTE_COMMA_QUOTE);
 151         sb.append(add.user);
 152         sb.append(QUOTE_PAREN_SEMI);
 153         return Result.ok(add);
 154     }
 155
 156     @Override
 157     public Result<FutureDAO.Data> insert(AuthzTrans trans, FutureDAO.Data fdd) {
 158         StringBuilder sb = cqlBatch.inc();
 159         sb.append("INSERT INTO authz.future (id,construct,expires,memo,start,target,target_key,target_date) VALUES (");
 160         sb.append(fdd.id.toString());
 161         sb.append(',');
 162         fdd.construct.hasArray();
 163         sb.append(Hash.toHex(fdd.construct.array()));
 164         sb.append(COMMA_QUOTE);
 165         sb.append(Chrono.utcStamp(fdd.expires));
 166         sb.append(QUOTE_COMMA_QUOTE);
 167         sb.append(fdd.memo.replace("'", "''"));
 168         sb.append(QUOTE_COMMA_QUOTE);
 169         sb.append(Chrono.utcStamp(fdd.expires));
 170         sb.append(QUOTE_COMMA_QUOTE);
 171         sb.append(fdd.target);
 172         if(fdd.target_key==null) {
 173             sb.append("',,'");
 174         } else {
 175             sb.append(QUOTE_COMMA_QUOTE);
 176             sb.append(fdd.target_key==null?"":fdd.target_key);
 177             sb.append(QUOTE_COMMA_QUOTE);
 178         }
 179         sb.append(Chrono.utcStamp(fdd.target_date));
 180         sb.append(QUOTE_PAREN_SEMI);
 181         return Result.ok(fdd);
 182     }
 183
 184     @Override
 185     public void flush() {
 186         cqlBatch.flush();
 187     }
 188 }