-
Notifications
You must be signed in to change notification settings - Fork 776
/
tracker.go
116 lines (98 loc) · 2.57 KB
/
tracker.go
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
/*
Copyright 2021 The Fission Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package tracker
import (
"bytes"
"context"
"errors"
"fmt"
"net/http"
"net/url"
"os"
"time"
"github.com/fission/fission/pkg/utils/uuid"
)
const (
HTTP_TIMEOUT time.Duration = 5 * time.Second
GA_TRACKING_ID string = "GA_TRACKING_ID"
GA_API_URL string = "GA_API_URL"
)
type (
Tracker struct {
gaPropertyID string
cid string
gaAPIURL string
}
Event struct {
Category string
Action string
Label string
Value string
}
)
func NewTracker() (*Tracker, error) {
id := uuid.NewString()
gaTrackingID := os.Getenv(GA_TRACKING_ID)
if gaTrackingID == "" {
return nil, errors.New("tracker.NewTracker: GA_TRACKING_ID env not set")
}
gaAPIURL := os.Getenv(GA_API_URL)
if gaAPIURL == "" {
gaAPIURL = "https://www.google-analytics.com/collect"
}
tracker := &Tracker{
gaPropertyID: gaTrackingID,
cid: id,
gaAPIURL: gaAPIURL,
}
return tracker, nil
}
func (t *Tracker) SendEvent(ctx context.Context, e Event) error {
if e.Action == "" || e.Category == "" {
return errors.New("tracker.SendEvent: category and action are required")
}
v := url.Values{
"v": {"1"},
"tid": {t.gaPropertyID},
"cid": {t.cid},
"t": {"event"},
"ec": {e.Category},
"ea": {e.Action},
}
if e.Label != "" {
v.Add("el", e.Label)
}
if e.Value != "" {
v.Add("ev", e.Value)
}
buf := bytes.NewBufferString(v.Encode())
req, err := http.NewRequestWithContext(ctx, "POST", t.gaAPIURL, buf)
req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
req.Header.Add("User-Agent", "ga-tracker/1.0")
if err != nil {
return err
}
ctx, cancel := context.WithTimeoutCause(req.Context(), HTTP_TIMEOUT, fmt.Errorf("tracker request timeout (%f)s exceeded", HTTP_TIMEOUT.Seconds()))
defer cancel()
req = req.WithContext(ctx)
client := http.DefaultClient
resp, err := client.Do(req)
if err != nil {
return err
}
if resp.StatusCode != http.StatusOK {
return errors.New("tracker.SendEvent: analytics response status not ok")
}
defer resp.Body.Close()
return err
}