request.go

package main

import(
	"net/http"
	"sync"

	"encoding/json"
	"strings"

	"io"
	"io/ioutil"
)

type Request struct {
	Header http.Header `json:"headers"`
	ContentLength int64 `json:"content_length"`
	Body string	`json:"body"`
	Method string `json:"method"`
	Path string `json:"path"`
	Query string `json:"query"`
}


type RequestDatabase struct {
	*sync.RWMutex
	requests []*Request
	maxRequests int
	Count int
}

func MakeRequest(req *http.Request) *Request {
	r := new(Request)

	r.Header = make(http.Header)

	for k, v := range req.Header {
		r.Header[k] = v
	}

	r.ContentLength = req.ContentLength
	r.Method = req.Method
	r.Path = req.URL.Path
	r.Query = req.URL.RawQuery

	body, _ := ioutil.ReadAll(req.Body)
	r.Body = string(body)

	return r
}

func (r *Request) ToJson() ([]byte, error) {
	return json.Marshal(r)
}

func (r *Request) Forward(client *http.Client, url string) {
	body := strings.NewReader(r.Body)
	req, _ := http.NewRequest(r.Method, url, body)

	for header, vals := range r.Header {
		for _, val := range vals {
			req.Header.Add(header, val)
		}
	}

	resp, _ := client.Do(req)
	io.Copy(ioutil.Discard, resp.Body)
	resp.Body.Close()
}

/////////////////////////////
// Request Database
/////////////////////////////

func (d *RequestDatabase) Insert(req *http.Request) *Request {
	d.Lock()
		r := MakeRequest(req)
		d.requests = append([]*Request{r}, d.requests...)

		if len(d.requests) >= d.maxRequests {
			d.requests = d.requests[0:d.maxRequests]
		}

		d.Count += 1

	d.Unlock()

	return r
}

func (d *RequestDatabase) Clear() {
	d.Lock()
		d.requests = make([]*Request, 0, d.maxRequests)
		d.Count = 0
	d.Unlock();
}

func (d *RequestDatabase) ToJson() ([]byte, error) {
	return json.Marshal(d.requests)
}

func MakeRequestDatabase(capacity int) *RequestDatabase {
	db := &RequestDatabase{new(sync.RWMutex), make([]*Request, 0, capacity), capacity, 0}
	return db
}