http urls monitor.

reader.go 789B

12345678910111213141516171819202122232425262728293031323334353637
  1. package render
  2. import (
  3. "io"
  4. "net/http"
  5. "strconv"
  6. )
  7. type Reader struct {
  8. ContentType string
  9. ContentLength int64
  10. Reader io.Reader
  11. Headers map[string]string
  12. }
  13. // Render (Reader) writes data with custom ContentType and headers.
  14. func (r Reader) Render(w http.ResponseWriter) (err error) {
  15. r.WriteContentType(w)
  16. r.Headers["Content-Length"] = strconv.FormatInt(r.ContentLength, 10)
  17. r.writeHeaders(w, r.Headers)
  18. _, err = io.Copy(w, r.Reader)
  19. return
  20. }
  21. func (r Reader) WriteContentType(w http.ResponseWriter) {
  22. writeContentType(w, []string{r.ContentType})
  23. }
  24. func (r Reader) writeHeaders(w http.ResponseWriter, headers map[string]string) {
  25. header := w.Header()
  26. for k, v := range headers {
  27. if val := header[k]; len(val) == 0 {
  28. header[k] = []string{v}
  29. }
  30. }
  31. }