chaas/traefik-rybbit-feeder
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update worker to send api key in header

Browse source
This commit is contained in:
FoxxMD 2026-02-26 15:31:18 -05:00
parent bb2da5f403
commit 140d4dc873
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
umami_worker.go
View file

@ -9,7 +9,6 @@ import (
) )
type RybbitEvent struct { type RybbitEvent struct {
APIKey string `json:"api_key"`
SiteID string `json:"site_id"` SiteID string `json:"site_id"`
Type string `json:"type"` Type string `json:"type"`
Pathname string `json:"pathname"` Pathname string `json:"pathname"`
@ -25,6 +24,7 @@ type RybbitEvent struct {
type SendBody struct { type SendBody struct {
Payload *RybbitEvent `json:"payload"` Payload *RybbitEvent `json:"payload"`
Type string `json:"type"` Type string `json:"type"`
ApiKey string
} }
func (h *UmamiFeeder) submitToFeed(req *http.Request, code int) { func (h *UmamiFeeder) submitToFeed(req *http.Request, code int) {
@ -37,7 +37,6 @@ func (h *UmamiFeeder) submitToFeed(req *http.Request, code int) {
} }
rEvent := &RybbitEvent{ rEvent := &RybbitEvent{
APIKey: h.apiKey,
SiteID: websiteId, SiteID: websiteId,
Type: "pageview", Type: "pageview",
Pathname: req.URL.Path, Pathname: req.URL.Path,
@ -89,7 +88,7 @@ func (h *UmamiFeeder) umamiEventFeeder(ctx context.Context) (err error) {
return nil return nil
case event := <-h.queue: case event := <-h.queue:
batch = append(batch, &SendBody{Payload: event, Type: "event"}) batch = append(batch, &SendBody{Payload: event, Type: "event", ApiKey: h.apiKey})
if len(batch) >= h.batchSize { if len(batch) >= h.batchSize {
h.reportEventsToUmami(ctx, batch) h.reportEventsToUmami(ctx, batch)
batch = make([]*SendBody, 0, h.batchSize) batch = make([]*SendBody, 0, h.batchSize)
@ -109,7 +108,10 @@ func (h *UmamiFeeder) umamiEventFeeder(ctx context.Context) (err error) {
func (h *UmamiFeeder) reportEventsToUmami(ctx context.Context, events []*SendBody) { func (h *UmamiFeeder) reportEventsToUmami(ctx context.Context, events []*SendBody) {
h.debug("reporting %d events", len(events)) h.debug("reporting %d events", len(events))
for _, value := range events { for _, value := range events {
resp, err := sendRequest(ctx, h.host+"/api/track", value.Payload, nil) headers := map[string][]string{
"Authorization": {"Bearer " + value.ApiKey},
}
resp, err := sendRequest(ctx, h.host+"/api/track", value.Payload, headers)
if err != nil { if err != nil {
h.error("failed to send tracking: " + err.Error()) h.error("failed to send tracking: " + err.Error())
return return