origin/network/http_server.go

package network

import (
	"crypto/tls"
	"github.com/duanhf2012/origin/log"
	"net/http"
	"time"
)

var Default_MaxHeaderBytes int = 1<<20

type CAFile struct {
	Certfile string
	Keyfile string
}

type HttpServer struct {
	listenAddr string
	readTimeout  time.Duration
	writeTimeout time.Duration

	handler      http.Handler
	caFileList     []CAFile

	httpServer *http.Server
}

func (slf *HttpServer) Init(listenAddr string, handler http.Handler, readTimeout time.Duration, writeTimeout time.Duration) {
	slf.listenAddr = listenAddr
	slf.handler = handler
	slf.readTimeout = readTimeout
	slf.writeTimeout = writeTimeout
}

func (slf *HttpServer) Start() {
	go slf.startListen()
}

func (slf *HttpServer) startListen() error {
	var tlsCaList []tls.Certificate
	var tlsConfig *tls.Config
	for _, caFile := range slf.caFileList {
		cer, err := tls.LoadX509KeyPair(caFile.Certfile, caFile.Keyfile)
		if err != nil {
			log.Fatal("Load CA  [%s]-[%s] file is fail:%s", caFile.Certfile, caFile.Keyfile, err.Error())
			return err
		}
		tlsCaList = append(tlsCaList, cer)
	}

	if len(tlsCaList) > 0 {
		tlsConfig = &tls.Config{Certificates: tlsCaList}
	}

	slf.httpServer = &http.Server{
		Addr:           slf.listenAddr,
		Handler:        slf.handler,
		ReadTimeout:    slf.readTimeout,
		WriteTimeout:   slf.writeTimeout,
		MaxHeaderBytes: Default_MaxHeaderBytes,
		TLSConfig:      tlsConfig,
	}

	var err error
	if len(tlsCaList) > 0 {
		err = slf.httpServer.ListenAndServeTLS("", "")
	} else {
		err = slf.httpServer.ListenAndServe()
	}

	if err != nil {
		log.Fatal("Listen for address %s failure:%+v.",slf.listenAddr,err)
		return err
	}

	return nil
}


func (slf *HttpServer) SetCAFile(caFile []CAFile) {
	slf.caFileList = caFile
}