126 lines
2.5 KiB
Go
126 lines
2.5 KiB
Go
|
package commonservice
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"crypto/rand"
|
|||
|
|
"encoding/hex"
|
|||
|
|
"fmt"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"nl-pms-api/internal/config"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// AdminUserID 是 code_count 体系的超级管理员账号（users.id=1）。
|
|||
|
|
const AdminUserID int64 = 1
|
|||
|
|
|
|||
|
|
// ParseID 把字符串解析为非负 int64；非法或负数视为 0。
|
|||
|
|
func ParseID(s string) int64 {
|
|||
|
|
n, _ := strconv.ParseInt(s, 10, 64)
|
|||
|
|
if n < 0 {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
return n
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Clip 按字节截断字符串到最多 n 字节。
|
|||
|
|
func Clip(s string, n int) string {
|
|||
|
|
if len(s) > n {
|
|||
|
|
return s[:n]
|
|||
|
|
}
|
|||
|
|
return s
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ClipRunes 按 rune 截断到最多 n 个字符。
|
|||
|
|
func ClipRunes(s string, n int) string {
|
|||
|
|
r := []rune(strings.TrimSpace(s))
|
|||
|
|
if len(r) > n {
|
|||
|
|
return string(r[:n])
|
|||
|
|
}
|
|||
|
|
return string(r)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NowRFC 返回当前 UTC 的 RFC3339 时间串（与 view 库惯例一致）。
|
|||
|
|
func NowRFC() string {
|
|||
|
|
return time.Now().UTC().Format(time.RFC3339)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// PublicURL 拼接文件公开访问地址：优先配置的 base_url，否则用 requestHost（可含 scheme://host）。
|
|||
|
|
func PublicURL(cfg *config.Config, requestHost, name string) string {
|
|||
|
|
base := cfg.BaseURL
|
|||
|
|
if base == "" {
|
|||
|
|
base = strings.TrimRight(requestHost, "/")
|
|||
|
|
if !strings.Contains(base, "://") {
|
|||
|
|
base = "http://" + base
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return base + "/files/" + name
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NormalizeKind 规范化文件用途：avatar | content，其余为空。
|
|||
|
|
func NormalizeKind(k string) string {
|
|||
|
|
if k == "avatar" || k == "content" {
|
|||
|
|
return k
|
|||
|
|
}
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// StoredName 生成不可枚举的存储相对路径（日期目录 + 128 位随机 hex）。
|
|||
|
|
func StoredName(ext string) string {
|
|||
|
|
b := make([]byte, 16)
|
|||
|
|
_, _ = rand.Read(b)
|
|||
|
|
return time.Now().UTC().Format("2006/01/02") + "/" + hex.EncodeToString(b) + ext
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MapStr 从通用 JSON map 取字符串字段。
|
|||
|
|
func MapStr(m map[string]any, key string) string {
|
|||
|
|
v, ok := m[key]
|
|||
|
|
if !ok || v == nil {
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
switch x := v.(type) {
|
|||
|
|
case string:
|
|||
|
|
return x
|
|||
|
|
case []byte:
|
|||
|
|
return string(x)
|
|||
|
|
case float64:
|
|||
|
|
if x == float64(int64(x)) {
|
|||
|
|
return strconv.FormatInt(int64(x), 10)
|
|||
|
|
}
|
|||
|
|
return strconv.FormatFloat(x, 'f', -1, 64)
|
|||
|
|
case bool:
|
|||
|
|
if x {
|
|||
|
|
return "1"
|
|||
|
|
}
|
|||
|
|
return "0"
|
|||
|
|
default:
|
|||
|
|
return fmt.Sprint(x)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MapInt64 从通用 JSON map 取 int64 字段。
|
|||
|
|
func MapInt64(m map[string]any, key string) int64 {
|
|||
|
|
v, ok := m[key]
|
|||
|
|
if !ok || v == nil {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
switch x := v.(type) {
|
|||
|
|
case float64:
|
|||
|
|
return int64(x)
|
|||
|
|
case int64:
|
|||
|
|
return x
|
|||
|
|
case int:
|
|||
|
|
return int64(x)
|
|||
|
|
case string:
|
|||
|
|
n, _ := strconv.ParseInt(x, 10, 64)
|
|||
|
|
return n
|
|||
|
|
case bool:
|
|||
|
|
if x {
|
|||
|
|
return 1
|
|||
|
|
}
|
|||
|
|
return 0
|
|||
|
|
default:
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
}
|