Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
1 contributor

Users who have contributed to this file

128 lines (101 sloc) 2.88 KB
package logdna
import "bytes"
import "encoding/json"
import "net/http"
import "net/url"
import "strconv"
import "time"
// IngestBaseURL is the base URL for the LogDNA ingest API.
const IngestBaseURL = "https://logs.logdna.com/logs/ingest"
// DefaultFlushLimit is the number of log lines before we flush to LogDNA
const DefaultFlushLimit = 5000
// Config is used by NewClient to configure new clients.
type Config struct {
APIKey string
LogFile string
Hostname string
FlushLimit int
}
// Client is a client to the LogDNA logging service.
type Client struct {
config Config
payload payloadJSON
apiURL url.URL
}
// logLineJSON represents a log line in the LogDNA ingest API JSON payload.
type logLineJSON struct {
Timestamp int64 `json:"timestamp"`
Line string `json:"line"`
File string `json:"file"`
}
// payloadJSON is the complete JSON payload that will be sent to the LogDNA
// ingest API.
type payloadJSON struct {
Lines []logLineJSON `json:"lines"`
}
// makeIngestURL creats a new URL to the a full LogDNA ingest API endpoint with
// API key and requierd parameters.
func makeIngestURL(cfg Config) url.URL {
u, _ := url.Parse(IngestBaseURL)
u.User = url.User(cfg.APIKey)
values := url.Values{}
values.Set("hostname", cfg.Hostname)
values.Set("now", strconv.FormatInt(time.Time{}.UnixNano(), 10))
u.RawQuery = values.Encode()
return *u
}
// NewClient returns a Client configured to send logs to the LogDNA ingest API.
func NewClient(cfg Config) *Client {
if cfg.FlushLimit == 0 {
cfg.FlushLimit = DefaultFlushLimit
}
var client Client
client.apiURL = makeIngestURL(cfg)
client.config = cfg
return &client
}
// Log adds a new log line to Client's payload.
//
// To actually send the logs, Flush() needs to be called.
//
// Flush is called automatically if we reach the client's flush limit.
func (c *Client) Log(t time.Time, msg string) {
if c.Size() == c.config.FlushLimit {
c.Flush()
}
// Ingest API wants timestamp in milliseconds so we need to round timestamp
// down from nanoseconds.
logLine := logLineJSON{
Timestamp: t.UnixNano() / 1000000,
Line: msg,
File: c.config.LogFile,
}
c.payload.Lines = append(c.payload.Lines, logLine)
}
// Size returns the number of lines waiting to be sent.
func (c *Client) Size() int {
return len(c.payload.Lines)
}
// Flush sends any buffered logs to LogDNA and clears the buffered logs.
func (c *Client) Flush() error {
// Return immediately if no logs to send
if c.Size() == 0 {
return nil
}
jsonPayload, err := json.Marshal(c.payload)
if err != nil {
return err
}
jsonReader := bytes.NewReader(jsonPayload)
resp, err := http.Post(c.apiURL.String(), "application/json", jsonReader)
if err != nil {
return err
}
defer resp.Body.Close()
c.payload = payloadJSON{}
return err
}
// Close closes the client. It also sends any buffered logs.
func (c *Client) Close() error {
return c.Flush()
}
You can’t perform that action at this time.