001 /**
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements. See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership. The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License. You may obtain a copy of the License at
009 *
010 * http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018
019 package org.apache.hadoop.record;
020
021 import java.io.IOException;
022 import java.util.TreeMap;
023 import java.util.ArrayList;
024 import java.io.PrintStream;
025 import java.io.OutputStream;
026 import java.io.UnsupportedEncodingException;
027
028 import org.apache.hadoop.classification.InterfaceAudience;
029 import org.apache.hadoop.classification.InterfaceStability;
030
031 /**
032 * @deprecated Replaced by <a href="https://hadoop.apache.org/avro/">Avro</a>.
033 */
034 @Deprecated
035 @InterfaceAudience.Public
036 @InterfaceStability.Stable
037 public class CsvRecordOutput implements RecordOutput {
038
039 private PrintStream stream;
040 private boolean isFirst = true;
041
042 private void throwExceptionOnError(String tag) throws IOException {
043 if (stream.checkError()) {
044 throw new IOException("Error serializing "+tag);
045 }
046 }
047
048 private void printCommaUnlessFirst() {
049 if (!isFirst) {
050 stream.print(",");
051 }
052 isFirst = false;
053 }
054
055 /** Creates a new instance of CsvRecordOutput */
056 public CsvRecordOutput(OutputStream out) {
057 try {
058 stream = new PrintStream(out, true, "UTF-8");
059 } catch (UnsupportedEncodingException ex) {
060 throw new RuntimeException(ex);
061 }
062 }
063
064 @Override
065 public void writeByte(byte b, String tag) throws IOException {
066 writeLong((long)b, tag);
067 }
068
069 @Override
070 public void writeBool(boolean b, String tag) throws IOException {
071 printCommaUnlessFirst();
072 String val = b ? "T" : "F";
073 stream.print(val);
074 throwExceptionOnError(tag);
075 }
076
077 @Override
078 public void writeInt(int i, String tag) throws IOException {
079 writeLong((long)i, tag);
080 }
081
082 @Override
083 public void writeLong(long l, String tag) throws IOException {
084 printCommaUnlessFirst();
085 stream.print(l);
086 throwExceptionOnError(tag);
087 }
088
089 @Override
090 public void writeFloat(float f, String tag) throws IOException {
091 writeDouble((double)f, tag);
092 }
093
094 @Override
095 public void writeDouble(double d, String tag) throws IOException {
096 printCommaUnlessFirst();
097 stream.print(d);
098 throwExceptionOnError(tag);
099 }
100
101 @Override
102 public void writeString(String s, String tag) throws IOException {
103 printCommaUnlessFirst();
104 stream.print(Utils.toCSVString(s));
105 throwExceptionOnError(tag);
106 }
107
108 @Override
109 public void writeBuffer(Buffer buf, String tag)
110 throws IOException {
111 printCommaUnlessFirst();
112 stream.print(Utils.toCSVBuffer(buf));
113 throwExceptionOnError(tag);
114 }
115
116 @Override
117 public void startRecord(Record r, String tag) throws IOException {
118 if (tag != null && ! tag.isEmpty()) {
119 printCommaUnlessFirst();
120 stream.print("s{");
121 isFirst = true;
122 }
123 }
124
125 @Override
126 public void endRecord(Record r, String tag) throws IOException {
127 if (tag == null || tag.isEmpty()) {
128 stream.print("\n");
129 isFirst = true;
130 } else {
131 stream.print("}");
132 isFirst = false;
133 }
134 }
135
136 @Override
137 public void startVector(ArrayList v, String tag) throws IOException {
138 printCommaUnlessFirst();
139 stream.print("v{");
140 isFirst = true;
141 }
142
143 @Override
144 public void endVector(ArrayList v, String tag) throws IOException {
145 stream.print("}");
146 isFirst = false;
147 }
148
149 @Override
150 public void startMap(TreeMap v, String tag) throws IOException {
151 printCommaUnlessFirst();
152 stream.print("m{");
153 isFirst = true;
154 }
155
156 @Override
157 public void endMap(TreeMap v, String tag) throws IOException {
158 stream.print("}");
159 isFirst = false;
160 }
161 }