usdt

internal/job/airdrop/apt.go

@@ -0,0 +1,94 @@
+package apt
+
+import (
+	"github.com/aptos-labs/aptos-go-sdk"
+	"github.com/aptos-labs/aptos-go-sdk/crypto"
+	"github.com/zeromicro/go-zero/core/threading"
+)
+
+type Apt struct {
+	client *aptos.Client
+	sender *aptos.Account
+
+	payloads chan aptos.TransactionBuildPayload
+	results  chan aptos.TransactionSubmissionResponse
+}
+
+func NewApt(privateKeyStr string, isTest bool, resultHandler func(aptos.TransactionSubmissionResponse)) (*Apt, error) {
+	var networkConfig aptos.NetworkConfig
+	if isTest {
+		networkConfig = aptos.TestnetConfig
+	} else {
+		networkConfig = aptos.MainnetConfig
+	}
+	client, err := aptos.NewClient(networkConfig)
+	if err != nil {
+		return nil, err
+	}
+
+	privateKey := &crypto.Ed25519PrivateKey{}
+	err = privateKey.FromHex(privateKeyStr)
+	if err != nil {
+		return nil, err
+	}
+
+	sender, err := aptos.NewAccountFromSigner(privateKey)
+	if err != nil {
+		return nil, err
+	}
+
+	payloads := make(chan aptos.TransactionBuildPayload, 100)
+	results := make(chan aptos.TransactionSubmissionResponse, 100)
+
+	threading.GoSafe(func() {
+		client.BuildSignAndSubmitTransactions(sender, payloads, results)
+	})
+
+	a := &Apt{
+		client:   client,
+		sender:   sender,
+		payloads: payloads,
+		results:  results,
+	}
+
+	threading.GoSafe(func() {
+		for result := range results {
+			a.handleResult(result)
+		}
+	})
+
+	return a, nil
+}
+
+func (a *Apt) transferUsdt(id uint64, toAddress string, amount uint64) error {
+	receiver := aptos.AccountAddress{}
+	err := receiver.ParseStringRelaxed(toAddress)
+	if err != nil {
+		return err
+	}
+
+	p, err := aptos.CoinTransferPayload(nil, receiver, amount)
+	if err != nil {
+		return err
+	}
+
+	a.payloads <- aptos.TransactionBuildPayload{
+		Id:    id,
+		Type:  aptos.TransactionSubmissionTypeSingle,
+		Inner: aptos.TransactionPayload{Payload: p},
+	}
+	return nil
+}
+
+func (a *Apt) handleResult(result aptos.TransactionSubmissionResponse) {
+
+}
+
+func (a *Apt) Start() {
+
+}
+
+func (a *Apt) Stop() {
+	close(a.payloads)
+	close(a.results)
+}