1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.hadoop.hbase.rest.provider.producer;
21
22 import java.io.IOException;
23 import java.io.OutputStream;
24 import java.lang.annotation.Annotation;
25 import java.lang.reflect.Type;
26 import java.nio.charset.Charset;
27
28 import javax.ws.rs.Produces;
29 import javax.ws.rs.WebApplicationException;
30 import javax.ws.rs.core.MediaType;
31 import javax.ws.rs.core.MultivaluedMap;
32 import javax.ws.rs.ext.MessageBodyWriter;
33 import javax.ws.rs.ext.Provider;
34
35 import org.apache.hadoop.hbase.classification.InterfaceAudience;
36 import org.apache.hadoop.hbase.rest.Constants;
37
38
39
40
41
42
43
44 @Provider
45 @Produces(Constants.MIMETYPE_TEXT)
46 @InterfaceAudience.Private
47 public class PlainTextMessageBodyProducer
48 implements MessageBodyWriter<Object> {
49
50 private ThreadLocal<byte[]> buffer = new ThreadLocal<byte[]>();
51
52 @Override
53 public boolean isWriteable(Class<?> arg0, Type arg1, Annotation[] arg2,
54 MediaType arg3) {
55 return true;
56 }
57
58 @Override
59 public long getSize(Object object, Class<?> type, Type genericType,
60 Annotation[] annotations, MediaType mediaType) {
61 byte[] bytes = object.toString().getBytes(Charset.forName("UTF-8"));
62 buffer.set(bytes);
63 return bytes.length;
64 }
65
66 @Override
67 public void writeTo(Object object, Class<?> type, Type genericType,
68 Annotation[] annotations, MediaType mediaType,
69 MultivaluedMap<String, Object> httpHeaders, OutputStream outStream)
70 throws IOException, WebApplicationException {
71 byte[] bytes = buffer.get();
72 outStream.write(bytes);
73 buffer.remove();
74 }
75 }