83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
package activity
|
|
|
|
import (
|
|
"context"
|
|
influxdb2 "github.com/influxdata/influxdb-client-go/v2"
|
|
"github.com/influxdata/influxdb-client-go/v2/api"
|
|
"github.com/influxdata/influxdb-client-go/v2/api/write"
|
|
"github.com/wormhole-foundation/wormhole-explorer/common/dbconsts"
|
|
"github.com/wormhole-foundation/wormhole-explorer/jobs/jobs/protocols/activity/internal/repositories"
|
|
"go.uber.org/zap"
|
|
"strconv"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// NewProtocolActivityJob creates an instance of the job implementation.
|
|
func NewProtocolActivityJob(statsDB api.WriteAPIBlocking, logger *zap.Logger, version string, activityFetchers ...ClientActivity) *ProtocolsActivityJob {
|
|
return &ProtocolsActivityJob{
|
|
statsDB: statsDB,
|
|
logger: logger.With(zap.String("module", "ProtocolsActivityJob")),
|
|
activityFetchers: activityFetchers,
|
|
version: version,
|
|
}
|
|
}
|
|
|
|
func (m *ProtocolsActivityJob) Run(ctx context.Context) error {
|
|
|
|
clientsQty := len(m.activityFetchers)
|
|
wg := sync.WaitGroup{}
|
|
wg.Add(clientsQty)
|
|
errs := make(chan error, clientsQty)
|
|
ts := time.Now().UTC().Truncate(time.Hour) // make minutes and seconds zero, so we only work with date and hour
|
|
from := time.Unix(0, 0).UTC()
|
|
m.logger.Info("running protocols activity job ", zap.Time("from", from), zap.Time("to", ts))
|
|
for _, cs := range m.activityFetchers {
|
|
go func(c ClientActivity) {
|
|
defer wg.Done()
|
|
activity, err := c.Get(ctx, from, ts)
|
|
if err != nil {
|
|
errs <- err
|
|
return
|
|
}
|
|
errs <- m.updateActivity(ctx, c.ProtocolName(), m.version, activity, ts)
|
|
}(cs)
|
|
}
|
|
|
|
wg.Wait()
|
|
close(errs)
|
|
for err := range errs {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *ProtocolsActivityJob) updateActivity(ctx context.Context, protocol, version string, activity repositories.ProtocolActivity, ts time.Time) error {
|
|
|
|
points := make([]*write.Point, 0, len(activity.Activities))
|
|
|
|
for i := range activity.Activities {
|
|
point := influxdb2.
|
|
NewPointWithMeasurement(dbconsts.ProtocolsActivityMeasurement).
|
|
AddTag("protocol", protocol).
|
|
AddTag("emitter_chain_id", strconv.FormatUint(activity.Activities[i].EmitterChainID, 10)).
|
|
AddTag("destination_chain_id", strconv.FormatUint(activity.Activities[i].DestinationChainID, 10)).
|
|
AddTag("version", version).
|
|
AddField("total_value_secure", activity.TotalValueSecure).
|
|
AddField("total_value_transferred", activity.TotalValueTransferred).
|
|
AddField("txs", activity.Activities[i].Txs).
|
|
AddField("total_usd", activity.Activities[i].TotalUSD).
|
|
SetTime(ts)
|
|
points = append(points, point)
|
|
}
|
|
|
|
err := m.statsDB.WritePoint(ctx, points...)
|
|
if err != nil {
|
|
m.logger.Error("failed updating protocol Activities in influxdb", zap.Error(err), zap.String("protocol", protocol))
|
|
}
|
|
return err
|
|
}
|