调整 wg Done 的位置

Signed-off-by: allan716 <[email protected]>
allanpk716 · Aug 9, 2022 · f7a6458 · f7a6458
1 parent 6e65987
commit f7a6458
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/scan_tools.go b/pkg/scan_tools.go
@@ -46,7 +46,7 @@ func (s ScanTools) Scan(protocolType ProtocolType, inputInfo InputInfo, showProg
 				println(protocolType.String(), deliveryInfo.Host, deliveryInfo.Port, checkResult.Success)
 			}
 			deliveryInfo.CheckResultChan <- checkResult
-			deliveryInfo.Wg.Done()
+			//deliveryInfo.Wg.Done()
 		}()
 
 		switch protocolType {
@@ -131,6 +131,7 @@ func (s ScanTools) Scan(protocolType ProtocolType, inputInfo InputInfo, showProg
 	outputInfo := OutputInfo{
 		ProtocolType: protocolType,
 	}
+	wg := &sync.WaitGroup{}
 	outputInfo.SuccessMapString = make(map[string][]string, 0)
 	outputInfo.FailedMapString = make(map[string][]string, 0)
 	go func() {
@@ -150,14 +151,13 @@ func (s ScanTools) Scan(protocolType ProtocolType, inputInfo InputInfo, showProg
 						outputInfo.SuccessMapString[revCheckResult.Host] = []string{revCheckResult.Port}
 					}
 				}
+				wg.Done()
 			case <-exitRevResultChan:
 				return
 			}
 		}
 	}()
 	// --------------------------------------------------
-	wg := &sync.WaitGroup{}
-
 	for _, ipRangeInfo := range ipRangeInfos {
 
 		if ipRangeInfo.CICR != nil {