把工程根目录中和"配置加载"、"数据存储"两个领域相关的全部 Go 文件迁到 internal/ 下的子包,按功能分组的第一阶段成果。 internal/config/ - 从 config.go 抽出 Config / MQTTConfig / WebConfig / DatabaseConfig 等 类型并大写导出,函数改名 Default/Load/Write/Validate/BuildTLS 等。 - 测试随被测代码迁移成 package config 的内部测试。 - 根目录留 config.go 作桥接:用 type alias 让旧的小写名(config / mqttConfig / databaseConfig 等)继续可用,避免修改 30+ 处调用方。 internal/store/ - 把 db.go、store_query.go、db_write_queue.go 与 13 个 *_store.go 一并 迁入;26 个 *Record 类型与 store 同包以避免循环依赖。 - store -> Store;50+ 标识符从小写未导出改为大写导出(包括 record、 ListOptions、错误变量、bot/llm/runtime 常量、helpers 等)。 - 新增 DB() / Driver() 访问器供 ai 子系统使用,避免直接访问私有字段。 - bot_pki_store.go 独立出来,把 PKI 解密所需的 store 方法集中归类。 - helpers.go 提供 hashPassword / uint32FromRecord / printJSON 等以前在 其他根目录文件中的辅助;test_helpers_test.go 提供 verifyPassword 与 publicMapTileSourceDTO 让测试可以本地运行而不依赖 main 包。 根目录新增: - store_bridge.go:完整 type-alias / 函数包装层,把 internal/store 的 导出名映射回旧的小写名,让 admin_*_routes.go、web.go、bot_service.go 等仍未迁出的文件继续编译。后续步骤把它们迁到各自领域包后可逐步删除。 - test_helpers_test.go:根目录测试沿用 openTestStore 的入口。 go build ./... 与 go test ./... 全部通过;测试数量与重构前一致。 Co-Authored-By: Claude <noreply@anthropic.com>
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package store
|
|
|
|
import "golang.org/x/crypto/bcrypt"
|
|
|
|
// AdminRole 是管理员账号在用户表里的角色字符串。其它包通过这个常量与
|
|
// `users.role` 字段对齐,避免硬编码。
|
|
const AdminRole = "admin"
|
|
|
|
// printJSON 是 store 包内部的诊断输出钩子。当前实现为 noop——保持与
|
|
// 重构前 main.go 的行为一致;如需启用调试,可在调用方替换。
|
|
func printJSON(record map[string]any) {
|
|
_ = record
|
|
}
|
|
|
|
// hashPassword 与 auth.go 中的散列实现保持一致(bcrypt 默认 cost)。
|
|
func hashPassword(password string) (string, error) {
|
|
hash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(hash), nil
|
|
}
|
|
|
|
// uint32FromRecord 把 map[string]any 中的整型字段安全转换为 uint32。
|
|
func uint32FromRecord(value any) (uint32, bool) {
|
|
switch v := value.(type) {
|
|
case uint32:
|
|
return v, true
|
|
case int:
|
|
if v >= 0 {
|
|
return uint32(v), true
|
|
}
|
|
case int64:
|
|
if v >= 0 {
|
|
return uint32(v), true
|
|
}
|
|
case uint64:
|
|
return uint32(v), true
|
|
case float64:
|
|
if v >= 0 {
|
|
return uint32(v), true
|
|
}
|
|
}
|
|
return 0, false
|
|
}
|