napnap/server.go at master · 0x5487/napnap · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package napnap

import (
	"net/http"
	"time"
)

// Config ...
type Config struct {
	Addr          string
	Domain        string // abc123.com, abc456.com
	CertCachePath string
	TLSCertFile   string
	TLSKeyFile    string
	ReadTimeout   time.Duration
	WriteTimeout  time.Duration
}

// Server ...
type Server struct {
	*http.Server
	Config *Config
}

// NewHTTPEngine ...
func NewHTTPEngine(addr string) *Server {
	s := &Server{
		Server: new(http.Server),
	}
	s.Addr = addr
	return s
}

// NewHTTPEngineWithConfig ...
func NewHTTPEngineWithConfig(c *Config) *Server {
	s := &Server{
		Server: new(http.Server),
		Config: c,
	}
	s.Addr = c.Addr
	s.ReadTimeout = c.ReadTimeout
	s.WriteTimeout = c.WriteTimeout
	return s
}