func UpdateNetworkById(ctx context.Context, w rest.ResponseWriter, r *rest.Request) {
repo, ok := middleware.RepoFromContext(ctx)
if !ok {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "内部服务器错误"})
return
}
logger, ok := middleware.LoggerFromContext(ctx)
if !ok {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "内部服务器错误"})
return
}
var info struct {
ID uint
Network string
Netmask string
Gateway string
Vlan string
Trunk string
Bonding string
AccessToken string
}
if err := r.DecodeJSONPayload(&info); err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "参数错误" + err.Error()})
return
}
info.Network = strings.TrimSpace(info.Network)
info.Netmask = strings.TrimSpace(info.Netmask)
info.Gateway = strings.TrimSpace(info.Gateway)
info.Vlan = strings.TrimSpace(info.Vlan)
info.Trunk = strings.TrimSpace(info.Trunk)
info.Bonding = strings.TrimSpace(info.Bonding)
info.AccessToken = strings.TrimSpace(info.AccessToken)
_, errVerify := VerifyAccessPurview(info.AccessToken, ctx, true, w, r)
if errVerify != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": errVerify.Error()})
return
}
if info.Network == "" || info.Netmask == "" || info.Gateway == "" {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "请将信息填写完整!"})
return
}
isValidate, err := regexp.MatchString("^((2[0-4]\\d|25[0-5]|[01]?\\d\\d?)\\.){3}(2[0-4]\\d|25[0-5]|[01]?\\d\\d?)$", info.Netmask)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "参数错误" + err.Error()})
return
}
if !isValidate {
w.WriteJSON(map[string]interface{}{"Status": "failure", "Message": "掩码格式不正确!", "Content": ""})
return
}
isValidateGageway, err := regexp.MatchString("^((2[0-4]\\d|25[0-5]|[01]?\\d\\d?)\\.){3}(2[0-4]\\d|25[0-5]|[01]?\\d\\d?)$", info.Gateway)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "参数错误" + err.Error()})
return
}
if !isValidateGageway {
w.WriteJSON(map[string]interface{}{"Status": "failure", "Message": "网关格式不正确!", "Content": ""})
return
}
count, err := repo.CountNetworkByNetworkAndId(info.Network, info.ID)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
if count > 0 {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "该网段已存在!"})
return
}
oldNetwork, err := repo.GetNetworkById(info.ID)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
mod, err := repo.UpdateNetworkById(info.ID, info.Network, info.Netmask, info.Gateway, info.Vlan, info.Trunk, info.Bonding)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
//网段发生更改的情况下，重新分配IP
if oldNetwork.Network != info.Network {
//处理网段
network, err := util.GetCidrInfo(info.Network, logger)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
ipList, err := util.GetIPListByMinAndMaxIP(network["MinIP"], network["MaxIP"])
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
_, errDelete := repo.DeleteIpByNetworkId(info.ID)
if errDelete != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": errDelete.Error()})
return
}
for _, ip := range ipList {
_, err := repo.AddIp(info.ID, ip)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
}
}
w.WriteJSON(map[string]interface{}{"Status": "success", "Message": "操作成功", "Content": mod})
}
//添加
func AddNetwork(ctx context.Context, w rest.ResponseWriter, r *rest.Request) {
repo, ok := middleware.RepoFromContext(ctx)
if !ok {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "内部服务器错误"})
return
}
var info struct {
Network string
Netmask string
Gateway string
Vlan string
Trunk string
Bonding string
}
if err := r.DecodeJSONPayload(&info); err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "参数错误"})
return
}
info.Network = strings.TrimSpace(info.Network)
info.Netmask = strings.TrimSpace(info.Netmask)
info.Gateway = strings.TrimSpace(info.Gateway)
info.Vlan = strings.TrimSpace(info.Vlan)
info.Trunk = strings.TrimSpace(info.Trunk)
info.Bonding = strings.TrimSpace(info.Bonding)
if info.Network == "" || info.Netmask == "" || info.Gateway == "" {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "请将信息填写完整!"})
return
}
isValidate, err := regexp.MatchString("^((2[0-4]\\d|25[0-5]|[01]?\\d\\d?)\\.){3}(2[0-4]\\d|25[0-5]|[01]?\\d\\d?)$", info.Netmask)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "参数错误" + err.Error()})
return
}
if !isValidate {
w.WriteJSON(map[string]interface{}{"Status": "failure", "Message": "掩码格式不正确!", "Content": ""})
return
}
isValidateGageway, err := regexp.MatchString("^((2[0-4]\\d|25[0-5]|[01]?\\d\\d?)\\.){3}(2[0-4]\\d|25[0-5]|[01]?\\d\\d?)$", info.Gateway)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "参数错误" + err.Error()})
return
}
if !isValidateGageway {
w.WriteJSON(map[string]interface{}{"Status": "failure", "Message": "网关格式不正确!", "Content": ""})
return
}
//fmt.Println("test")
//处理网段
network, err := util.GetCidrInfo(info.Network)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
ipList, err := util.GetIPListByMinAndMaxIP(network["MinIP"], network["MaxIP"])
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
count, err := repo.CountNetworkByNetwork(info.Network)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
if count > 0 {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": "该网段已存在!"})
return
}
newNetwork, errAdd := repo.AddNetwork(info.Network, info.Netmask, info.Gateway, info.Vlan, info.Trunk, info.Bonding)
_, errDelete := repo.DeleteIpByNetworkId(newNetwork.ID)
if errDelete != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": errDelete.Error()})
return
}
for _, ip := range ipList {
_, err := repo.AddIp(newNetwork.ID, ip)
if err != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": err.Error()})
return
}
}
if errAdd != nil {
w.WriteJSON(map[string]interface{}{"Status": "error", "Message": errAdd.Error()})
return
}
w.WriteJSON(map[string]interface{}{"Status": "success", "Message": "操作成功"})
}