Подготовить HY2XS к production-сборке

apps/dao/account.go

@@ -0,0 +1,119 @@
+package dao
+
+import (
+	"errors"
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
+	"hy2xs-admin/model/constant"
+	"hy2xs-admin/model/dto"
+	"hy2xs-admin/model/entity"
+	"time"
+)
+
+func SaveAccount(account entity.Account) (int64, error) {
+	if tx := sqliteDB.Save(&account); tx.Error != nil {
+		logrus.Errorf("%v", tx.Error)
+		return 0, errors.New(constant.SysError)
+	}
+	return *account.Id, nil
+}
+
+func DeleteAccount(ids []int64) error {
+	if tx := sqliteDB.Where("id in ?", ids).Delete(&entity.Account{}); tx.Error != nil {
+		logrus.Errorf("%v", tx.Error)
+		return errors.New(constant.SysError)
+	}
+	return nil
+}
+
+func UpdateAccount(ids []int64, updates map[string]interface{}) error {
+	if len(updates) > 0 {
+		updates["update_time"] = time.Now().Format("2006-01-02 15:04:05")
+		if tx := sqliteDB.Model(&entity.Account{}).
+			Where("id in ?", ids).
+			Updates(updates); tx.Error != nil {
+			logrus.Errorf("%v", tx.Error)
+			return errors.New(constant.SysError)
+		}
+	}
+	return nil
+}
+
+func UpsertAccount(accounts []entity.Account) error {
+	if tx := sqliteDB.Model(&entity.Account{}).Clauses(clause.OnConflict{
+		Columns:   []clause.Column{{Name: "username"}},
+		DoUpdates: clause.AssignmentColumns([]string{"pass", "con_pass", "quota", "download", "upload", "expire_time", "kick_util_time", "device_no", "role", "deleted", "create_time", "update_time", "login_at", "con_at", "remark"}),
+	}).Create(accounts); tx.Error != nil {
+		logrus.Errorf("%v", tx.Error)
+		return errors.New(constant.SysError)
+	}
+	return nil
+}
+
+func UpdateAccountTraffic(username string, download int64, upload int64) error {
+	if upload != 0 || download != 0 {
+		updates := map[string]interface{}{}
+		if download != 0 {
+			updates["download"] = gorm.Expr("download + ?", download)
+		}
+		if upload != 0 {
+			updates["upload"] = gorm.Expr("upload + ?", upload)
+		}
+		updates["update_time"] = time.Now().Format("2006-01-02 15:04:05")
+		if tx := sqliteDB.Model(&entity.Account{}).
+			Where("username = ?", username).
+			Updates(updates); tx.Error != nil {
+			logrus.Errorf("%v", tx.Error)
+			return errors.New(constant.SysError)
+		}
+	}
+	return nil
+}
+
+func GetAccount(query interface{}, args ...interface{}) (entity.Account, error) {
+	var account entity.Account
+	if tx := sqliteDB.Model(&entity.Account{}).
+		Where(query, args...).First(&account); tx.Error != nil {
+		if tx.Error == gorm.ErrRecordNotFound {
+			return account, errors.New(constant.WrongPassword)
+		}
+		logrus.Errorf("%v", tx.Error)
+		return account, errors.New(constant.SysError)
+	}
+	return account, nil
+}
+
+func PageAccount(accountPageDto dto.AccountPageDto) ([]entity.Account, int64, error) {
+	var accounts []entity.Account
+	var total int64
+	tx := sqliteDB.Model(&entity.Account{})
+	if accountPageDto.Username != nil && *accountPageDto.Username != "" {
+		tx.Where("username like ?", fmt.Sprintf("%%%s%%", *accountPageDto.Username))
+	}
+	if accountPageDto.Deleted != nil {
+		tx.Where("deleted = ?", *accountPageDto.Deleted)
+	}
+	if accountPageDto.Remark != nil && *accountPageDto.Remark != "" {
+		tx.Where("remark like ?", fmt.Sprintf("%%%s%%", *accountPageDto.Remark))
+	}
+	tx.Count(&total)
+	if tx.Scopes(Paginate(accountPageDto.PageNum, accountPageDto.PageSize)).
+		Order("role,create_time desc").
+		Find(&accounts); tx.Error != nil {
+		logrus.Errorf("%v", tx.Error)
+		return accounts, 0, errors.New(constant.SysError)
+	}
+	return accounts, total, nil
+}
+
+func ListAccount(query interface{}, args ...interface{}) ([]entity.Account, error) {
+	var accounts []entity.Account
+	if tx := sqliteDB.Model(&entity.Account{}).
+		Where(query, args...).Order("role,create_time desc").Find(&accounts); tx.Error != nil {
+		logrus.Errorf("%v", tx.Error)
+		return accounts, errors.New(constant.SysError)
+	}
+	return accounts, nil
+}