fix cache set

This commit is contained in:
Boshi Lian 2022-10-23 08:07:30 +00:00
parent cae06825ce
commit 4c8baddefc
2 changed files with 10 additions and 4 deletions

View file

@ -211,7 +211,8 @@ func (p *plugin) createUpstream(conn libplugin.ConnMetadata, to pipeConfigTo, or
// password found
if pass != "" {
u.Auth = libplugin.CreatePasswordAuth([]byte(pass))
return u, p.cache.Add(conn.UniqueID(), &to, cache.DefaultExpiration)
p.cache.Set(conn.UniqueID(), &to, cache.DefaultExpiration)
return u, nil
}
// try private key
@ -225,7 +226,8 @@ func (p *plugin) createUpstream(conn libplugin.ConnMetadata, to pipeConfigTo, or
if data != nil {
u.Auth = libplugin.CreatePrivateKeyAuth(data)
return u, p.cache.Add(conn.UniqueID(), &to, cache.DefaultExpiration)
p.cache.Set(conn.UniqueID(), &to, cache.DefaultExpiration)
return u, nil
}
return nil, fmt.Errorf("no password or private key found")