package premium_dm

import (
	"context"

	dm "github.com/pingcap/tiflow/dm/openapi"
)

func createPremiumDMTaskWithMetricLabels(
	ctx context.Context, proxy *PremiumDMProxy, req *dm.CreateTaskRequest, clusterID string,
) (*dm.OperateTaskResponse, error) {
	req.Task.MetricLabels = &dm.Task_MetricLabels{AdditionalProperties: map[string]string{
		"keyspace_name": clusterID,
	}}
	return proxy.CreatePremiumDMTask(ctx, req, clusterID)
}