/*
 * @Author: hongyu guo
 * @Description:
 * @Version: 1.0.0
 * @Date: 15:05 2021/06/15
 * @Copyright: MIN-Group；国家重大科技基础设施——未来网络北大实验室；深圳市信息论与未来网络重点实验室
 */

package minhttp

import (
	"context"
	"minhttp/common"
	"net"
	"time"
)

//func ListenAndServe(addr string, handler Handler) error {
//	server := &Server{Addr: addr, Handler: handler}
//	return server.ListenAndServe()
//}

// 新版ListenAndServe
func ListenAndServe(iAddrFunc, iAddrPasswdFunc func(...interface{}) string, portFunc func(...interface{}) uint64, handler Handler) error {
	server := &Server{IdentifierAddr: iAddrFunc(), IdentifierPasswd: iAddrPasswdFunc(), Handler: handler}
	// return server.ListenAndServe(newKeyChain, getIdentityByNameAndPasswd, portFunc)
	return server.ListenAndServe(newKeyChain, createIdentityByNameAndPasswd, portFunc)
}

func (srv *Server) Serve(l net.Listener) error {
	//if fn := testHookServerServe; fn != nil {
	//	fn(srv, l) // call hook with unwrapped listener
	//}

	origListener := l
	l = &onceCloseListener{Listener: l}
	defer l.Close()

	if !srv.trackListener(&l, true) {
		return ErrServerClosed
	}
	defer srv.trackListener(&l, false)

	baseCtx := context.Background()
	if srv.BaseContext != nil {
		baseCtx = srv.BaseContext(origListener)
		if baseCtx == nil {
			panic("BaseContext returned a nil context")
		}
	}

	var tempDelay time.Duration // how long to sleep on accept failure

	ctx := context.WithValue(baseCtx, ServerContextKey, srv)
	for {
		rw, err := l.Accept()
		if err != nil {
			select {
			case <-srv.getDoneChan():
				return ErrServerClosed
			default:
			}
			if ne, ok := err.(net.Error); ok && ne.Temporary() {
				if tempDelay == 0 {
					tempDelay = 5 * time.Millisecond
				} else {
					tempDelay *= 2
				}
				if max := 1 * time.Second; tempDelay > max {
					tempDelay = max
				}
				srv.logf("minhttp: Accept error: %v; retrying in %v", err, tempDelay)
				time.Sleep(tempDelay)
				continue
			}
			return err
		}
		connCtx := ctx
		if cc := srv.ConnContext; cc != nil {
			connCtx = cc(connCtx, rw)
			if connCtx == nil {
				panic("ConnContext returned nil")
			}
		}
		tempDelay = 0
		c := srv.newConn(rw)
		c.setState(c.minConn, StateNew) // before Serve can return
		common.LogDebug("serve the conn")
		go c.serve(connCtx)
	}
}