481fb923c5
full diff: https://github.com/emicklei/go-restful/compare/v2.9.5...v3.7.3 - Switch to using go modules - Add check for wildcard to fix CORS filter - Add check on writer to prevent compression of response twice - Add OPTIONS shortcut WebService receiver - Add Route metadata to request attributes or allow adding attributes to routes - Add wroteHeader set - Enable content encoding on Handle and ServeHTTP - Feat: support google custom verb - Feature: override list of method allowed without content-type - Fix Allow header not set on '405: Method Not Allowed' responses - Fix Go 1.15: conversion from int to string yields a string of one rune - Fix WriteError return value - Fix: use request/response resulting from filter chain - handle path params with prefixes and suffixes - HTTP response body was broken, if struct to be converted to JSON has boolean value - List available representations in 406 body - Support describing response headers - Unwrap function in filter chain + remove unused dispatchWithFilters Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package restful
|
|
|
|
// Copyright 2015 Ernest Micklei. All rights reserved.
|
|
// Use of this source code is governed by a license
|
|
// that can be found in the LICENSE file.
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"compress/zlib"
|
|
"sync"
|
|
)
|
|
|
|
// SyncPoolCompessors is a CompressorProvider that use the standard sync.Pool.
|
|
type SyncPoolCompessors struct {
|
|
GzipWriterPool *sync.Pool
|
|
GzipReaderPool *sync.Pool
|
|
ZlibWriterPool *sync.Pool
|
|
}
|
|
|
|
// NewSyncPoolCompessors returns a new ("empty") SyncPoolCompessors.
|
|
func NewSyncPoolCompessors() *SyncPoolCompessors {
|
|
return &SyncPoolCompessors{
|
|
GzipWriterPool: &sync.Pool{
|
|
New: func() interface{} { return newGzipWriter() },
|
|
},
|
|
GzipReaderPool: &sync.Pool{
|
|
New: func() interface{} { return newGzipReader() },
|
|
},
|
|
ZlibWriterPool: &sync.Pool{
|
|
New: func() interface{} { return newZlibWriter() },
|
|
},
|
|
}
|
|
}
|
|
|
|
func (s *SyncPoolCompessors) AcquireGzipWriter() *gzip.Writer {
|
|
return s.GzipWriterPool.Get().(*gzip.Writer)
|
|
}
|
|
|
|
func (s *SyncPoolCompessors) ReleaseGzipWriter(w *gzip.Writer) {
|
|
s.GzipWriterPool.Put(w)
|
|
}
|
|
|
|
func (s *SyncPoolCompessors) AcquireGzipReader() *gzip.Reader {
|
|
return s.GzipReaderPool.Get().(*gzip.Reader)
|
|
}
|
|
|
|
func (s *SyncPoolCompessors) ReleaseGzipReader(r *gzip.Reader) {
|
|
s.GzipReaderPool.Put(r)
|
|
}
|
|
|
|
func (s *SyncPoolCompessors) AcquireZlibWriter() *zlib.Writer {
|
|
return s.ZlibWriterPool.Get().(*zlib.Writer)
|
|
}
|
|
|
|
func (s *SyncPoolCompessors) ReleaseZlibWriter(w *zlib.Writer) {
|
|
s.ZlibWriterPool.Put(w)
|
|
}
|
|
|
|
func newGzipWriter() *gzip.Writer {
|
|
// create with an empty bytes writer; it will be replaced before using the gzipWriter
|
|
writer, err := gzip.NewWriterLevel(new(bytes.Buffer), gzip.BestSpeed)
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
return writer
|
|
}
|
|
|
|
func newGzipReader() *gzip.Reader {
|
|
// create with an empty reader (but with GZIP header); it will be replaced before using the gzipReader
|
|
// we can safely use currentCompressProvider because it is set on package initialization.
|
|
w := currentCompressorProvider.AcquireGzipWriter()
|
|
defer currentCompressorProvider.ReleaseGzipWriter(w)
|
|
b := new(bytes.Buffer)
|
|
w.Reset(b)
|
|
w.Flush()
|
|
w.Close()
|
|
reader, err := gzip.NewReader(bytes.NewReader(b.Bytes()))
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
return reader
|
|
}
|
|
|
|
func newZlibWriter() *zlib.Writer {
|
|
writer, err := zlib.NewWriterLevel(new(bytes.Buffer), gzip.BestSpeed)
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
return writer
|
|
}
|