engineering/sshpiper
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix cache set

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

8
plugin/kubernetes/kubernetes.go
View file

@ -7,6 +7,7 @@ import (
"fmt"
"net"
"regexp"
"time"
gocache "github.com/patrickmn/go-cache"
"github.com/tg123/sshpiper/libplugin"
@ -69,6 +70,7 @@ func newKubernetesPlugin() (*plugin, error) {
k8sclient: k8sclient.CoreV1(),
lister: lister,
stop: stop,
cache: gocache.New(1*time.Minute, 10*time.Minute),
}, nil
}
@ -154,7 +156,8 @@ func (p *plugin) createUpstream(conn libplugin.ConnMetadata, pipe *piperv1beta1.
if originPassword != "" {
u.Auth = libplugin.CreatePasswordAuth([]byte(originPassword))
return u, p.cache.Add(conn.UniqueID(), pipe, gocache.DefaultExpiration)
p.cache.Set(conn.UniqueID(), pipe, gocache.DefaultExpiration)
return u, nil
}
secret, err := p.k8sclient.Secrets(pipe.Namespace).Get(context.Background(), to.PrivateKeySecret.Name, metav1.GetOptions{})
@ -164,7 +167,8 @@ func (p *plugin) createUpstream(conn libplugin.ConnMetadata, pipe *piperv1beta1.
data := secret.Data["privatekey"]
if data != nil {
return u, p.cache.Add(conn.UniqueID(), pipe, gocache.DefaultExpiration)
p.cache.Set(conn.UniqueID(), pipe, gocache.DefaultExpiration)
return u, nil
}
return nil, fmt.Errorf("no password or private key found")