rw.go

   1 // godlighty -- highly-customizable HTTP, HTTP/2, HTTPS server
   2 // Copyright (C) 2021-2024 Sergey Matveev <stargrave@stargrave.org>
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU General Public License as published by
   6 // the Free Software Foundation, version 3 of the License.
   7 //
   8 // This program is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 // GNU General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU General Public License
  14 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15
  16 package godlighty
  17
  18 import (
  19         "io"
  20         "net/http"
  21 )
  22
  23 type gzipResponseWriter struct {
  24         io.Writer
  25         http.ResponseWriter
  26         status int
  27 }
  28
  29 func (w *gzipResponseWriter) WriteHeader(status int) {
  30         w.status = status
  31 }
  32
  33 func (w *gzipResponseWriter) Write(b []byte) (int, error) {
  34         return w.Writer.Write(b)
  35 }
  36
  37 type CountResponseWriter struct {
  38         http.ResponseWriter
  39         Status int
  40         Size   int64
  41 }
  42
  43 func (w *CountResponseWriter) WriteHeader(status int) {
  44         w.Status = status
  45         w.ResponseWriter.WriteHeader(status)
  46 }
  47
  48 func (w *CountResponseWriter) Write(b []byte) (int, error) {
  49         n, err := w.ResponseWriter.Write(b)
  50         w.Size += int64(n)
  51         return n, err
  52 }