add sync-once func for authz when restart

Signed-off-by: superCharge-xsy <[email protected]>
kmesh-net · Sep 6, 2024 · 7eb35a1 · 7eb35a1
1 parent f83283c
commit 7eb35a1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/pkg/bpf/bpf.go b/pkg/bpf/bpf.go
@@ -98,7 +98,6 @@ func (l *BpfLoader) StartAdsMode() (err error) {
 	l.bpfLogLevel = l.obj.SockConn.BpfLogLevel
 	ret := C.deserial_init()
 	if ret != 0 {
-		l.Stop()
 		return fmt.Errorf("deserial_init failed:%v", ret)
 	}
 	return nil

diff --git a/pkg/bpf/bpf_kmesh_l4_workload.go b/pkg/bpf/bpf_kmesh_l4_workload.go
@@ -91,7 +91,6 @@ func (l *BpfLoader) StartWorkloadMode() error {
 
 	ret := C.deserial_init()
 	if ret != 0 {
-		l.Stop()
 		return fmt.Errorf("deserial_init failed:%v", ret)
 	}
 	l.bpfLogLevel = l.workloadObj.SockConn.BpfLogLevel

diff --git a/pkg/controller/workload/workload_processor.go b/pkg/controller/workload/workload_processor.go
@@ -58,7 +58,8 @@ type Processor struct {
 	WorkloadCache cache.WorkloadCache
 	ServiceCache  cache.ServiceCache
 
-	once sync.Once
+	once      sync.Once
+	authzOnce sync.Once
 }
 
 func newProcessor(workloadMap bpf2go.KmeshCgroupSockWorkloadMaps) *Processor {
@@ -660,7 +661,7 @@ func (p *Processor) handleAuthorizationTypeResponse(rsp *service_discovery_v3.De
 		log.Debugf("remove authorization policy %s", resourceName)
 	}
 
-	p.once.Do(func() {
+	p.authzOnce.Do(func() {
 		p.handleRemovedAuthzPolicyDuringRestart(rbac)
 	})
 	return nil