style: 系统初始引导函数优化

This commit is contained in:
TsMask
2024-05-07 19:14:44 +08:00
parent 1708b12bd1
commit 9abfe9b30e
2 changed files with 31 additions and 28 deletions

View File

@@ -1,13 +1,9 @@
package controller
import (
"runtime"
"time"
adminConstants "be.ems/src/framework/constants/admin"
"be.ems/src/framework/constants/common"
tokenConstants "be.ems/src/framework/constants/token"
"be.ems/src/framework/utils/cmd"
"be.ems/src/framework/utils/ctx"
"be.ems/src/framework/utils/machine"
tokenUtils "be.ems/src/framework/utils/token"
@@ -97,12 +93,8 @@ func (s *BootloaderController) BootloaderDone(c *gin.Context) {
return
}
// 记录完成时间
err := machine.SetLaunchInfo(map[string]any{
common.LAUNCH_BOOTLOADER: false, // 启动引导
common.LAUNCH_BOOTLOADER + "Time": time.Now().UnixMilli(), // 引导完成时间
})
if err != nil {
// 标记引导完成
if err := machine.Bootloader(false); err != nil {
c.JSON(200, result.ErrMsg(err.Error()))
return
}
@@ -127,24 +119,7 @@ func (s *BootloaderController) BootloaderReset(c *gin.Context) {
return
}
// 重置数据库
if runtime.GOOS == "windows" {
c.JSON(200, result.ErrMsg("Does not support window operations"))
return
} else {
_, err := cmd.ExecWithCheck("sudo", "/usr/local/omc/bin/setomc.sh", "-m", "install")
if err != nil {
c.JSON(200, result.ErrMsg(err.Error()))
return
}
}
// 重置引导标记
err := machine.SetLaunchInfo(map[string]any{
"bootloader": true, // 启动引导
"bootloaderTime": 0, // 引导完成时间
})
if err != nil {
if err := machine.Reset(); err != nil {
c.JSON(200, result.ErrMsg(err.Error()))
return
}