Skip to content

Commit a5df054

Browse files
committed
update: 优化主控删除和实例删除的逻辑
1 parent 27425e1 commit a5df054

2 files changed

Lines changed: 14 additions & 1 deletion

File tree

internal/endpoint/service.go

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ func (s *Service) DeleteEndpoint(id int64) error {
240240
return err
241241
}
242242

243-
// 2) 删除关联隧道
243+
// 2) 删除SSE日志
244244
if _, err := tx.Exec(`DELETE FROM "EndpointSSE" WHERE endpointId = ?`, id); err != nil {
245245
tx.Rollback()
246246
return err
@@ -252,6 +252,14 @@ func (s *Service) DeleteEndpoint(id int64) error {
252252
tx.Rollback()
253253
return err
254254
}
255+
256+
// 3) 删除回收站
257+
res, err = tx.Exec(`DELETE FROM "TunnelRecycle" WHERE endpointId = ?`, id)
258+
if err != nil {
259+
tx.Rollback()
260+
return err
261+
}
262+
255263
affected, err := res.RowsAffected()
256264
if err != nil {
257265
tx.Rollback()

internal/tunnel/service.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -906,11 +906,16 @@ func (s *Service) DeleteTunnelAndWait(instanceID string, timeout time.Duration,
906906
if err != nil {
907907
return err
908908
}
909+
909910
rows, _ := result.RowsAffected()
910911
if rows == 0 {
911912
return errors.New("隧道删除失败")
912913
}
913914

915+
if !recycle {
916+
_, _ = s.db.Exec(`DELETE FROM "EndpointSSE" WHERE instanceId = ?`, instanceID)
917+
}
918+
914919
// 更新端点隧道计数
915920
_, _ = s.db.Exec(`UPDATE "Endpoint" SET tunnelCount = (
916921
SELECT COUNT(*) FROM "Tunnel" WHERE endpointId = ?

0 commit comments

Comments
 (0)