cadi/core/src/main/java/org/onap/aaf/cadi/util/JsonOutputStream.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.cadi.util;
  23
  24 import java.io.IOException;
  25 import java.io.OutputStream;
  26
  27 public class JsonOutputStream extends OutputStream {
  28     private static final byte[] TWO_SPACE = "  ".getBytes();
  29     private OutputStream os;
  30     private boolean closeable;
  31     private int indent = 0;
  32     private int prev,ret=0;
  33
  34     public JsonOutputStream(OutputStream os) {
  35         // Don't close these, or dire consequences.
  36         closeable = !os.equals(System.out) && !os.equals(System.err);
  37         this.os = os;
  38     }
  39
  40     @Override
  41     public void write(int b) throws IOException {
  42         if (ret=='\n') {
  43             ret = 0;
  44             if (prev!=',' || (b!='{' && b!='[')) {
  45                 os.write('\n');
  46                 for (int i=0;i<indent;++i) {
  47                     os.write(TWO_SPACE);
  48                 }
  49             }
  50         }
  51         switch(b) {
  52             case '{':
  53             case '[':
  54                     ret = '\n';
  55                     ++indent;
  56                     break;
  57             case '}':
  58             case ']':
  59                     --indent;
  60                     os.write('\n');
  61                     for (int i=0;i<indent;++i) {
  62                         os.write(TWO_SPACE);
  63                     }
  64                     break;
  65             case ',':
  66                     ret = '\n';
  67                     break;
  68
  69         }
  70         os.write(b);
  71         prev = b;
  72     }
  73     public void resetIndent() {
  74         indent = 1;
  75     }
  76
  77     @Override
  78     public void flush() throws IOException {
  79         os.flush();
  80     }
  81
  82     @Override
  83     public void close() throws IOException {
  84         if (closeable) {
  85             os.close();
  86         }
  87     }
  88
  89 }