mirror of
https://github.com/NVIDIA/nvidia-container-toolkit
synced 2024-11-25 21:39:10 +00:00
Output YAML separator
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
parent
3d2328bdfd
commit
c793fc27d8
@ -122,7 +122,7 @@ func (m command) run(c *cli.Context, cfg *config) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = outputTo.Write(data)
|
err = writeToOutput(cfg.jsonMode, data, outputTo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to write output: %v", err)
|
return fmt.Errorf("failed to write output: %v", err)
|
||||||
}
|
}
|
||||||
@ -130,6 +130,22 @@ func (m command) run(c *cli.Context, cfg *config) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func writeToOutput(jsonMode bool, data []byte, output io.Writer) error {
|
||||||
|
if !jsonMode {
|
||||||
|
_, err := output.Write([]byte("---\n"))
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to write YAML separator: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
_, err := output.Write(data)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to write data: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (m command) generateSpec() (*specs.Spec, error) {
|
func (m command) generateSpec() (*specs.Spec, error) {
|
||||||
nvmllib := nvml.New()
|
nvmllib := nvml.New()
|
||||||
if r := nvmllib.Init(); r != nvml.SUCCESS {
|
if r := nvmllib.Init(); r != nvml.SUCCESS {
|
||||||
|
Loading…
Reference in New Issue
Block a user