Browse Source

修复细节问题

Signed-off-by: allan716 <[email protected]>
allan716 3 years ago
parent
commit
d4cfb77530

+ 3 - 1
internal/logic/downloader_helper/downloader_helper.go

@@ -24,6 +24,7 @@ func NewDownloaderHelper(settings settings.Settings, _subSupplierHub *subSupplie
 	}
 }
 
+// Start 开启任务
 func (d DownloaderHelper) Start() error {
 	var err error
 	// 下载实例
@@ -86,6 +87,7 @@ func (d DownloaderHelper) Start() error {
 	return nil
 }
 
-func (d DownloaderHelper) Stop() {
+// Cancel 提前取消任务的执行
+func (d DownloaderHelper) Cancel() {
 	d.downloader.Cancel()
 }

+ 6 - 1
internal/logic/pre_download_process/pre_download_proces.go

@@ -228,7 +228,12 @@ func (p *PreDownloadProcess) ReloadBrowser() *PreDownloadProcess {
 
 func (p *PreDownloadProcess) Wait() error {
 
-	return errors.New(p.stageName + " " + p.gError.Error())
+	if p.gError != nil {
+		return errors.New(p.stageName + " " + p.gError.Error())
+
+	} else {
+		return nil
+	}
 }
 
 const (