mirror of
https://github.com/usememos/memos.git
synced 2024-12-19 17:12:02 +03:00
41 lines
882 B
Go
41 lines
882 B
Go
package segment
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/segmentio/analytics-go"
|
|
metric "github.com/usememos/memos/plugin/metrics"
|
|
)
|
|
|
|
var _ metric.Collector = (*collector)(nil)
|
|
|
|
// collector is the metrics collector https://segment.com/.
|
|
type collector struct {
|
|
client analytics.Client
|
|
}
|
|
|
|
// NewCollector creates a new instance of segment.
|
|
func NewCollector(key string) metric.Collector {
|
|
client := analytics.New(key)
|
|
|
|
return &collector{
|
|
client: client,
|
|
}
|
|
}
|
|
|
|
// Collect will exec all the segment collector.
|
|
func (c *collector) Collect(metric *metric.Metric) error {
|
|
properties := analytics.NewProperties()
|
|
for key, value := range metric.Labels {
|
|
properties.Set(key, value)
|
|
}
|
|
|
|
return c.client.Enqueue(analytics.Track{
|
|
Event: string(metric.Name),
|
|
AnonymousId: uuid.NewString(),
|
|
Properties: properties,
|
|
Timestamp: time.Now().UTC(),
|
|
})
|
|
}
|