feat: 完善跟单模板管理功能

- 更新需求文档:移除每日最大亏损限制、跟单延迟、轮询和WebSocket配置
- 跟单比例改为百分比格式(10%-1000%),只允许整数
- 所有USDC字段支持最多4位小数
- 固定金额和最小金额校验(必须>=1)
- 模板名称支持编辑
- 复制模板改为Modal形式,可编辑后创建
- 模板列表按创建时间降序排序,显示创建时间
- 移动端适配:使用卡片式布局
- 合并跟单比例和固定金额列为跟单配置
- 支持卖出改为跟单卖出
- 菜单结构调整:跟单管理作为父菜单,包含跟单模板和跟单配置子菜单
This commit is contained in:
WrBug
2025-12-01 23:03:17 +08:00
parent 914983acd7
commit fd10a31780
28 changed files with 4298 additions and 234 deletions
@@ -0,0 +1,168 @@
package com.wrbug.polymarketbot.controller
import com.wrbug.polymarketbot.dto.*
import com.wrbug.polymarketbot.service.CopyTradingService
import org.slf4j.LoggerFactory
import org.springframework.http.ResponseEntity
import org.springframework.web.bind.annotation.*
/**
* 跟单配置管理控制器(钱包-模板关联)
*/
@RestController
@RequestMapping("/api/copy-trading")
class CopyTradingController(
private val copyTradingService: CopyTradingService
) {
private val logger = LoggerFactory.getLogger(CopyTradingController::class.java)
/**
* 创建跟单
*/
@PostMapping("/create")
fun createCopyTrading(@RequestBody request: CopyTradingCreateRequest): ResponseEntity<ApiResponse<CopyTradingDto>> {
return try {
if (request.accountId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("账户 ID 无效"))
}
if (request.templateId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("模板 ID 无效"))
}
if (request.leaderId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("Leader ID 无效"))
}
val result = copyTradingService.createCopyTrading(request)
result.fold(
onSuccess = { copyTrading ->
logger.info("成功创建跟单: ${copyTrading.id}")
ResponseEntity.ok(ApiResponse.success(copyTrading))
},
onFailure = { e ->
logger.error("创建跟单失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("创建跟单失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("创建跟单异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("创建跟单失败: ${e.message}"))
}
}
/**
* 查询跟单列表
*/
@PostMapping("/list")
fun getCopyTradingList(@RequestBody request: CopyTradingListRequest): ResponseEntity<ApiResponse<CopyTradingListResponse>> {
return try {
val result = copyTradingService.getCopyTradingList(request)
result.fold(
onSuccess = { response ->
ResponseEntity.ok(ApiResponse.success(response))
},
onFailure = { e ->
logger.error("查询跟单列表失败: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询跟单列表失败: ${e.message}"))
}
)
} catch (e: Exception) {
logger.error("查询跟单列表异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询跟单列表失败: ${e.message}"))
}
}
/**
* 更新跟单状态
*/
@PostMapping("/update-status")
fun updateCopyTradingStatus(@RequestBody request: CopyTradingUpdateStatusRequest): ResponseEntity<ApiResponse<CopyTradingDto>> {
return try {
if (request.copyTradingId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("跟单 ID 无效"))
}
val result = copyTradingService.updateCopyTradingStatus(request)
result.fold(
onSuccess = { copyTrading ->
logger.info("成功更新跟单状态: ${copyTrading.id}, enabled=${copyTrading.enabled}")
ResponseEntity.ok(ApiResponse.success(copyTrading))
},
onFailure = { e ->
logger.error("更新跟单状态失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
is IllegalStateException -> ResponseEntity.ok(ApiResponse.businessError(e.message ?: "业务逻辑错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("更新跟单状态失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("更新跟单状态异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("更新跟单状态失败: ${e.message}"))
}
}
/**
* 删除跟单
*/
@PostMapping("/delete")
fun deleteCopyTrading(@RequestBody request: CopyTradingDeleteRequest): ResponseEntity<ApiResponse<Unit>> {
return try {
if (request.copyTradingId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("跟单 ID 无效"))
}
val result = copyTradingService.deleteCopyTrading(request.copyTradingId)
result.fold(
onSuccess = {
logger.info("成功删除跟单: ${request.copyTradingId}")
ResponseEntity.ok(ApiResponse.success(Unit))
},
onFailure = { e ->
logger.error("删除跟单失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("删除跟单失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("删除跟单异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("删除跟单失败: ${e.message}"))
}
}
/**
* 查询钱包绑定的模板
*/
@PostMapping("/account-templates")
fun getAccountTemplates(@RequestBody request: AccountTemplatesRequest): ResponseEntity<ApiResponse<AccountTemplatesResponse>> {
return try {
if (request.accountId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("账户 ID 无效"))
}
val result = copyTradingService.getAccountTemplates(request.accountId)
result.fold(
onSuccess = { response ->
ResponseEntity.ok(ApiResponse.success(response))
},
onFailure = { e ->
logger.error("查询钱包绑定的模板失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("查询钱包绑定的模板失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("查询钱包绑定的模板异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询钱包绑定的模板失败: ${e.message}"))
}
}
}
@@ -0,0 +1,195 @@
package com.wrbug.polymarketbot.controller
import com.wrbug.polymarketbot.dto.*
import com.wrbug.polymarketbot.service.CopyTradingTemplateService
import org.slf4j.LoggerFactory
import org.springframework.http.ResponseEntity
import org.springframework.web.bind.annotation.*
/**
* 跟单模板管理控制器
*/
@RestController
@RequestMapping("/api/copy-trading/templates")
class CopyTradingTemplateController(
private val templateService: CopyTradingTemplateService
) {
private val logger = LoggerFactory.getLogger(CopyTradingTemplateController::class.java)
/**
* 创建模板
*/
@PostMapping("/create")
fun createTemplate(@RequestBody request: TemplateCreateRequest): ResponseEntity<ApiResponse<TemplateDto>> {
return try {
if (request.templateName.isBlank()) {
return ResponseEntity.ok(ApiResponse.paramError("模板名称不能为空"))
}
val result = templateService.createTemplate(request)
result.fold(
onSuccess = { template ->
logger.info("成功创建模板: ${template.id}")
ResponseEntity.ok(ApiResponse.success(template))
},
onFailure = { e ->
logger.error("创建模板失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("创建模板失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("创建模板异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("创建模板失败: ${e.message}"))
}
}
/**
* 更新模板
*/
@PostMapping("/update")
fun updateTemplate(@RequestBody request: TemplateUpdateRequest): ResponseEntity<ApiResponse<TemplateDto>> {
return try {
if (request.templateId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("模板 ID 无效"))
}
val result = templateService.updateTemplate(request)
result.fold(
onSuccess = { template ->
logger.info("成功更新模板: ${template.id}")
ResponseEntity.ok(ApiResponse.success(template))
},
onFailure = { e ->
logger.error("更新模板失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("更新模板失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("更新模板异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("更新模板失败: ${e.message}"))
}
}
/**
* 删除模板
*/
@PostMapping("/delete")
fun deleteTemplate(@RequestBody request: TemplateDeleteRequest): ResponseEntity<ApiResponse<Unit>> {
return try {
if (request.templateId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("模板 ID 无效"))
}
val result = templateService.deleteTemplate(request.templateId)
result.fold(
onSuccess = {
logger.info("成功删除模板: ${request.templateId}")
ResponseEntity.ok(ApiResponse.success(Unit))
},
onFailure = { e ->
logger.error("删除模板失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
is IllegalStateException -> ResponseEntity.ok(ApiResponse.businessError(e.message ?: "业务逻辑错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("删除模板失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("删除模板异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("删除模板失败: ${e.message}"))
}
}
/**
* 复制模板
*/
@PostMapping("/copy")
fun copyTemplate(@RequestBody request: TemplateCopyRequest): ResponseEntity<ApiResponse<TemplateDto>> {
return try {
if (request.templateId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("模板 ID 无效"))
}
if (request.templateName.isBlank()) {
return ResponseEntity.ok(ApiResponse.paramError("新模板名称不能为空"))
}
val result = templateService.copyTemplate(request)
result.fold(
onSuccess = { template ->
logger.info("成功复制模板: ${template.id}")
ResponseEntity.ok(ApiResponse.success(template))
},
onFailure = { e ->
logger.error("复制模板失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("复制模板失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("复制模板异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("复制模板失败: ${e.message}"))
}
}
/**
* 查询模板列表
*/
@PostMapping("/list")
fun getTemplateList(): ResponseEntity<ApiResponse<TemplateListResponse>> {
return try {
val result = templateService.getTemplateList()
result.fold(
onSuccess = { response ->
ResponseEntity.ok(ApiResponse.success(response))
},
onFailure = { e ->
logger.error("查询模板列表失败: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询模板列表失败: ${e.message}"))
}
)
} catch (e: Exception) {
logger.error("查询模板列表异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询模板列表失败: ${e.message}"))
}
}
/**
* 查询模板详情
*/
@PostMapping("/detail")
fun getTemplateDetail(@RequestBody request: TemplateDetailRequest): ResponseEntity<ApiResponse<TemplateDto>> {
return try {
if (request.templateId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("模板 ID 无效"))
}
val result = templateService.getTemplateDetail(request.templateId)
result.fold(
onSuccess = { template ->
ResponseEntity.ok(ApiResponse.success(template))
},
onFailure = { e ->
logger.error("查询模板详情失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("查询模板详情失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("查询模板详情异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询模板详情失败: ${e.message}"))
}
}
}
@@ -0,0 +1,170 @@
package com.wrbug.polymarketbot.controller
import com.wrbug.polymarketbot.dto.*
import com.wrbug.polymarketbot.service.LeaderService
import org.slf4j.LoggerFactory
import org.springframework.http.ResponseEntity
import org.springframework.web.bind.annotation.*
/**
* Leader 管理控制器
*/
@RestController
@RequestMapping("/api/copy-trading/leaders")
class LeaderController(
private val leaderService: LeaderService
) {
private val logger = LoggerFactory.getLogger(LeaderController::class.java)
/**
* 添加被跟单者
*/
@PostMapping("/add")
fun addLeader(@RequestBody request: LeaderAddRequest): ResponseEntity<ApiResponse<LeaderDto>> {
return try {
if (request.leaderAddress.isBlank()) {
return ResponseEntity.ok(ApiResponse.paramError("Leader 地址不能为空"))
}
val result = leaderService.addLeader(request)
result.fold(
onSuccess = { leader ->
logger.info("成功添加 Leader: ${leader.id}")
ResponseEntity.ok(ApiResponse.success(leader))
},
onFailure = { e ->
logger.error("添加 Leader 失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
is IllegalStateException -> ResponseEntity.ok(ApiResponse.businessError(e.message ?: "业务逻辑错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("添加 Leader 失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("添加 Leader 异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("添加 Leader 失败: ${e.message}"))
}
}
/**
* 更新被跟单者
*/
@PostMapping("/update")
fun updateLeader(@RequestBody request: LeaderUpdateRequest): ResponseEntity<ApiResponse<LeaderDto>> {
return try {
if (request.leaderId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("Leader ID 无效"))
}
val result = leaderService.updateLeader(request)
result.fold(
onSuccess = { leader ->
logger.info("成功更新 Leader: ${leader.id}")
ResponseEntity.ok(ApiResponse.success(leader))
},
onFailure = { e ->
logger.error("更新 Leader 失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("更新 Leader 失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("更新 Leader 异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("更新 Leader 失败: ${e.message}"))
}
}
/**
* 删除被跟单者
*/
@PostMapping("/delete")
fun deleteLeader(@RequestBody request: LeaderDeleteRequest): ResponseEntity<ApiResponse<Unit>> {
return try {
if (request.leaderId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("Leader ID 无效"))
}
val result = leaderService.deleteLeader(request.leaderId)
result.fold(
onSuccess = {
logger.info("成功删除 Leader: ${request.leaderId}")
ResponseEntity.ok(ApiResponse.success(Unit))
},
onFailure = { e ->
logger.error("删除 Leader 失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
is IllegalStateException -> ResponseEntity.ok(ApiResponse.businessError(e.message ?: "业务逻辑错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("删除 Leader 失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("删除 Leader 异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("删除 Leader 失败: ${e.message}"))
}
}
/**
* 查询被跟单者列表
*/
@PostMapping("/list")
fun getLeaderList(@RequestBody request: LeaderListRequest): ResponseEntity<ApiResponse<LeaderListResponse>> {
return try {
val result = leaderService.getLeaderList(request)
result.fold(
onSuccess = { response ->
ResponseEntity.ok(ApiResponse.success(response))
},
onFailure = { e ->
logger.error("查询 Leader 列表失败: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询 Leader 列表失败: ${e.message}"))
}
)
} catch (e: Exception) {
logger.error("查询 Leader 列表异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询 Leader 列表失败: ${e.message}"))
}
}
/**
* 查询被跟单者详情
*/
@PostMapping("/detail")
fun getLeaderDetail(@RequestBody request: LeaderDetailRequest): ResponseEntity<ApiResponse<LeaderDto>> {
return try {
if (request.leaderId <= 0) {
return ResponseEntity.ok(ApiResponse.paramError("Leader ID 无效"))
}
val result = leaderService.getLeaderDetail(request.leaderId)
result.fold(
onSuccess = { leader ->
ResponseEntity.ok(ApiResponse.success(leader))
},
onFailure = { e ->
logger.error("查询 Leader 详情失败: ${e.message}", e)
when (e) {
is IllegalArgumentException -> ResponseEntity.ok(ApiResponse.paramError(e.message ?: "参数错误"))
else -> ResponseEntity.ok(ApiResponse.serverError("查询 Leader 详情失败: ${e.message}"))
}
}
)
} catch (e: Exception) {
logger.error("查询 Leader 详情异常: ${e.message}", e)
ResponseEntity.ok(ApiResponse.serverError("查询 Leader 详情失败: ${e.message}"))
}
}
}
/**
* Leader 详情请求
*/
data class LeaderDetailRequest(
val leaderId: Long
)
@@ -0,0 +1,91 @@
package com.wrbug.polymarketbot.dto
/**
* 跟单创建请求
*/
data class CopyTradingCreateRequest(
val accountId: Long,
val templateId: Long,
val leaderId: Long,
val enabled: Boolean = true
)
/**
* 跟单列表请求
*/
data class CopyTradingListRequest(
val accountId: Long? = null,
val templateId: Long? = null,
val leaderId: Long? = null,
val enabled: Boolean? = null
)
/**
* 跟单状态更新请求
*/
data class CopyTradingUpdateStatusRequest(
val copyTradingId: Long,
val enabled: Boolean
)
/**
* 跟单删除请求
*/
data class CopyTradingDeleteRequest(
val copyTradingId: Long
)
/**
* 查询钱包绑定的模板请求
*/
data class AccountTemplatesRequest(
val accountId: Long
)
/**
* 跟单信息响应
*/
data class CopyTradingDto(
val id: Long,
val accountId: Long,
val accountName: String?,
val walletAddress: String,
val templateId: Long,
val templateName: String,
val leaderId: Long,
val leaderName: String?,
val leaderAddress: String,
val enabled: Boolean,
val createdAt: Long,
val updatedAt: Long
)
/**
* 跟单列表响应
*/
data class CopyTradingListResponse(
val list: List<CopyTradingDto>,
val total: Long
)
/**
* 钱包绑定的模板信息
*/
data class AccountTemplateDto(
val templateId: Long,
val templateName: String,
val copyTradingId: Long,
val leaderId: Long,
val leaderName: String?,
val leaderAddress: String,
val enabled: Boolean
)
/**
* 钱包绑定的模板列表响应
*/
data class AccountTemplatesResponse(
val list: List<AccountTemplateDto>,
val total: Long
)
@@ -0,0 +1,114 @@
package com.wrbug.polymarketbot.dto
/**
* 模板创建请求
*/
data class TemplateCreateRequest(
val templateName: String,
val copyMode: String = "RATIO", // "RATIO" 或 "FIXED"
val copyRatio: String? = null, // 仅在 copyMode="RATIO" 时生效
val fixedAmount: String? = null, // 仅在 copyMode="FIXED" 时生效
val maxOrderSize: String? = null,
val minOrderSize: String? = null,
val maxDailyLoss: String? = null,
val maxDailyOrders: Int? = null,
val priceTolerance: String? = null,
val delaySeconds: Int? = null,
val pollIntervalSeconds: Int? = null,
val useWebSocket: Boolean? = null,
val websocketReconnectInterval: Int? = null,
val websocketMaxRetries: Int? = null,
val supportSell: Boolean? = null
)
/**
* 模板更新请求
*/
data class TemplateUpdateRequest(
val templateId: Long,
val templateName: String? = null, // 模板名称(可选)
val copyMode: String? = null,
val copyRatio: String? = null,
val fixedAmount: String? = null,
val maxOrderSize: String? = null,
val minOrderSize: String? = null,
val maxDailyLoss: String? = null,
val maxDailyOrders: Int? = null,
val priceTolerance: String? = null,
val delaySeconds: Int? = null,
val pollIntervalSeconds: Int? = null,
val useWebSocket: Boolean? = null,
val websocketReconnectInterval: Int? = null,
val websocketMaxRetries: Int? = null,
val supportSell: Boolean? = null
)
/**
* 模板删除请求
*/
data class TemplateDeleteRequest(
val templateId: Long
)
/**
* 模板复制请求
*/
data class TemplateCopyRequest(
val templateId: Long,
val templateName: String,
val copyMode: String? = null,
val copyRatio: String? = null,
val fixedAmount: String? = null,
val maxOrderSize: String? = null,
val minOrderSize: String? = null,
val maxDailyLoss: String? = null,
val maxDailyOrders: Int? = null,
val priceTolerance: String? = null,
val delaySeconds: Int? = null,
val pollIntervalSeconds: Int? = null,
val useWebSocket: Boolean? = null,
val websocketReconnectInterval: Int? = null,
val websocketMaxRetries: Int? = null,
val supportSell: Boolean? = null
)
/**
* 模板详情请求
*/
data class TemplateDetailRequest(
val templateId: Long
)
/**
* 模板信息响应
*/
data class TemplateDto(
val id: Long,
val templateName: String,
val copyMode: String,
val copyRatio: String,
val fixedAmount: String?,
val maxOrderSize: String,
val minOrderSize: String,
val maxDailyLoss: String,
val maxDailyOrders: Int,
val priceTolerance: String,
val delaySeconds: Int,
val pollIntervalSeconds: Int,
val useWebSocket: Boolean,
val websocketReconnectInterval: Int,
val websocketMaxRetries: Int,
val supportSell: Boolean,
val useCount: Long = 0, // 使用该模板的跟单数量
val createdAt: Long,
val updatedAt: Long
)
/**
* 模板列表响应
*/
data class TemplateListResponse(
val list: List<TemplateDto>,
val total: Long
)
@@ -0,0 +1,57 @@
package com.wrbug.polymarketbot.dto
/**
* Leader 添加请求
*/
data class LeaderAddRequest(
val leaderAddress: String,
val leaderName: String? = null,
val category: String? = null // sports 或 crypto
)
/**
* Leader 更新请求
*/
data class LeaderUpdateRequest(
val leaderId: Long,
val leaderName: String? = null,
val category: String? = null
)
/**
* Leader 删除请求
*/
data class LeaderDeleteRequest(
val leaderId: Long
)
/**
* Leader 列表请求
*/
data class LeaderListRequest(
val category: String? = null // sports 或 crypto
)
/**
* Leader 信息响应
*/
data class LeaderDto(
val id: Long,
val leaderAddress: String,
val leaderName: String?,
val category: String?,
val copyTradingCount: Long = 0, // 跟单关系数量
val totalOrders: Long? = null, // 总订单数(可选)
val totalPnl: String? = null, // 总盈亏(可选)
val createdAt: Long,
val updatedAt: Long
)
/**
* Leader 列表响应
*/
data class LeaderListResponse(
val list: List<LeaderDto>,
val total: Long
)
@@ -0,0 +1,38 @@
package com.wrbug.polymarketbot.entity
import jakarta.persistence.*
/**
* 跟单关系实体(钱包-模板关联,多对多关系)
*/
@Entity
@Table(
name = "copy_trading",
uniqueConstraints = [
UniqueConstraint(columnNames = ["account_id", "template_id", "leader_id"])
]
)
data class CopyTrading(
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
val id: Long? = null,
@Column(name = "account_id", nullable = false)
val accountId: Long, // 钱包账户ID
@Column(name = "template_id", nullable = false)
val templateId: Long, // 模板ID
@Column(name = "leader_id", nullable = false)
val leaderId: Long, // Leader ID
@Column(name = "enabled", nullable = false)
val enabled: Boolean = true, // 是否启用
@Column(name = "created_at", nullable = false)
val createdAt: Long = System.currentTimeMillis(),
@Column(name = "updated_at", nullable = false)
var updatedAt: Long = System.currentTimeMillis()
)
@@ -0,0 +1,68 @@
package com.wrbug.polymarketbot.entity
import jakarta.persistence.*
import java.math.BigDecimal
import com.wrbug.polymarketbot.util.toSafeBigDecimal
/**
* 跟单模板实体
*/
@Entity
@Table(name = "copy_trading_templates")
data class CopyTradingTemplate(
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
val id: Long? = null,
@Column(name = "template_name", unique = true, nullable = false, length = 100)
val templateName: String, // 模板名称
@Column(name = "copy_mode", nullable = false, length = 10)
val copyMode: String = "RATIO", // "RATIO" 或 "FIXED"
@Column(name = "copy_ratio", nullable = false, precision = 10, scale = 2)
val copyRatio: BigDecimal = BigDecimal.ONE, // 仅在 copyMode="RATIO" 时生效
@Column(name = "fixed_amount", precision = 20, scale = 8)
val fixedAmount: BigDecimal? = null, // 仅在 copyMode="FIXED" 时生效
@Column(name = "max_order_size", nullable = false, precision = 20, scale = 8)
val maxOrderSize: BigDecimal = "1000".toSafeBigDecimal(),
@Column(name = "min_order_size", nullable = false, precision = 20, scale = 8)
val minOrderSize: BigDecimal = "1".toSafeBigDecimal(),
@Column(name = "max_daily_loss", nullable = false, precision = 20, scale = 8)
val maxDailyLoss: BigDecimal = "10000".toSafeBigDecimal(),
@Column(name = "max_daily_orders", nullable = false)
val maxDailyOrders: Int = 100,
@Column(name = "price_tolerance", nullable = false, precision = 5, scale = 2)
val priceTolerance: BigDecimal = "5".toSafeBigDecimal(), // 百分比
@Column(name = "delay_seconds", nullable = false)
val delaySeconds: Int = 0,
@Column(name = "poll_interval_seconds", nullable = false)
val pollIntervalSeconds: Int = 5, // 轮询间隔(仅在 WebSocket 不可用时使用)
@Column(name = "use_websocket", nullable = false)
val useWebSocket: Boolean = true, // 是否优先使用 WebSocket 推送
@Column(name = "websocket_reconnect_interval", nullable = false)
val websocketReconnectInterval: Int = 5000, // WebSocket 重连间隔(毫秒)
@Column(name = "websocket_max_retries", nullable = false)
val websocketMaxRetries: Int = 10, // WebSocket 最大重试次数
@Column(name = "support_sell", nullable = false)
val supportSell: Boolean = true, // 是否支持跟单卖出
@Column(name = "created_at", nullable = false)
val createdAt: Long = System.currentTimeMillis(),
@Column(name = "updated_at", nullable = false)
var updatedAt: Long = System.currentTimeMillis()
)
@@ -0,0 +1,38 @@
package com.wrbug.polymarketbot.entity
import jakarta.persistence.*
import com.wrbug.polymarketbot.util.CategoryValidator
/**
* 被跟单者(Leader)实体
*/
@Entity
@Table(name = "copy_trading_leaders")
data class Leader(
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
val id: Long? = null,
@Column(name = "leader_address", unique = true, nullable = false, length = 42)
val leaderAddress: String, // 钱包地址
@Column(name = "leader_name", length = 100)
val leaderName: String? = null,
@Column(name = "category", length = 20)
val category: String? = null, // sports 或 cryptonull 表示不筛选
@Column(name = "created_at", nullable = false)
val createdAt: Long = System.currentTimeMillis(),
@Column(name = "updated_at", nullable = false)
var updatedAt: Long = System.currentTimeMillis()
) {
init {
// 验证分类
if (category != null) {
CategoryValidator.validate(category)
}
}
}
@@ -0,0 +1,62 @@
package com.wrbug.polymarketbot.repository
import com.wrbug.polymarketbot.entity.CopyTrading
import org.springframework.data.jpa.repository.JpaRepository
import org.springframework.stereotype.Repository
/**
* 跟单关系 Repository
*/
@Repository
interface CopyTradingRepository : JpaRepository<CopyTrading, Long> {
/**
* 根据账户ID查找跟单列表
*/
fun findByAccountId(accountId: Long): List<CopyTrading>
/**
* 根据模板ID查找跟单列表
*/
fun findByTemplateId(templateId: Long): List<CopyTrading>
/**
* 根据 Leader ID 查找跟单列表
*/
fun findByLeaderId(leaderId: Long): List<CopyTrading>
/**
* 根据账户ID和模板ID查找跟单列表
*/
fun findByAccountIdAndTemplateId(accountId: Long, templateId: Long): List<CopyTrading>
/**
* 根据账户ID、模板ID和Leader ID查找跟单
*/
fun findByAccountIdAndTemplateIdAndLeaderId(
accountId: Long,
templateId: Long,
leaderId: Long
): CopyTrading?
/**
* 查找所有启用的跟单
*/
fun findByEnabledTrue(): List<CopyTrading>
/**
* 根据账户ID查找启用的跟单
*/
fun findByAccountIdAndEnabledTrue(accountId: Long): List<CopyTrading>
/**
* 统计使用指定模板的跟单数量
*/
fun countByTemplateId(templateId: Long): Long
/**
* 统计指定 Leader 的跟单数量
*/
fun countByLeaderId(leaderId: Long): Long
}
@@ -0,0 +1,28 @@
package com.wrbug.polymarketbot.repository
import com.wrbug.polymarketbot.entity.CopyTradingTemplate
import org.springframework.data.jpa.repository.JpaRepository
import org.springframework.stereotype.Repository
/**
* 跟单模板 Repository
*/
@Repository
interface CopyTradingTemplateRepository : JpaRepository<CopyTradingTemplate, Long> {
/**
* 根据模板名称查找模板
*/
fun findByTemplateName(templateName: String): CopyTradingTemplate?
/**
* 检查模板名称是否存在
*/
fun existsByTemplateName(templateName: String): Boolean
/**
* 查找所有模板,按创建时间降序排序(最新的在前)
*/
fun findAllByOrderByCreatedAtDesc(): List<CopyTradingTemplate>
}
@@ -0,0 +1,33 @@
package com.wrbug.polymarketbot.repository
import com.wrbug.polymarketbot.entity.Leader
import org.springframework.data.jpa.repository.JpaRepository
import org.springframework.stereotype.Repository
/**
* Leader Repository
*/
@Repository
interface LeaderRepository : JpaRepository<Leader, Long> {
/**
* 根据钱包地址查找 Leader
*/
fun findByLeaderAddress(leaderAddress: String): Leader?
/**
* 检查钱包地址是否存在
*/
fun existsByLeaderAddress(leaderAddress: String): Boolean
/**
* 根据分类查找 Leader 列表
*/
fun findByCategory(category: String?): List<Leader>
/**
* 查找所有 Leader,按创建时间排序
*/
fun findAllByOrderByCreatedAtAsc(): List<Leader>
}
@@ -0,0 +1,257 @@
package com.wrbug.polymarketbot.service
import com.wrbug.polymarketbot.dto.*
import com.wrbug.polymarketbot.entity.CopyTrading
import com.wrbug.polymarketbot.repository.AccountRepository
import com.wrbug.polymarketbot.repository.CopyTradingRepository
import com.wrbug.polymarketbot.repository.CopyTradingTemplateRepository
import com.wrbug.polymarketbot.repository.LeaderRepository
import org.slf4j.LoggerFactory
import org.springframework.stereotype.Service
import org.springframework.transaction.annotation.Transactional
/**
* 跟单配置管理服务(钱包-模板关联)
*/
@Service
class CopyTradingService(
private val copyTradingRepository: CopyTradingRepository,
private val accountRepository: AccountRepository,
private val templateRepository: CopyTradingTemplateRepository,
private val leaderRepository: LeaderRepository
) {
private val logger = LoggerFactory.getLogger(CopyTradingService::class.java)
/**
* 创建跟单
*/
@Transactional
fun createCopyTrading(request: CopyTradingCreateRequest): Result<CopyTradingDto> {
return try {
// 1. 验证账户是否存在
val account = accountRepository.findById(request.accountId).orElse(null)
?: return Result.failure(IllegalArgumentException("账户不存在"))
// 2. 验证模板是否存在
val template = templateRepository.findById(request.templateId).orElse(null)
?: return Result.failure(IllegalArgumentException("模板不存在"))
// 3. 验证 Leader 是否存在
val leader = leaderRepository.findById(request.leaderId).orElse(null)
?: return Result.failure(IllegalArgumentException("Leader 不存在"))
// 4. 检查是否已存在相同的跟单关系
val existing = copyTradingRepository.findByAccountIdAndTemplateIdAndLeaderId(
request.accountId,
request.templateId,
request.leaderId
)
if (existing != null) {
return Result.failure(IllegalArgumentException("该跟单关系已存在"))
}
// 5. 创建跟单关系
val copyTrading = CopyTrading(
accountId = request.accountId,
templateId = request.templateId,
leaderId = request.leaderId,
enabled = request.enabled
)
val saved = copyTradingRepository.save(copyTrading)
logger.info("成功创建跟单: ${saved.id}, account=${request.accountId}, template=${request.templateId}, leader=${request.leaderId}")
Result.success(toDto(saved, account, template, leader))
} catch (e: Exception) {
logger.error("创建跟单失败", e)
Result.failure(e)
}
}
/**
* 查询跟单列表
*/
fun getCopyTradingList(request: CopyTradingListRequest): Result<CopyTradingListResponse> {
return try {
val copyTradings = when {
request.accountId != null && request.templateId != null && request.leaderId != null -> {
val found = copyTradingRepository.findByAccountIdAndTemplateIdAndLeaderId(
request.accountId,
request.templateId,
request.leaderId
)
if (found != null) listOf(found) else emptyList()
}
request.accountId != null && request.templateId != null -> {
copyTradingRepository.findByAccountIdAndTemplateId(request.accountId, request.templateId)
}
request.accountId != null -> {
copyTradingRepository.findByAccountId(request.accountId)
}
request.templateId != null -> {
copyTradingRepository.findByTemplateId(request.templateId)
}
request.leaderId != null -> {
copyTradingRepository.findByLeaderId(request.leaderId)
}
request.enabled != null && request.enabled -> {
copyTradingRepository.findByEnabledTrue()
}
else -> {
copyTradingRepository.findAll()
}
}
// 过滤启用状态
val filtered = if (request.enabled != null) {
copyTradings.filter { it.enabled == request.enabled }
} else {
copyTradings
}
val dtos = filtered.map { copyTrading ->
val account = accountRepository.findById(copyTrading.accountId).orElse(null)
val template = templateRepository.findById(copyTrading.templateId).orElse(null)
val leader = leaderRepository.findById(copyTrading.leaderId).orElse(null)
if (account == null || template == null || leader == null) {
logger.warn("跟单关系数据不完整: ${copyTrading.id}")
null
} else {
toDto(copyTrading, account, template, leader)
}
}.filterNotNull()
Result.success(
CopyTradingListResponse(
list = dtos,
total = dtos.size.toLong()
)
)
} catch (e: Exception) {
logger.error("查询跟单列表失败", e)
Result.failure(e)
}
}
/**
* 更新跟单状态
*/
@Transactional
fun updateCopyTradingStatus(request: CopyTradingUpdateStatusRequest): Result<CopyTradingDto> {
return try {
val copyTrading = copyTradingRepository.findById(request.copyTradingId).orElse(null)
?: return Result.failure(IllegalArgumentException("跟单关系不存在"))
val updated = copyTrading.copy(
enabled = request.enabled,
updatedAt = System.currentTimeMillis()
)
val saved = copyTradingRepository.save(updated)
logger.info("成功更新跟单状态: ${saved.id}, enabled=${saved.enabled}")
val account = accountRepository.findById(saved.accountId).orElse(null)
val template = templateRepository.findById(saved.templateId).orElse(null)
val leader = leaderRepository.findById(saved.leaderId).orElse(null)
if (account == null || template == null || leader == null) {
return Result.failure(IllegalStateException("跟单关系数据不完整"))
}
Result.success(toDto(saved, account, template, leader))
} catch (e: Exception) {
logger.error("更新跟单状态失败", e)
Result.failure(e)
}
}
/**
* 删除跟单
*/
@Transactional
fun deleteCopyTrading(copyTradingId: Long): Result<Unit> {
return try {
val copyTrading = copyTradingRepository.findById(copyTradingId).orElse(null)
?: return Result.failure(IllegalArgumentException("跟单关系不存在"))
copyTradingRepository.delete(copyTrading)
logger.info("成功删除跟单: $copyTradingId")
Result.success(Unit)
} catch (e: Exception) {
logger.error("删除跟单失败", e)
Result.failure(e)
}
}
/**
* 查询钱包绑定的模板
*/
fun getAccountTemplates(accountId: Long): Result<AccountTemplatesResponse> {
return try {
// 验证账户是否存在
val account = accountRepository.findById(accountId).orElse(null)
?: return Result.failure(IllegalArgumentException("账户不存在"))
val copyTradings = copyTradingRepository.findByAccountId(accountId)
val dtos = copyTradings.mapNotNull { copyTrading ->
val template = templateRepository.findById(copyTrading.templateId).orElse(null)
val leader = leaderRepository.findById(copyTrading.leaderId).orElse(null)
if (template == null || leader == null) {
logger.warn("跟单关系数据不完整: ${copyTrading.id}")
null
} else {
AccountTemplateDto(
templateId = template.id!!,
templateName = template.templateName,
copyTradingId = copyTrading.id!!,
leaderId = leader.id!!,
leaderName = leader.leaderName,
leaderAddress = leader.leaderAddress,
enabled = copyTrading.enabled
)
}
}
Result.success(
AccountTemplatesResponse(
list = dtos,
total = dtos.size.toLong()
)
)
} catch (e: Exception) {
logger.error("查询钱包绑定的模板失败", e)
Result.failure(e)
}
}
/**
* 转换为 DTO
*/
private fun toDto(
copyTrading: CopyTrading,
account: com.wrbug.polymarketbot.entity.Account,
template: com.wrbug.polymarketbot.entity.CopyTradingTemplate,
leader: com.wrbug.polymarketbot.entity.Leader
): CopyTradingDto {
return CopyTradingDto(
id = copyTrading.id!!,
accountId = account.id!!,
accountName = account.accountName,
walletAddress = account.walletAddress,
templateId = template.id!!,
templateName = template.templateName,
leaderId = leader.id!!,
leaderName = leader.leaderName,
leaderAddress = leader.leaderAddress,
enabled = copyTrading.enabled,
createdAt = copyTrading.createdAt,
updatedAt = copyTrading.updatedAt
)
}
}
@@ -0,0 +1,264 @@
package com.wrbug.polymarketbot.service
import com.wrbug.polymarketbot.dto.*
import com.wrbug.polymarketbot.entity.CopyTradingTemplate
import com.wrbug.polymarketbot.repository.CopyTradingRepository
import com.wrbug.polymarketbot.repository.CopyTradingTemplateRepository
import com.wrbug.polymarketbot.util.toSafeBigDecimal
import org.slf4j.LoggerFactory
import org.springframework.stereotype.Service
import org.springframework.transaction.annotation.Transactional
import java.math.BigDecimal
/**
* 跟单模板管理服务
*/
@Service
class CopyTradingTemplateService(
private val templateRepository: CopyTradingTemplateRepository,
private val copyTradingRepository: CopyTradingRepository
) {
private val logger = LoggerFactory.getLogger(CopyTradingTemplateService::class.java)
/**
* 创建模板
*/
@Transactional
fun createTemplate(request: TemplateCreateRequest): Result<TemplateDto> {
return try {
// 1. 验证模板名称
if (request.templateName.isBlank()) {
return Result.failure(IllegalArgumentException("模板名称不能为空"))
}
// 2. 检查模板名称是否已存在
if (templateRepository.existsByTemplateName(request.templateName)) {
return Result.failure(IllegalArgumentException("模板名称已存在"))
}
// 3. 验证 copyMode
if (request.copyMode !in listOf("RATIO", "FIXED")) {
return Result.failure(IllegalArgumentException("copyMode 必须是 RATIO 或 FIXED"))
}
// 4. 创建模板
val template = CopyTradingTemplate(
templateName = request.templateName,
copyMode = request.copyMode,
copyRatio = request.copyRatio?.toSafeBigDecimal() ?: BigDecimal.ONE,
fixedAmount = request.fixedAmount?.toSafeBigDecimal(),
maxOrderSize = request.maxOrderSize?.toSafeBigDecimal() ?: "1000".toSafeBigDecimal(),
minOrderSize = request.minOrderSize?.toSafeBigDecimal() ?: "1".toSafeBigDecimal(),
maxDailyLoss = request.maxDailyLoss?.toSafeBigDecimal() ?: "10000".toSafeBigDecimal(),
maxDailyOrders = request.maxDailyOrders ?: 100,
priceTolerance = request.priceTolerance?.toSafeBigDecimal() ?: "5".toSafeBigDecimal(),
delaySeconds = request.delaySeconds ?: 0,
pollIntervalSeconds = request.pollIntervalSeconds ?: 5,
useWebSocket = request.useWebSocket ?: true,
websocketReconnectInterval = request.websocketReconnectInterval ?: 5000,
websocketMaxRetries = request.websocketMaxRetries ?: 10,
supportSell = request.supportSell ?: true
)
val saved = templateRepository.save(template)
logger.info("成功创建模板: ${saved.id}, ${saved.templateName}")
Result.success(toDto(saved))
} catch (e: Exception) {
logger.error("创建模板失败", e)
Result.failure(e)
}
}
/**
* 更新模板
*/
@Transactional
fun updateTemplate(request: TemplateUpdateRequest): Result<TemplateDto> {
return try {
val template = templateRepository.findById(request.templateId).orElse(null)
?: return Result.failure(IllegalArgumentException("模板不存在"))
// 如果提供了模板名称,验证名称唯一性
if (request.templateName != null) {
if (request.templateName.isBlank()) {
return Result.failure(IllegalArgumentException("模板名称不能为空"))
}
// 如果新名称与当前名称不同,检查是否已存在
if (request.templateName != template.templateName) {
if (templateRepository.existsByTemplateName(request.templateName)) {
return Result.failure(IllegalArgumentException("模板名称已存在"))
}
}
}
// 验证 copyMode
if (request.copyMode != null && request.copyMode !in listOf("RATIO", "FIXED")) {
return Result.failure(IllegalArgumentException("copyMode 必须是 RATIO 或 FIXED"))
}
val updated = template.copy(
templateName = request.templateName ?: template.templateName,
copyMode = request.copyMode ?: template.copyMode,
copyRatio = request.copyRatio?.toSafeBigDecimal() ?: template.copyRatio,
fixedAmount = request.fixedAmount?.toSafeBigDecimal() ?: template.fixedAmount,
maxOrderSize = request.maxOrderSize?.toSafeBigDecimal() ?: template.maxOrderSize,
minOrderSize = request.minOrderSize?.toSafeBigDecimal() ?: template.minOrderSize,
maxDailyLoss = request.maxDailyLoss?.toSafeBigDecimal() ?: template.maxDailyLoss,
maxDailyOrders = request.maxDailyOrders ?: template.maxDailyOrders,
priceTolerance = request.priceTolerance?.toSafeBigDecimal() ?: template.priceTolerance,
delaySeconds = request.delaySeconds ?: template.delaySeconds,
pollIntervalSeconds = request.pollIntervalSeconds ?: template.pollIntervalSeconds,
useWebSocket = request.useWebSocket ?: template.useWebSocket,
websocketReconnectInterval = request.websocketReconnectInterval ?: template.websocketReconnectInterval,
websocketMaxRetries = request.websocketMaxRetries ?: template.websocketMaxRetries,
supportSell = request.supportSell ?: template.supportSell,
updatedAt = System.currentTimeMillis()
)
val saved = templateRepository.save(updated)
logger.info("成功更新模板: ${saved.id}")
Result.success(toDto(saved))
} catch (e: Exception) {
logger.error("更新模板失败", e)
Result.failure(e)
}
}
/**
* 删除模板
*/
@Transactional
fun deleteTemplate(templateId: Long): Result<Unit> {
return try {
val template = templateRepository.findById(templateId).orElse(null)
?: return Result.failure(IllegalArgumentException("模板不存在"))
// 检查是否有跟单正在使用该模板
val useCount = copyTradingRepository.countByTemplateId(templateId)
if (useCount > 0) {
return Result.failure(IllegalStateException("该模板还有 $useCount 个跟单关系在使用,请先删除跟单关系"))
}
templateRepository.delete(template)
logger.info("成功删除模板: $templateId")
Result.success(Unit)
} catch (e: Exception) {
logger.error("删除模板失败", e)
Result.failure(e)
}
}
/**
* 复制模板
*/
@Transactional
fun copyTemplate(request: TemplateCopyRequest): Result<TemplateDto> {
return try {
val sourceTemplate = templateRepository.findById(request.templateId).orElse(null)
?: return Result.failure(IllegalArgumentException("源模板不存在"))
// 检查新模板名称是否已存在
if (templateRepository.existsByTemplateName(request.templateName)) {
return Result.failure(IllegalArgumentException("模板名称已存在"))
}
// 创建新模板
val newTemplate = CopyTradingTemplate(
templateName = request.templateName,
copyMode = request.copyMode ?: sourceTemplate.copyMode,
copyRatio = request.copyRatio?.toSafeBigDecimal() ?: sourceTemplate.copyRatio,
fixedAmount = request.fixedAmount?.toSafeBigDecimal() ?: sourceTemplate.fixedAmount,
maxOrderSize = request.maxOrderSize?.toSafeBigDecimal() ?: sourceTemplate.maxOrderSize,
minOrderSize = request.minOrderSize?.toSafeBigDecimal() ?: sourceTemplate.minOrderSize,
maxDailyLoss = request.maxDailyLoss?.toSafeBigDecimal() ?: sourceTemplate.maxDailyLoss,
maxDailyOrders = request.maxDailyOrders ?: sourceTemplate.maxDailyOrders,
priceTolerance = request.priceTolerance?.toSafeBigDecimal() ?: sourceTemplate.priceTolerance,
delaySeconds = request.delaySeconds ?: sourceTemplate.delaySeconds,
pollIntervalSeconds = request.pollIntervalSeconds ?: sourceTemplate.pollIntervalSeconds,
useWebSocket = request.useWebSocket ?: sourceTemplate.useWebSocket,
websocketReconnectInterval = request.websocketReconnectInterval ?: sourceTemplate.websocketReconnectInterval,
websocketMaxRetries = request.websocketMaxRetries ?: sourceTemplate.websocketMaxRetries,
supportSell = request.supportSell ?: sourceTemplate.supportSell
)
val saved = templateRepository.save(newTemplate)
logger.info("成功复制模板: ${sourceTemplate.id} -> ${saved.id}")
Result.success(toDto(saved))
} catch (e: Exception) {
logger.error("复制模板失败", e)
Result.failure(e)
}
}
/**
* 查询模板列表
*/
fun getTemplateList(): Result<TemplateListResponse> {
return try {
val templates = templateRepository.findAllByOrderByCreatedAtDesc()
val templateDtos = templates.map { template ->
val useCount = copyTradingRepository.countByTemplateId(template.id!!)
toDto(template, useCount)
}
Result.success(
TemplateListResponse(
list = templateDtos,
total = templateDtos.size.toLong()
)
)
} catch (e: Exception) {
logger.error("查询模板列表失败", e)
Result.failure(e)
}
}
/**
* 查询模板详情
*/
fun getTemplateDetail(templateId: Long): Result<TemplateDto> {
return try {
val template = templateRepository.findById(templateId).orElse(null)
?: return Result.failure(IllegalArgumentException("模板不存在"))
val useCount = copyTradingRepository.countByTemplateId(templateId)
Result.success(toDto(template, useCount))
} catch (e: Exception) {
logger.error("查询模板详情失败", e)
Result.failure(e)
}
}
/**
* 转换为 DTO
*/
private fun toDto(template: CopyTradingTemplate, useCount: Long = 0): TemplateDto {
return TemplateDto(
id = template.id!!,
templateName = template.templateName,
copyMode = template.copyMode,
copyRatio = template.copyRatio.toPlainString(),
fixedAmount = template.fixedAmount?.toPlainString(),
maxOrderSize = template.maxOrderSize.toPlainString(),
minOrderSize = template.minOrderSize.toPlainString(),
maxDailyLoss = template.maxDailyLoss.toPlainString(),
maxDailyOrders = template.maxDailyOrders,
priceTolerance = template.priceTolerance.toPlainString(),
delaySeconds = template.delaySeconds,
pollIntervalSeconds = template.pollIntervalSeconds,
useWebSocket = template.useWebSocket,
websocketReconnectInterval = template.websocketReconnectInterval,
websocketMaxRetries = template.websocketMaxRetries,
supportSell = template.supportSell,
useCount = useCount,
createdAt = template.createdAt,
updatedAt = template.updatedAt
)
}
}
@@ -0,0 +1,194 @@
package com.wrbug.polymarketbot.service
import com.wrbug.polymarketbot.dto.*
import com.wrbug.polymarketbot.entity.Leader
import com.wrbug.polymarketbot.repository.AccountRepository
import com.wrbug.polymarketbot.repository.CopyTradingRepository
import com.wrbug.polymarketbot.repository.LeaderRepository
import com.wrbug.polymarketbot.util.CategoryValidator
import org.slf4j.LoggerFactory
import org.springframework.stereotype.Service
import org.springframework.transaction.annotation.Transactional
/**
* Leader 管理服务
*/
@Service
class LeaderService(
private val leaderRepository: LeaderRepository,
private val accountRepository: AccountRepository,
private val copyTradingRepository: CopyTradingRepository
) {
private val logger = LoggerFactory.getLogger(LeaderService::class.java)
/**
* 添加被跟单者
*/
@Transactional
fun addLeader(request: LeaderAddRequest): Result<LeaderDto> {
return try {
// 1. 验证地址格式
if (!isValidWalletAddress(request.leaderAddress)) {
return Result.failure(IllegalArgumentException("无效的钱包地址格式"))
}
// 2. 验证分类
if (request.category != null) {
CategoryValidator.validate(request.category)
}
// 3. 检查是否已存在
if (leaderRepository.existsByLeaderAddress(request.leaderAddress)) {
return Result.failure(IllegalArgumentException("该 Leader 地址已存在"))
}
// 4. 验证 Leader 地址不能与自己的地址相同
if (accountRepository.existsByWalletAddress(request.leaderAddress)) {
return Result.failure(IllegalArgumentException("Leader 地址不能与自己的账户地址相同"))
}
// 5. 创建 Leader
val leader = Leader(
leaderAddress = request.leaderAddress,
leaderName = request.leaderName,
category = request.category
)
val saved = leaderRepository.save(leader)
logger.info("成功添加 Leader: ${saved.id}, ${saved.leaderAddress}")
Result.success(toDto(saved))
} catch (e: Exception) {
logger.error("添加 Leader 失败", e)
Result.failure(e)
}
}
/**
* 更新被跟单者
*/
@Transactional
fun updateLeader(request: LeaderUpdateRequest): Result<LeaderDto> {
return try {
val leader = leaderRepository.findById(request.leaderId).orElse(null)
?: return Result.failure(IllegalArgumentException("Leader 不存在"))
// 验证分类
if (request.category != null) {
CategoryValidator.validate(request.category)
}
val updated = leader.copy(
leaderName = request.leaderName ?: leader.leaderName,
category = request.category ?: leader.category,
updatedAt = System.currentTimeMillis()
)
val saved = leaderRepository.save(updated)
logger.info("成功更新 Leader: ${saved.id}")
Result.success(toDto(saved))
} catch (e: Exception) {
logger.error("更新 Leader 失败", e)
Result.failure(e)
}
}
/**
* 删除被跟单者
*/
@Transactional
fun deleteLeader(leaderId: Long): Result<Unit> {
return try {
val leader = leaderRepository.findById(leaderId).orElse(null)
?: return Result.failure(IllegalArgumentException("Leader 不存在"))
// 检查是否有跟单关系
val copyTradingCount = copyTradingRepository.countByLeaderId(leaderId)
if (copyTradingCount > 0) {
return Result.failure(IllegalStateException("该 Leader 还有 $copyTradingCount 个跟单关系,请先删除跟单关系"))
}
leaderRepository.delete(leader)
logger.info("成功删除 Leader: $leaderId")
Result.success(Unit)
} catch (e: Exception) {
logger.error("删除 Leader 失败", e)
Result.failure(e)
}
}
/**
* 查询 Leader 列表
*/
fun getLeaderList(request: LeaderListRequest): Result<LeaderListResponse> {
return try {
// 验证分类
if (request.category != null) {
CategoryValidator.validate(request.category)
}
val leaders = if (request.category != null) {
leaderRepository.findByCategory(request.category)
} else {
leaderRepository.findAllByOrderByCreatedAtAsc()
}
val leaderDtos = leaders.map { leader ->
val copyTradingCount = copyTradingRepository.countByLeaderId(leader.id!!)
toDto(leader, copyTradingCount)
}
Result.success(
LeaderListResponse(
list = leaderDtos,
total = leaderDtos.size.toLong()
)
)
} catch (e: Exception) {
logger.error("查询 Leader 列表失败", e)
Result.failure(e)
}
}
/**
* 查询 Leader 详情
*/
fun getLeaderDetail(leaderId: Long): Result<LeaderDto> {
return try {
val leader = leaderRepository.findById(leaderId).orElse(null)
?: return Result.failure(IllegalArgumentException("Leader 不存在"))
val copyTradingCount = copyTradingRepository.countByLeaderId(leaderId)
Result.success(toDto(leader, copyTradingCount))
} catch (e: Exception) {
logger.error("查询 Leader 详情失败", e)
Result.failure(e)
}
}
/**
* 转换为 DTO
*/
private fun toDto(leader: Leader, copyTradingCount: Long = 0): LeaderDto {
return LeaderDto(
id = leader.id!!,
leaderAddress = leader.leaderAddress,
leaderName = leader.leaderName,
category = leader.category,
copyTradingCount = copyTradingCount,
createdAt = leader.createdAt,
updatedAt = leader.updatedAt
)
}
/**
* 验证钱包地址格式
*/
private fun isValidWalletAddress(address: String): Boolean {
return address.startsWith("0x") && address.length == 42
}
}
@@ -0,0 +1,12 @@
-- 创建被跟单者(Leader)表
CREATE TABLE IF NOT EXISTS copy_trading_leaders (
id BIGINT AUTO_INCREMENT PRIMARY KEY,
leader_address VARCHAR(42) NOT NULL UNIQUE COMMENT '被跟单者的钱包地址',
leader_name VARCHAR(100) NULL COMMENT '被跟单者名称',
category VARCHAR(20) NULL COMMENT '分类筛选(sports/crypto),null表示不筛选',
created_at BIGINT NOT NULL COMMENT '创建时间(毫秒时间戳)',
updated_at BIGINT NOT NULL COMMENT '更新时间(毫秒时间戳)',
INDEX idx_leader_address (leader_address),
INDEX idx_category (category)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='被跟单者表';
@@ -0,0 +1,23 @@
-- 创建跟单模板表
CREATE TABLE IF NOT EXISTS copy_trading_templates (
id BIGINT AUTO_INCREMENT PRIMARY KEY,
template_name VARCHAR(100) NOT NULL UNIQUE COMMENT '模板名称',
copy_mode VARCHAR(10) NOT NULL DEFAULT 'RATIO' COMMENT '跟单金额模式(RATIO/FIXED',
copy_ratio DECIMAL(10, 2) NOT NULL DEFAULT 1.00 COMMENT '跟单比例(仅在copyMode=RATIO时生效)',
fixed_amount DECIMAL(20, 8) NULL COMMENT '固定跟单金额(仅在copyMode=FIXED时生效)',
max_order_size DECIMAL(20, 8) NOT NULL DEFAULT 1000.00000000 COMMENT '单笔订单最大金额(USDC',
min_order_size DECIMAL(20, 8) NOT NULL DEFAULT 1.00000000 COMMENT '单笔订单最小金额(USDC',
max_daily_loss DECIMAL(20, 8) NOT NULL DEFAULT 10000.00000000 COMMENT '每日最大亏损限制(USDC',
max_daily_orders INT NOT NULL DEFAULT 100 COMMENT '每日最大跟单订单数',
price_tolerance DECIMAL(5, 2) NOT NULL DEFAULT 5.00 COMMENT '价格容忍度(百分比,0-100',
delay_seconds INT NOT NULL DEFAULT 0 COMMENT '跟单延迟(秒,默认0立即跟单)',
poll_interval_seconds INT NOT NULL DEFAULT 5 COMMENT '轮询间隔(秒,仅在WebSocket不可用时使用)',
use_websocket BOOLEAN NOT NULL DEFAULT TRUE COMMENT '是否优先使用WebSocket推送',
websocket_reconnect_interval INT NOT NULL DEFAULT 5000 COMMENT 'WebSocket重连间隔(毫秒)',
websocket_max_retries INT NOT NULL DEFAULT 10 COMMENT 'WebSocket最大重试次数',
support_sell BOOLEAN NOT NULL DEFAULT TRUE COMMENT '是否支持跟单卖出',
created_at BIGINT NOT NULL COMMENT '创建时间(毫秒时间戳)',
updated_at BIGINT NOT NULL COMMENT '更新时间(毫秒时间戳)',
INDEX idx_template_name (template_name)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='跟单模板表';
@@ -0,0 +1,19 @@
-- 创建跟单关系表(钱包-模板关联,多对多关系)
CREATE TABLE IF NOT EXISTS copy_trading (
id BIGINT AUTO_INCREMENT PRIMARY KEY,
account_id BIGINT NOT NULL COMMENT '钱包账户ID',
template_id BIGINT NOT NULL COMMENT '模板ID',
leader_id BIGINT NOT NULL COMMENT 'Leader ID',
enabled BOOLEAN NOT NULL DEFAULT TRUE COMMENT '是否启用',
created_at BIGINT NOT NULL COMMENT '创建时间(毫秒时间戳)',
updated_at BIGINT NOT NULL COMMENT '更新时间(毫秒时间戳)',
UNIQUE KEY uk_account_template_leader (account_id, template_id, leader_id),
INDEX idx_account_id (account_id),
INDEX idx_template_id (template_id),
INDEX idx_leader_id (leader_id),
INDEX idx_enabled (enabled),
FOREIGN KEY (account_id) REFERENCES copy_trading_accounts(id) ON DELETE CASCADE,
FOREIGN KEY (template_id) REFERENCES copy_trading_templates(id) ON DELETE RESTRICT,
FOREIGN KEY (leader_id) REFERENCES copy_trading_leaders(id) ON DELETE CASCADE
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='跟单关系表(钱包-模板关联)';