Added labels and annotations string flags. Parse json to map for labels

and annotations

go.mod

@@ -42,6 +42,7 @@ replace (
 	kubevirt.io/client-go => github.com/kubevirt/client-go v0.40.0-rc.2
 	kubevirt.io/containerized-data-importer => github.com/rancher/kubevirt-containerized-data-importer v1.26.1-0.20210303063201-9e7a78643487
 	sigs.k8s.io/structured-merge-diff => sigs.k8s.io/structured-merge-diff v0.0.0-20190302045857-e85c7b244fd2
+	github.com/harvester/go-harvester => ../go-harvester
 )
 
 require (

harvester/create.go

@@ -1,6 +1,7 @@
 package harvester
 
 import (
+	"encoding/json"
 	"fmt"
 	"io/ioutil"
 	"strings"
@@ -104,13 +105,26 @@ func (d *Driver) Create() error {
 		ImageID:          fmt.Sprintf("%s/%s", d.Namespace, d.ImageName),
 	}
 
+	templateAnnotations := make(map[string]string)
+	j_err := json.Unmarshal([]byte(d.Annotations), &templateAnnotations)
+	if j_err != nil {
+		panic(j_err)
+	}
+	templateLabels := make(map[string]string)
+	l_err := json.Unmarshal([]byte(d.Labels), &templateLabels)
+	if l_err != nil {
+		panic(l_err)
+	}
+
 	// create vm
 	vmBuilder := builder.NewVMBuilder("docker-machine-driver-harvester").
 		Namespace(d.Namespace).Name(d.MachineName).
 		CPU(d.CPU).Memory(d.MemorySize).
 		Image(d.DiskSize, d.DiskBus, dataVolumeOption).
 		EvictionStrategy(true).
 		DefaultPodAntiAffinity().
+		SpecTemplateAnnotations(templateAnnotations).
+		SpecTemplateLabels(templateLabels).
 		CloudInit(userData, networkData)
 
 	if d.KeyPairName != "" {

harvester/flags.go

@@ -125,6 +125,16 @@ func (d *Driver) GetCreateFlags() []mcnflag.Flag {
 			Name:   "harvester-network-name",
 			Usage:  "harvester network name",
 		},
+		mcnflag.StringFlag{
+			EnvVar: "HARVESTER_ANNOTATIONS",
+			Name:   "harvester-annotations",
+			Usage:  "harvester annotations",
+		},
+		mcnflag.StringFlag{
+			EnvVar: "HARVESTER_ANNOTATIONS",
+			Name:   "harvester-labels",
+			Usage:  "harvester labels",
+		},
 		mcnflag.StringFlag{
 			EnvVar: "HARVESTER_NETWORK_MODEL",
 			Name:   "harvester-network-model",
@@ -161,6 +171,9 @@ func (d *Driver) SetConfigFromFlags(flags drivers.DriverOptions) error {
 	d.NetworkName = flags.String("harvester-network-name")
 	d.NetworkModel = flags.String("harvester-network-model")
 
+	d.Annotations = flags.String("harvester-annotations")
+	d.Labels = flags.String("harvester-labels")
+
 	d.SetSwarmConfigFromFlags(flags)
 
 	return d.checkConfig()

harvester/harvester.go

@@ -52,6 +52,9 @@ type Driver struct {
 
 	NetworkName  string
 	NetworkModel string
+
+	Annotations string
+	Labels      string
 }
 
 func NewDriver(hostName, storePath string) *Driver {