mirror of https://github.com/ehang-io/nps
62 lines
1.1 KiB
Go
62 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"C"
|
|
"github.com/astaxie/beego/logs"
|
|
"github.com/cnlh/nps/client"
|
|
"github.com/cnlh/nps/lib/common"
|
|
"github.com/cnlh/nps/lib/version"
|
|
"time"
|
|
)
|
|
|
|
var status int
|
|
var closeBefore int
|
|
var cl *client.TRPClient
|
|
|
|
//export StartClientByVerifyKey
|
|
func StartClientByVerifyKey(serverAddr, verifyKey, connType, proxyUrl *C.char) int {
|
|
if cl != nil {
|
|
closeBefore = 1
|
|
cl.Close()
|
|
}
|
|
cl = client.NewRPClient(C.GoString(serverAddr), C.GoString(verifyKey), C.GoString(connType), C.GoString(proxyUrl), nil)
|
|
closeBefore = 0
|
|
go func() {
|
|
for {
|
|
status = 1
|
|
cl.Start()
|
|
status = 0
|
|
if closeBefore == 1 {
|
|
return
|
|
}
|
|
time.Sleep(time.Second * 5)
|
|
}
|
|
}()
|
|
return 1
|
|
}
|
|
|
|
//export GetClientStatus
|
|
func GetClientStatus() int {
|
|
return status
|
|
}
|
|
|
|
//export CloseClient
|
|
func CloseClient() {
|
|
closeBefore = 1
|
|
cl.Close()
|
|
}
|
|
|
|
//export Version
|
|
func Version() *C.char {
|
|
return C.CString(version.VERSION)
|
|
}
|
|
|
|
func Logs() *C.char {
|
|
return C.CString(common.GetLogMsg())
|
|
}
|
|
|
|
func main() {
|
|
// Need a main function to make CGO compile package as C shared library
|
|
logs.SetLogger("store")
|
|
}
|