76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
package service
|
|
|
|
import (
|
|
"next-terminal/server/global/gateway"
|
|
"next-terminal/server/log"
|
|
"next-terminal/server/model"
|
|
"next-terminal/server/repository"
|
|
"next-terminal/server/term"
|
|
)
|
|
|
|
type AccessGatewayService struct {
|
|
accessGatewayRepository *repository.AccessGatewayRepository
|
|
}
|
|
|
|
func NewAccessGatewayService(accessGatewayRepository *repository.AccessGatewayRepository) *AccessGatewayService {
|
|
accessGatewayService = &AccessGatewayService{accessGatewayRepository: accessGatewayRepository}
|
|
return accessGatewayService
|
|
}
|
|
|
|
func (r AccessGatewayService) GetGatewayAndReconnectById(accessGatewayId string) (g *gateway.Gateway, err error) {
|
|
g = gateway.GlobalGatewayManager.GetById(accessGatewayId)
|
|
if g == nil || !g.Connected {
|
|
accessGateway, err := r.accessGatewayRepository.FindById(accessGatewayId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
g = r.ReConnect(&accessGateway)
|
|
}
|
|
return g, nil
|
|
}
|
|
|
|
func (r AccessGatewayService) GetGatewayById(accessGatewayId string) (g *gateway.Gateway, err error) {
|
|
g = gateway.GlobalGatewayManager.GetById(accessGatewayId)
|
|
if g == nil {
|
|
accessGateway, err := r.accessGatewayRepository.FindById(accessGatewayId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
g = r.ReConnect(&accessGateway)
|
|
}
|
|
return g, nil
|
|
}
|
|
|
|
func (r AccessGatewayService) ReConnectAll() error {
|
|
gateways, err := r.accessGatewayRepository.FindAll()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(gateways) > 0 {
|
|
for i := range gateways {
|
|
r.ReConnect(&gateways[i])
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r AccessGatewayService) ReConnect(m *model.AccessGateway) *gateway.Gateway {
|
|
log.Debugf("重建接入网关「%v」中...", m.Name)
|
|
r.DisconnectById(m.ID)
|
|
sshClient, err := term.NewSshClient(m.IP, m.Port, m.Username, m.Password, m.PrivateKey, m.Passphrase)
|
|
var g *gateway.Gateway
|
|
if err != nil {
|
|
g = gateway.NewGateway(m.ID, m.Localhost, false, err.Error(), nil)
|
|
} else {
|
|
g = gateway.NewGateway(m.ID, m.Localhost, true, "", sshClient)
|
|
}
|
|
gateway.GlobalGatewayManager.Add <- g
|
|
log.Debugf("重建接入网关「%v」完成", m.Name)
|
|
return g
|
|
}
|
|
|
|
func (r AccessGatewayService) DisconnectById(accessGatewayId string) {
|
|
gateway.GlobalGatewayManager.Del <- accessGatewayId
|
|
}
|