You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							98 lines
						
					
					
						
							2.5 KiB
						
					
					
				
			
		
		
		
			
			
			
				
					
				
				
					
				
			
		
		
	
	
							98 lines
						
					
					
						
							2.5 KiB
						
					
					
				
								// Copyright 2015 Matthew Holt and The Caddy Authors
							 | 
						|
								//
							 | 
						|
								// Licensed under the Apache License, Version 2.0 (the "License");
							 | 
						|
								// you may not use this file except in compliance with the License.
							 | 
						|
								// You may obtain a copy of the License at
							 | 
						|
								//
							 | 
						|
								//     http://www.apache.org/licenses/LICENSE-2.0
							 | 
						|
								//
							 | 
						|
								// Unless required by applicable law or agreed to in writing, software
							 | 
						|
								// distributed under the License is distributed on an "AS IS" BASIS,
							 | 
						|
								// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
							 | 
						|
								// See the License for the specific language governing permissions and
							 | 
						|
								// limitations under the License.
							 | 
						|
								
							 | 
						|
								package caddygzip
							 | 
						|
								
							 | 
						|
								import (
							 | 
						|
									"fmt"
							 | 
						|
									"strconv"
							 | 
						|
								
							 | 
						|
									"github.com/caddyserver/caddy/v2"
							 | 
						|
									"github.com/caddyserver/caddy/v2/caddyconfig/caddyfile"
							 | 
						|
									"github.com/caddyserver/caddy/v2/modules/caddyhttp/encode"
							 | 
						|
									"github.com/klauspost/compress/gzip"
							 | 
						|
								)
							 | 
						|
								
							 | 
						|
								func init() {
							 | 
						|
									caddy.RegisterModule(Gzip{})
							 | 
						|
								}
							 | 
						|
								
							 | 
						|
								// Gzip can create gzip encoders.
							 | 
						|
								type Gzip struct {
							 | 
						|
									Level int `json:"level,omitempty"`
							 | 
						|
								}
							 | 
						|
								
							 | 
						|
								// CaddyModule returns the Caddy module information.
							 | 
						|
								func (Gzip) CaddyModule() caddy.ModuleInfo {
							 | 
						|
									return caddy.ModuleInfo{
							 | 
						|
										ID:  "http.encoders.gzip",
							 | 
						|
										New: func() caddy.Module { return new(Gzip) },
							 | 
						|
									}
							 | 
						|
								}
							 | 
						|
								
							 | 
						|
								// UnmarshalCaddyfile sets up the handler from Caddyfile tokens.
							 | 
						|
								func (g *Gzip) UnmarshalCaddyfile(d *caddyfile.Dispenser) error {
							 | 
						|
									for d.Next() {
							 | 
						|
										if !d.NextArg() {
							 | 
						|
											continue
							 | 
						|
										}
							 | 
						|
										levelStr := d.Val()
							 | 
						|
										level, err := strconv.Atoi(levelStr)
							 | 
						|
										if err != nil {
							 | 
						|
											return err
							 | 
						|
										}
							 | 
						|
										g.Level = level
							 | 
						|
									}
							 | 
						|
									return nil
							 | 
						|
								}
							 | 
						|
								
							 | 
						|
								// Provision provisions g's configuration.
							 | 
						|
								func (g *Gzip) Provision(ctx caddy.Context) error {
							 | 
						|
									if g.Level == 0 {
							 | 
						|
										g.Level = defaultGzipLevel
							 | 
						|
									}
							 | 
						|
									return nil
							 | 
						|
								}
							 | 
						|
								
							 | 
						|
								// Validate validates g's configuration.
							 | 
						|
								func (g Gzip) Validate() error {
							 | 
						|
									if g.Level < gzip.StatelessCompression {
							 | 
						|
										return fmt.Errorf("quality too low; must be >= %d", gzip.StatelessCompression)
							 | 
						|
									}
							 | 
						|
									if g.Level > gzip.BestCompression {
							 | 
						|
										return fmt.Errorf("quality too high; must be <= %d", gzip.BestCompression)
							 | 
						|
									}
							 | 
						|
									return nil
							 | 
						|
								}
							 | 
						|
								
							 | 
						|
								// AcceptEncoding returns the name of the encoding as
							 | 
						|
								// used in the Accept-Encoding request headers.
							 | 
						|
								func (Gzip) AcceptEncoding() string { return "gzip" }
							 | 
						|
								
							 | 
						|
								// NewEncoder returns a new gzip writer.
							 | 
						|
								func (g Gzip) NewEncoder() encode.Encoder {
							 | 
						|
									writer, _ := gzip.NewWriterLevel(nil, g.Level)
							 | 
						|
									return writer
							 | 
						|
								}
							 | 
						|
								
							 | 
						|
								// Informed from http://blog.klauspost.com/gzip-performance-for-go-webservers/
							 | 
						|
								var defaultGzipLevel = 5
							 | 
						|
								
							 | 
						|
								// Interface guards
							 | 
						|
								var (
							 | 
						|
									_ encode.Encoding       = (*Gzip)(nil)
							 | 
						|
									_ caddy.Provisioner     = (*Gzip)(nil)
							 | 
						|
									_ caddy.Validator       = (*Gzip)(nil)
							 | 
						|
									_ caddyfile.Unmarshaler = (*Gzip)(nil)
							 | 
						|
								)
							 | 
						|
								
							 |