更改环境变量，增加NODE_ID

zhangqian

2023-12-26 a2667454aa39ea72b0e3660dca7dc4f468d712a2

更改环境变量，增加NODE_ID

 src/k8s/create.go

@@ -39,6 +39,7 @@
    ServiceName    string                // Service 名称
    ALHost         string                // 算法Host
    Host           string                // 服务Host
    NodeID         string                // Nsq节点ID
}

func create_test() {
@@ -184,6 +185,61 @@
func createDeployment(config Config) error {
    fmt.Println("\033[1;37;40mCreating Deployment:", config.DeploymentName, "\033[0m")

    envs := []apiv1.EnvVar{
        {
            Name:  "GRPC_PORT",
            Value: fmt.Sprint(config.RpcPort),
        },
        {
            Name: "DB_HOST",
            ValueFrom: &apiv1.EnvVarSource{
                FieldRef: &apiv1.ObjectFieldSelector{
                    APIVersion: "v1",
                    FieldPath:  "status.hostIP",
                },
            },
        },
        {
            Name: "HOST",
            ValueFrom: &apiv1.EnvVarSource{
                FieldRef: &apiv1.ObjectFieldSelector{
                    APIVersion: "v1",
                    FieldPath:  "status.hostIP",
                },
            },
        },
        {
            Name:  "DB_NAME",
            Value: config.DBName,
        },
        {
            Name:  "DB_PORT",
            Value: strconv.Itoa(6446),
        },
        {
            Name:  "DB_USER",
            Value: config.NameSpace,
        },
        {
            Name:  "DB_PASSWD",
            Value: config.NameSpace + "@Basic2023",
        },
    }

    if config.ALHost != "" {
        envs = append(envs, apiv1.EnvVar{
            Name:  "AL_HOST",
            Value: config.ALHost,
        })
    }

    if config.NodeID != "" {
        envs = append(envs, apiv1.EnvVar{
            Name:  "NODE_ID",
            Value: config.NodeID,
        })
    }

    deployment := &appsv1.Deployment{
        ObjectMeta: metav1.ObjectMeta{
            Name: config.DeploymentName,
@@ -216,52 +272,9 @@
                    },
                    Containers: []apiv1.Container{
                        {
                            Name:  config.NameSpace,
                            Image: config.Image,
                            Env: []apiv1.EnvVar{
                                {
                                    Name:  "GRPC_PORT",
                                    Value: fmt.Sprint(config.RpcPort),
                                },
                                {
                                    Name: "DB_HOST",
                                    ValueFrom: &apiv1.EnvVarSource{
                                        FieldRef: &apiv1.ObjectFieldSelector{
                                            APIVersion: "v1",
                                            FieldPath:  "status.hostIP",
                                        },
                                    },
                                },
                                {
                                    Name: "HOST",
                                    ValueFrom: &apiv1.EnvVarSource{
                                        FieldRef: &apiv1.ObjectFieldSelector{
                                            APIVersion: "v1",
                                            FieldPath:  "status.hostIP",
                                        },
                                    },
                                },
                                {
                                    Name:  "AL_HOST",
                                    Value: config.ALHost,
                                },
                                {
                                    Name:  "DB_NAME",
                                    Value: config.DBName,
                                },
                                {
                                    Name:  "DB_PORT",
                                    Value: strconv.Itoa(6446),
                                },
                                {
                                    Name:  "DB_USER",
                                    Value: config.NameSpace,
                                },
                                {
                                    Name:  "DB_PASSWD",
                                    Value: config.NameSpace + "@Basic2023",
                                },
                            },
                            Name:            config.NameSpace,
                            Image:           config.Image,
                            Env:             envs,
                            ImagePullPolicy: apiv1.PullAlways, // 设置镜像拉取策略为 Always
                        },
                    },