增加批量导入资产功能

pkg/model/asset.go

@@ -9,8 +9,8 @@ import (
 type Asset struct {
 	ID           string         `gorm:"primary_key " json:"id"`
 	Name         string         `json:"name"`
-	IP           string         `json:"ip"`
 	Protocol     string         `json:"protocol"`
+	IP           string         `json:"ip"`
 	Port         int            `json:"port"`
 	AccountType  string         `json:"accountType"`
 	Username     string         `json:"username"`

pkg/model/login-log.go

@@ -81,7 +81,7 @@ func FindLoginLogById(id string) (o LoginLog, err error) {
 	return
 }
 
-func Logout(token string) {
+func Logout(token string) (err error) {
 
 	loginLog, err := FindLoginLogById(token)
 	if err != nil {
@@ -89,7 +89,10 @@ func Logout(token string) {
 		return
 	}
 
-	global.DB.Table("login_logs").Where("id = ?", token).Update("logout_time", utils.NowJsonTime())
+	err = global.DB.Updates(&LoginLog{LogoutTime: utils.NowJsonTime(), ID: token}).Error
+	if err != nil {
+		return err
+	}
 
 	loginLogs, err := FindAliveLoginLogsByUserId(loginLog.UserId)
 	if err != nil {
@@ -97,6 +100,7 @@ func Logout(token string) {
 	}
 
 	if len(loginLogs) == 0 {
-		UpdateUserById(&User{Online: false}, loginLog.UserId)
+		err = UpdateUserOnline(false, loginLog.UserId)
 	}
+	return
 }

pkg/model/user.go

@@ -129,6 +129,17 @@ func UpdateUserById(o *User, id string) {
 	global.DB.Updates(o)
 }
 
+func UpdateUserOnline(online bool, id string) (err error) {
+	sql := "update users set online = ? where id = ?"
+	err = global.DB.Exec(sql, online, id).Error
+	return
+}
+
+func FindOnlineUsers() (o []User, err error) {
+	err = global.DB.Where("online = ?", true).Find(&o).Error
+	return
+}
+
 func DeleteUserById(id string) {
 	global.DB.Where("id = ?", id).Delete(&User{})
 	// 删除用户组中的用户关系