refactor: separate brc20 processing logic to smaller funcs

modules/brc20/processor_brc20.go

@@ -97,25 +97,16 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 	newEventTransferTransfers := make([]*entity.EventTransferTransfer, 0)
 	newBalances := make(map[string]map[string]*entity.Balance)
 
-	for _, payload := range payloads {
-		tickEntry := tickEntries[payload.Tick]
-
-		if payload.Transfer.SentAsFee && payload.Transfer.OldSatPoint == (ordinals.SatPoint{}) {
-			// skip inscriptions inscribed as fee
-			continue
-		}
-
-		switch payload.Op {
-		case brc20.OperationDeploy:
+	handleEventDeploy := func(payload *brc20.Payload, tickEntry *entity.TickEntry) {
 		if payload.Transfer.TransferCount > 1 {
 			// skip used deploy inscriptions
-				continue
+			return
 		}
 		if tickEntry != nil {
 			// skip deploy inscriptions for duplicate ticks
-				continue
+			return
 		}
-			tickEntry := &entity.TickEntry{
+		newEntry := &entity.TickEntry{
 			Tick:                payload.Tick,
 			OriginalTick:        payload.OriginalTick,
 			TotalSupply:         payload.Max,
@@ -130,10 +121,10 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 			CompletedAt:         time.Time{},
 			CompletedAtHeight:   0,
 		}
-			newTickEntries[payload.Tick] = tickEntry
-			newTickEntryStates[payload.Tick] = tickEntry
+		newTickEntries[payload.Tick] = newEntry
+		newTickEntryStates[payload.Tick] = newEntry
 		// update entries for other operations in same block
-			tickEntries[payload.Tick] = tickEntry
+		tickEntries[payload.Tick] = newEntry
 
 		newEventDeploys = append(newEventDeploys, &entity.EventDeploy{
 			Id:                latestEventId + 1,
@@ -153,26 +144,27 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 			IsSelfMint:        payload.SelfMint,
 		})
 		latestEventId++
-		case brc20.OperationMint:
+	}
+	handleEventMint := func(payload *brc20.Payload, tickEntry *entity.TickEntry) {
 		if payload.Transfer.TransferCount > 1 {
 			// skip used mint inscriptions that are already used
-				continue
+			return
 		}
 		if tickEntry == nil {
 			// skip mint inscriptions for non-existent ticks
-				continue
+			return
 		}
 		if -payload.Amt.Exponent() > int32(tickEntry.Decimals) {
 			// skip mint inscriptions with decimals greater than allowed
-				continue
+			return
 		}
 		if tickEntry.MintedAmount.GreaterThanOrEqual(tickEntry.TotalSupply) {
 			// skip mint inscriptions for ticks with completed mints
-				continue
+			return
 		}
 		if payload.Amt.GreaterThan(tickEntry.LimitPerMint) {
 			// skip mint inscriptions with amount greater than limit per mint
-				continue
+			return
 		}
 		mintableAmount := tickEntry.TotalSupply.Sub(tickEntry.MintedAmount)
 		if payload.Amt.GreaterThan(mintableAmount) {
@@ -183,16 +175,17 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 			parentIdValue, ok := inscriptionIdsToParent[payload.Transfer.InscriptionId]
 			if !ok {
 				// skip mint inscriptions for self mint ticks without parent inscription
-					continue
+				return
 			}
 			if parentIdValue != tickEntry.DeployInscriptionId {
 				// skip mint inscriptions for self mint ticks with invalid parent inscription
-					continue
+				return
 			}
 			parentId = &parentIdValue
 		}
 
 		tickEntry.MintedAmount = tickEntry.MintedAmount.Add(payload.Amt)
+		// mark as completed if this mint completes the total supply
 		if tickEntry.MintedAmount.GreaterThanOrEqual(tickEntry.TotalSupply) {
 			tickEntry.CompletedAt = blockHeader.Timestamp
 			tickEntry.CompletedAtHeight = payload.Transfer.BlockHeight
@@ -215,21 +208,8 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 			ParentId:          parentId,
 		})
 		latestEventId++
-		case brc20.OperationTransfer:
-			if payload.Transfer.TransferCount > 2 {
-				// skip used transfer inscriptions
-				continue
 	}
-			if tickEntry == nil {
-				// skip transfer inscriptions for non-existent ticks
-				continue
-			}
-			if -payload.Amt.Exponent() > int32(tickEntry.Decimals) {
-				// skip transfer inscriptions with decimals greater than allowed
-				continue
-			}
-
-			if payload.Transfer.OldSatPoint == (ordinals.SatPoint{}) {
+	handleEventInscribeTransfer := func(payload *brc20.Payload) {
 		// inscribe transfer event
 		pkScriptHex := hex.EncodeToString(payload.Transfer.NewPkScript)
 		balance, ok := balances[pkScriptHex][payload.Tick]
@@ -244,7 +224,7 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 		}
 		if payload.Amt.GreaterThan(balance.AvailableBalance) {
 			// skip inscribe transfer event if amount exceeds available balance
-					continue
+			return
 		}
 		// update balance state
 		balance.BlockHeight = uint64(blockHeader.Height)
@@ -277,15 +257,8 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 		latestEventId++
 		eventInscribeTransfers[payload.Transfer.InscriptionId] = event
 		newEventInscribeTransfers = append(newEventInscribeTransfers, event)
-			} else {
-				// transfer transfer event
-				inscribeTransfer, ok := eventInscribeTransfers[payload.Transfer.InscriptionId]
-				if !ok {
-					// skip transfer transfer event if prior inscribe transfer event does not exist
-					continue
 	}
-
-				if payload.Transfer.SentAsFee {
+	handleEventTransferTransferAsFee := func(payload *brc20.Payload, inscribeTransfer *entity.EventInscribeTransfer) {
 		// return balance to sender
 		fromPkScriptHex := hex.EncodeToString(inscribeTransfer.PkScript)
 		fromBalance, ok := balances[fromPkScriptHex][payload.Tick]
@@ -328,13 +301,14 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 			SpentAsFee:        true,
 			Amount:            payload.Amt,
 		})
-				} else {
+	}
+	handleEventTransferTransferNormal := func(payload *brc20.Payload, tickEntry *entity.TickEntry, inscribeTransfer *entity.EventInscribeTransfer) {
 		// subtract balance from sender
 		fromPkScriptHex := hex.EncodeToString(inscribeTransfer.PkScript)
 		fromBalance, ok := balances[fromPkScriptHex][payload.Tick]
 		if !ok {
 			// skip transfer transfer event if from balance does not exist
-						continue
+			return
 		}
 		fromBalance.BlockHeight = uint64(blockHeader.Height)
 		fromBalance.OverallBalance = fromBalance.OverallBalance.Sub(payload.Amt)
@@ -398,6 +372,49 @@ func (p *Processor) processBRC20States(ctx context.Context, transfers []*entity.
 			Amount:            payload.Amt,
 		})
 	}
+
+	for _, payload := range payloads {
+		tickEntry := tickEntries[payload.Tick]
+
+		if payload.Transfer.SentAsFee && payload.Transfer.OldSatPoint == (ordinals.SatPoint{}) {
+			// skip inscriptions inscribed as fee
+			continue
+		}
+
+		switch payload.Op {
+		case brc20.OperationDeploy:
+			handleEventDeploy(payload, tickEntry)
+		case brc20.OperationMint:
+			handleEventMint(payload, tickEntry)
+		case brc20.OperationTransfer:
+			if payload.Transfer.TransferCount > 2 {
+				// skip used transfer inscriptions
+				continue
+			}
+			if tickEntry == nil {
+				// skip transfer inscriptions for non-existent ticks
+				continue
+			}
+			if -payload.Amt.Exponent() > int32(tickEntry.Decimals) {
+				// skip transfer inscriptions with decimals greater than allowed
+				continue
+			}
+
+			if payload.Transfer.OldSatPoint == (ordinals.SatPoint{}) {
+				handleEventInscribeTransfer(payload)
+			} else {
+				// transfer transfer event
+				inscribeTransfer, ok := eventInscribeTransfers[payload.Transfer.InscriptionId]
+				if !ok {
+					// skip transfer transfer event if prior inscribe transfer event does not exist
+					continue
+				}
+
+				if payload.Transfer.SentAsFee {
+					handleEventTransferTransferAsFee(payload, inscribeTransfer)
+				} else {
+					handleEventTransferTransferNormal(payload, tickEntry, inscribeTransfer)
+				}
 			}
 		}
 	}