[minor] store AWS agent environment in separate struct

agent.go

@@ -36,18 +36,23 @@ type Agent struct {
 
 // AgentEnv stores basic information of the endpoint
 type AgentEnv struct {
-	Init            string   `json:"init,omitempty"`
-	Ident           string   `json:"ident,omitempty"`
-	OS              string   `json:"os,omitempty"`
-	Arch            string   `json:"arch,omitempty"`
-	IsProxied       bool     `json:"isproxied"`
-	Proxy           string   `json:"proxy,omitempty"`
-	Addresses       []string `json:"addresses,omitempty"`
-	PublicIP        string   `json:"publicip,omitempty"`
-	AWSInstanceID   string   `json:"aws_instanceid,omitempty"`
-	AWSLocalIPV4    string   `json:"aws_localipv4,omitempty"`
-	AWSAMIID        string   `json:"aws_amiid,omitempty"`
-	AWSInstanceType string   `json:"aws_instancetype,omitempty"`
+	Init      string      `json:"init,omitempty"`
+	Ident     string      `json:"ident,omitempty"`
+	OS        string      `json:"os,omitempty"`
+	Arch      string      `json:"arch,omitempty"`
+	IsProxied bool        `json:"isproxied"`
+	Proxy     string      `json:"proxy,omitempty"`
+	Addresses []string    `json:"addresses,omitempty"`
+	PublicIP  string      `json:"publicip,omitempty"`
+	AWS       AgentEnvAWS `json:"aws,omitempty"`
+}
+
+// Stores AWS specific agent environment values
+type AgentEnvAWS struct {
+	AWSInstanceID   string `json:"aws_instanceid,omitempty"`
+	AWSLocalIPV4    string `json:"aws_localipv4,omitempty"`
+	AWSAMIID        string `json:"aws_amiid,omitempty"`
+	AWSInstanceType string `json:"aws_instancetype,omitempty"`
 }
 
 type AgentsStats struct {

mig-agent/aws.go

@@ -62,7 +62,7 @@ func addAWSInstanceID(orig_ctx Context) (ctx Context, err error) {
 	if err != nil {
 		return
 	}
-	ctx.Agent.Env.AWSInstanceID = res
+	ctx.Agent.Env.AWS.AWSInstanceID = res
 	return
 }
 
@@ -73,7 +73,7 @@ func addAWSLocalIPV4(orig_ctx Context) (ctx Context, err error) {
 	if err != nil {
 		return
 	}
-	ctx.Agent.Env.AWSLocalIPV4 = res
+	ctx.Agent.Env.AWS.AWSLocalIPV4 = res
 	return
 }
 
@@ -84,7 +84,7 @@ func addAWSAMIID(orig_ctx Context) (ctx Context, err error) {
 	if err != nil {
 		return
 	}
-	ctx.Agent.Env.AWSAMIID = res
+	ctx.Agent.Env.AWS.AWSAMIID = res
 	return
 }
 
@@ -95,7 +95,7 @@ func addAWSInstanceType(orig_ctx Context) (ctx Context, err error) {
 	if err != nil {
 		return
 	}
-	ctx.Agent.Env.AWSInstanceType = res
+	ctx.Agent.Env.AWS.AWSInstanceType = res
 	return
 }