github/troubleshoot
1
0
Fork 0
mirror of https://github.com/replicatedhq/troubleshoot.git synced 2026-02-14 18:29:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

initial attempt

Browse Source 
Signed-off-by: Dan Jones <danj@replicated.com>
This commit is contained in:
Dan Jones
2022-08-22 16:57:54 +01:00
parent 414bf84c68
commit 3b3ae4e7bf
3 changed files with 92 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/troubleshoot/cli/root.go
View File

@@ -34,7 +34,7 @@ from a server that can be used to assist when troubleshooting a Kubernetes clust
v := viper.GetViper()
logger.SetQuiet(v.GetBool("quiet"))
return runTroubleshoot(v, args[0])
return runTroubleshoot(v, args)
},
}

107
cmd/troubleshoot/cli/run.go
View File

@@ -31,7 +31,7 @@ import (
"k8s.io/client-go/rest"
)
func runTroubleshoot(v *viper.Viper, arg string) error {
func runTroubleshoot(v *viper.Viper, arg []string) error {
interactive := v.GetBool("interactive") && isatty.IsTerminal(os.Stdout.Fd())
if interactive {
@@ -68,23 +68,62 @@ func runTroubleshoot(v *viper.Viper, arg string) error {
})
}
collectorContent, err := supportbundle.LoadSupportBundleSpec(arg)
if err != nil {
return errors.Wrap(err, "failed to load collector spec")
}
// collectorContent, err := supportbundle.LoadSupportBundleSpec(arg)
// if err != nil {
// return errors.Wrap(err, "failed to load collector spec")
// }
//
// multidocs := strings.Split(string(collectorContent), "\n---\n")
//
// // we support both raw collector kinds and supportbundle kinds here
// supportBundle, err := supportbundle.ParseSupportBundleFromDoc([]byte(multidocs[0]))
// if err != nil {
// return errors.Wrap(err, "failed to parse collector")
// }
multidocs := strings.Split(string(collectorContent), "\n---\n")
// we support both raw collector kinds and supportbundle kinds here
supportBundle, err := supportbundle.ParseSupportBundleFromDoc([]byte(multidocs[0]))
if err != nil {
return errors.Wrap(err, "failed to parse collector")
}
var mainBundle *troubleshootv1beta2.SupportBundle
troubleshootclientsetscheme.AddToScheme(scheme.Scheme)
decode := scheme.Codecs.UniversalDeserializer().Decode
additionalRedactors := &troubleshootv1beta2.Redactor{}
for i, v := range arg {
collectorContent, err := supportbundle.LoadSupportBundleSpec(v)
if err != nil {
return errors.Wrap(err, "failed to load collector spec")
}
multidocs := strings.Split(string(collectorContent), "\n---\n")
supportBundle, err := supportbundle.ParseSupportBundleFromDoc([]byte(multidocs[0]))
if err != nil {
return errors.Wrap(err, "failed to parse collector")
}
if i == 0 {
mainBundle = supportBundle
} else {
mainBundle.concatSpec(supportBundle)
}
for i, additionalDoc := range multidocs {
if i == 0 {
continue
}
additionalDoc, err := docrewrite.ConvertToV1Beta2([]byte(additionalDoc))
if err != nil {
return errors.Wrap(err, "failed to convert to v1beta2")
}
obj, _, err := decode(additionalDoc, nil, nil)
if err != nil {
return errors.Wrapf(err, "failed to parse additional doc %d", i)
}
multidocRedactors, ok := obj.(*troubleshootv1beta2.Redactor)
if !ok {
continue
}
additionalRedactors.Spec.Redactors = append(additionalRedactors.Spec.Redactors, multidocRedactors.Spec.Redactors...)
}
}
for idx, redactor := range v.GetStringSlice("redactors") {
redactorObj, err := supportbundle.GetRedactorFromURI(redactor)
if err != nil {
@@ -96,24 +135,24 @@ func runTroubleshoot(v *viper.Viper, arg string) error {
}
}
for i, additionalDoc := range multidocs {
if i == 0 {
continue
}
additionalDoc, err := docrewrite.ConvertToV1Beta2([]byte(additionalDoc))
if err != nil {
return errors.Wrap(err, "failed to convert to v1beta2")
}
obj, _, err := decode(additionalDoc, nil, nil)
if err != nil {
return errors.Wrapf(err, "failed to parse additional doc %d", i)
}
multidocRedactors, ok := obj.(*troubleshootv1beta2.Redactor)
if !ok {
continue
}
additionalRedactors.Spec.Redactors = append(additionalRedactors.Spec.Redactors, multidocRedactors.Spec.Redactors...)
}
// for i, additionalDoc := range multidocs {
// if i == 0 {
// continue
// }
// additionalDoc, err := docrewrite.ConvertToV1Beta2([]byte(additionalDoc))
// if err != nil {
// return errors.Wrap(err, "failed to convert to v1beta2")
// }
// obj, _, err := decode(additionalDoc, nil, nil)
// if err != nil {
// return errors.Wrapf(err, "failed to parse additional doc %d", i)
// }
// multidocRedactors, ok := obj.(*troubleshootv1beta2.Redactor)
// if !ok {
// continue
// }
// additionalRedactors.Spec.Redactors = append(additionalRedactors.Spec.Redactors, multidocRedactors.Spec.Redactors...)
// }
var collectorCB func(chan interface{}, string)
progressChan := make(chan interface{}) // non-zero buffer can result in missed messages
@@ -193,7 +232,7 @@ func runTroubleshoot(v *viper.Viper, arg string) error {
c.Println(fmt.Sprintf("\r%s\r", cursor.ClearEntireLine()))
}
response, err := supportbundle.CollectSupportBundleFromSpec(&supportBundle.Spec, additionalRedactors, createOpts)
response, err := supportbundle.CollectSupportBundleFromSpec(&mainBundle.Spec, additionalRedactors, createOpts)
if err != nil {
return errors.Wrap(err, "failed to run collect and analyze process")
}
@@ -202,7 +241,7 @@ func runTroubleshoot(v *viper.Viper, arg string) error {
close(finishedCh) // this removes the spinner
isFinishedChClosed = true
if err := showInteractiveResults(supportBundle.Name, response.AnalyzerResults); err != nil {
if err := showInteractiveResults(mainBundle.Name, response.AnalyzerResults); err != nil {
interactive = false
}
} else {
@@ -211,7 +250,7 @@ func runTroubleshoot(v *viper.Viper, arg string) error {
}
if !response.FileUploaded {
if appName := supportBundle.Labels["applicationName"]; appName != "" {
if appName := mainBundle.Labels["applicationName"]; appName != "" {
f := `A support bundle for %s has been created in this directory
named %s. Please upload it on the Troubleshoot page of
the %s Admin Console to begin analysis.`

18
pkg/apis/troubleshoot/v1beta2/supportbundle_types.go
View File

@@ -48,6 +48,24 @@ type SupportBundle struct {
Status SupportBundleStatus `json:"status,omitempty"`
}
func (s *SupportBundle) concatSpec(bundle *SupportBundle) {
for _, v := range bundle.Spec.Collectors {
s.Spec.Collectors = append(bundle.Spec.Collectors, v)
}
for _, v := range bundle.Spec.AfterCollection {
s.Spec.AfterCollection = append(bundle.Spec.AfterCollection, v)
}
for _, v := range bundle.Spec.HostCollectors {
s.Spec.HostCollectors = append(bundle.Spec.HostCollectors, v)
}
for _, v := range bundle.Spec.HostAnalyzers {
s.Spec.HostAnalyzers = append(bundle.Spec.HostAnalyzers, v)
}
for _, v := range bundle.Spec.Analyzers {
s.Spec.Analyzers = append(bundle.Spec.Analyzers, v)
}
}
// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
// SupportBundleList contains a list of SupportBundle