better log for upstream

This commit is contained in:
Boshi Lian 2018-12-29 03:48:56 -08:00
parent 50bf31f9e6
commit bc1afde4b8
3 changed files with 13 additions and 4 deletions

View file

@ -10,7 +10,8 @@ import (
func (p *plugin) findUpstream(conn ssh.ConnMetadata) (net.Conn, *ssh.AuthPipe, error) {
d, err := lookupDownstreamWithFallback(p.db, conn.User())
user := conn.User()
d, err := lookupDownstreamWithFallback(p.db, user)
if err != nil {
return nil, nil, err
@ -18,6 +19,8 @@ func (p *plugin) findUpstream(conn ssh.ConnMetadata) (net.Conn, *ssh.AuthPipe, e
addr := d.Upstream.Server.Address
logger.Printf("mapping user [%v] to [%v@%v]", user, d.Username, addr)
c, err := dial(addr)
if err != nil {

View file

@ -34,7 +34,9 @@ func (p *plugin) Init(glogger *log.Logger) error {
return err
}
db.AutoMigrate(
logger.Printf("upstream provider: Database driver [%v] initializing", db.Dialect().GetName())
err = db.AutoMigrate(
new(keydata),
new(privateKey),
new(hostKey),
@ -43,7 +45,11 @@ func (p *plugin) Init(glogger *log.Logger) error {
new(authorizedKey),
new(downstream),
new(config),
)
).Error
if err != nil {
logger.Printf("AutoMigrate error: %v", err)
}
p.db = db

View file

@ -27,7 +27,7 @@ func (p *plugin) Init(glogger *log.Logger) error {
logger = glogger
logger.Printf("upstream provider: workingdir %v init", config.WorkingDir)
logger.Printf("upstream provider: workingdir from path [%v] initializing", config.WorkingDir)
return nil
}