Changed how the data is outputted to stdout and fixed xname issue

This commit is contained in:
David J. Allen 2023-10-25 12:13:21 -06:00
parent e039099263
commit 47f773df2a

View file

@ -122,16 +122,8 @@ func CollectAll(probeStates *[]ScannedResult, l *log.Logger, q *QueryParams) err
done := make(chan struct{}, q.Threads+1) done := make(chan struct{}, q.Threads+1)
chanProbeState := make(chan ScannedResult, q.Threads+1) chanProbeState := make(chan ScannedResult, q.Threads+1)
// generate custom xnames for bmcs
node := xnames.Node{
Cabinet: 1000,
Chassis: 1,
ComputeModule: 7,
NodeBMC: -1,
}
// collect bmc information asynchronously // collect bmc information asynchronously
var offset = 0
var wg sync.WaitGroup var wg sync.WaitGroup
wg.Add(q.Threads) wg.Add(q.Threads)
for i := 0; i < q.Threads; i++ { for i := 0; i < q.Threads; i++ {
@ -145,12 +137,19 @@ func CollectAll(probeStates *[]ScannedResult, l *log.Logger, q *QueryParams) err
q.Host = ps.Host q.Host = ps.Host
q.Port = ps.Port q.Port = ps.Port
node.NodeBMC += 1 // generate custom xnames for bmcs
node := xnames.Node{
bmclibClient, err := NewClient(l, q) Cabinet: 1000,
if err != nil { Chassis: 1,
l.Log.Errorf("could not make client: %v", err) ComputeModule: 7,
NodeBMC: offset,
} }
offset += 1
// bmclibClient, err := NewClient(l, q)
// if err != nil {
// l.Log.Errorf("could not make client: %v", err)
// }
gofishClient, err := connectGofish(q) gofishClient, err := connectGofish(q)
if err != nil { if err != nil {
@ -173,14 +172,14 @@ func CollectAll(probeStates *[]ScannedResult, l *log.Logger, q *QueryParams) err
var rm map[string]json.RawMessage var rm map[string]json.RawMessage
// inventories // inventories
if bmclibClient != nil { // if bmclibClient != nil {
inventory, err := CollectInventory(bmclibClient, q) // inventory, err := CollectInventory(bmclibClient, q)
if err != nil { // if err != nil {
l.Log.Errorf("could not query inventory (%v:%v): %v", q.Host, q.Port, err) // l.Log.Errorf("could not query inventory (%v:%v): %v", q.Host, q.Port, err)
} // }
json.Unmarshal(inventory, &rm) // json.Unmarshal(inventory, &rm)
data["Inventory"] = rm["Inventory"] // data["Inventory"] = rm["Inventory"]
} // }
// chassis // chassis
if gofishClient != nil { if gofishClient != nil {
@ -211,25 +210,29 @@ func CollectAll(probeStates *[]ScannedResult, l *log.Logger, q *QueryParams) err
headers := make(map[string]string) headers := make(map[string]string)
headers["Content-Type"] = "application/json" headers["Content-Type"] = "application/json"
b, err := json.MarshalIndent(data, "", " ") body, err := json.MarshalIndent(data, "", " ")
if err != nil { if err != nil {
l.Log.Errorf("could not marshal JSON: %v", err) l.Log.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("%v\n", string(body))
}
// write JSON data to file // write JSON data to file
err = os.WriteFile(path.Clean(outputPath + "/" + q.Host + ".json"), b, os.ModePerm) err = os.WriteFile(path.Clean(outputPath + "/" + q.Host + ".json"), body, os.ModePerm)
if err != nil { if err != nil {
l.Log.Errorf("could not write data to file: %v", err) l.Log.Errorf("could not write data to file: %v", err)
} }
// add all endpoints to smd // add all endpoints to smd
err = smd.AddRedfishEndpoint(b, headers) err = smd.AddRedfishEndpoint(body, headers)
if err != nil { if err != nil {
l.Log.Error(err) l.Log.Error(err)
// try updating instead // try updating instead
if q.ForceUpdate { if q.ForceUpdate {
err = smd.UpdateRedfishEndpoint(data["ID"].(string), b, headers) err = smd.UpdateRedfishEndpoint(data["ID"].(string), body, headers)
if err != nil { if err != nil {
l.Log.Error(err) l.Log.Error(err)
} }
@ -296,9 +299,6 @@ func CollectMetadata(client *bmclib.Client, q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("Metadata: %v\n", string(b))
}
ctxCancel() ctxCancel()
return b, nil return b, nil
} }
@ -328,9 +328,6 @@ func CollectInventory(client *bmclib.Client, q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
ctxCancel() ctxCancel()
return b, nil return b, nil
} }
@ -358,9 +355,6 @@ func CollectPowerState(client *bmclib.Client, q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
ctxCancel() ctxCancel()
return b, nil return b, nil
@ -392,11 +386,7 @@ func CollectUsers(client *bmclib.Client, q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
// return b, nil
ctxCancel() ctxCancel()
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
return b, nil return b, nil
} }
@ -406,9 +396,6 @@ func CollectBios(client *bmclib.Client, q *QueryParams) ([]byte, error) {
// return nil, fmt.Errorf("could not make query: %v", err) // return nil, fmt.Errorf("could not make query: %v", err)
// } // }
b, err := makeRequest(client, client.GetBiosConfiguration, q.Timeout) b, err := makeRequest(client, client.GetBiosConfiguration, q.Timeout)
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
return b, err return b, err
} }
@ -437,9 +424,6 @@ func CollectEthernetInterfaces(c *gofish.APIClient, q *QueryParams, systemID str
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
// if q.Verbose {
// fmt.Printf("%v\n", string(b))
// }
return b, nil return b, nil
} }
@ -455,9 +439,6 @@ func CollectChassis(c *gofish.APIClient, q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
return b, nil return b, nil
} }
@ -483,9 +464,6 @@ func CollectStorage(c *gofish.APIClient, q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
return b, nil return b, nil
} }
@ -516,9 +494,6 @@ func CollectSystems(c *gofish.APIClient, q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
return b, nil return b, nil
} }
@ -534,9 +509,6 @@ func CollectRegisteries(c *gofish.APIClient, q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
return b, nil return b, nil
} }
@ -577,9 +549,6 @@ func CollectProcessors(q *QueryParams) ([]byte, error) {
return nil, fmt.Errorf("could not marshal JSON: %v", err) return nil, fmt.Errorf("could not marshal JSON: %v", err)
} }
if q.Verbose {
fmt.Printf("%v\n", string(b))
}
return b, nil return b, nil
} }