config.go

package gohttp

// Config config
type Config struct {
	Debug             bool     `json:"debug" yaml:"debug"`
	Addr              string   `json:"addr" yaml:"addr"`
	ReadTimeout       int      `json:"readtimeout" yaml:"readtimeout"`
	ReadHeaderTimeout int      `json:"readheadertime" yaml:"readheadertime"`
	WriteTimeout      int      `json:"writetime" yaml:"writetime"`
	MaxHeaderBytes    int      `json:"maxheaderbytes" yaml:"maxheaderbytes"`
	CertFile          string   `json:"cert" yaml:"cert"`
	KeyFile           string   `json:"key" yaml:"key"`
	WebPath           string   `json:"web_path" yaml:"web_path"`
	StaticPath        []string `json:"static_path" yaml:"static_path"`
	LogFormat         string   `json:"log_format" yaml:"log_format"`
}

// SetDebug set debug model
func (cfg *Config) SetDebug(debug bool) {
	cfg.Debug = debug
}

// SetLogFormat set log format
func (cfg *Config) SetLogFormat(format string) {
	cfg.LogFormat = format
}

// SetAddress set address
func (cfg *Config) SetAddress(addr string) {
	cfg.Addr = addr
}

// SetTimeout set time out
func (cfg *Config) SetTimeout(timeout int) {
	cfg.ReadTimeout = timeout
	cfg.ReadHeaderTimeout = timeout
	cfg.WriteTimeout = timeout
}

// SetMaxHeaderBytes set max header bytes
func (cfg *Config) SetMaxHeaderBytes(max int) {
	cfg.MaxHeaderBytes = max
}

// SetTLS set tls
func (cfg *Config) SetTLS(cert, key string) {
	cfg.CertFile = cert
	cfg.KeyFile = key
}

// SetWebPath set web path
func (cfg *Config) SetWebPath(web string) {
	cfg.WebPath = web
}

// SetStaticPath set static path
func (cfg *Config) SetStaticPath(static ...string) {
	cfg.StaticPath = append(cfg.StaticPath, static...)
}

// InitConfig init config
func InitConfig() *Config {
	cfg := new(Config)
	cfg.SetDebug(true)
	cfg.SetLogFormat("%3d %s %s (%s) %s")
	cfg.SetAddress(":18081")
	cfg.SetTimeout(30)
	cfg.SetMaxHeaderBytes(1 << 20) //DefaultMaxHeaderBytes 1MB
	cfg.SetStaticPath(".")
	cfg.SetWebPath(".")
	return cfg
}