amplitude-go/client.go at master · msingleton/amplitude-go · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// Package amplitude provides access to the Amplitude API
package amplitude

import (
	"encoding/json"
	"net/http"
	"net/url"
)

const eventEndpoint = "https://api.amplitude.com/httpapi"
const identifyEndpoint = "https://api.amplitude.com/identify"

// Client manages the communication to the Amplitude API
type Client struct {
	eventEndpoint    string
	identifyEndpoint string
	key              string
	client           *http.Client
}

type Event struct {
	Adid               string                 `json:"adid,omitempty"`
	AppVersion         string                 `json:"app_version,omitempty"`
	Carrier            string                 `json:"carrier,omitempty"`
	City               string                 `json:"city,omitempty"`
	Country            string                 `json:"country,omitempty"`
	DeviceBrand        string                 `json:"device_brand,omitempty"`
	DeviceId           string                 `json:"device_id,omitempty"`
	DeviceManufacturer string                 `json:"device_manufacturer,omitempty"`
	DeviceModel        string                 `json:"device_model,omitempty"`
	DeviceType         string                 `json:"device_type,omitempty"`
	Dma                string                 `json:"dma,omitempty"`
	EventId            int                    `json:"event_id,omitempty"`
	EventProperties    map[string]interface{} `json:"event_properties,omitempty"`
	EventType          string                 `json:"event_type,omitempty"`
	Groups             map[string]interface{} `json:"groups,omitempty"`
	Ifda               string                 `json:"ifda,omitempty"`
	InsertId           string                 `json:"insert_id,omitempty"`
	Ip                 string                 `json:"ip,omitempty"`
	Language           string                 `json:"language,omitempty"`
	LocationLat        string                 `json:"location_lat,omitempty"`
	LocationLng        string                 `json:"location_lng,omitempty"`
	OsName             string                 `json:"os_name,omitempty"`
	OsVersion          string                 `json:"os_version,omitempty"`
	Paying             string                 `json:"paying,omitempty"`
	Platform           string                 `json:"platform,omitempty"`
	Price              float64                `json:"price,omitempty"`
	ProductId          string                 `json:"productId,omitempty"`
	Quantity           int                    `json:"quantity,omitempty"`
	Region             string                 `json:"region,omitempty"`
	Revenue            float64                `json:"revenue,omitempty"`
	RevenueType        string                 `json:"revenueType,omitempty"`
	SessionId          int64                  `json:"session_id,omitempty"`
	StartVersion       string                 `json:"start_version,omitempty"`
	Time               int64                  `json:"time,omitempty"`
	UserId             string                 `json:"user_id,omitempty"`
	UserProperties     map[string]interface{} `json:"user_properties,omitempty"`
}

type Identify struct {
	AppVersion         string                 `json:"app_version,omitempty"`
	Carrier            string                 `json:"carrier,omitempty"`
	City               string                 `json:"city,omitempty"`
	Country            string                 `json:"country,omitempty"`
	DeviceBrand        string                 `json:"device_brand,omitempty"`
	DeviceId           string                 `json:"device_id,omitempty"`
	DeviceManufacturer string                 `json:"device_manufacturer,omitempty"`
	DeviceModel        string                 `json:"device_model,omitempty"`
	DeviceType         string                 `json:"device_type,omitempty"`
	Dma                string                 `json:"dma,omitempty"`
	Groups             map[string]interface{} `json:"groups,omitempty"`
	Language           string                 `json:"language,omitempty"`
	OsName             string                 `json:"os_name,omitempty"`
	OsVersion          string                 `json:"os_version,omitempty"`
	Paying             string                 `json:"paying,omitempty"`
	Platform           string                 `json:"platform,omitempty"`
	Region             string                 `json:"region,omitempty"`
	StartVersion       string                 `json:"start_version,omitempty"`
	UserId             string                 `json:"user_id,omitempty"`
	UserProperties     map[string]interface{} `json:"user_properties,omitempty"`
}

// New client with API key
func New(key string) *Client {
	return &Client{
		eventEndpoint:    eventEndpoint,
		identifyEndpoint: identifyEndpoint,
		key:              key,
		client:           new(http.Client),
	}
}

func (c *Client) SetClient(client *http.Client) {
	c.client = client
}

func (c *Client) Event(msg Event) error {
	msgJson, err := json.Marshal(msg)
	if err != nil {
		return err
	}

	return c.send(c.eventEndpoint, "event", string(msgJson))
}

func (c *Client) Identify(msg Identify) error {
	msgJson, err := json.Marshal(msg)
	if err != nil {
		return err
	}

	return c.send(c.identifyEndpoint, "identification", string(msgJson))
}

func (c *Client) send(endpoint string, key string, value string) error {
	values := url.Values{}
	values.Add("api_key", c.key)
	values.Add(key, value)

	resp, err := c.client.PostForm(endpoint, values)
	if err == nil {
		resp.Body.Close()
	}
	return err
}