nft质押任务逻辑

internal/svc/service_context.go

@@ -3,6 +3,7 @@ package svc
 import (
 	ea "github.com/earn-alliance/earnalliance-go"
 	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/core/stores/redis"
 	"github.com/zeromicro/go-zero/core/stores/sqlx"
 	"nova_task/internal/config"
 	"nova_task/internal/model"
@@ -24,9 +25,14 @@ type ServiceContext struct {
 	ConfigModel             model.NhSystemConfigModel
 	NftHolderModel          model.NhNftHolderModel
 	NftHolderChangeLogModel model.NhNftHolderChangeLogModel
+	StakeNftModel           model.NhTaskNftStakeModel
+	StakeNftLogModel        model.NhTaskNftStakeLogModel
+	StakeRewardModel        model.NhTaskNftStakeRewardModel
+	GamePitModel            model.NhGamePitModel
 
 	Earn   *ea.Client
 	DBConn sqlx.SqlConn
+	Redis  *redis.Redis
 }
 
 func NewServiceContext(c config.Config) *ServiceContext {
@@ -47,9 +53,14 @@ func NewServiceContext(c config.Config) *ServiceContext {
 		ConfigModel:             model.NewNhSystemConfigModel(dbConn, c.Cache),
 		NftHolderModel:          model.NewNhNftHolderModel(dbConn),
 		NftHolderChangeLogModel: model.NewNhNftHolderChangeLogModel(dbConn),
+		StakeNftModel:           model.NewNhTaskNftStakeModel(dbConn),
+		StakeNftLogModel:        model.NewNhTaskNftStakeLogModel(dbConn),
+		StakeRewardModel:        model.NewNhTaskNftStakeRewardModel(dbConn),
+		GamePitModel:            model.NewNhGamePitModel(dbConn),
 
 		Earn:   c.Earn.BuildEarnClient(),
 		DBConn: dbConn,
+		Redis:  redis.MustNewRedis(c.Redis),
 	}
 }