summaryrefslogtreecommitdiff
path: root/recording/config.go
blob: 975887b7f638921f76dc4bcb2fafbf8fe644a20c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package recording

import (
	"bilibili-livestream-archiver/common"
	"fmt"
)

type TaskConfig struct {
	RoomId    common.RoomId   `mapstructure:"room_id"`
	Transport TransportConfig `mapstructure:"transport"`
	Download  DownloadConfig  `mapstructure:"download"`
}

type TransportConfig struct {
	SocketTimeoutSeconds int `mapstructure:"socket_timeout_seconds"`
	RetryIntervalSeconds int `mapstructure:"retry_interval_seconds"`
	MaxRetryTimes        int `mapstructure:"max_retry_times"`
}

type DownloadConfig struct {
	SaveDirectory        string `mapstructure:"save_directory"`
	DiskWriteBufferBytes int    `mapstructure:"disk_write_buffer_bytes"`
}

func DefaultTransportConfig() TransportConfig {
	return TransportConfig{
		SocketTimeoutSeconds: 10,
		RetryIntervalSeconds: 2,
		MaxRetryTimes:        5,
	}
}

func (t TaskConfig) String() string {
	return fmt.Sprintf("Room ID: %v, %v, %v", t.RoomId, t.Transport.String(), t.Download.String())
}

func (t TransportConfig) String() string {
	return fmt.Sprintf("Socket timeout: %vs, Retry interval: %vs, Max retry times: %v",
		t.SocketTimeoutSeconds, t.RetryIntervalSeconds, t.MaxRetryTimes)
}

func (d DownloadConfig) String() string {
	return fmt.Sprintf("Save directory: \"%v\"", d.SaveDirectory)
}