AT&T 2.0.19 Code drop, stage 3

[aaf/authz.git] / auth / auth-batch / src / main / java / org / onap / aaf / auth / helpers / MonthData.java

diff --git a/auth/auth-batch/src/main/java/org/onap/aaf/auth/helpers/MonthData.java b/auth/auth-batch/src/main/java/org/onap/aaf/auth/helpers/MonthData.java
new file mode 100644 (file)
index 0000000..13a4c92
--- /dev/null
+++ b/auth/auth-batch/src/main/java/org/onap/aaf/auth/helpers/MonthData.java
@@ -0,0 +1,121 @@
+/**
+ * ============LICENSE_START====================================================
+ * org.onap.aaf
+ * ===========================================================================
+ * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
+ * ===========================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END====================================================
+ *
+ */
+
+package org.onap.aaf.auth.helpers;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+import java.io.PrintStream;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import org.onap.aaf.misc.env.util.Split;
+
+import java.util.Set;
+import java.util.TreeMap;
+
+public class MonthData {
+    public final Map<Integer,Set<Row>> data = 
+               new TreeMap<Integer,Set<Row>>();
+       private File f;
+    
+    public MonthData(String env) throws IOException {
+       f = new File("Monthly"+env+".dat");
+       
+       if(f.exists()) {
+               BufferedReader br = new BufferedReader(new FileReader(f));
+               try {
+                       String line;
+                       String[] split;
+                       while((line=br.readLine())!=null) {
+                               if(!line.startsWith("#")) {
+                                       split = Split.split(',', line);
+                                       if(split.length==5) {
+                                               add(Integer.parseInt(split[0]),split[1],
+                                                       Integer.parseInt(split[2]),
+                                                       Integer.parseInt(split[3]),
+                                                       Integer.parseInt(split[4])
+                                               );
+                                       }
+                               }
+                       }
+               } finally {
+                       br.close();
+               }
+       }
+    }
+    
+    public void add(int yr_mon, String target, long total, long adds, long drops) {
+               Set<Row> row = data.get(yr_mon);
+               if(row==null) {
+                       data.put(yr_mon, (row=new HashSet<Row>()));
+               }
+               row.add(new Row(target,total,adds,drops));
+       }
+    
+    public boolean notExists(int yr_mon) {
+       return data.get(yr_mon)==null;
+    }
+    
+       public static class Row implements Comparable<Row> {
+       public final String target;
+       public final long total;
+       public final long adds;
+       public final long drops;
+       
+       public Row(String t, long it, long a, long d) {
+               target = t;
+               total = it;
+               adds = a;
+               drops = d;
+       }
+
+               @Override
+               public int compareTo(Row o) {
+                       return target.compareTo(o.target);
+               }
+               
+               public String toString() {
+                       return target + '|' + total + '|' + drops + '|' + adds;
+               }
+    }
+
+    public void write() throws IOException {
+       if(f.exists()) {
+               File bu = new File(f.getName()+".bak");
+               f.renameTo(bu);
+       }
+               PrintStream ps = new PrintStream(f);
+               try {
+                       for( Entry<Integer, Set<Row>> rows : data.entrySet()) {
+                               for(Row row : rows.getValue()) {
+                                       ps.printf("%d,%s,%d,%d,%d\n",rows.getKey(),row.target,row.total,row.adds,row.drops);
+                               }
+                       }
+               } finally {
+                       ps.close();
+               }
+    }
+
+}