Update Batch from Testing
[aaf/authz.git] / auth / auth-batch / src / main / java / org / onap / aaf / auth / batch / actions / URPunt.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
22 package org.onap.aaf.auth.batch.actions;
23
24 import java.io.IOException;
25 import java.util.Date;
26 import java.util.List;
27
28 import org.onap.aaf.auth.batch.helpers.UserRole;
29 import org.onap.aaf.auth.dao.cass.UserRoleDAO;
30 import org.onap.aaf.auth.dao.cass.UserRoleDAO.Data;
31 import org.onap.aaf.auth.env.AuthzTrans;
32 import org.onap.aaf.auth.layer.Result;
33 import org.onap.aaf.misc.env.APIException;
34 import org.onap.aaf.misc.env.util.Chrono;
35
36 import com.datastax.driver.core.Cluster;
37
38 public class URPunt extends ActionPuntDAO<UserRole,Void,String> {
39     public URPunt(AuthzTrans trans, Cluster cluster, int months, int range, boolean dryRun) throws APIException, IOException {
40         super(trans,cluster, months, dryRun);
41     }
42
43     public URPunt(AuthzTrans trans, ActionDAO<?,?,?> adao, int months, int range) {
44         super(trans, adao, months);
45     }
46
47     public Result<Void> exec(AuthzTrans trans, UserRole ur, String text) {
48         if (dryRun) {
49             trans.info().log("Would Update User",ur.user(),"and Role", ur.role(), text);
50             return Result.ok();
51         } else {
52             Result<List<Data>> read = q.userRoleDAO.read(trans, ur.user(), ur.role());
53             if (read.isOK()) {
54                 for (UserRoleDAO.Data data : read.value) {
55                     Date from = data.expires;
56                     data.expires = puntDate(from);
57                     if (data.expires.compareTo(from)<=0) {
58                         trans.debug().printf("Error: %s is same or before %s", Chrono.dateOnlyStamp(data.expires), Chrono.dateOnlyStamp(from));
59                     } else {
60                         trans.info().log("Updating User",ur.user(),"and Role", ur.role(), "from",Chrono.dateOnlyStamp(from),"to",Chrono.dateOnlyStamp(data.expires), text);
61                         q.userRoleDAO.update(trans, data);
62                     }
63                 }
64                 return Result.ok();
65             } else {
66                 return Result.err(read);
67             }
68         }
69     }
70 }