fathom/pkg/aggregator/store.go

package aggregator

import (
	"strings"
	"time"

	"github.com/usefathom/fathom/pkg/datastore"
	"github.com/usefathom/fathom/pkg/models"
)

func (agg *aggregator) getSiteStats(r *results, t time.Time) (*models.SiteStats, error) {
	// get from map
	date := t.Format("2006-01-02")
	if stats, ok := r.Sites[date]; ok {
		return stats, nil

	}

	// get from db
	stats, err := agg.database.GetSiteStats(0, t)
	if err != nil && err != datastore.ErrNoResults {
		return nil, err
	}

	// create in db
	if stats == nil {
		stats = &models.SiteStats{
			Date: t,
		}

		err = agg.database.InsertSiteStats(stats)
		if err != nil {
			return nil, err
		}
	}

	r.Sites[date] = stats
	return stats, nil
}

func (agg *aggregator) getPageStats(r *results, t time.Time, hostname string, pathname string) (*models.PageStats, error) {
	date := t.Format("2006-01-02")
	if stats, ok := r.Pages[date+hostname+pathname]; ok {
		return stats, nil
	}

	stats, err := agg.database.GetPageStats(0, t, hostname, pathname)
	if err != nil && err != datastore.ErrNoResults {
		return nil, err
	}

	if stats == nil {
		stats = &models.PageStats{
			Hostname: hostname,
			Pathname: pathname,
			Date:     t,
		}
		err = agg.database.InsertPageStats(stats)
		if err != nil {
			return nil, err
		}
	}

	r.Pages[date+hostname+pathname] = stats
	return stats, nil
}

func (agg *aggregator) getReferrerStats(r *results, t time.Time, hostname string, pathname string) (*models.ReferrerStats, error) {
	date := t.Format("2006-01-02")
	if stats, ok := r.Referrers[date+hostname+pathname]; ok {
		return stats, nil
	}

	// get from db
	stats, err := agg.database.GetReferrerStats(0, t, hostname, pathname)
	if err != nil && err != datastore.ErrNoResults {
		return nil, err
	}

	// create in db
	if stats == nil {
		stats = &models.ReferrerStats{
			Hostname: hostname,
			Pathname: pathname,
			Date:     t,
			Group:    "",
		}

		// TODO: Abstract this
		if strings.Contains(stats.Hostname, "www.google.") {
			stats.Group = "Google"
		}

		err = agg.database.InsertReferrerStats(stats)
		if err != nil {
			return nil, err
		}
	}

	r.Referrers[date+hostname+pathname] = stats
	return stats, nil
}