Skip to content

Commit

Permalink
Rollback capture with empty UUID (#71)
Browse files Browse the repository at this point in the history
  • Loading branch information
evgeek authored Aug 26, 2024
1 parent f37d6f7 commit 1409f23
Show file tree
Hide file tree
Showing 12 changed files with 46 additions and 60 deletions.
12 changes: 0 additions & 12 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -62,18 +62,6 @@ func main() {
Properties: posthog.NewProperties().
Set("$current_url", "https://example.com"),
})

// Capture event with calculated uuid to deduplicate repeated events.
// The library github.com/google/uuid is used
key := myEvent.Id + myEvent.Project
uid := uuid.NewSHA1(uuid.NameSpaceX500, []byte(key)).String()
client.Enqueue(posthog.Capture{
Uuid: uid,
DistinctId: "test-user",
Event: "$pageview",
Properties: posthog.NewProperties().
Set("$current_url", "https://example.com"),
})

// Check if a feature flag is enabled
isMyFlagEnabled, err := client.IsFeatureEnabled(
Expand Down
6 changes: 1 addition & 5 deletions capture.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,7 @@ type Capture struct {
// This field is exported for serialization purposes and shouldn't be set by
// the application, its value is always overwritten by the library.
Type string
// You don't usually need to specify this field - Posthog will generate it automatically.
// Use it only when necessary - for example, to prevent duplicate events.
Uuid string

DistinctId string
Event string
Timestamp time.Time
Expand Down Expand Up @@ -46,7 +44,6 @@ func (msg Capture) Validate() error {

type CaptureInApi struct {
Type string `json:"type"`
Uuid string `json:"uuid"`
Library string `json:"library"`
LibraryVersion string `json:"library_version"`
Timestamp time.Time `json:"timestamp"`
Expand Down Expand Up @@ -75,7 +72,6 @@ func (msg Capture) APIfy() APIMessage {

apified := CaptureInApi{
Type: msg.Type,
Uuid: msg.Uuid,
Library: library,
LibraryVersion: libraryVersion,
Timestamp: msg.Timestamp,
Expand Down
19 changes: 19 additions & 0 deletions config.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@ package posthog
import (
"net/http"
"time"

"github.com/google/uuid"
)

// Instances of this type carry the different configuration options that may
Expand Down Expand Up @@ -77,6 +79,12 @@ type Config struct {
// If not set the client will fallback to use a default retry policy.
RetryAfter func(int) time.Duration

// A function called by the client to generate unique message identifiers.
// The client uses a UUID generator if none is provided.
// This field is not exported and only exposed internally to let unit tests
// mock the id generation.
uid func() string

// A function called by the client to get the current time, `time.Now` is
// used by default.
// This field is not exported and only exposed internally to let unit tests
Expand Down Expand Up @@ -165,6 +173,10 @@ func makeConfig(c Config) Config {
c.RetryAfter = DefaultBacko().Duration
}

if c.uid == nil {
c.uid = uid
}

if c.now == nil {
c.now = time.Now
}
Expand All @@ -175,3 +187,10 @@ func makeConfig(c Config) Config {

return c
}

// This function returns a string representation of a UUID, it's the default
// function used for generating unique IDs.
func uid() string {
new_uuid, _ := uuid.NewRandom()
return new_uuid.String()
}
22 changes: 0 additions & 22 deletions fixtures/test-enqueue-capture-with-uuid.json

This file was deleted.

3 changes: 1 addition & 2 deletions fixtures/test-enqueue-capture.json
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,7 @@
},
"send_feature_flags": false,
"timestamp": "2009-11-10T23:00:00Z",
"type": "capture",
"uuid": ""
"type": "capture"
}
]
}
3 changes: 1 addition & 2 deletions fixtures/test-interval-capture.json
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,7 @@
},
"send_feature_flags": false,
"timestamp": "2009-11-10T23:00:00Z",
"type": "capture",
"uuid": ""
"type": "capture"
}
]
}
9 changes: 3 additions & 6 deletions fixtures/test-many-capture.json
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,7 @@
},
"send_feature_flags": false,
"timestamp": "2009-11-10T23:00:00Z",
"type": "capture",
"uuid": ""
"type": "capture"
},
{
"distinct_id": "123456",
Expand All @@ -30,8 +29,7 @@
},
"send_feature_flags": false,
"timestamp": "2009-11-10T23:00:00Z",
"type": "capture",
"uuid": ""
"type": "capture"
},
{
"distinct_id": "123456",
Expand All @@ -46,8 +44,7 @@
},
"send_feature_flags": false,
"timestamp": "2009-11-10T23:00:00Z",
"type": "capture",
"uuid": ""
"type": "capture"
}
]
}
3 changes: 1 addition & 2 deletions fixtures/test-merge-capture.json
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,7 @@
},
"send_feature_flags": false,
"timestamp": "2015-07-10T23:00:00Z",
"type": "capture",
"uuid": ""
"type": "capture"
}
]
}
3 changes: 1 addition & 2 deletions fixtures/test-timestamp-capture.json
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,7 @@
},
"send_feature_flags": false,
"timestamp": "2015-07-10T23:00:00Z",
"type": "capture",
"uuid": ""
"type": "capture"
}
]
}
5 changes: 4 additions & 1 deletion go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,10 @@ module github.com/posthog/posthog-go

go 1.18

require github.com/urfave/cli v1.22.5
require (
github.com/google/uuid v1.3.0
github.com/urfave/cli v1.22.5
)

require (
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d // indirect
Expand Down
4 changes: 3 additions & 1 deletion go.sum
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d h1:U+s90UTSYgptZMwQh2aRr3LuazLJIa+Pg3Kc1ylSYVY=
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=
github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/russross/blackfriday/v2 v2.0.1 h1:lPqVAte+HuHNfhJ/0LC98ESWRz8afy9tM/0RK8m9o+Q=
Expand All @@ -10,4 +12,4 @@ github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeV
github.com/urfave/cli v1.22.5 h1:lNq9sAHXK2qfdI8W+GRItjCEkI+2oR4d+MEHy1CKXoU=
github.com/urfave/cli v1.22.5/go.mod h1:Gos4lmkARVdJ6EkW0WaNv/tZAAMe9V7XWyB60NtXRu0=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
17 changes: 12 additions & 5 deletions posthog_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -152,6 +152,8 @@ func fixture(name string) string {
return string(b)
}

func mockId() string { return "I'm unique" }

func mockTime() time.Time {
// time.Unix(0, 0) fails on Circle
return time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC)
Expand Down Expand Up @@ -189,11 +191,11 @@ func ExampleCapture() {
Endpoint: server.URL,
BatchSize: 1,
now: mockTime,
uid: mockId,
})
defer client.Close()

client.Enqueue(Capture{
Uuid: "00000000-0000-0000-0000-000000000000",
Event: "Download",
DistinctId: "123456",
Properties: Properties{
Expand Down Expand Up @@ -223,8 +225,7 @@ func ExampleCapture() {
// },
// "send_feature_flags": false,
// "timestamp": "2009-11-10T23:00:00Z",
// "type": "capture",
// "uuid": "00000000-0000-0000-0000-000000000000"
// "type": "capture"
// }
// ]
// }
Expand All @@ -245,6 +246,7 @@ func TestCaptureNoProperties(t *testing.T) {
Endpoint: server.URL,
BatchSize: 1,
now: mockTime,
uid: mockId,
DefaultEventProperties: NewProperties().Set("service", "api"),
})
defer client.Close()
Expand Down Expand Up @@ -321,9 +323,8 @@ func TestEnqueue(t *testing.T) {
},

"*capture": {
strings.TrimSpace(fixture("test-enqueue-capture-with-uuid.json")),
strings.TrimSpace(fixture("test-enqueue-capture.json")),
&Capture{
Uuid: "11111111-1111-1111-1111-111111111111",
Event: "Download",
DistinctId: "123456",
Properties: Properties{
Expand All @@ -345,6 +346,7 @@ func TestEnqueue(t *testing.T) {
Logger: t,
BatchSize: 1,
now: mockTime,
uid: mockId,
})
defer client.Close()

Expand Down Expand Up @@ -402,6 +404,7 @@ func TestCaptureWithInterval(t *testing.T) {
Verbose: true,
Logger: t,
now: mockTime,
uid: mockId,
})
defer client.Close()

Expand Down Expand Up @@ -438,6 +441,7 @@ func TestCaptureWithTimestamp(t *testing.T) {
Logger: t,
BatchSize: 1,
now: mockTime,
uid: mockId,
})
defer client.Close()

Expand Down Expand Up @@ -471,6 +475,7 @@ func TestCaptureWithDefaultProperties(t *testing.T) {
Logger: t,
BatchSize: 1,
now: mockTime,
uid: mockId,
})
defer client.Close()

Expand Down Expand Up @@ -503,6 +508,7 @@ func TestCaptureMany(t *testing.T) {
Logger: t,
BatchSize: 3,
now: mockTime,
uid: mockId,
})
defer client.Close()

Expand Down Expand Up @@ -1705,6 +1711,7 @@ func TestCaptureSendFlags(t *testing.T) {
Logger: t,
BatchSize: 1,
now: mockTime,
uid: mockId,

PersonalApiKey: "some very secret key",
})
Expand Down

0 comments on commit 1409f23

Please sign in to comment.