heimoshuiyu/msw-open-music
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

prepare play

Browse Source
This commit is contained in:
黑墨水鱼
2021-05-25 12:49:51 +08:00
parent 29b9b22759
commit b7e1c757f6
5 changed files with 277 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

142
internal/pkg/api/api.go
View File

@@ -7,6 +7,7 @@ import (
"io"
"log"
"msw-open-music/internal/pkg/database"
"msw-open-music/internal/pkg/tmpfs"
"net/http"
"os"
"os/exec"
@@ -20,6 +21,7 @@ type API struct {
Server http.Server
token string
APIConfig APIConfig
Tmpfs *tmpfs.Tmpfs
}
type FfmpegConfigs struct {
@@ -345,23 +347,39 @@ func (api *API) HandleGetFileInfo(w http.ResponseWriter, r *http.Request) {
}
}
func (api *API) HandleGetFileStream(w http.ResponseWriter, r *http.Request) {
func (api *API) CheckGetFileStream(w http.ResponseWriter, r *http.Request) (error) {
var err error
q := r.URL.Query()
ids := q["id"]
if len(ids) == 0 {
api.HandleErrorString(w, r, `parameter "id" can't be empty`)
return
err = errors.New(`parameter "id" can't be empty`)
api.HandleError(w, r, err)
return err
}
id, err := strconv.Atoi(ids[0])
_, err = strconv.Atoi(ids[0])
if err != nil {
api.HandleErrorString(w, r, `parameter "id" should be an integer`)
return
err = errors.New(`parameter "id" should be an integer`)
api.HandleError(w, r, err)
return err
}
configs := q["config"]
if len(configs) == 0 {
api.HandleErrorString(w, r, `parameter "config" can't be empty`)
err = errors.New(`parameter "config" can't be empty`)
api.HandleError(w, r, err)
return err
}
return nil
}
func (api *API) HandleGetFileStream(w http.ResponseWriter, r *http.Request) {
err := api.CheckGetFileStream(w, r)
if err != nil {
return
}
q := r.URL.Query()
ids := q["id"]
id, err := strconv.Atoi(ids[0])
configs := q["config"]
configName := configs[0]
file, err := api.Db.GetFile(int64(id))
if err != nil {
@@ -384,7 +402,7 @@ func (api *API) HandleGetFileStream(w http.ResponseWriter, r *http.Request) {
}
args := strings.Split(ffmpegConfig.Args, " ")
startArgs := []string {"-i", path}
endArgs := []string {"-vn", "-f", "matroska", "-"}
endArgs := []string {"-vn", "-f", "ogg", "-"}
ffmpegArgs := append(startArgs, args...)
ffmpegArgs = append(ffmpegArgs, endArgs...)
cmd := exec.Command("ffmpeg", ffmpegArgs...)
@@ -396,6 +414,111 @@ func (api *API) HandleGetFileStream(w http.ResponseWriter, r *http.Request) {
}
}
type PrepareFileStreamDirectRequest struct {
ID int64 `json:"id"`
ConfigName string `json:"config_name"`
}
type PrepareFileStreamDirectResponse struct {
Filesize int64 `json:"filesize"`
}
func (api *API) HandlePrepareFileStreamDirect(w http.ResponseWriter, r *http.Request) {
prepareFileStreamDirectRequst := &PrepareFileStreamDirectRequest{
ID: -1,
}
err := json.NewDecoder(r.Body).Decode(prepareFileStreamDirectRequst)
if err != nil {
api.HandleError(w, r, err)
return
}
// check empty
if prepareFileStreamDirectRequst.ID < 0 {
api.HandleErrorString(w, r, `"id" can't be none or negative`)
return
}
if prepareFileStreamDirectRequst.ConfigName == "" {
api.HandleErrorString(w, r, `"config_name" can't be empty`)
return
}
file, err := api.Db.GetFile(prepareFileStreamDirectRequst.ID)
if err != nil {
api.HandleError(w, r, err)
return
}
srcPath, err := file.Path()
if err != nil {
api.HandleError(w, r, err)
return
}
log.Println("[api] Prepare stream direct file", srcPath, prepareFileStreamDirectRequst.ConfigName)
ffmpegConfig, ok := api.APIConfig.FfmpegConfigs[prepareFileStreamDirectRequst.ConfigName]
if !ok {
api.HandleErrorStringCode(w, r, `ffmpeg config not found`, 404)
return
}
objPath := api.Tmpfs.GetObjFilePath(prepareFileStreamDirectRequst.ID, prepareFileStreamDirectRequst.ConfigName)
// check obj file exists
exists := api.Tmpfs.Exits(objPath)
if exists {
fileInfo, err := os.Stat(objPath)
if err != nil {
api.HandleError(w, r, err)
return
}
prepareFileStreamDirectResponse := &PrepareFileStreamDirectResponse{
Filesize: fileInfo.Size(),
}
json.NewEncoder(w).Encode(prepareFileStreamDirectResponse)
return
}
api.Tmpfs.Record(objPath)
args := strings.Split(ffmpegConfig.Args, " ")
startArgs := []string {"-i", srcPath}
endArgs := []string {"-vn", "-y", objPath}
ffmpegArgs := append(startArgs, args...)
ffmpegArgs = append(ffmpegArgs, endArgs...)
cmd := exec.Command("ffmpeg", ffmpegArgs...)
err = cmd.Run()
if err != nil {
api.HandleError(w, r, err)
return
}
fileInfo, err := os.Stat(objPath)
if err != nil {
api.HandleError(w, r, err)
return
}
prepareFileStreamDirectResponse := &PrepareFileStreamDirectResponse{
Filesize: fileInfo.Size(),
}
json.NewEncoder(w).Encode(prepareFileStreamDirectResponse)
}
func (api *API) HandleGetFileStreamDirect(w http.ResponseWriter, r *http.Request) {
err := api.CheckGetFileStream(w, r)
if err != nil {
return
}
q := r.URL.Query()
ids := q["id"]
id, err := strconv.Atoi(ids[0])
configs := q["config"]
configName := configs[0]
path := api.Tmpfs.GetObjFilePath(int64(id), configName)
log.Println("[api] Get direct cached file", path)
http.ServeFile(w, r, path)
}
func (api *API) HandleGetFileDirect(w http.ResponseWriter, r *http.Request) {
q := r.URL.Query()
ids := q["id"]
@@ -572,6 +695,7 @@ func NewAPI(apiConfig APIConfig) (*API, error) {
},
APIConfig: apiConfig,
}
api.Tmpfs = tmpfs.NewTmpfs()
// mount api
apiMux.HandleFunc("/hello", api.HandleOK)
@@ -585,6 +709,8 @@ func NewAPI(apiConfig APIConfig) (*API, error) {
apiMux.HandleFunc("/get_ffmpeg_config_list", api.HandleGetFfmpegConfigs)
apiMux.HandleFunc("/feedback", api.HandleFeedback)
apiMux.HandleFunc("/get_file_info", api.HandleGetFileInfo)
apiMux.HandleFunc("/get_file_stream_direct", api.HandleGetFileStreamDirect)
apiMux.HandleFunc("/prepare_file_stream_direct", api.HandlePrepareFileStreamDirect)
// below needs token
apiMux.HandleFunc("/walk", api.HandleWalk)
apiMux.HandleFunc("/reset", api.HandleReset)

62
internal/pkg/tmpfs/tmpfs.go Normal file
View File

@@ -0,0 +1,62 @@
package tmpfs
import (
"log"
"os"
"path/filepath"
"strconv"
"sync"
"time"
)
type Tmpfs struct {
record map[string]int64
Root string
FileLifeTime int64
CleanerInternal int64
wg sync.WaitGroup
}
func (tmpfs *Tmpfs) GetObjFilePath(id int64, configName string) (string) {
return filepath.Join(tmpfs.Root, strconv.FormatInt(id, 10) + "." + configName + ".ogg")
}
func NewTmpfs() *Tmpfs {
tmpfs := &Tmpfs{
record: make(map[string]int64),
FileLifeTime: 10*60, // ! important
CleanerInternal: 1,
Root: "/tmp/",
}
tmpfs.wg.Add(1)
go tmpfs.Cleaner()
return tmpfs
}
func (tmpfs *Tmpfs) Record(filename string) {
tmpfs.record[filename] = time.Now().Unix()
}
func (tmpfs *Tmpfs) Exits(filename string) (bool) {
_, ok := tmpfs.record[filename]
return ok
}
func (tmpfs *Tmpfs) Cleaner() {
var err error
for {
now := time.Now().Unix()
for key, value := range tmpfs.record {
if now - value > tmpfs.FileLifeTime {
err = os.Remove(key)
if err != nil {
log.Println("[tmpfs] Failed to remove file", err)
}
log.Println("[tmpfs] Deleted file", key)
delete(tmpfs.record, key)
}
}
time.Sleep(time.Second)
}
}

12
internal/pkg/tmpfs/tmpfs_test.go Normal file
View File

@@ -0,0 +1,12 @@
package tmpfs
import "testing"
func TestTmpfs(t *testing.T) {
t.Log("Starting ...")
tmpfs := NewTmpfs()
tmpfs.FileLifeTime = 1
tmpfs.Record("/tmp/testfile")
t.Log(tmpfs.record)
tmpfs.wg.Wait()
}