refactor

NTF-marketplace · Sep 3, 2024 · 36ed4f4 · 36ed4f4
1 parent b933241
commit 36ed4f4
Show file tree

Hide file tree

Showing 16 changed files with 238 additions and 170 deletions.
diff --git a/src/main/kotlin/com/api/admin/controller/AdminController.kt b/src/main/kotlin/com/api/admin/controller/AdminController.kt
@@ -23,10 +23,10 @@ class AdminController(
         @RequestParam address: String,
         @RequestBody request: DepositRequest,
     ): Mono<ResponseEntity<Void>> {
-        return transferService.getTransferData(address, request.chainType, request.transactionHash, AccountType.DEPOSIT)
+        return transferService.getTransferData(address, request.chainType, request.transactionHash, AccountType.DEPOSIT, request.accountLogId)
             .then(Mono.just(ResponseEntity.ok().build<Void>()))
             .onErrorResume {
-                Mono.just(ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build())
+                Mono.just(ResponseEntity.status(HttpStatus.BAD_REQUEST).build())
             }
     }
 
@@ -37,13 +37,9 @@ class AdminController(
         @RequestBody request: WithdrawERC20Request,
     ): Mono<ResponseEntity<Void>> {
         println("Received withdraw request for address: $address with amount: ${request.amount}")
-        return web3jService.processTransactionERC20(address, request.amount, request.chainType)
+        return web3jService.createTransactionERC20(address, request.amount, request.chainType,request.accountLogId)
         .doOnSuccess { println("Transaction successful") }
-        .then(Mono.just(ResponseEntity.ok().build<Void>()))
-        .doOnError { e ->
-            println("Error in withdrawERC20: ${e.message}")
-            e.printStackTrace()
-        }
+        .then(Mono.just(ResponseEntity.ok().build()))
     }
 
 
@@ -53,11 +49,8 @@ class AdminController(
         @RequestParam address: String,
         @RequestBody request: WithdrawERC721Request,
     ): Mono<ResponseEntity<Void>> {
-        return web3jService.processTransactionERC721(address, request.nftId)
-            .then(Mono.just(ResponseEntity.ok().build<Void>()))
-            .onErrorResume {
-                Mono.just(ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build())
-            }
+        return web3jService.createTransactionERC721(address, request.nftId,request.accountLogId)
+            .then(Mono.just(ResponseEntity.ok().build()))
     }
 
 

diff --git a/src/main/kotlin/com/api/admin/controller/dto/DepositRequest.kt b/src/main/kotlin/com/api/admin/controller/dto/DepositRequest.kt
@@ -5,5 +5,6 @@ import com.api.admin.enums.ChainType
 data class DepositRequest(
     val chainType: ChainType,
     val transactionHash: String,
+    val accountLogId: Long,
 )
 
diff --git a/src/main/kotlin/com/api/admin/controller/dto/WithdrawERC20Request.kt b/src/main/kotlin/com/api/admin/controller/dto/WithdrawERC20Request.kt
@@ -7,4 +7,5 @@ import java.math.BigInteger
 data class WithdrawERC20Request(
     val chainType: ChainType,
     val amount: BigDecimal,
+    val accountLogId: Long,
     )
diff --git a/src/main/kotlin/com/api/admin/controller/dto/WithdrawERC721Request.kt b/src/main/kotlin/com/api/admin/controller/dto/WithdrawERC721Request.kt
@@ -2,4 +2,5 @@ package com.api.admin.controller.dto
 
 data class WithdrawERC721Request(
     val nftId: Long,
+    val accountLogId: Long,
 )
diff --git a/src/main/kotlin/com/api/admin/domain/transferFailLog/TransferFailLog.kt b/src/main/kotlin/com/api/admin/domain/transferFailLog/TransferFailLog.kt
@@ -1,5 +1,6 @@
 package com.api.admin.domain.transferFailLog
 
+import com.api.admin.enums.TransferType
 import org.springframework.data.annotation.Id
 import org.springframework.data.relational.core.mapping.Table
 import java.sql.Timestamp
@@ -11,4 +12,5 @@ data class TransferFailLog(
     val timestamp: Long? = System.currentTimeMillis(),
     val transactionHash: String?,
     val errorMessage: String,
+    val transferType: TransferType,
     )
diff --git a/src/main/kotlin/com/api/admin/enums/Enum.kt b/src/main/kotlin/com/api/admin/enums/Enum.kt
@@ -24,7 +24,7 @@ enum class TokenType {
     MATIC, ETH
 }
 
-enum class TransactionStatus {
+enum class TransactionStatusType {
     SUCCESS, FAILURE
 }
 
diff --git a/src/main/kotlin/com/api/admin/rabbitMQ/event/dto/AdminTransferDetailResponse.kt b/src/main/kotlin/com/api/admin/rabbitMQ/event/dto/AdminTransferDetailResponse.kt
@@ -0,0 +1,10 @@
+package com.api.admin.rabbitMQ.event.dto
+
+import com.api.admin.enums.TransferType
+import java.math.BigDecimal
+
+data class AdminTransferDetailResponse(
+    val nftId: Long?,
+    val transferType: TransferType,
+    val balance: BigDecimal?,
+)
diff --git a/src/main/kotlin/com/api/admin/rabbitMQ/event/dto/AdminTransferResponse.kt b/src/main/kotlin/com/api/admin/rabbitMQ/event/dto/AdminTransferResponse.kt
@@ -1,18 +1,43 @@
 package com.api.admin.rabbitMQ.event.dto
 
+import com.api.admin.domain.transfer.Transfer
+import com.api.admin.domain.transferFailLog.TransferFailLog
 import com.api.admin.enums.AccountType
 import com.api.admin.enums.ChainType
+import com.api.admin.enums.TransactionStatusType
 import com.api.admin.enums.TransferType
+import com.api.admin.rabbitMQ.event.dto.AdminTransferResponse.Companion.toResponse
 import java.math.BigDecimal
 
 
 data class AdminTransferResponse(
-    val id: Long,
-    val walletAddress: String,
-    val nftId: Long?,
-    val timestamp: Long,
+    val accountLogId: Long,
     val accountType: AccountType,
     val transferType: TransferType,
-    val balance: BigDecimal?,
-    val chainType: ChainType,
-)
+    val transactionStatusType: TransactionStatusType,
+    val adminTransferDetailResponse: AdminTransferDetailResponse?
+) {
+    companion object {
+        fun Transfer.toResponse(accountId: Long) = AdminTransferResponse(
+            accountLogId = accountId,
+            accountType = this.accountType,
+            transferType = this.transferType,
+            transactionStatusType = TransactionStatusType.SUCCESS,
+            adminTransferDetailResponse =  AdminTransferDetailResponse(
+                nftId = this.nftId,
+                transferType =this.transferType,
+                balance = this.balance
+            )
+        )
+
+        fun TransferFailLog.toResponse(accountId: Long,accountType: AccountType) = AdminTransferResponse(
+            accountLogId = accountId,
+            accountType = accountType,
+            transferType = this.transferType,
+            transactionStatusType = TransactionStatusType.FAILURE,
+            adminTransferDetailResponse = null
+
+        )
+
+    }
+}
diff --git a/src/main/kotlin/com/api/admin/service/TransferFailLogService.kt b/src/main/kotlin/com/api/admin/service/TransferFailLogService.kt
diff --git a/src/main/kotlin/com/api/admin/service/TransferFailService.kt b/src/main/kotlin/com/api/admin/service/TransferFailService.kt
@@ -0,0 +1,38 @@
+package com.api.admin.service
+
+import com.api.admin.domain.transferFailLog.TransferFailLog
+import com.api.admin.domain.transferFailLog.TransferFailLogRepository
+import com.api.admin.enums.AccountType
+import com.api.admin.enums.TransferType
+import com.api.admin.rabbitMQ.event.dto.AdminTransferCreatedEvent
+import com.api.admin.rabbitMQ.event.dto.AdminTransferResponse.Companion.toResponse
+import org.springframework.context.ApplicationEventPublisher
+import org.springframework.stereotype.Service
+import reactor.core.publisher.Mono
+
+@Service
+class TransferFailService(
+    private val transferFailLogRepository: TransferFailLogRepository,
+    private val eventPublisher: ApplicationEventPublisher,
+) {
+
+    fun save(accountId: Long,
+             address: String,
+             transactionHash: String?,
+             message: String,
+             transferType: TransferType,
+             accountType: AccountType
+    ): Mono<Void> {
+        return transferFailLogRepository.save(
+            TransferFailLog(
+                wallet = address,
+                transactionHash = transactionHash,
+                errorMessage = message,
+                transferType = transferType
+            )
+        ).doOnSuccess {
+            eventPublisher.publishEvent(AdminTransferCreatedEvent(this, it.toResponse(accountId,accountType)))
+        }.then()
+
+    }
+}
diff --git a/src/main/kotlin/com/api/admin/service/TransferService.kt b/src/main/kotlin/com/api/admin/service/TransferService.kt
@@ -7,12 +7,13 @@ import com.api.admin.enums.ChainType
 import com.api.admin.enums.TransferType
 import com.api.admin.properties.AdminInfoProperties
 import com.api.admin.rabbitMQ.event.dto.AdminTransferCreatedEvent
-import com.api.admin.rabbitMQ.event.dto.AdminTransferResponse
+import com.api.admin.rabbitMQ.event.dto.AdminTransferResponse.Companion.toResponse
 import com.api.admin.service.dto.InfuraTransferDetail
 import org.springframework.context.ApplicationEventPublisher
 import org.springframework.stereotype.Service
 import reactor.core.publisher.Flux
 import reactor.core.publisher.Mono
+import reactor.core.scheduler.Schedulers
 import java.math.BigDecimal
 import java.math.BigInteger
 import java.time.Instant
@@ -25,32 +26,36 @@ class TransferService(
     private val nftService: NftService,
     private val adminInfoProperties: AdminInfoProperties,
 ) {
-
-    private val transferEventSignature = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
-    private val nativeTransferEventSignature = "0xe6497e3ee548a3372136af2fcb0696db31fc6cf20260707645068bd3fe97f3c4"
-
+    companion object{
+        const val transferEventSignature = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
+        const val nativeTransferEventSignature = "0xe6497e3ee548a3372136af2fcb0696db31fc6cf20260707645068bd3fe97f3c4"
+    }
 
     fun getTransferData(
         wallet: String,
         chainType: ChainType,
         transactionHash: String,
         accountType: AccountType,
+        accountLogId: Long
     ): Mono<Void> {
         return transferRepository.existsByTransactionHash(transactionHash)
-            .flatMap {
-                if (it) {
+            .flatMap { exists ->
+                if (exists) {
                     Mono.error(IllegalStateException("Transaction already exists"))
                 } else {
-                    saveTransfer(wallet, chainType, transactionHash, accountType)
-                        .doOnNext { transfer ->
-                            eventPublisher.publishEvent(AdminTransferCreatedEvent(this, transfer.toResponse()))
-                        }
-                        .then()
+                    Mono.defer { saveTransfer(wallet, chainType, transactionHash, accountType,accountLogId).then() }
+                        .subscribeOn(Schedulers.boundedElastic())
+                        .then(Mono.empty())
                 }
             }
     }
 
-    fun saveTransfer(wallet: String, chainType: ChainType, transactionHash: String, accountType: AccountType): Flux<Transfer> {
+    fun saveTransfer(wallet: String,
+                     chainType: ChainType,
+                     transactionHash: String,
+                     accountType: AccountType,
+                     accountId: Long
+    ): Flux<Void> {
         return infuraApiService.getTransferLog(chainType, transactionHash)
             .flatMapMany { response ->
                 val result = response.result
@@ -61,30 +66,21 @@ class TransferService(
                     Flux.error(IllegalStateException("Transaction logs not found for transaction hash: $transactionHash"))
                 }
             }
-            .flatMap { transfer -> transferRepository.save(transfer) }
+            .flatMap { transfer -> transferRepository.save(transfer).doOnNext { println("having ?") }
+                .doOnSuccess {transfer ->
+                    eventPublisher.publishEvent(AdminTransferCreatedEvent(this, transfer.toResponse(accountId)))
+                }.then()
+        }
     }
-
-
-    private fun Transfer.toResponse() = AdminTransferResponse(
-        id = this.id!!,
-        walletAddress = this.wallet,
-        nftId = this.nftId,
-        timestamp = this.timestamp,
-        accountType = this.accountType,
-        transferType = this.transferType,
-        balance = this.balance,
-        chainType = this.chainType,
-    )
-
     fun InfuraTransferDetail.toEntity(wallet: String, accountType: AccountType, chainType: ChainType): Mono<Transfer> {
         return Mono.just(this)
             .flatMap { log ->
                 println("log : " + log.toString())
                 when {
                     log.topics[0] == nativeTransferEventSignature ->
                         handleERC20Transfer(log, wallet, accountType, chainType,TransferType.NATIVE)
-//                    log.topics[0] == transferEventSignature && log.topics.size == 3 ->
-//                        handleERC20Transfer(log, wallet, accountType, chainType, TransferType.ERC20)
+                   // log.topics[0] == transferEventSignature && log.topics.size == 3 ->
+                   //     handleERC20Transfer(log, wallet, accountType, chainType, TransferType.ERC20)
                     log.topics[0] == transferEventSignature && log.topics.size == 4 ->
                         handleERC721Transfer(log, wallet, accountType, chainType)
                     else -> Mono.empty()
@@ -97,7 +93,7 @@ class TransferService(
         val to = parseAddress(log.topics[3])
         val amount = when (transferType) {
             TransferType.NATIVE -> parseNativeTransferAmount(log.data)
-//            TransferType.ERC20 -> toBigDecimal(log.data)
+           // TransferType.ERC20 -> toBigDecimal(log.data)
             else -> BigDecimal.ZERO
         }