feat: kubesphere 4.0 (#6115)
* feat: kubesphere 4.0 Signed-off-by: ci-bot <ci-bot@kubesphere.io> * feat: kubesphere 4.0 Signed-off-by: ci-bot <ci-bot@kubesphere.io> --------- Signed-off-by: ci-bot <ci-bot@kubesphere.io> Co-authored-by: ks-ci-bot <ks-ci-bot@example.com> Co-authored-by: joyceliu <joyceliu@yunify.com>
This commit is contained in:
committed by
GitHub
parent
b5015ec7b9
commit
447a51f08b
79
pkg/controller/telemetry/telemetry_runnable.go
Normal file
79
pkg/controller/telemetry/telemetry_runnable.go
Normal file
@@ -0,0 +1,79 @@
|
||||
/*
|
||||
* Please refer to the LICENSE file in the root directory of the project.
|
||||
* https://github.com/kubesphere/kubesphere/blob/master/LICENSE
|
||||
*/
|
||||
|
||||
package telemetry
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os/exec"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"k8s.io/klog/v2"
|
||||
"k8s.io/utils/ptr"
|
||||
clusterv1alpha1 "kubesphere.io/api/cluster/v1alpha1"
|
||||
"sigs.k8s.io/controller-runtime/pkg/manager"
|
||||
|
||||
kscontroller "kubesphere.io/kubesphere/pkg/controller"
|
||||
"kubesphere.io/kubesphere/pkg/controller/options"
|
||||
)
|
||||
|
||||
var _ manager.LeaderElectionRunnable = &Runnable{}
|
||||
var _ manager.Runnable = &Runnable{}
|
||||
var _ kscontroller.Controller = &Runnable{}
|
||||
|
||||
const (
|
||||
runnableName = "telemetry"
|
||||
// defaultPeriod for collect data
|
||||
defaultPeriod = time.Hour * 24
|
||||
)
|
||||
|
||||
type Runnable struct {
|
||||
*options.TelemetryOptions
|
||||
}
|
||||
|
||||
func (r *Runnable) Name() string {
|
||||
return runnableName
|
||||
}
|
||||
|
||||
func (r *Runnable) SetupWithManager(mgr *kscontroller.Manager) error {
|
||||
if mgr.TelemetryOptions == nil || mgr.TelemetryOptions.KSCloudURL == "" {
|
||||
klog.V(4).Infof("telemetry runnable is disabled")
|
||||
return nil
|
||||
}
|
||||
r.TelemetryOptions = mgr.TelemetryOptions
|
||||
if r.TelemetryOptions.Period == nil {
|
||||
r.TelemetryOptions.Period = ptr.To[time.Duration](defaultPeriod)
|
||||
}
|
||||
return mgr.Add(r)
|
||||
}
|
||||
|
||||
func (r *Runnable) Enabled(clusterRole string) bool {
|
||||
return strings.EqualFold(clusterRole, string(clusterv1alpha1.ClusterRoleHost))
|
||||
}
|
||||
|
||||
func (r *Runnable) Start(ctx context.Context) error {
|
||||
t := time.NewTicker(*r.Period)
|
||||
for {
|
||||
select {
|
||||
case <-t.C:
|
||||
var args = []string{
|
||||
"--url", r.KSCloudURL,
|
||||
}
|
||||
|
||||
cmd := exec.CommandContext(ctx, "telemetry", args...)
|
||||
if _, err := cmd.CombinedOutput(); err != nil {
|
||||
klog.Errorf("failed to exec command for telemetry %v", err)
|
||||
}
|
||||
case <-ctx.Done():
|
||||
t.Stop()
|
||||
return nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (r *Runnable) NeedLeaderElection() bool {
|
||||
return true
|
||||
}
|
||||
Reference in New Issue
Block a user