Add more db tracers

handlers/badges.go

@@ -23,6 +23,7 @@ import (
 	"github.com/sirupsen/logrus"
 	"github.com/wahyd4/badger/models"
 	"github.com/wahyd4/badger/utils"
+	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
 )
 
 var (
@@ -311,6 +312,9 @@ func (handler *BadgesHandler) generateSVG(c *gin.Context, colorScheme ColorSchem
 }
 
 func (handler *BadgesHandler) queryBadgeFromDB(queryName string) (*models.Badge, error) {
+	span := tracer.StartSpan("db.queryBadge")
+	defer span.Finish()
+
 	var badge models.Badge
 	err := handler.DB.Find(&badge, "ulid = ?", queryName).Error
 	if err != nil {
@@ -427,6 +431,9 @@ func (handler *BadgesHandler) queryMetric(badge *models.Badge, internalRequest b
 }
 
 func (handler *BadgesHandler) fetchBadgeValueFromDB(badge *models.Badge) (*models.BadgeAPIValue, error) {
+	span := tracer.StartSpan("db.fetchBadgeValue")
+	defer span.Finish()
+
 	var badgeValue models.BadgeAPIValue
 	err := handler.DB.Order("id DESC").First(&badgeValue, "ulid = ?", badge.ULID).Error
 	if err != nil {
@@ -439,6 +446,9 @@ func (handler *BadgesHandler) fetchBadgeValueFromDB(badge *models.Badge) (*model
 }
 
 func (handler *BadgesHandler) fetchViewCountFromDB(badge *models.Badge) (*models.BadgeViewCount, error) {
+	span := tracer.StartSpan("db.fetchViewCount")
+	defer span.Finish()
+
 	var viewCount models.BadgeViewCount
 	err := handler.DB.Find(&viewCount, "ulid = ?", badge.ULID).Error
 	if err != nil {

handlers/stats.go

@@ -38,6 +38,9 @@ func (handler *StatsHandler) GetStats(c *gin.Context) {
 }
 
 func (handler *StatsHandler) badgeStats(badge *models.Badge) ([]*StatsResult, error) {
+	span := tracer.StartSpan("db.badgeStats")
+	defer span.Finish()
+
 	var result = make([]*StatsResult, 0)
 	err := handler.DB.Raw(`select
 		date_trunc('day', created_at AT TIME ZONE 'Australia/Melbourne') as date, -- or hour, day, week, month, year