(cmd *cobra.Command, cmds []AgentCmdExecute)
| 268 | } |
| 269 | |
| 270 | func agentInitCmd(cmd *cobra.Command, cmds []AgentCmdExecute) { |
| 271 | conn := agentGetConn(cmd) |
| 272 | if conn == nil { |
| 273 | return |
| 274 | } |
| 275 | defer conn.Close() |
| 276 | name := "" |
| 277 | groupID := agentParamData.GroupID |
| 278 | clusterID := agentParamData.ClusterID |
| 279 | teamID := agentParamData.TeamID |
| 280 | k8sWatchPolicy := agent.KubernetesWatchPolicy(agentParamData.K8SWatchPolicy) |
| 281 | grpcBufferSize := agentParamData.GRPCBufferSize * 1024 * 1024 |
| 282 | fmt.Printf("request trisolaris(%s), params(%+v)\n", conn.Target(), agentParamData) |
| 283 | c := agent.NewSynchronizerClient(conn) |
| 284 | reqData := &agent.SyncRequest{ |
| 285 | CtrlIp: &agentParamData.CtrlIP, |
| 286 | CtrlMac: &agentParamData.CtrlMac, |
| 287 | CurrentGrpcBufferSize: &grpcBufferSize, |
| 288 | AgentGroupIdRequest: &groupID, |
| 289 | KubernetesClusterId: &clusterID, |
| 290 | KubernetesWatchPolicy: &k8sWatchPolicy, |
| 291 | ProcessName: &name, |
| 292 | TeamId: &teamID, |
| 293 | } |
| 294 | var response *agent.SyncResponse |
| 295 | var err error |
| 296 | response, err = c.Sync(context.Background(), reqData) |
| 297 | if err != nil { |
| 298 | fmt.Println(err) |
| 299 | return |
| 300 | } |
| 301 | if agentParamData.GRPCBufferSize == 0 { |
| 302 | fmt.Printf("revision: %s\n", response.GetRevision()) |
| 303 | } else { |
| 304 | fmt.Printf("current grpc buffer size (byte): %d\n", grpcBufferSize) |
| 305 | } |
| 306 | |
| 307 | for _, cmd := range cmds { |
| 308 | cmd(response) |
| 309 | } |
| 310 | } |
| 311 | |
| 312 | func AgpidAgentResponse(cmd *cobra.Command) { |
| 313 | conn := agentGetConn(cmd) |
no test coverage detected