更改环境变量，增加NODE_ID

zhangqian

2023-12-26 a2667454aa39ea72b0e3660dca7dc4f468d712a2

 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
                  },
               },