e8961f14c3b9fe24f63759088a887ea1c0e5de71
[aaf/authz.git] / auth / auth-batch / src / main / java / org / onap / aaf / auth / batch / approvalsets / ApprovalSet.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.approvalsets;
22
23 import java.nio.ByteBuffer;
24 import java.util.ArrayList;
25 import java.util.GregorianCalendar;
26 import java.util.HashSet;
27 import java.util.List;
28 import java.util.Set;
29
30 import org.onap.aaf.auth.dao.cass.ApprovalDAO;
31 import org.onap.aaf.auth.dao.cass.FutureDAO;
32 import org.onap.aaf.auth.env.AuthzTrans;
33 import org.onap.aaf.auth.layer.Result;
34 import org.onap.aaf.misc.env.util.Chrono;
35
36 public class ApprovalSet {
37     private DataView dataview;
38     protected FutureDAO.Data fdd;
39     protected List<ApprovalDAO.Data> ladd;
40     
41     public ApprovalSet(final GregorianCalendar start, final String target, final DataView dv) {
42         dataview = dv;
43         fdd = new FutureDAO.Data();
44         fdd.id = Chrono.dateToUUID(System.currentTimeMillis());
45         fdd.target = target;
46         fdd.start = start.getTime();
47         ladd = new ArrayList<>();
48     }
49     
50     protected void setConstruct(final ByteBuffer bytes) {
51         fdd.construct = bytes;
52     }
53
54     protected void setMemo(final String memo) {
55         fdd.memo = memo;
56     }
57     
58     protected void setExpires(final GregorianCalendar expires) {
59         fdd.expires = expires.getTime();
60     }
61     
62     public Result<Void> write(AuthzTrans trans) {
63         StringBuilder errs = null;
64         if(ladd == null || ladd.isEmpty()) {
65             errs = new StringBuilder("No Approvers for ");
66             errs .append(fdd.memo);
67         } else {
68             Result<FutureDAO.Data> rf = dataview.insert(trans, fdd);
69             if(rf.notOK()) {
70                 errs = new StringBuilder();
71                 errs.append(rf.errorString());
72             } else {
73                 for(ApprovalDAO.Data add : ladd) {
74                     Result<ApprovalDAO.Data> af = dataview.insert(trans, add);
75                     if(af.notOK()) {
76                         if(errs==null) {
77                             errs = new StringBuilder();
78                         } else {
79                             errs.append('\n');
80                         }
81                         errs.append(af.errorString());
82                     }
83                 }
84             }
85         }
86         return errs==null?Result.ok():Result.err(Result.ERR_Backend,errs.toString());
87     }
88
89     public boolean hasApprovals() {
90         return !ladd.isEmpty();
91     }
92     
93     public Set<String> approvers() {
94         Set<String> rv = new HashSet<>();
95         for(ApprovalDAO.Data app : ladd) {
96             rv.add(app.approver);
97         }
98         return rv;
99     }
100 }