152 lines
2.9 KiB
Go
152 lines
2.9 KiB
Go
/*
|
|
Copyright NetFoundry Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
https://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package stringz
|
|
|
|
import "fmt"
|
|
|
|
func EqualSlices(a []string, b []string) bool {
|
|
if len(a) != len(b) {
|
|
return false
|
|
}
|
|
for idx, aVal := range a {
|
|
if aVal != b[idx] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func Contains(a []string, val string) bool {
|
|
_, found := FirstIndexOf(a, val)
|
|
return found
|
|
}
|
|
|
|
func Remove(a []string, val string) []string {
|
|
var result []string
|
|
for _, next := range a {
|
|
if next != val {
|
|
result = append(result, next)
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
func ContainsAll(a []string, values ...string) bool {
|
|
for _, val := range values {
|
|
if !Contains(a, val) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func ContainsAny(a []string, values ...string) bool {
|
|
for _, val := range values {
|
|
if Contains(a, val) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func FirstIndexOf(a []string, val string) (int, bool) {
|
|
for idx, s := range a {
|
|
if s == val {
|
|
return idx, true
|
|
}
|
|
}
|
|
return -1, false
|
|
}
|
|
|
|
func Difference(a, b []string) []string {
|
|
mb := map[string]bool{}
|
|
for _, x := range b {
|
|
mb[x] = true
|
|
}
|
|
var ab []string
|
|
for _, x := range a {
|
|
if _, ok := mb[x]; !ok {
|
|
ab = append(ab, x)
|
|
}
|
|
}
|
|
return ab
|
|
}
|
|
|
|
func ToStringSlice(a [][]byte) []string {
|
|
result := make([]string, len(a))
|
|
for i := 0; i < len(a); i++ {
|
|
result[i] = string(a[i])
|
|
}
|
|
return result
|
|
}
|
|
|
|
func OrEmpty(val *string) string {
|
|
if val == nil {
|
|
return ""
|
|
}
|
|
return *val
|
|
}
|
|
|
|
func Permutations(v []string, f func([]string)) {
|
|
PermutationWith([]string{}, v, f)
|
|
}
|
|
|
|
func PermutationWith(base, v []string, f func([]string)) {
|
|
for i := 0; i < len(v); i++ {
|
|
var result []string
|
|
if len(base) > 0 {
|
|
result = append(result, base...)
|
|
}
|
|
result = append(result, v[i])
|
|
f(result)
|
|
PermutationWith(result, v[i+1:], f)
|
|
}
|
|
}
|
|
|
|
func SetToSlice(values map[string]struct{}) []string {
|
|
var result []string
|
|
for val := range values {
|
|
result = append(result, val)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func SliceToSet(values []string) map[string]struct{} {
|
|
result := map[string]struct{}{}
|
|
for _, val := range values {
|
|
result[val] = struct{}{}
|
|
}
|
|
return result
|
|
}
|
|
|
|
var units = []string{"B", "K", "M", "G", "T", "P"}
|
|
|
|
func FormatBytes(val uint64) string {
|
|
var i int
|
|
var target uint64
|
|
for i = range units {
|
|
target = 1 << uint(10*(i+1))
|
|
if val < target {
|
|
break
|
|
}
|
|
}
|
|
if i > 0 {
|
|
return fmt.Sprintf("%0.2f%s", float64(val)/(float64(target)/1024), units[i])
|
|
}
|
|
return fmt.Sprintf("%dB", val)
|
|
}
|