github.com/kaydxh/golang@v0.0.131/go/net/http/response_writer.go (about) 1 /* 2 *Copyright (c) 2022, kaydxh 3 * 4 *Permission is hereby granted, free of charge, to any person obtaining a copy 5 *of this software and associated documentation files (the "Software"), to deal 6 *in the Software without restriction, including without limitation the rights 7 *to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 8 *copies of the Software, and to permit persons to whom the Software is 9 *furnished to do so, subject to the following conditions: 10 * 11 *The above copyright notice and this permission notice shall be included in all 12 *copies or substantial portions of the Software. 13 * 14 *THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15 *IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16 *FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 17 *AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 18 *LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 19 *OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 20 *SOFTWARE. 21 */ 22 package http 23 24 import ( 25 "bytes" 26 "fmt" 27 "net/http" 28 ) 29 30 // ResponseWriterWrapper 31 type ResponseWriterWrapper struct { 32 w http.ResponseWriter 33 body bytes.Buffer 34 statusCode int 35 } 36 37 func NewResponseWriterWrapper(w http.ResponseWriter) *ResponseWriterWrapper { 38 return &ResponseWriterWrapper{ 39 w: w, 40 statusCode: http.StatusOK, 41 } 42 } 43 44 func (rww *ResponseWriterWrapper) Write(buf []byte) (int, error) { 45 rww.body.Write(buf) 46 return rww.w.Write(buf) 47 } 48 49 // Header function overwrites the http.ResponseWriter Header() function 50 func (rww *ResponseWriterWrapper) Header() http.Header { 51 return rww.w.Header() 52 } 53 54 // WriteHeader function overwrites the http.ResponseWriter WriteHeader() function 55 func (rww *ResponseWriterWrapper) WriteHeader(statusCode int) { 56 rww.statusCode = statusCode 57 rww.w.WriteHeader(statusCode) 58 } 59 60 // String function pack respose header, http status code and body 61 func (rww *ResponseWriterWrapper) String() string { 62 var buf bytes.Buffer 63 64 buf.WriteString("Headers:") 65 for k, v := range rww.w.Header() { 66 buf.WriteString(fmt.Sprintf("%s: %v", k, v)) 67 } 68 69 buf.WriteString(fmt.Sprintf(" Status Code:%d", rww.statusCode)) 70 71 buf.WriteString(" Body:") 72 buf.WriteString(rww.body.String()) 73 return buf.String() 74 }