kairos-agent/pkg/utils/strings.go
2022-08-18 13:14:12 +00:00

42 lines
703 B
Go

package utils
import (
"encoding/json"
"fmt"
"math/rand"
"strings"
"time"
"gopkg.in/yaml.v1"
)
func init() {
rand.Seed(time.Now().UnixNano())
}
var letterRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
func RandStringRunes(n int) string {
b := make([]rune, n)
for i := range b {
b[i] = letterRunes[rand.Intn(len(letterRunes))]
}
return string(b)
}
func ListOutput(rels []string, output string) []string {
switch strings.ToLower(output) {
case "yaml":
d, _ := yaml.Marshal(rels)
return []string{string(d)}
case "json":
d, _ := json.Marshal(rels)
return []string{string(d)}
default:
for _, r := range rels {
fmt.Println(r)
}
return rels
}
}