hanfuit
645b64fe21
服务端（Go）： - GM 塞背包/带属性装备发放：在线角色推送 op14 增量物品行，背包实时刷新，无需回角色选择界面 - GM 改金币/改等级：去掉在线拒绝，落库后推送 op14 钱包/op37 等级状态，在线实时刷新 - 等级端点统一切换到新版 GMSetCharacterLevel（自带在线刷新） - 无限/锁死疲劳模式 + 燃烧疲劳：每房间消耗 2 点、打怪与通关经验双倍 - 燃烧疲劳开关时给所有在线角色推送系统提示（op12 mode 0） - 深渊史诗掉率：GM 全局倍率 + 城镇增益药剂窗口倍率，真实接入掉落结算 启动器（Python/JS）： - 标题栏自绘、窗口 1204x717、禁止拖拽改大小、仅最小化/关闭 - GM 工具：疲劳与燃烧、深渊史诗掉率、拍卖行上架（单一定价对齐） - 拍卖行上架前端对齐服务端单一价格语义，修复 400 报错
104 lines
2.2 KiB
Go
104 lines
2.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
)
|
|
|
|
// WriteFileAtomically 通过同目录临时文件、fsync 和原子替换持久化写入单个文件。
|
|
func WriteFileAtomically(ctx context.Context, path string, data []byte, perm fs.FileMode) error {
|
|
ctx = contextOrBackground(ctx)
|
|
if err := ctx.Err(); err != nil {
|
|
return err
|
|
}
|
|
if path == "" {
|
|
return fmt.Errorf("atomic file path is required")
|
|
}
|
|
dir := filepath.Dir(path)
|
|
if err := os.MkdirAll(dir, 0o750); err != nil {
|
|
return err
|
|
}
|
|
tmp, err := os.CreateTemp(dir, "."+filepath.Base(path)+".*.tmp")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tmpName := tmp.Name()
|
|
keepTmp := true
|
|
defer func() {
|
|
if keepTmp {
|
|
_ = os.Remove(tmpName)
|
|
}
|
|
}()
|
|
if err := writeAndSyncTempFile(tmp, data, perm); err != nil {
|
|
_ = tmp.Close()
|
|
return err
|
|
}
|
|
if err := tmp.Close(); err != nil {
|
|
return err
|
|
}
|
|
if err := ctx.Err(); err != nil {
|
|
return err
|
|
}
|
|
if err := replaceFile(tmpName, path); err != nil {
|
|
return err
|
|
}
|
|
keepTmp = false
|
|
return syncDir(dir)
|
|
}
|
|
|
|
func writeAndSyncTempFile(file *os.File, data []byte, perm fs.FileMode) error {
|
|
written, err := file.Write(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if written != len(data) {
|
|
return io.ErrShortWrite
|
|
}
|
|
if err := file.Chmod(perm); err != nil {
|
|
return err
|
|
}
|
|
return file.Sync()
|
|
}
|
|
|
|
func replaceFile(tmpName, path string) error {
|
|
if err := os.Rename(tmpName, path); err == nil {
|
|
return nil
|
|
} else if _, statErr := os.Stat(path); statErr != nil {
|
|
return err
|
|
}
|
|
|
|
backupPath := tmpName + ".bak"
|
|
if err := os.Rename(path, backupPath); err != nil {
|
|
return err
|
|
}
|
|
if err := os.Rename(tmpName, path); err != nil {
|
|
restoreErr := os.Rename(backupPath, path)
|
|
return errors.Join(err, restoreErr)
|
|
}
|
|
if err := os.Remove(backupPath); err != nil && !errors.Is(err, os.ErrNotExist) {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func syncDir(dir string) error {
|
|
file, err := os.Open(dir) //nolint:gosec // G304：目录来自调用点限定的持久化路径。
|
|
if err != nil {
|
|
if runtime.GOOS == "windows" {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
defer func() { _ = file.Close() }()
|
|
if err := file.Sync(); err != nil && runtime.GOOS != "windows" {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|