server.go

package http

import (
	"fmt"
	"net/http"
	"sync"
	"sync/atomic"
	"time"
)

// HTTPRequestKeyProvider represents request key provider to extract a request field.
type HTTPRequestKeyProvider func(source interface{}) (string, error)

// HTTPRequestKeyProviders represents key providers
var HTTPRequestKeyProviders = make(map[string]HTTPRequestKeyProvider)

type Server struct {
	http.Server
	*httpHandler
	trips            map[string]*HTTPResponses
	mux              sync.Mutex
	rotate           bool
	indexKeys        []string
	requestTemplate  string
	responseTemplate string
}

func (s *Server) Append(trips *HTTPServerTrips) {
	s.mux.Lock()
	defer s.mux.Unlock()
	if len(s.trips) > 0 {
		for k, v := range s.trips {
			if _, ok := trips.Trips[k]; ok {
				continue
			}
			trips.Trips[k] = v
		}
	}
	s.httpHandler.handler = getServerHandler(&s.Server, s.httpHandler, trips)
}

// StartServer starts http request, the server has ability to replay recorded  HTTP trips with https://github.com/viant/toolbox/blob/master/bridge/http_bridge_recording_util.go#L82
func StartServer(port int, trips *HTTPServerTrips, reqTemplate, respTemplate string) (*Server, error) {
	err := trips.Init(reqTemplate, respTemplate)
	if err != nil {
		return nil, fmt.Errorf("failed to start http server :%v, %v", port, err)
	}

	var httpHandler = &httpHandler{
		running: 1,
	}

	server := &Server{
		rotate:           trips.Rotate,
		indexKeys:        trips.IndexKeys,
		httpHandler:      httpHandler,
		trips:            trips.Trips,
		Server:           http.Server{Addr: fmt.Sprintf(":%v", port), Handler: httpHandler},
		requestTemplate:  reqTemplate,
		responseTemplate: respTemplate,
	}
	httpHandler.handler = getServerHandler(&server.Server, httpHandler, trips)

	errorNotification := make(chan bool, 1)
	go func() {
		fmt.Printf("Starting server on %v\n", port)
		err = server.Server.ListenAndServe()
		atomic.StoreInt32(&httpHandler.running, 0)
		errorNotification <- true
		if err != nil {
			err = fmt.Errorf("failed to start http server on port %v, %v", port, err)
		}
	}()

	//if there is error in starting server quite immediately
	select {
	case <-errorNotification:
	case <-time.After(time.Second * 2):
	}
	return server, err
}