diff options
| -rw-r--r-- | recording/config.go | 3 | ||||
| -rw-r--r-- | recording/runner.go | 15 |
2 files changed, 15 insertions, 3 deletions
diff --git a/recording/config.go b/recording/config.go index bce9a8f..975887b 100644 --- a/recording/config.go +++ b/recording/config.go @@ -18,7 +18,8 @@ type TransportConfig struct { } type DownloadConfig struct { - SaveDirectory string `mapstructure:"save_directory"` + SaveDirectory string `mapstructure:"save_directory"` + DiskWriteBufferBytes int `mapstructure:"disk_write_buffer_bytes"` } func DefaultTransportConfig() TransportConfig { diff --git a/recording/runner.go b/recording/runner.go index 7b82a5c..c95e559 100644 --- a/recording/runner.go +++ b/recording/runner.go @@ -8,6 +8,7 @@ package recording import ( "bilibili-livestream-archiver/bilibili" "bilibili-livestream-archiver/common" + "bufio" "context" "encoding/json" "errors" @@ -165,15 +166,25 @@ func record( ) filePath := path.Join(task.Download.SaveDirectory, fileName) file, err := os.OpenFile(filePath, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644) - defer func() { _ = file.Close() }() if err != nil { logger.Printf("ERROR: Cannot open file for writing: %v", err) cancelled = true return } + defer func() { _ = file.Close() }() + + // buffered writer + fWriter := bufio.NewWriterSize(file, task.Download.DiskWriteBufferBytes) + defer func() { + err := fWriter.Flush() + if err != nil { + logger := log.Default() + logger.Printf("Failed to flush buffered file write data: %v\n", err) + } + }() logger.Printf("Recording live stream to file \"%v\"...", filePath) - err = bi.CopyLiveStream(ctx, task.RoomId, streamSource, file) + err = bi.CopyLiveStream(ctx, task.RoomId, streamSource, fWriter) cancelled = err == nil || errors.Is(err, context.Canceled) if !cancelled { // real error happens |