Dask dashboard should have a separate log config

Signed-off-by: Andrew Dye <[email protected]>

flyteplugins/go/tasks/plugins/k8s/dask/config.go

@@ -0,0 +1,29 @@
+package dask
+
+import (
+	pluginsConfig "github.com/flyteorg/flyte/flyteplugins/go/tasks/config"
+	"github.com/flyteorg/flyte/flyteplugins/go/tasks/logs"
+)
+
+//go:generate pflags Config --default-var=defaultConfig
+
+var (
+	defaultConfig = Config{
+		Logs: logs.DefaultConfig,
+	}
+
+	configSection = pluginsConfig.MustRegisterSubSection("dask", &defaultConfig)
+)
+
+// Config is config for 'dask' plugin
+type Config struct {
+	Logs logs.LogConfig `json:"logs,omitempty"`
+}
+
+func GetConfig() *Config {
+	return configSection.GetConfig().(*Config)
+}
+
+func SetConfig(cfg *Config) error {
+	return configSection.SetConfig(cfg)
+}

flyteplugins/go/tasks/plugins/k8s/dask/dask.go

@@ -279,7 +279,7 @@ func createJobSpec(workerSpec daskAPI.WorkerSpec, schedulerSpec daskAPI.Schedule
 }
 
 func (p daskResourceHandler) GetTaskPhase(ctx context.Context, pluginContext k8s.PluginContext, r client.Object) (pluginsCore.PhaseInfo, error) {
-	logPlugin, err := logs.InitializeLogPlugins(logs.GetLogConfig())
+	logPlugin, err := logs.InitializeLogPlugins(&GetConfig().Logs)
 	if err != nil {
 		return pluginsCore.PhaseInfoUndefined, err
 	}