feat: get onchain asset average cost (#1398)

cmd/update-user-onchain-asset-average-cost/main.go

@@ -0,0 +1,33 @@
+package main
+
+import (
+	"time"
+
+	"github.com/defipod/mochi/pkg/config"
+	"github.com/defipod/mochi/pkg/entities"
+	"github.com/defipod/mochi/pkg/job"
+	"github.com/defipod/mochi/pkg/logger"
+	"github.com/defipod/mochi/pkg/service"
+)
+
+func main() {
+	cfg := config.LoadConfig(config.DefaultConfigLoaders())
+	log := logger.NewLogrusLogger()
+	// *** entities ***
+	err := entities.Init(cfg, log)
+	if err != nil {
+		log.Fatal(err, "failed to init entities")
+	}
+
+	service, err := service.NewService(cfg, log)
+	if err != nil {
+		log.Fatal(err, "failed to init service")
+	}
+	defer service.Sentry.Flush(2 * time.Second)
+
+	if err := job.NewUpdateUserOnchainAssetAvgCostJob(entities.Get()).Run(); err != nil {
+		log.Fatal(err, "failed to run job")
+	}
+
+	log.Info("done")
+}

migrations/schemas/20240531090511-add-onchain-asset-avg-costs-table.sql

@@ -0,0 +1,15 @@
+-- +migrate Up
+CREATE TABLE IF NOT EXISTS onchain_asset_avg_costs (
+    wallet_address TEXT NOT NULL,
+    token_address TEXT NOT NULL,
+    symbol TEXT NOT NULL,
+    blockchain TEXT NOT NULL,
+    average_cost FLOAT NOT NULL,
+    created_at TIMESTAMPTZ DEFAULT now(),
+    updated_at TIMESTAMPTZ DEFAULT now()
+);
+
+CREATE UNIQUE INDEX "uidx_wallet_address_token_address_chain" ON onchain_asset_avg_costs (wallet_address, token_address, blockchain);
+
+-- +migrate Down
+DELETE TABLE IF EXISTS onchain_asset_avg_costs;

pkg/config/config.go

@@ -146,6 +146,8 @@ type Config struct {
 
 	SentryDsn    string
 	TonConfigURL string
+
+	DuneApiKey string
 }
 
 type MarketplaceBaseUrl struct {
@@ -383,6 +385,8 @@ func generateConfigFromViper(v *viper.Viper) Config {
 
 		SentryDsn:    v.GetString("SENTRY_DSN"),
 		TonConfigURL: v.GetString("TON_CONFIG_URL"),
+
+		DuneApiKey: v.GetString("DUNE_API_KEY"),
 	}
 }
 

pkg/entities/onchain_asset_avg_cost.go

@@ -0,0 +1,23 @@
+package entities
+
+import (
+	"github.com/defipod/mochi/pkg/model"
+)
+
+func (e *Entity) UpsertManyOnchainAssetAvgCost(assets []model.OnchainAssetAvgCost) error {
+	err := e.repo.OnchainAssetAverageCost.UpsertMany(assets)
+	if err != nil {
+		e.log.Error(err, "[entities.UpsertManyOnchainAssetAvgCost] - repo.OnchainAssetAverageCost.UpsertMany failed")
+		return err
+	}
+	return nil
+}
+
+func (e *Entity) GetOnchainAssetAvgCostByWalletAddress(walletAddress string) ([]model.OnchainAssetAvgCost, error) {
+	assets, err := e.repo.OnchainAssetAverageCost.GetByWalletAddr(walletAddress)
+	if err != nil {
+		e.log.Error(err, "[entities.GetOnchainAssetAvgCostByWalletAddress] - repo.OnchainAssetAverageCost.GetByWalletAddress failed")
+		return nil, err
+	}
+	return assets, nil
+}

pkg/entities/swap.go

@@ -216,3 +216,14 @@ func (e *Entity) Swap(req request.SwapRequest) (interface{}, error) {
 	}
 	return nil, nil
 }
+
+func (e *Entity) GetOnchainAssetAvgCost(req request.GetOnchainAssetAvgCost) ([]model.OnchainAssetAvgCost, error) {
+	// get all onchain asset avg cost
+	assets, err := e.repo.OnchainAssetAverageCost.GetByWalletAddr(req.WalletAddress)
+	if err != nil {
+		e.log.Fields(logger.Fields{"req": req}).Error(err, "[entity.GetOnchainAssetAvgCost] - repo.OnchainAssetAverageCost.GetByWalletAddr() failed")
+		return nil, err
+	}
+
+	return assets, nil
+}

pkg/handler/swap/interface.go

@@ -6,4 +6,5 @@ type IHandler interface {
 	GetSwapRoutes(c *gin.Context)
 	ExecuteSwapRoutes(c *gin.Context)
 	OnchainData(c *gin.Context)
+	GetOnchainAssetAvgCost(c *gin.Context)
 }

pkg/handler/swap/swap.go

@@ -104,3 +104,21 @@ func (h *Handler) OnchainData(c *gin.Context) {
 
 	c.JSON(http.StatusOK, response.CreateResponse[any](data, nil, nil, nil))
 }
+
+func (h *Handler) GetOnchainAssetAvgCost(c *gin.Context) {
+	var req request.GetOnchainAssetAvgCost
+	if err := c.ShouldBindQuery(&req); err != nil {
+		h.log.Fields(logger.Fields{"req": req}).Error(err, "[handler.GetOnchainAssetAvgCost] - failed to read JSON")
+		c.JSON(http.StatusBadRequest, response.CreateResponse[any](nil, nil, err, nil))
+		return
+	}
+
+	data, err := h.entities.GetOnchainAssetAvgCost(req)
+	if err != nil {
+		h.log.Fields(logger.Fields{"wallet_address": req.WalletAddress}).Error(err, "[handler.GetOnchainAssetAvgCost] - entity.GetOnchainAssetAvgCost() failed")
+		c.JSON(baseerrs.GetStatusCode(err), response.CreateResponse[any](nil, nil, err, nil))
+		return
+	}
+
+	c.JSON(http.StatusOK, response.CreateResponse[any](data, nil, nil, nil))
+}