From ea1e08032e500030b1eb4d6e05ab80e10248d427 Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Mon, 24 Jun 2024 11:32:02 +0800
Subject: [PATCH] 1

---
 server/backtask/task/spec_transfer721/hero.go | 22 ++++++++++++++++---
 .../backtask/task/spec_transfer721/types.go   |  6 -----
 server/jccommon/types.go                      |  6 +++++
 3 files changed, 25 insertions(+), 9 deletions(-)

diff --git a/server/backtask/task/spec_transfer721/hero.go b/server/backtask/task/spec_transfer721/hero.go
index 8d3967b2..e9e67008 100644
--- a/server/backtask/task/spec_transfer721/hero.go
+++ b/server/backtask/task/spec_transfer721/hero.go
@@ -5,11 +5,12 @@ import (
 	"main/service"
 	"main/constant"
 	"mt"
+	"jccommon"
 	//"fmt"
 )
 
 type hero struct {
-	mailCfgHash *q5.ConcurrentMap[string, *mailConfig]
+	mailCfgHash *q5.ConcurrentMap[string, *jccommon.MailConfig]
 }
 
 func (this* hero) onMint(dbIdx int64, netId int32, contractAddress string, tokenId string,
@@ -47,11 +48,15 @@ func (this* hero) internalSendMail(dbIdx int64, accountAddress string, mailName
 	if mailMeta == nil {
 		return true
 	}
+	mailCfg := this.getMailConfigMeta(mailName)
+	if mailCfg == nil {
+		return true
+	}
 	//uniKey := fmt.Sprintf("%d_%s_%s", dbIdx, mailName, tokenId)
 	return true
 }
 
-func (this *hero) getMailConfigMeta(mailName string) *mailConfig {
+func (this *hero) getMailConfigMeta(mailName string) *jccommon.MailConfig {
 	if v, ok := this.mailCfgHash.Load(mailName); ok {
 		return *v
 	} else {
@@ -59,8 +64,19 @@ func (this *hero) getMailConfigMeta(mailName string) *mailConfig {
 	}
 }
 
+func (this *hero) registerMailConfig(mailName string, tag1 int32, tag2 int32) {
+	p := new(jccommon.MailConfig)
+	p.MailName = mailName
+	p.Tag1 = tag1
+	p.Tag2 = tag2
+	this.mailCfgHash.Store(mailName, p)
+}
+
 func newHero() *hero {
 	p := new(hero)
-	p.mailCfgHash = new(q5.ConcurrentMap[string, *mailConfig])
+	p.mailCfgHash = new(q5.ConcurrentMap[string, *jccommon.MailConfig])
+	p.registerMailConfig(constant.MAIL_HERO_MINT, jccommon.MAIL_TAG1_HERO, jccommon.MAIL_TAG2_HERO_MINT)
+	p.registerMailConfig(constant.MAIL_HERO_LOCK, jccommon.MAIL_TAG1_HERO, jccommon.MAIL_TAG2_HERO_LOCK)
+	p.registerMailConfig(constant.MAIL_HERO_UNLOCK, jccommon.MAIL_TAG1_HERO, jccommon.MAIL_TAG2_HERO_UNLOCK)
 	return p
 }
diff --git a/server/backtask/task/spec_transfer721/types.go b/server/backtask/task/spec_transfer721/types.go
index ee612117..b26dc1ea 100644
--- a/server/backtask/task/spec_transfer721/types.go
+++ b/server/backtask/task/spec_transfer721/types.go
@@ -8,9 +8,3 @@ type specTransfer721Handle interface {
 	onUnlock(dbIdx int64, netId int32, contractAddress string, tokenId string,
 		from string, to string) bool
 }
-
-type mailConfig struct {
-	mailName string
-	tag1 int32
-	tag2 int32
-}
diff --git a/server/jccommon/types.go b/server/jccommon/types.go
index 91ccf67c..4420806e 100644
--- a/server/jccommon/types.go
+++ b/server/jccommon/types.go
@@ -41,3 +41,9 @@ type OrderUpdatedEvent struct {
 		UpdatedAt interface{} `json:"updated_at"`
 	} `json:"data"`
 }
+
+type MailConfig struct {
+	MailName string
+	Tag1 int32
+	Tag2 int32
+}