]>
Commit | Line | Data |
---|---|---|
f67539c2 TL |
1 | /* |
2 | * Licensed to the Apache Software Foundation (ASF) under one | |
3 | * or more contributor license agreements. See the NOTICE file | |
4 | * distributed with this work for additional information | |
5 | * regarding copyright ownership. The ASF licenses this file | |
6 | * to you under the Apache License, Version 2.0 (the | |
7 | * "License"); you may not use this file except in compliance | |
8 | * with the License. You may obtain a copy of the License at | |
9 | * | |
10 | * http://www.apache.org/licenses/LICENSE-2.0 | |
11 | * | |
12 | * Unless required by applicable law or agreed to in writing, | |
13 | * software distributed under the License is distributed on an | |
14 | * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
15 | * KIND, either express or implied. See the License for the | |
16 | * specific language governing permissions and limitations | |
17 | * under the License. | |
18 | */ | |
19 | ||
20 | package thrift | |
21 | ||
22 | import ( | |
23 | "compress/gzip" | |
24 | "io" | |
25 | "net/http" | |
26 | "strings" | |
27 | ) | |
28 | ||
29 | // NewThriftHandlerFunc is a function that create a ready to use Apache Thrift Handler function | |
30 | func NewThriftHandlerFunc(processor TProcessor, | |
31 | inPfactory, outPfactory TProtocolFactory) func(w http.ResponseWriter, r *http.Request) { | |
32 | ||
33 | return gz(func(w http.ResponseWriter, r *http.Request) { | |
34 | w.Header().Add("Content-Type", "application/x-thrift") | |
35 | ||
36 | transport := NewStreamTransport(r.Body, w) | |
37 | processor.Process(r.Context(), inPfactory.GetProtocol(transport), outPfactory.GetProtocol(transport)) | |
38 | }) | |
39 | } | |
40 | ||
41 | // gz transparently compresses the HTTP response if the client supports it. | |
42 | func gz(handler http.HandlerFunc) http.HandlerFunc { | |
43 | return func(w http.ResponseWriter, r *http.Request) { | |
44 | if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") { | |
45 | handler(w, r) | |
46 | return | |
47 | } | |
48 | w.Header().Set("Content-Encoding", "gzip") | |
49 | gz := gzip.NewWriter(w) | |
50 | defer gz.Close() | |
51 | gzw := gzipResponseWriter{Writer: gz, ResponseWriter: w} | |
52 | handler(gzw, r) | |
53 | } | |
54 | } | |
55 | ||
56 | type gzipResponseWriter struct { | |
57 | io.Writer | |
58 | http.ResponseWriter | |
59 | } | |
60 | ||
61 | func (w gzipResponseWriter) Write(b []byte) (int, error) { | |
62 | return w.Writer.Write(b) | |
63 | } |