107 lines
2.5 KiB
Smarty
107 lines
2.5 KiB
Smarty
/*
|
||
Copyright 2021 Qiniu Cloud (qiniu.com)
|
||
Licensed under the Apache License, Version 2.0 (the "License");
|
||
you may not use this file except in compliance with the License.
|
||
You may obtain a copy of the License at
|
||
http://www.apache.org/licenses/LICENSE-2.0
|
||
Unless required by applicable law or agreed to in writing, software
|
||
distributed under the License is distributed on an "AS IS" BASIS,
|
||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
See the License for the specific language governing permissions and
|
||
limitations under the License.
|
||
*/
|
||
|
||
package cover
|
||
|
||
import (
|
||
"fmt"
|
||
"time"
|
||
"os"
|
||
"log"
|
||
"strconv"
|
||
"net/url"
|
||
|
||
"{{.GlobalCoverVarImportPath}}/websocket"
|
||
|
||
_cover "{{.GlobalCoverVarImportPath}}"
|
||
)
|
||
|
||
func init() {
|
||
// init host
|
||
host_env := os.Getenv("GOC_CUSTOM_HOST")
|
||
if host_env != "" {
|
||
host = host_env
|
||
}
|
||
|
||
var dialer = websocket.DefaultDialer
|
||
|
||
go func() {
|
||
for {
|
||
// 获取进程元信息用于注册
|
||
ps, err := getRegisterInfo()
|
||
if err != nil {
|
||
time.Sleep(waitDelay)
|
||
continue
|
||
}
|
||
|
||
// 注册，直接将元信息放在 ws 地址中
|
||
v := url.Values{}
|
||
v.Set("hostname", ps.hostname)
|
||
v.Set("pid", strconv.Itoa(ps.pid))
|
||
v.Set("cmdline", ps.cmdline)
|
||
v.Encode()
|
||
|
||
watchstreamUrl := fmt.Sprintf("ws://%v/v2/internal/ws/watchstream?%v", host, v.Encode())
|
||
ws, _, err := dialer.Dial(watchstreamUrl, nil)
|
||
if err != nil {
|
||
log.Printf("[goc][Error] watch fail to dial to goc server: %v", err)
|
||
time.Sleep(waitDelay)
|
||
continue
|
||
}
|
||
|
||
// 连接成功
|
||
_cover.WatchEnabled_{{.Random}} = true
|
||
log.Printf("[goc][Info] watch connected to goc server")
|
||
|
||
ticker := time.NewTicker(time.Second)
|
||
closeFlag := false
|
||
go func() {
|
||
for {
|
||
// 必须调用一下以触发 ping 的自动处理
|
||
_, _, err := ws.ReadMessage()
|
||
if err != nil {
|
||
break
|
||
}
|
||
}
|
||
closeFlag = true
|
||
}()
|
||
|
||
Loop:
|
||
for {
|
||
select {
|
||
case block := <-_cover.WatchChannel_{{.Random}}:
|
||
i := block.I
|
||
|
||
cov := fmt.Sprintf("%s:%d.%d,%d.%d %d %d", block.Name,
|
||
block.Pos[3*i+0], uint16(block.Pos[3*i+2]),
|
||
block.Pos[3*i+1], uint16(block.Pos[3*i+2] >> 16),
|
||
block.Stmts,
|
||
1)
|
||
|
||
err = ws.WriteMessage(websocket.TextMessage, []byte(cov))
|
||
if err != nil {
|
||
_cover.WatchEnabled_{{.Random}} = false
|
||
log.Println("[goc][Error] push coverage failed: %v", err)
|
||
time.Sleep(waitDelay)
|
||
break Loop
|
||
}
|
||
case <-ticker.C:
|
||
if closeFlag == true {
|
||
break Loop
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}()
|
||
}
|