Add linter script

[vaultmon/.git] / http.go

package main

import (
	"bytes"
	"crypto/tls"
	"net/http"
	"time"
)

const (
	connectionTimeout = time.Second * 10
)

var (
	transport = &http.Transport{
		TLSClientConfig: &tls.Config{
			InsecureSkipVerify: true,
		},
	}
	client = &http.Client{
		Timeout:   connectionTimeout,
		Transport: transport,
	}
)

type HeaderMap = map[string]string

type Connection struct {
	url     string
	headers HeaderMap
}

func (c *Connection) Post(message string) (*http.Response, error) {
	req, err := http.NewRequest("POST", c.url, bytes.NewBufferString(message))
	if err != nil {
		return nil, err
	}

	for k, v := range c.headers {
		req.Header.Set(k, v)
	}

	resp, err := client.Do(req)
	return resp, err
}

func (c *Connection) Get() (*http.Response, error) {
	req, err := http.NewRequest("GET", c.url, nil)
	if err != nil {
		return nil, err
	}

	resp, err := client.Do(req)
	return resp, err
}

func (c *Connection) Head() (*http.Response, error) {
	req, err := http.NewRequest("HEAD", c.url, nil)
	if err != nil {
		return nil, err
	}

	resp, err := client.Do(req)
	return resp, err
}

func (c *Connection) SetHeaders(headers HeaderMap) {
	c.headers = headers
}

func NewConnection(url string) *Connection {
	return &Connection{url: url}
}