github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-06-24 15:24:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added kubernetes provider singleton (#599)

Browse Source
This commit is contained in:
RoyUP9 2022-01-09 10:50:58 +02:00 committed by GitHub
parent e3e9681110
commit c49c344c2a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 29 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
agent/pkg/controllers/config_controller.go
View File

@ -45,7 +45,7 @@ func PostTapConfig(c *gin.Context) {
podRegex, _ := regexp.Compile(".*") podRegex, _ := regexp.Compile(".*")
kubernetesProvider, err := kubernetes.NewProviderInCluster() kubernetesProvider, err := providers.GetKubernetesProvider()
if err != nil { if err != nil {
c.JSON(http.StatusInternalServerError, err) c.JSON(http.StatusInternalServerError, err)
return return
@ -66,7 +66,7 @@ func PostTapConfig(c *gin.Context) {
} }
func GetTapConfig(c *gin.Context) { func GetTapConfig(c *gin.Context) {
kubernetesProvider, err := kubernetes.NewProviderInCluster() kubernetesProvider, err := providers.GetKubernetesProvider()
if err != nil { if err != nil {
c.JSON(http.StatusInternalServerError, err) c.JSON(http.StatusInternalServerError, err)
return return

27
agent/pkg/providers/kubernetes_provider.go Normal file
View File

@ -0,0 +1,27 @@
package providers
import (
"github.com/up9inc/mizu/shared/kubernetes"
"sync"
)
var lock = &sync.Mutex{}
var kubernetesProvider *kubernetes.Provider
func GetKubernetesProvider() (*kubernetes.Provider, error) {
if kubernetesProvider == nil {
lock.Lock()
defer lock.Unlock()
if kubernetesProvider == nil {
var err error
kubernetesProvider, err = kubernetes.NewProviderInCluster()
if err != nil {
return nil, err
}
}
}
return kubernetesProvider, nil
}