feat(manager): make check status from flink batch

internal/manager/manager.go

@@ -5,39 +5,69 @@ import (
 	"flink-kube-operator/internal/managed_job"
 	"time"
 
+	"gitea.com/logicamp/lc"
 	api "github.com/logi-camp/go-flink-client"
+	"github.com/samber/lo"
 	"github.com/tidwall/buntdb"
+	"go.uber.org/zap"
 	"k8s.io/apimachinery/pkg/types"
 )
 
-var managedJobs = map[types.UID]managed_job.ManagedJob{}
+type Manager struct {
+	client      *api.Client
+	managedJobs map[types.UID]managed_job.ManagedJob
+}
 
-func Setup(client *api.Client, db *buntdb.DB, crdInstance *crd.Crd) {
+func NewManager(client *api.Client, db *buntdb.DB, crdInstance *crd.Crd) Manager {
 	ticker := time.NewTicker(5 * time.Second)
 	quit := make(chan struct{})
+	mgr := Manager{
+		client:      client,
+		managedJobs: map[types.UID]managed_job.ManagedJob{},
+	}
 
 	go func() {
 		for {
 			select {
 			case <-ticker.C:
-				cycle(client, db, crdInstance)
+				mgr.cycle(client, db, crdInstance)
 			case <-quit:
 				ticker.Stop()
 				return
 			}
 		}
 	}()
+	return mgr
 }
 
-func cycle(client *api.Client, db *buntdb.DB, crdInstance *crd.Crd) {
+func (mgr *Manager) cycle(client *api.Client, db *buntdb.DB, crdInstance *crd.Crd) {
+	jobsOverviews, err := mgr.client.JobsOverview()
+	if err != nil {
+		lc.Logger.Error("[manager] [cycle] cannot check flink jobs status", zap.Error(err))
+	}
+	//lc.Logger.Debug("[manager] [cycle] overviews", zap.Any("overviews", jobsOverviews))
+
 	for _, uid := range crd.GetAllJobKeys() {
 		def := crd.GetJob(uid)
-		managedJob, ok := managedJobs[uid]
+		managedJob, ok := mgr.managedJobs[uid]
 		if ok {
 			managedJob.Update(def)
 		} else {
-			managedJob := managed_job.NewManagedJob(client, db, def, crdInstance)
-			managedJobs[uid] = *managedJob
+			managedJob = *managed_job.NewManagedJob(client, db, def, crdInstance)
+			//mgr.managedJobs[uid] = managedJob
 		}
+		jobOverview, ok := lo.Find(jobsOverviews.Jobs, func(job api.JobOverview) bool {
+			jobId := managedJob.GetJobId()
+			if jobId != nil {
+				return job.ID == *jobId
+			}
+			return false
+		})
+		if ok {
+			lc.Logger.Debug("[manager] read status from flink", zap.String("name", jobOverview.Name), zap.String("state", jobOverview.State))
+			managedJob.SetStatus(managed_job.JobStatus(jobOverview.State))
+		}
+		managedJob.Cycle()
+		mgr.managedJobs[uid] = managedJob
 	}
 }