mirror of
https://github.com/tavenli/port-forward.git
synced 2024-04-21 12:42:13 +00:00
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package Service
|
|
|
|
import (
|
|
"fmt"
|
|
"forward-core/Models"
|
|
|
|
"github.com/astaxie/beego"
|
|
"github.com/astaxie/beego/orm"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var (
|
|
OrmerS orm.Ormer
|
|
//ForWardServ = new(ForWardServer)
|
|
ForWardServ = NewForWardServer()
|
|
MagicServ = NewMagicServiceV1()
|
|
//MagicServ = new(MagicServer)
|
|
ConsoleServ = new(ConsoleServer)
|
|
SysDataS = new(SysDataService)
|
|
|
|
ForWardDebug = true
|
|
)
|
|
|
|
func init() {
|
|
//开启DEBUG模式，输出SQL信息
|
|
orm.Debug = true
|
|
|
|
//_ "github.com/mattn/go-sqlite3"
|
|
orm.RegisterDriver("sqlite3", orm.DRSqlite)
|
|
orm.RegisterDataBase("default", "sqlite3", "file:data/data.db?cache=shared&loc=auto")
|
|
|
|
OrmerS = orm.NewOrm()
|
|
OrmerS.Using("default")
|
|
|
|
onstartForward := beego.AppConfig.String("onstart.forward")
|
|
if onstartForward == "true" {
|
|
onStartForward()
|
|
}
|
|
}
|
|
|
|
func onStartForward() {
|
|
forwards := SysDataS.GetAllPortForwardList(1)
|
|
for _, entity := range forwards {
|
|
resultChan := make(chan Models.FuncResult)
|
|
config := SysDataS.ToForwardConfig(entity)
|
|
go ForWardServ.OpenForward(config, resultChan)
|
|
|
|
fmt.Println(<-resultChan)
|
|
}
|
|
}
|