mirror of https://github.com/q191201771/lal.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
154 lines
3.5 KiB
Go
154 lines
3.5 KiB
Go
// Copyright 2019, Chef. All rights reserved.
|
|
// https://github.com/q191201771/lal
|
|
//
|
|
// Use of this source code is governed by a MIT-style license
|
|
// that can be found in the License file.
|
|
//
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
package logic
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
|
|
"github.com/q191201771/lal/pkg/httpflv"
|
|
|
|
"github.com/q191201771/lal/pkg/hls"
|
|
"github.com/q191201771/naza/pkg/nazajson"
|
|
"github.com/q191201771/naza/pkg/nazalog"
|
|
)
|
|
|
|
//var ErrMissKeyField = errors.New("missing key field in config file")
|
|
|
|
type Config struct {
|
|
RTMPConfig RTMPConfig `json:"rtmp"`
|
|
HTTPFLVConfig HTTPFLVConfig `json:"httpflv"`
|
|
HLSConfig HLSConfig `json:"hls"`
|
|
HTTPTSConfig HTTPTSConfig `json:"httpts"`
|
|
RTSPConfig RTSPConfig `json:"rtsp"`
|
|
RelayPushConfig RelayPushConfig `json:"relay_push"`
|
|
RelayPullConfig RelayPullConfig `json:"relay_pull"`
|
|
|
|
HTTPAPIConfig HTTPAPIConfig `json:"http_api"`
|
|
ServerID string `json:"server_id"`
|
|
HTTPNotifyConfig HTTPNotifyConfig `json:"http_notify"`
|
|
PProfConfig PProfConfig `json:"pprof"`
|
|
LogConfig nazalog.Option `json:"log"`
|
|
}
|
|
|
|
type RTMPConfig struct {
|
|
Enable bool `json:"enable"`
|
|
Addr string `json:"addr"`
|
|
GOPNum int `json:"gop_num"`
|
|
}
|
|
|
|
type HTTPFLVConfig struct {
|
|
httpflv.ServerConfig
|
|
GOPNum int `json:"gop_num"`
|
|
}
|
|
|
|
type HTTPTSConfig struct {
|
|
Enable bool `json:"enable"`
|
|
SubListenAddr string `json:"sub_listen_addr"`
|
|
}
|
|
|
|
type HLSConfig struct {
|
|
SubListenAddr string `json:"sub_listen_addr"`
|
|
hls.MuxerConfig
|
|
}
|
|
|
|
type RTSPConfig struct {
|
|
Enable bool `json:"enable"`
|
|
Addr string `json:"addr"`
|
|
}
|
|
|
|
type RelayPushConfig struct {
|
|
Enable bool `json:"enable"`
|
|
AddrList []string `json:"addr_list"`
|
|
}
|
|
|
|
type RelayPullConfig struct {
|
|
Enable bool `json:"enable"`
|
|
Addr string `json:"addr"`
|
|
}
|
|
|
|
type HTTPAPIConfig struct {
|
|
Enable bool `json:"enable"`
|
|
Addr string `json:"addr"`
|
|
}
|
|
|
|
type HTTPNotifyConfig struct {
|
|
Enable bool `json:"enable"`
|
|
UpdateIntervalSec int `json:"update_interval_sec"`
|
|
OnUpdate string `json:"on_update"`
|
|
OnPubStart string `json:"on_pub_start"`
|
|
OnPubStop string `json:"on_pub_stop"`
|
|
OnSubStart string `json:"on_sub_start"`
|
|
OnSubStop string `json:"on_sub_stop"`
|
|
OnRTMPConnect string `json:"on_rtmp_connect"`
|
|
}
|
|
|
|
type PProfConfig struct {
|
|
Enable bool `json:"enable"`
|
|
Addr string `json:"addr"`
|
|
}
|
|
|
|
func LoadConf(confFile string) (*Config, error) {
|
|
var config Config
|
|
rawContent, err := ioutil.ReadFile(confFile)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err = json.Unmarshal(rawContent, &config); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
j, err := nazajson.New(rawContent)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 检查一级配置项
|
|
keyFieldList := []string{
|
|
"rtmp",
|
|
"httpflv",
|
|
"hls",
|
|
"httpts",
|
|
"rtsp",
|
|
"relay_push",
|
|
"relay_pull",
|
|
"http_api",
|
|
"http_notify",
|
|
"pprof",
|
|
"log",
|
|
}
|
|
for _, kf := range keyFieldList {
|
|
if !j.Exist(kf) {
|
|
nazalog.Warnf("missing config item %s", kf)
|
|
}
|
|
}
|
|
|
|
// 配置不存在时，设置默认值
|
|
if !j.Exist("log.level") {
|
|
config.LogConfig.Level = nazalog.LevelDebug
|
|
}
|
|
if !j.Exist("log.filename") {
|
|
config.LogConfig.Filename = "./logs/lalserver.log"
|
|
}
|
|
if !j.Exist("log.is_to_stdout") {
|
|
config.LogConfig.IsToStdout = true
|
|
}
|
|
if !j.Exist("log.is_rotate_daily") {
|
|
config.LogConfig.IsRotateDaily = true
|
|
}
|
|
if !j.Exist("log.short_file_flag") {
|
|
config.LogConfig.ShortFileFlag = true
|
|
}
|
|
if !j.Exist("log.assert_behavior") {
|
|
config.LogConfig.AssertBehavior = nazalog.AssertError
|
|
}
|
|
|
|
return &config, nil
|
|
}
|