Function addDockerComposeCommonParam

agent/app/service/app_utils.go:1800–1897

(composeMap map[string]interface{}, serviceName string, req request.AppContainerConfig, params map[string]interface{})

}

func addDockerComposeCommonParam(composeMap map[string]interface{}, serviceName string, req request.AppContainerConfig, params map[string]interface{}) error {
	services, serviceValid := composeMap["services"].(map[string]interface{})
	if !serviceValid {
		return buserr.New("ErrFileParse")
	}
	imagePreFix := xpack.MultiNodeProvider.GetImagePrefix()
	if imagePreFix != "" {
		for _, service := range services {
			serviceValue := service.(map[string]interface{})
			if image, ok := serviceValue["image"]; ok {
				imageStr := image.(string)
				lastSlashIndex := strings.LastIndex(imageStr, "/")
				if lastSlashIndex != -1 {
					imageStr = imageStr[lastSlashIndex+1:]
				}
				imageStr = imagePreFix + "/" + imageStr
				serviceValue["image"] = imageStr
			}
		}
	}

	service, serviceExist := services[serviceName]
	if !serviceExist {
		return buserr.New("ErrFileParse")
	}
	serviceValue := service.(map[string]interface{})

	deploy := map[string]interface{}{}
	if de, ok := serviceValue["deploy"]; ok {
		deploy = de.(map[string]interface{})
	}
	resource := map[string]interface{}{}
	if res, ok := deploy["resources"]; ok {
		resource = res.(map[string]interface{})
	}
	resource["limits"] = map[string]interface{}{
		"cpus":   "${CPUS}",
		"memory": "${MEMORY_LIMIT}",
	}
	if req.GpuConfig {
		resource["reservations"] = map[string]interface{}{
			"devices": []map[string]interface{}{
				{
					"driver":       "nvidia",
					"count":        "all",
					"capabilities": []string{"gpu"},
				},
			},
		}
	}
	deploy["resources"] = resource
	serviceValue["deploy"] = deploy

	if req.RestartPolicy != "" {
		if req.RestartPolicy == "on-failure" {
			serviceValue["restart"] = "on-failure:5"
		} else {
			serviceValue["restart"] = req.RestartPolicy