auth/auth-cass/src/main/java/org/onap/aaf/auth/dao/Loader.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.dao;
  23
  24 import java.io.DataInputStream;
  25 import java.io.DataOutputStream;
  26 import java.io.IOException;
  27 import java.util.ArrayList;
  28 import java.util.Collection;
  29 import java.util.HashMap;
  30 import java.util.HashSet;
  31 import java.util.List;
  32 import java.util.Map;
  33 import java.util.Map.Entry;
  34 import java.util.Set;
  35
  36 import com.datastax.driver.core.Row;
  37
  38 public abstract class Loader<DATA> {
  39     private int keylimit;
  40     public Loader(int keylimit) {
  41         this.keylimit = keylimit;
  42     }
  43
  44     public int keylimit() {
  45         return keylimit;
  46     }
  47
  48     protected abstract DATA load(DATA data, Row row);
  49     protected abstract void key(DATA data, int idx, Object[] obj);
  50     protected abstract void body(DATA data, int idx, Object[] obj);
  51
  52     public final Object[] extract(DATA data, int size, CassDAOImpl.CRUD type) {
  53         Object[] rv=null;
  54         switch(type) {
  55             case delete:
  56                 rv = new Object[keylimit()];
  57                 key(data,0,rv);
  58                 break;
  59             case update:
  60                 rv = new Object[size];
  61                 body(data,0,rv);
  62                 int body = size-keylimit();
  63                 if (body>0) {
  64                     key(data,body,rv);
  65                 }
  66                 break;
  67             default:
  68                 rv = new Object[size];
  69                 key(data,0,rv);
  70                 if (size>keylimit()) {
  71                     body(data,keylimit(),rv);
  72                 }
  73                 break;
  74         }
  75         return rv;
  76     }
  77
  78     public static void writeString(DataOutputStream os, String s) throws IOException {
  79         if (s==null) {
  80             os.writeInt(-1);
  81         } else {
  82             switch(s.length()) {
  83                 case 0:
  84                     os.writeInt(0);
  85                     break;
  86                 default:
  87                     byte[] bytes = s.getBytes();
  88                     os.writeInt(bytes.length);
  89                     os.write(bytes);
  90             }
  91         }
  92     }
  93
  94
  95     /**
  96      * We use bytes here to set a Maximum
  97      *
  98      * @param is
  99      * @param MAX
 100      * @return
 101      * @throws IOException
 102      */
 103     public static String readString(DataInputStream is, byte[] _buff) throws IOException {
 104         int l = is.readInt();
 105         byte[] buff = _buff;
 106         switch(l) {
 107             case -1: return null;
 108             case  0: return "";
 109             default:
 110                 // Cover case where there is a large string, without always allocating a large buffer.
 111                 if (l>buff.length) {
 112                     buff = new byte[l];
 113                 }
 114                 is.read(buff,0,l);
 115                 return new String(buff,0,l);
 116         }
 117     }
 118
 119     /**
 120      * Write a set with proper sizing
 121      *
 122      * Note: at the moment, this is just String.  Probably can develop system where types
 123      * are supported too... but not now.
 124      *
 125      * @param os
 126      * @param set
 127      * @throws IOException
 128      */
 129     public static void writeStringSet(DataOutputStream os, Collection<String> set) throws IOException {
 130         if (set==null) {
 131             os.writeInt(-1);
 132         } else {
 133             os.writeInt(set.size());
 134             for (String s : set) {
 135                 writeString(os, s);
 136             }
 137         }
 138
 139     }
 140
 141     public static Set<String> readStringSet(DataInputStream is, byte[] buff) throws IOException {
 142         int l = is.readInt();
 143         if (l<0) {
 144             return null;
 145         }
 146         Set<String> set = new HashSet<>(l);
 147         for (int i=0;i<l;++i) {
 148             set.add(readString(is,buff));
 149         }
 150         return set;
 151     }
 152
 153     public static List<String> readStringList(DataInputStream is, byte[] buff) throws IOException {
 154         int l = is.readInt();
 155         if (l<0) {
 156             return null;
 157         }
 158         List<String> list = new ArrayList<>(l);
 159         for (int i=0;i<l;++i) {
 160             list.add(Loader.readString(is,buff));
 161         }
 162         return list;
 163     }
 164
 165     /**
 166      * Write a map
 167      * @param os
 168      * @param map
 169      * @throws IOException
 170      */
 171     public static void writeStringMap(DataOutputStream os, Map<String,String> map) throws IOException {
 172         if (map==null) {
 173             os.writeInt(-1);
 174         } else {
 175             Set<Entry<String, String>> es = map.entrySet();
 176             os.writeInt(es.size());
 177             for (Entry<String,String> e : es) {
 178                 writeString(os, e.getKey());
 179                 writeString(os, e.getValue());
 180             }
 181         }
 182
 183     }
 184
 185     public static Map<String,String> readStringMap(DataInputStream is, byte[] buff) throws IOException {
 186         int l = is.readInt();
 187         if (l<0) {
 188             return null;
 189         }
 190         Map<String,String> map = new HashMap<>(l);
 191         for (int i=0;i<l;++i) {
 192             String key = readString(is,buff);
 193             map.put(key,readString(is,buff));
 194         }
 195         return map;
 196     }
 197     public static void writeHeader(DataOutputStream os, int magic, int version) throws IOException {
 198         os.writeInt(magic);
 199         os.writeInt(version);
 200     }
 201
 202     public static int readHeader(DataInputStream is, final int magic, final int version) throws IOException {
 203         if (is.readInt()!=magic) {
 204             throw new IOException("Corrupted Data Stream");
 205         }
 206         int v = is.readInt();
 207         if (version<0 || v>version) {
 208             throw new IOException("Unsupported Data Version: " + v);
 209         }
 210         return v;
 211     }
 212
 213 }
 214