mirror of
https://github.com/gohouse/gorose.git
synced 2024-05-03 08:45:07 +08:00
68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
package gorose
|
||
|
||
import (
|
||
"database/sql"
|
||
"fmt"
|
||
"github.com/gohouse/gorose/v3/builder"
|
||
"math/rand"
|
||
"regexp"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
func init() {
|
||
rand.New(rand.NewSource(time.Now().UnixNano()))
|
||
}
|
||
|
||
func As(table any, alias string) builder.TableClause {
|
||
return builder.TableClause{
|
||
Tables: table,
|
||
Alias: alias,
|
||
}
|
||
}
|
||
func GetRandomInt(num int) int {
|
||
return rand.Intn(num)
|
||
}
|
||
func GetRandomWeightedIndex(weights []int) int {
|
||
if len(weights) == 0 {
|
||
return 0
|
||
}
|
||
if len(weights) == 1 {
|
||
return 0
|
||
}
|
||
totalWeight := 0
|
||
for _, w := range weights {
|
||
totalWeight += w
|
||
}
|
||
if totalWeight == 0 {
|
||
return rand.Intn(len(weights))
|
||
}
|
||
|
||
rnd := rand.Intn(totalWeight)
|
||
|
||
currentWeight := 0
|
||
for i, w := range weights {
|
||
currentWeight += w
|
||
if rnd < currentWeight {
|
||
return i
|
||
}
|
||
}
|
||
return -1 // 如果权重都为 0，或者总权重为 0，则返回 -1
|
||
}
|
||
|
||
//////////// struct field ptr 4 orm helpers ////////////
|
||
|
||
func Ptr[T any](arg T) *T {
|
||
return &arg
|
||
}
|
||
|
||
//////////// sql.Null* type helpers ////////////
|
||
|
||
func Null[T any](arg T) sql.Null[T] {
|
||
return sql.Null[T]{V: arg, Valid: true}
|
||
}
|
||
|
||
func NamedSprintf(format string, a ...any) string {
|
||
return strings.TrimSpace(regexp.MustCompile(`\s{2,}`).ReplaceAllString(fmt.Sprintf(regexp.MustCompile(`:\w+`).ReplaceAllString(format, "%s"), a...), " "))
|
||
}
|