misc/env/src/main/java/org/onap/aaf/misc/env/jaxb/JAXBStringifier.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.misc.env.jaxb;
  23
  24 import java.io.OutputStream;
  25 import java.io.StringWriter;
  26 import java.io.Writer;
  27
  28 import javax.xml.bind.JAXBException;
  29 import javax.xml.namespace.QName;
  30
  31 import org.onap.aaf.misc.env.APIException;
  32 import org.onap.aaf.misc.env.Env;
  33 import org.onap.aaf.misc.env.TimeTaken;
  34 import org.onap.aaf.misc.env.old.IOStringifier;
  35
  36 public class JAXBStringifier<T> implements IOStringifier<T> {
  37         private JAXBmar jmar;
  38
  39         public JAXBStringifier(Class<?>... classes) throws APIException {
  40                 try {
  41                         jmar = new JAXBmar(classes);
  42                 } catch (JAXBException e) {
  43                         throw new APIException(e);
  44                 }
  45         }
  46
  47         public JAXBStringifier(QName qname, Class<?>... classes)
  48                         throws APIException {
  49                 try {
  50                         jmar = new JAXBmar(qname, classes);
  51                 } catch (JAXBException e) {
  52                         throw new APIException(e);
  53                 }
  54         }
  55
  56         // package on purpose
  57         JAXBStringifier(JAXBmar jmar) {
  58                 this.jmar = jmar;
  59         }
  60
  61         // // @Override
  62         public void stringify(Env env, T input, Writer writer, boolean ... options)
  63                         throws APIException {
  64                 TimeTaken tt = env.start("JAXB Marshal", Env.XML);
  65                 try {
  66                         jmar.marshal(env.debug(), input, writer, options);
  67                 } catch (JAXBException e) {
  68                         throw new APIException(e);
  69                 } finally {
  70                         tt.done();
  71                 }
  72         }
  73
  74         // @Override
  75         public void stringify(Env env, T input, OutputStream os, boolean ... options)
  76                         throws APIException {
  77                 // TODO create an OutputStream that Counts?
  78                 TimeTaken tt = env.start("JAXB Marshal", Env.XML);
  79                 try {
  80                         jmar.marshal(env.debug(), input, os, options);
  81                 } catch (JAXBException e) {
  82                         throw new APIException(e);
  83                 } finally {
  84                         tt.done();
  85                 }
  86         }
  87
  88         // @Override
  89         public String stringify(Env env, T input, boolean ... options) throws APIException {
  90                 TimeTaken tt = env.start("JAXB Marshal", Env.XML);
  91                 StringWriter sw = new StringWriter();
  92                 try {
  93                         jmar.marshal(env.debug(), input, sw, options);
  94                         String rv = sw.toString();
  95                         tt.size(rv.length());
  96                         return rv;
  97                 } catch (JAXBException e) {
  98                         tt.size(0);
  99                         throw new APIException(e);
 100                 } finally {
 101                         tt.done();
 102                 }
 103         }
 104
 105         // // @Override
 106         public void servicePrestart(Env env) throws APIException {
 107         }
 108
 109         // // @Override
 110         public void threadPrestart(Env env) throws APIException {
 111         }
 112
 113         // // @Override
 114         public void refresh(Env env) throws APIException {
 115         }
 116
 117         // // @Override
 118         public void threadDestroy(Env env) throws APIException {
 119         }
 120
 121         // // @Override
 122         public void serviceDestroy(Env env) throws APIException {
 123         }
 124
 125         // @Override
 126         public JAXBStringifier<T> pretty(boolean pretty) {
 127                 jmar.pretty(pretty);
 128                 return this;
 129         }
 130
 131         // @Override
 132         public JAXBStringifier<T> asFragment(boolean fragment) {
 133                 jmar.asFragment(fragment);
 134                 return this;
 135         }
 136
 137 }