extract ip for net inherit replace (#424)

Author: jschwinger233

engine/docker/container.go

@@ -82,7 +82,7 @@ func loadRawArgs(b []byte) (*RawArgs, error) {
 }
 
 // VirtualizationCreate create a workload
-func (e *Engine) VirtualizationCreate(ctx context.Context, opts *enginetypes.VirtualizationCreateOptions) (*enginetypes.VirtualizationCreated, error) {
+func (e *Engine) VirtualizationCreate(ctx context.Context, opts *enginetypes.VirtualizationCreateOptions) (*enginetypes.VirtualizationCreated, error) { // nolint
 	r := &enginetypes.VirtualizationCreated{}
 	// memory should more than 4MiB
 	if opts.Memory > 0 && opts.Memory < minMemory || opts.Memory < 0 {
@@ -244,6 +244,11 @@ func (e *Engine) VirtualizationCreate(ctx context.Context, opts *enginetypes.Vir
 		EndpointsConfig: map[string]*dockernetwork.EndpointSettings{},
 	}
 	for networkID, ipv4 := range networks {
+		if useCNI(opts.Labels) && ipv4 != "" {
+			config.Labels["ipv4"] = ipv4
+			break
+		}
+
 		endpointSetting, err := e.makeIPV4EndpointSetting(ipv4)
 		if err != nil {
 			return r, err

engine/docker/helper.go

@@ -282,3 +282,12 @@ func dumpFromString(ctx context.Context, ca, cert, key *os.File, caStr, certStr,
 	log.Debug(ctx, "[dumpFromString] Dump ca.pem, cert.pem, key.pem from string")
 	return nil
 }
+
+func useCNI(labels map[string]string) bool {
+	for k, v := range labels {
+		if k == "cni" && v == "1" {
+			return true
+		}
+	}
+	return false
+}