misc/env/src/main/java/org/onap/aaf/misc/env/StoreImpl.java

   1 /**\r
   2  * ============LICENSE_START====================================================\r
   3  * org.onap.aaf\r
   4  * ===========================================================================\r
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.\r
   6  * ===========================================================================\r
   7  * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * you may not use this file except in compliance with the License.\r
   9  * You may obtain a copy of the License at\r
  10  * \r
  11  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  * \r
  13  * Unless required by applicable law or agreed to in writing, software\r
  14  * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * See the License for the specific language governing permissions and\r
  17  * limitations under the License.\r
  18  * ============LICENSE_END====================================================\r
  19  *\r
  20  */\r
  21 \r
  22 package org.onap.aaf.misc.env;\r
  23 \r
  24 import java.io.File;\r
  25 import java.io.FileInputStream;\r
  26 import java.io.IOException;\r
  27 import java.lang.reflect.GenericArrayType;\r
  28 import java.util.ArrayList;\r
  29 import java.util.HashMap;\r
  30 import java.util.List;\r
  31 import java.util.Map.Entry;\r
  32 \r
  33 import org.onap.aaf.misc.env.util.Split;\r
  34 \r
  35 import java.util.Properties;\r
  36 \r
  37 \r
  38 public class StoreImpl implements Store {\r
  39         /*\r
  40          * The re-adjustment factor for growing the Static State array. \r
  41          */\r
  42         private static final int growSize = 10;\r
  43         \r
  44         /*\r
  45          * The index reference for Slot assignment.\r
  46          */\r
  47         private int local;\r
  48         \r
  49         /*\r
  50          * The index reference for StaticSlot assignment. \r
  51          */\r
  52         private int stat;\r
  53         \r
  54         /*\r
  55          * The name/slot map for local (transaction specific) State.\r
  56          */\r
  57         private HashMap<String, Slot> localMap;\r
  58         \r
  59         /*\r
  60          * The name/slot map for Static State.\r
  61          */\r
  62         private HashMap<String, StaticSlot> staticMap;\r
  63 \r
  64         private Object[] staticState;\r
  65         \r
  66         public StoreImpl() {\r
  67                  staticState = new Object[growSize];\r
  68                  staticMap = new HashMap<>();\r
  69                  localMap = new HashMap<>();\r
  70         }\r
  71         \r
  72         public StoreImpl(String tag) {\r
  73                  staticState = new Object[growSize];\r
  74                  staticMap = new HashMap<>();\r
  75                  localMap = new HashMap<>();\r
  76         }\r
  77 \r
  78         \r
  79         public StoreImpl(String tag, String[] args) {\r
  80                  staticState = new Object[growSize];\r
  81                  staticMap = new HashMap<>();\r
  82                  localMap = new HashMap<>();\r
  83 \r
  84                  if(tag!=null) {\r
  85                         String tequals = tag + '=';\r
  86                         for(String arg : args) {\r
  87                                 if(arg.startsWith(tequals) && !arg.equals(tequals)) { // needs to have something after =\r
  88                                         Properties props = new Properties();\r
  89                                         for(String f : Split.split(File.pathSeparatorChar,arg.substring(tequals.length()))) {\r
  90                                                 moreProps(new File(f),props);\r
  91                                         }\r
  92                                         for(Entry<Object, Object> es : props.entrySet()) {\r
  93                                                 put(staticSlot(es.getKey().toString()),es.getValue());\r
  94                                         }\r
  95                                 }\r
  96                         }\r
  97                  }\r
  98 \r
  99                 // Make sure properties on command line override those in Props\r
 100                 propsFromArgs(tag,args);\r
 101         }\r
 102         \r
 103         public StoreImpl(String tag, Properties props) {\r
 104                  staticState = new Object[growSize];\r
 105                  staticMap = new HashMap<>();\r
 106                  localMap = new HashMap<>();\r
 107                  \r
 108                  if(tag!=null) {\r
 109                          String fname = props.getProperty(tag);\r
 110                          if(fname!=null) {\r
 111                                  for(String f : Split.split(File.pathSeparatorChar,fname)) {\r
 112                                          if(!moreProps(new File(f),props)) {\r
 113                                                 System.err.println("Unable to load Properties from " + f); \r
 114                                          }\r
 115                                  }\r
 116                          }\r
 117                  }\r
 118 \r
 119                  for(Entry<Object, Object> es : props.entrySet()) {\r
 120                          put(staticSlot(es.getKey().toString()),es.getValue());\r
 121                  }\r
 122         }\r
 123 \r
 124         public void propsFromArgs(String tag, String[] args) {\r
 125                 if(tag!=null) {\r
 126                         for(String arg : args) {\r
 127                                 String sarg[] = Split.split('=',arg);\r
 128                                 if(sarg.length==2) {\r
 129                                         if(tag.equals(sarg[0])) {\r
 130                                                 for(String fname : Split.split(File.pathSeparatorChar,sarg[1])) {\r
 131                                                         moreProps(new File(fname),null /* no target */);\r
 132                                                 }\r
 133                                         }\r
 134                                         put(staticSlot(sarg[0]),sarg[1]);\r
 135                                 }\r
 136                         }\r
 137                 }\r
 138         }\r
 139 \r
 140         private boolean moreProps(File f, Properties target) {\r
 141                  if(f.exists()) {\r
 142                          Properties props = new Properties();\r
 143                          try {\r
 144                                  FileInputStream fis = new FileInputStream(f);\r
 145                                  try {\r
 146                                          props.load(fis);\r
 147                                          if(target!=null) {\r
 148                                                  target.load(fis);\r
 149                                          }\r
 150                                  } finally {\r
 151                                          fis.close();\r
 152                                  }\r
 153                          } catch(IOException e) {\r
 154                                  System.err.println(e);\r
 155                          }\r
 156                          for(Entry<Object, Object> es : props.entrySet()) {\r
 157                                  put(staticSlot(es.getKey().toString()),es.getValue());\r
 158                          }\r
 159                          return true;\r
 160                  } else {\r
 161                          return false;\r
 162                  }\r
 163         }\r
 164 \r
 165         public Object[] newTransState() {\r
 166                 return new Object[local];\r
 167         }\r
 168 \r
 169         /* (non-Javadoc)\r
 170          * @see com.att.env.Store#slot(java.lang.String)\r
 171          */\r
 172         public synchronized Slot slot(String name) {\r
 173                 name = name == null ? "" : name.trim();\r
 174                 Slot slot = localMap.get(name);\r
 175                 if (slot == null)  {\r
 176                         slot = new Slot(local++, name);\r
 177                         localMap.put(name, slot);\r
 178                 }\r
 179                 return slot;\r
 180         }\r
 181         \r
 182         \r
 183         /* (non-Javadoc)\r
 184          * @see com.att.env.Store#existingSlot(java.lang.String)\r
 185          */\r
 186         public Slot existingSlot(String name) {\r
 187                 return localMap.get(name);\r
 188         }\r
 189         \r
 190         /* (non-Javadoc)\r
 191          * @see com.att.env.Store#existingSlotNames()\r
 192          */\r
 193         public List<String> existingSlotNames() {\r
 194                 return new ArrayList<>(localMap.keySet());\r
 195         }\r
 196 \r
 197         /* (non-Javadoc)\r
 198          * @see com.att.env.Store#staticSlot(java.lang.String)\r
 199          */\r
 200         public synchronized StaticSlot staticSlot(String name) {\r
 201                 name = name == null ? "" : name.trim();\r
 202                 StaticSlot slot = staticMap.get(name);\r
 203                 if (slot == null)  {\r
 204                         if (stat%growSize == 0) {\r
 205                                 Object[] temp = staticState;\r
 206                                 staticState = new Object[temp.length+growSize];\r
 207                                 System.arraycopy(temp, 0, staticState, 0, temp.length);\r
 208                         }\r
 209                         slot = new StaticSlot(stat++, name);\r
 210                         staticMap.put(name, slot);\r
 211                 }\r
 212                 return slot;\r
 213         }\r
 214         \r
 215         /* (non-Javadoc)\r
 216          * @see com.att.env.Store#put(com.att.env.StaticSlot, java.lang.Object)\r
 217          */\r
 218         public void put(StaticSlot slot, Object value) {\r
 219                 staticState[slot.slot] = value;\r
 220         }\r
 221         \r
 222         /* (non-Javadoc)\r
 223          * @see com.att.env.Store#get(com.att.env.StaticSlot T defaultObject)\r
 224          */\r
 225         @SuppressWarnings("unchecked")\r
 226         public<T> T get(StaticSlot sslot,T dflt) {\r
 227                 T t = (T)staticState[sslot.slot];\r
 228                 return t==null?dflt:t;\r
 229         }\r
 230 \r
 231         @SuppressWarnings("unchecked")\r
 232         public <T> T get(StaticSlot sslot) {\r
 233                 return (T)staticState[sslot.slot];\r
 234         }\r
 235 \r
 236         public List<String> existingStaticSlotNames() {\r
 237                 return new ArrayList<>(staticMap.keySet());\r
 238         }\r
 239 }\r
 240 \r