mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-23 01:40:29 +08:00
阶段性提交
This commit is contained in:
GoEdgeLab
@@ -2,7 +2,6 @@ package tcp
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
||||
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/serverutils"
|
||||
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||||
@@ -23,21 +22,22 @@ func (this *IndexAction) Init() {
|
||||
func (this *IndexAction) RunGet(params struct {
|
||||
ServerId int64
|
||||
}) {
|
||||
server, config, isOk := serverutils.FindServer(&this.ParentAction, params.ServerId)
|
||||
server, _, isOk := serverutils.FindServer(&this.ParentAction, params.ServerId)
|
||||
if !isOk {
|
||||
return
|
||||
}
|
||||
if config.TCP == nil {
|
||||
this.ErrorPage(errors.New("there is no tcp setting"))
|
||||
return
|
||||
}
|
||||
|
||||
if config.TCP.Listen == nil {
|
||||
config.TCP.Listen = []*serverconfigs.NetworkAddressConfig{}
|
||||
tcpConfig := &serverconfigs.TCPProtocolConfig{}
|
||||
if len(server.TcpJSON) > 0 {
|
||||
err := json.Unmarshal(server.TcpJSON, tcpConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
}
|
||||
} else {
|
||||
tcpConfig.IsOn = true
|
||||
}
|
||||
|
||||
this.Data["serverType"] = server.Type
|
||||
this.Data["addresses"] = config.TCP.Listen
|
||||
this.Data["tcpConfig"] = tcpConfig
|
||||
|
||||
this.Show()
|
||||
}
|
||||
@@ -49,8 +49,6 @@ func (this *IndexAction) RunPost(params struct {
|
||||
|
||||
Must *actions.Must
|
||||
}) {
|
||||
serverId := params.ServerId
|
||||
|
||||
server, _, isOk := serverutils.FindServer(&this.ParentAction, params.ServerId)
|
||||
if !isOk {
|
||||
return
|
||||
@@ -62,188 +60,30 @@ func (this *IndexAction) RunPost(params struct {
|
||||
this.Fail("端口地址解析失败：" + err.Error())
|
||||
}
|
||||
|
||||
switch server.Type {
|
||||
case serverconfigs.ServerTypeHTTPProxy, serverconfigs.ServerTypeHTTPWeb:
|
||||
var httpConfig = &serverconfigs.HTTPProtocolConfig{}
|
||||
if len(server.HttpJSON) > 0 {
|
||||
err = json.Unmarshal(server.HttpJSON, httpConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
httpConfig.Listen = []*serverconfigs.NetworkAddressConfig{}
|
||||
} else {
|
||||
httpConfig.IsOn = true
|
||||
tcpConfig := &serverconfigs.TCPProtocolConfig{}
|
||||
if len(server.TcpJSON) > 0 {
|
||||
err := json.Unmarshal(server.TcpJSON, tcpConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
}
|
||||
} else {
|
||||
tcpConfig.IsOn = true
|
||||
}
|
||||
tcpConfig.Listen = addresses
|
||||
|
||||
var httpsConfig = &serverconfigs.HTTPSProtocolConfig{}
|
||||
if len(server.HttpsJSON) > 0 {
|
||||
err = json.Unmarshal(server.HttpsJSON, httpsConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
httpsConfig.Listen = []*serverconfigs.NetworkAddressConfig{}
|
||||
} else {
|
||||
httpsConfig.IsOn = true
|
||||
}
|
||||
configData, err := json.Marshal(tcpConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
|
||||
for _, addr := range addresses {
|
||||
switch addr.Protocol.Primary() {
|
||||
case serverconfigs.ProtocolHTTP:
|
||||
httpConfig.AddListen(addr)
|
||||
case serverconfigs.ProtocolHTTPS:
|
||||
httpsConfig.AddListen(addr)
|
||||
}
|
||||
}
|
||||
|
||||
httpData, err := json.Marshal(httpConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
httpsData, err := json.Marshal(httpsConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
_, err = this.RPC().ServerRPC().UpdateServerHTTP(this.AdminContext(), &pb.UpdateServerHTTPRequest{
|
||||
ServerId: serverId,
|
||||
Config: httpData,
|
||||
})
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
_, err = this.RPC().ServerRPC().UpdateServerHTTPS(this.AdminContext(), &pb.UpdateServerHTTPSRequest{
|
||||
ServerId: serverId,
|
||||
Config: httpsData,
|
||||
})
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
case serverconfigs.ServerTypeTCPProxy:
|
||||
tcpProxy := &serverconfigs.TCPProtocolConfig{}
|
||||
if len(server.TcpJSON) > 0 {
|
||||
err = json.Unmarshal(server.TcpJSON, tcpProxy)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
tcpProxy.Listen = []*serverconfigs.NetworkAddressConfig{}
|
||||
} else {
|
||||
tcpProxy.IsOn = true
|
||||
}
|
||||
|
||||
tlsProxy := &serverconfigs.TLSProtocolConfig{}
|
||||
if len(server.TlsJSON) > 0 {
|
||||
err = json.Unmarshal(server.TlsJSON, tlsProxy)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
tlsProxy.Listen = []*serverconfigs.NetworkAddressConfig{}
|
||||
} else {
|
||||
tlsProxy.IsOn = true
|
||||
}
|
||||
|
||||
for _, addr := range addresses {
|
||||
switch addr.Protocol.Primary() {
|
||||
case serverconfigs.ProtocolTCP:
|
||||
tcpProxy.AddListen(addr)
|
||||
case serverconfigs.ProtocolTLS:
|
||||
tlsProxy.AddListen(addr)
|
||||
}
|
||||
}
|
||||
|
||||
tcpData, err := json.Marshal(tcpProxy)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
|
||||
tlsData, err := json.Marshal(tlsProxy)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
|
||||
_, err = this.RPC().ServerRPC().UpdateServerTCP(this.AdminContext(), &pb.UpdateServerTCPRequest{
|
||||
ServerId: serverId,
|
||||
Config: tcpData,
|
||||
})
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
|
||||
_, err = this.RPC().ServerRPC().UpdateServerTLS(this.AdminContext(), &pb.UpdateServerTLSRequest{
|
||||
ServerId: serverId,
|
||||
Config: tlsData,
|
||||
})
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
case serverconfigs.ServerTypeUnixProxy:
|
||||
unixConfig := &serverconfigs.UnixProtocolConfig{}
|
||||
if len(server.UnixJSON) > 0 {
|
||||
err = json.Unmarshal(server.UnixJSON, unixConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
unixConfig.Listen = []*serverconfigs.NetworkAddressConfig{}
|
||||
}
|
||||
for _, addr := range addresses {
|
||||
switch addr.Protocol.Primary() {
|
||||
case serverconfigs.ProtocolUnix:
|
||||
unixConfig.AddListen(addr)
|
||||
}
|
||||
}
|
||||
unixData, err := json.Marshal(unixConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
_, err = this.RPC().ServerRPC().UpdateServerUnix(this.AdminContext(), &pb.UpdateServerUnixRequest{
|
||||
ServerId: serverId,
|
||||
Config: unixData,
|
||||
})
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
case serverconfigs.ServerTypeUDPProxy:
|
||||
udpConfig := &serverconfigs.UDPProtocolConfig{}
|
||||
if len(server.UdpJSON) > 0 {
|
||||
err = json.Unmarshal(server.UdpJSON, udpConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
udpConfig.Listen = []*serverconfigs.NetworkAddressConfig{}
|
||||
}
|
||||
for _, addr := range addresses {
|
||||
switch addr.Protocol.Primary() {
|
||||
case serverconfigs.ProtocolUDP:
|
||||
udpConfig.AddListen(addr)
|
||||
}
|
||||
}
|
||||
udpData, err := json.Marshal(udpConfig)
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
_, err = this.RPC().ServerRPC().UpdateServerUDP(this.AdminContext(), &pb.UpdateServerUDPRequest{
|
||||
ServerId: serverId,
|
||||
Config: udpData,
|
||||
})
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
_, err = this.RPC().ServerRPC().UpdateServerTCP(this.AdminContext(), &pb.UpdateServerTCPRequest{
|
||||
ServerId: params.ServerId,
|
||||
Config: configData,
|
||||
})
|
||||
if err != nil {
|
||||
this.ErrorPage(err)
|
||||
return
|
||||
}
|
||||
|
||||
this.Success()
|
||||
|
||||
Reference in New Issue
Block a user