You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package lo
|
|
|
|
import (
|
|
"unicode/utf8"
|
|
)
|
|
|
|
// Substring return part of a string.
|
|
// Play: https://go.dev/play/p/TQlxQi82Lu1
|
|
func Substring[T ~string](str T, offset int, length uint) T {
|
|
size := len(str)
|
|
|
|
if offset < 0 {
|
|
offset = size + offset
|
|
if offset < 0 {
|
|
offset = 0
|
|
}
|
|
}
|
|
|
|
if offset > size {
|
|
return Empty[T]()
|
|
}
|
|
|
|
if length > uint(size)-uint(offset) {
|
|
length = uint(size - offset)
|
|
}
|
|
|
|
return str[offset : offset+int(length)]
|
|
}
|
|
|
|
// ChunkString returns an array of strings split into groups the length of size. If array can't be split evenly,
|
|
// the final chunk will be the remaining elements.
|
|
// Play: https://go.dev/play/p/__FLTuJVz54
|
|
func ChunkString[T ~string](str T, size int) []T {
|
|
if size <= 0 {
|
|
panic("lo.ChunkString: Size parameter must be greater than 0")
|
|
}
|
|
|
|
if len(str) == 0 {
|
|
return []T{""}
|
|
}
|
|
|
|
if size >= len(str) {
|
|
return []T{str}
|
|
}
|
|
|
|
var chunks []T = make([]T, 0, ((len(str)-1)/size)+1)
|
|
currentLen := 0
|
|
currentStart := 0
|
|
for i := range str {
|
|
if currentLen == size {
|
|
chunks = append(chunks, str[currentStart:i])
|
|
currentLen = 0
|
|
currentStart = i
|
|
}
|
|
currentLen++
|
|
}
|
|
chunks = append(chunks, str[currentStart:])
|
|
return chunks
|
|
}
|
|
|
|
// RuneLength is an alias to utf8.RuneCountInString which returns the number of runes in string.
|
|
// Play: https://go.dev/play/p/tuhgW_lWY8l
|
|
func RuneLength(str string) int {
|
|
return utf8.RuneCountInString(str)
|
|
}
|