github.com/cockroachdb/cockroach@v20.2.0-alpha.1+incompatible/pkg/server/debug/pprofui/response_writer.go (about) 1 // Copyright 2018 The Cockroach Authors. 2 // 3 // Use of this software is governed by the Business Source License 4 // included in the file licenses/BSL.txt. 5 // 6 // As of the Change Date specified in that file, in accordance with 7 // the Business Source License, use of this software will be governed 8 // by the Apache License, Version 2.0, included in the file 9 // licenses/APL.txt. 10 11 package pprofui 12 13 import ( 14 "io" 15 "net/http" 16 ) 17 18 // responseBridge is a helper for fetching from the pprof profile handlers. 19 // Their interface wants a http.ResponseWriter, so we give it one. The writes 20 // are passed through to an `io.Writer` of our choosing. 21 type responseBridge struct { 22 target io.Writer 23 statusCode int 24 } 25 26 var _ http.ResponseWriter = &responseBridge{} 27 28 func (r *responseBridge) Header() http.Header { 29 return http.Header{} 30 } 31 32 func (r *responseBridge) Write(b []byte) (int, error) { 33 return r.target.Write(b) 34 } 35 36 func (r *responseBridge) WriteHeader(statusCode int) { 37 r.statusCode = statusCode 38 }