加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
main.go 1.69 KB
一键复制 编辑 原始数据 按行查看 历史
langhuihui 提交于 2020-04-11 20:19 . 增加彩色日志输出功能
package engine
import (
"encoding/json"
"fmt"
"io/ioutil"
"log"
"os"
"path/filepath"
"runtime"
"strings"
"time" // colorable
"github.com/BurntSushi/toml"
. "github.com/logrusorgru/aurora"
)
var (
// ConfigRaw 配置信息的原始数据
ConfigRaw []byte
// Version 引擎版本号
Version string
// EngineInfo 引擎信息
EngineInfo = &struct {
Version *string
StartTime time.Time //启动时间
}{&Version, time.Now()}
)
// Run 启动Monibuca引擎
func Run(configFile string) (err error) {
if runtime.GOOS == "windows" {
ioutil.WriteFile("shutdown.bat", []byte(fmt.Sprintf("taskkill /pid %d -t -f", os.Getpid())), 0777)
} else {
ioutil.WriteFile("shutdown.sh", []byte(fmt.Sprintf("kill -9 %d", os.Getpid())), 0777)
}
_, enginePath, _, _ := runtime.Caller(0)
if parts := strings.Split(filepath.Dir(enginePath), "@"); len(parts) > 1 {
Version = parts[len(parts)-1]
}
if ConfigRaw, err = ioutil.ReadFile(configFile); err != nil {
Print(Red("read config file error:"), err)
return
}
Print(Green("start monibuca"), BrightBlue(Version))
go Summary.StartSummary()
var cg map[string]interface{}
if _, err = toml.Decode(string(ConfigRaw), &cg); err == nil {
if cfg, ok := cg["Monibuca"]; ok {
b, _ := json.Marshal(cfg)
if err = json.Unmarshal(b, config); err != nil {
log.Println(err)
}
}
for name, config := range Plugins {
if cfg, ok := cg[name]; ok {
b, _ := json.Marshal(cfg)
if err = json.Unmarshal(b, config.Config); err != nil {
log.Println(err)
continue
}
} else if config.Config != nil {
continue
}
if config.Run != nil {
go config.Run()
}
}
} else {
Print(Red("decode config file error:"), err)
}
return
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化