diff options
| author | Keuin <[email protected]> | 2022-09-08 00:31:26 +0800 |
|---|---|---|
| committer | Keuin <[email protected]> | 2022-09-08 00:31:26 +0800 |
| commit | a720c2c16b442b668db465fbbc70740cfc7ddee4 (patch) | |
| tree | aac1ba37e4a19c3e9c34f70e9297b0a9e3803db6 /recording/runner.go | |
| parent | d00c97e9dbfb59672ced042af8a6e849efab98cc (diff) | |
Load config from file or cli.
Diffstat (limited to 'recording/runner.go')
| -rw-r--r-- | recording/runner.go | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/recording/runner.go b/recording/runner.go index a4693fb..d4e5a96 100644 --- a/recording/runner.go +++ b/recording/runner.go @@ -10,10 +10,12 @@ import ( "bilibili-livestream-archiver/common" "context" "encoding/json" + "errors" "fmt" "log" "os" "path" + "sync" "time" ) @@ -25,7 +27,8 @@ type TaskResult struct { // RunTask start a monitor&download task and // put its execution result into a channel. -func RunTask(ctx context.Context, task *TaskConfig, chTaskResult chan<- TaskResult) { +func RunTask(ctx context.Context, wg *sync.WaitGroup, task *TaskConfig, chTaskResult chan<- TaskResult) { + defer wg.Done() err := doTask(ctx, task) chTaskResult <- TaskResult{ Task: task, @@ -146,7 +149,7 @@ func record( fileName := fmt.Sprintf( "%s.%s", GenerateFileName(profile.Data.Title, time.Now()), - common.Optional[string](common.GetFileExtensionFromUrl(streamSource.URL)).OrElse("flv"), + common.Errorable[string](common.GetFileExtensionFromUrl(streamSource.URL)).OrElse("flv"), ) filePath := path.Join(task.Download.SaveDirectory, fileName) file, err := os.OpenFile(filePath, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644) |