104 lines
2.7 KiB
Go
104 lines
2.7 KiB
Go
|
package verifyMgr
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"sync"
|
|||
|
|
|
|||
|
|
"Framework/managecenterMgr"
|
|||
|
|
"goutil/webUtil"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
verifyUrlMap = make(map[string]VerifyType)
|
|||
|
|
mutex sync.RWMutex
|
|||
|
|
con_SuccessStatusCode = 200
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 初始化需要验证的Url及对应的验证方式
|
|||
|
|
func Init(url string, _type VerifyType) {
|
|||
|
|
mutex.Lock()
|
|||
|
|
defer mutex.Unlock()
|
|||
|
|
|
|||
|
|
if _, exists := verifyUrlMap[url]; exists {
|
|||
|
|
panic(fmt.Errorf("%s已经存在，请检查", url))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
verifyUrlMap[url] = _type
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证
|
|||
|
|
func Verify() (errList []error) {
|
|||
|
|
errList1 := verifyInternal()
|
|||
|
|
errList2 := verifyServerGroup()
|
|||
|
|
|
|||
|
|
if len(errList1) > 0 {
|
|||
|
|
errList = append(errList, errList1...)
|
|||
|
|
}
|
|||
|
|
if len(errList2) > 0 {
|
|||
|
|
errList = append(errList, errList2...)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证内部的url
|
|||
|
|
func verifyInternal() (errList []error) {
|
|||
|
|
// 验证单独指定的url
|
|||
|
|
mutex.RLock()
|
|||
|
|
defer mutex.RUnlock()
|
|||
|
|
|
|||
|
|
header := webUtil.GetFormHeader()
|
|||
|
|
transport := webUtil.NewTransport()
|
|||
|
|
transport.DisableKeepAlives = true
|
|||
|
|
transport = webUtil.GetTimeoutTransport(transport, 30)
|
|||
|
|
|
|||
|
|
for url, _type := range verifyUrlMap {
|
|||
|
|
switch _type {
|
|||
|
|
case Con_Get:
|
|||
|
|
if statusCode, _, err := webUtil.GetWebData3(url, "", header, transport); statusCode != con_SuccessStatusCode || err != nil {
|
|||
|
|
errList = append(errList, fmt.Errorf("access %s failed. StatusCode:%d, err:%s", url, statusCode, err))
|
|||
|
|
}
|
|||
|
|
case Con_Post:
|
|||
|
|
if statusCode, _, err := webUtil.PostMapData(url, nil, header, transport); statusCode != con_SuccessStatusCode || err != nil {
|
|||
|
|
errList = append(errList, fmt.Errorf("access %s failed. StatusCode:%d, err:%s", url, statusCode, err))
|
|||
|
|
}
|
|||
|
|
default:
|
|||
|
|
errList = append(errList, fmt.Errorf("the type of %s is wrong, not it's %d", url, _type))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证服务器组的url
|
|||
|
|
func verifyServerGroup() (errList []error) {
|
|||
|
|
// 验证配置在ManageCenter中的所有ServerGroup
|
|||
|
|
serverGroupList := managecenterMgr.GetServerGroupList()
|
|||
|
|
chList := make([]chan bool, 0, len(serverGroupList))
|
|||
|
|
|
|||
|
|
header := webUtil.GetFormHeader()
|
|||
|
|
transport := webUtil.NewTransport()
|
|||
|
|
transport.DisableKeepAlives = true
|
|||
|
|
transport = webUtil.GetTimeoutTransport(transport, 30)
|
|||
|
|
|
|||
|
|
for _, item := range serverGroupList {
|
|||
|
|
url := item.GetGSCallbackUrl("Verify.ashx")
|
|||
|
|
ch := make(chan bool)
|
|||
|
|
chList = append(chList, ch)
|
|||
|
|
go func(_url string, _ch chan bool) {
|
|||
|
|
fmt.Printf("VerifyUrl:%s\n", _url)
|
|||
|
|
if statusCode, _, err := webUtil.PostMapData(_url, nil, header, transport); statusCode != con_SuccessStatusCode || err != nil {
|
|||
|
|
errList = append(errList, fmt.Errorf("access %s failed. StatusCode:%d, err:%s", _url, statusCode, err))
|
|||
|
|
}
|
|||
|
|
_ch <- true
|
|||
|
|
}(url, ch)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 等待所有的ch返回
|
|||
|
|
for _, ch := range chList {
|
|||
|
|
<-ch
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|