Mohammadreza Khani
9f36dca7c9
This fix prevents undefined behavior and runtime crashes in multithreaded use cases.
30 lines
600 B
Go
30 lines
600 B
Go
package crd
|
|
|
|
import (
|
|
"flink-kube-operator/internal/crd/v1alpha1"
|
|
"flink-kube-operator/pkg"
|
|
|
|
"k8s.io/apimachinery/pkg/types"
|
|
)
|
|
|
|
// var jobs = map[types.UID]*v1alpha1.FlinkJob{}
|
|
var jobs = pkg.SafeMap[types.UID, *v1alpha1.FlinkJob]{}
|
|
|
|
func (crd *Crd) repsert(job *v1alpha1.FlinkJob) {
|
|
jobs.Store(job.GetUID(), job)
|
|
}
|
|
|
|
func GetJob(uid types.UID) v1alpha1.FlinkJob {
|
|
job, _ := jobs.Load(uid)
|
|
return *job.DeepCopy()
|
|
}
|
|
|
|
func GetAllJobKeys() []types.UID {
|
|
keys := []types.UID{}
|
|
jobs.Range(func(k types.UID, value *v1alpha1.FlinkJob) bool {
|
|
keys = append(keys, k)
|
|
return true
|
|
})
|
|
return keys
|
|
}
|