misc/env/src/main/java/org/onap/aaf/misc/env/jaxb/JAXBDF.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.jaxb;\r
  23 \r
  24 import java.io.InputStream;\r
  25 import java.io.OutputStream;\r
  26 import java.io.Reader;\r
  27 import java.io.StringWriter;\r
  28 import java.io.Writer;\r
  29 \r
  30 import javax.xml.bind.JAXBException;\r
  31 import javax.xml.namespace.QName;\r
  32 import javax.xml.validation.Schema;\r
  33 \r
  34 import org.onap.aaf.misc.env.APIException;\r
  35 import org.onap.aaf.misc.env.BaseDataFactory;\r
  36 import org.onap.aaf.misc.env.Data;\r
  37 import org.onap.aaf.misc.env.Env;\r
  38 import org.onap.aaf.misc.env.EnvJAXB;\r
  39 import org.onap.aaf.misc.env.TimeTaken;\r
  40 import org.onap.aaf.misc.env.old.IOObjectifier;\r
  41 import org.onap.aaf.misc.env.old.IOStringifier;\r
  42 import org.onap.aaf.misc.env.old.OldDataFactory;\r
  43 \r
  44 public class JAXBDF<T> extends BaseDataFactory implements OldDataFactory<T>,IOObjectifier<T>, IOStringifier<T> {\r
  45         // Package on purpose\r
  46         EnvJAXB primaryEnv;\r
  47         JAXBumar jumar;\r
  48         JAXBmar jmar;\r
  49 \r
  50         public JAXBDF(EnvJAXB env, Class<?> ... classes) throws APIException {\r
  51                 try {\r
  52                         primaryEnv = env;\r
  53                         jumar = new JAXBumar(classes);\r
  54                         jmar = new JAXBmar(classes) ;\r
  55                 } catch (JAXBException e) {\r
  56                         throw new APIException(e);\r
  57                 }\r
  58         }\r
  59         \r
  60         public JAXBDF(EnvJAXB env, Schema schema, Class<?> ... classes) throws APIException {\r
  61                 try {\r
  62                         primaryEnv = env;\r
  63                         jumar = new JAXBumar(schema, classes);\r
  64                         jmar = new JAXBmar(classes);\r
  65                 } catch (JAXBException e) {\r
  66                         throw new APIException(e);\r
  67                 }\r
  68         }\r
  69         \r
  70         public JAXBDF(EnvJAXB env, QName qname, Class<?> ... classes) throws APIException {\r
  71                 try {\r
  72                         primaryEnv = env;\r
  73                         jumar = new JAXBumar(classes);\r
  74                         jmar = new JAXBmar(qname, classes);\r
  75                 } catch (JAXBException e) {\r
  76                         throw new APIException(e);\r
  77                 }\r
  78         }\r
  79 \r
  80         public JAXBDF(EnvJAXB env, Schema schema, QName qname, Class<?> ... classes) throws APIException {\r
  81                 try {\r
  82                         primaryEnv = env;\r
  83                         jumar = new JAXBumar(schema, classes);\r
  84                         jmar = new JAXBmar(qname, classes);\r
  85                 } catch (JAXBException e) {\r
  86                         throw new APIException(e);\r
  87                 }\r
  88         }\r
  89         \r
  90         // @Override\r
  91         public T newInstance() throws APIException {\r
  92                 try {\r
  93                         return jumar.newInstance();\r
  94                 } catch (Exception e) {\r
  95                         throw new APIException(e);\r
  96                 }\r
  97         }\r
  98 \r
  99         // @Override\r
 100         public IOStringifier<T> pretty(boolean pretty) {\r
 101                 jmar.pretty(pretty);\r
 102                 return this;\r
 103         }\r
 104 \r
 105         // @Override\r
 106         public IOStringifier<T> asFragment(boolean fragment) {\r
 107                 jmar.asFragment(fragment);\r
 108                 return this;\r
 109         }\r
 110 \r
 111         // @Override\r
 112         public void servicePrestart(Env env) throws APIException {\r
 113         }\r
 114 \r
 115         // @Override\r
 116         public void threadPrestart(Env env) throws APIException {\r
 117         }\r
 118 \r
 119         // @Override\r
 120         public void refresh(Env env) throws APIException {\r
 121         }\r
 122 \r
 123         // @Override\r
 124         public void threadDestroy(Env env) throws APIException {\r
 125         }\r
 126 \r
 127         // @Override\r
 128         public void serviceDestroy(Env env) throws APIException {\r
 129         }\r
 130 \r
 131         @SuppressWarnings("unchecked")\r
 132         // @Override\r
 133         public Data<T> newData() {\r
 134                 return new JAXBData<T>(primaryEnv, this, new JAXBStringifier<T>(jmar), new JAXBObjectifier<T>(jumar),"",(Class<T>)jmar.getMarshalClass());\r
 135         }\r
 136 \r
 137         @SuppressWarnings("unchecked")\r
 138         // @Override\r
 139         public Data<T> newData(Env env) {\r
 140                 return new JAXBData<T>(env, this,new JAXBStringifier<T>(jmar), new JAXBObjectifier<T>(jumar),"",(Class<T>)jmar.getMarshalClass());\r
 141         }\r
 142 \r
 143         // @Override\r
 144         public Data<T> newData(T type) {\r
 145                 return new JAXBData<T>(primaryEnv, this, new JAXBStringifier<T>(jmar), new JAXBObjectifier<T>(jumar), type);\r
 146         }\r
 147 \r
 148         // @Override\r
 149         public Data<T> newDataFromStream(Env env, InputStream input) throws APIException {\r
 150                 //TODO Write an unvalidated String using STAX checking for end of Doc?\r
 151                 // perhaps key evaluation as well.\r
 152                 try {\r
 153                         T t = jumar.unmarshal(env.debug(), input);\r
 154                         return new JAXBData<T>(primaryEnv, this, new JAXBStringifier<T>(jmar), new JAXBObjectifier<T>(jumar),t);\r
 155                 } catch(JAXBException e) {\r
 156                         throw new APIException(e);\r
 157                 }\r
 158         }\r
 159 \r
 160         @SuppressWarnings("unchecked")\r
 161         // @Override\r
 162         public Data<T> newDataFromString(String string) {\r
 163                 return new JAXBData<T>(primaryEnv, this,new JAXBStringifier<T>(jmar), new JAXBObjectifier<T>(jumar), string,(Class<T>)jmar.getMarshalClass());\r
 164         }\r
 165 \r
 166         /////////// Old DataFactory Interface \r
 167         // @Override\r
 168         public String stringify(T type) throws APIException {\r
 169                 try {\r
 170                         StringWriter sw = new StringWriter();\r
 171                         jmar.marshal(primaryEnv.debug(), type, sw);\r
 172                         return sw.toString();\r
 173                 } catch (JAXBException e) {\r
 174                         throw new APIException(e);\r
 175                 }       \r
 176         }\r
 177 \r
 178         // @Override\r
 179         public void stringify(T type, Writer writer) throws APIException {\r
 180                 try {\r
 181                         jmar.marshal(primaryEnv.debug(), type, writer);\r
 182                 } catch (JAXBException e) {\r
 183                         throw new APIException(e);\r
 184                 }       \r
 185         }\r
 186 \r
 187         // @Override\r
 188         public void stringify(T type, OutputStream os) throws APIException {\r
 189                 try {\r
 190                         jmar.marshal(primaryEnv.debug(), type, os);\r
 191                 } catch (JAXBException e) {\r
 192                         throw new APIException(e);\r
 193                 }       \r
 194         }\r
 195 \r
 196         /////////// New DataFactory Interface \r
 197         // @Override\r
 198         public String stringify(Env env, T input, boolean ... options) throws APIException {\r
 199                 try {\r
 200                         StringWriter sw = new StringWriter();\r
 201                         TimeTaken tt = env.start("JAXB Stringify", Env.XML);\r
 202                         try {\r
 203                                 jmar.marshal(env.debug(), input, sw, options);\r
 204                         } finally {\r
 205                                 tt.done();\r
 206                         }\r
 207                         String str = sw.toString();\r
 208                         tt.size(str.getBytes().length);\r
 209                         return str;\r
 210                 } catch (JAXBException e) {\r
 211                         throw new APIException(e);\r
 212                 }\r
 213         }\r
 214 \r
 215         // @Override\r
 216         public void stringify(Env env, T input, Writer writer, boolean ... options) throws APIException {\r
 217                 TimeTaken tt = env.start("JAXB Stringify", Env.XML);\r
 218                 try {\r
 219                         jmar.marshal(env.debug(), input, writer, options);\r
 220                 } catch (JAXBException e) {\r
 221                         throw new APIException(e);\r
 222                 } finally {\r
 223                         tt.done();\r
 224                 }\r
 225         }\r
 226 \r
 227         // @Override\r
 228         public void stringify(Env env, T input, OutputStream os, boolean ... options) throws APIException {\r
 229                 TimeTaken tt = env.start("JAXB Stringify", Env.XML);\r
 230                 try {\r
 231                         jmar.marshal(env.debug(), input, os, options);\r
 232                 } catch (JAXBException e) {\r
 233                         throw new APIException(e);\r
 234                 } finally {\r
 235                         tt.done();\r
 236                 }\r
 237         }\r
 238 \r
 239         // @Override\r
 240         public T objectify(Env env, Reader rdr) throws APIException {\r
 241                 TimeTaken tt = env.start("JAXB Objectify", Env.XML);\r
 242                 try {\r
 243                         return jumar.unmarshal(env.debug(), rdr);\r
 244                 } catch (JAXBException e) {\r
 245                         throw new APIException(e);\r
 246                 } finally {\r
 247                         tt.done();\r
 248                 }\r
 249         }\r
 250 \r
 251         // @Override\r
 252         public T objectify(Reader rdr) throws APIException {\r
 253                 try {\r
 254                         return jumar.unmarshal(primaryEnv.debug(), rdr);\r
 255                 } catch (JAXBException e) {\r
 256                         throw new APIException(e);\r
 257                 }       \r
 258         }\r
 259 \r
 260         // @Override\r
 261         public T objectify(Env env, InputStream is) throws APIException {\r
 262                 TimeTaken tt = env.start("JAXB Objectify", Env.XML);\r
 263                 try {\r
 264                         return jumar.unmarshal(env.debug(), is);\r
 265                 } catch (JAXBException e) {\r
 266                         throw new APIException(e);\r
 267                 } finally {\r
 268                         tt.done();\r
 269                 }\r
 270         }\r
 271 \r
 272         // @Override\r
 273         public T objectify(InputStream is) throws APIException {\r
 274                 try {\r
 275                         return jumar.unmarshal(primaryEnv.debug(), is);\r
 276                 } catch (JAXBException e) {\r
 277                         throw new APIException(e);\r
 278                 }       \r
 279         }\r
 280 \r
 281         // @Override\r
 282         public T objectify(Env env, String input) throws APIException {\r
 283                 TimeTaken tt = env.start("JAXB Objectify", Env.XML);\r
 284                 tt.size(input.getBytes().length);\r
 285                 try {\r
 286                         return jumar.unmarshal(env.debug(), input);\r
 287                 } catch (JAXBException e) {\r
 288                         throw new APIException(e);\r
 289                 } finally {\r
 290                         tt.done();\r
 291                 }\r
 292         }\r
 293 \r
 294         // @Override\r
 295         public T objectify(String text) throws APIException {\r
 296                 try {\r
 297                         return jumar.unmarshal(primaryEnv.debug(), text);\r
 298                 } catch (JAXBException e) {\r
 299                         throw new APIException(e);\r
 300                 }       \r
 301         }\r
 302 \r
 303         @SuppressWarnings("unchecked")\r
 304         // @Override\r
 305         public Class<T> getTypeClass() {\r
 306                 return (Class<T>)jmar.getMarshalClass();\r
 307         }\r
 308 \r
 309 }\r