EasyGBS后台操作日志功能实现方式分享

//插入记录
func InsertOperation(operation *models.Operation) error {
	return database.Create(operation).Error
}
//获取所有操作记录
func GetAllOperation() ([]models.Operation, error) {
	var operations []models.Operation
	err := database.Find(&operations).Error
	return operations, err
}

func GetOperationCount() int {
	var count int
	database.Model(models.Operation{}).Count(&count)
	return count
}
//根据条件获取记录
func GetOperations(req OperationReq) ([]models.Operation, error) {
	var operations []models.Operation
	table := database.Table(models.Operation{}.TableName())
	if req.Type != "" {
		table = table.Where("operation_type = ?", req.Type)
	}
	if req.Start != 0 {
		table = table.Offset(req.Start)
	}
	if req.StartTime != "" {
		start, err := time.ParseInLocation(etime.DateTimeLayout, req.StartTime, time.Local)
		if err != nil {
			return operations, errors.New("开始时间有误")
		}
		table = table.Where("operation_time >= ?", start)
	}
	if req.StartTime != "" {
		end, err := time.ParseInLocation(etime.DateTimeLayout, req.EndTime, time.Local)
		if err != nil {
			return operations, errors.New("结束时间有误")
		}
		table = table.Where("operation_time <= ?", end)
	}
	if req.Limit != 0 {
		table = table.Limit(req.Limit)
	}
	err := table.Find(&operations).Error
	return operations, err
}
//获取单条操作记录
func GetOneOperation(id int) (models.Operation, error) {
	var operation models.Operation
	err := database.Where("id = ?", id).Find(&operation).Error
	return operation, err
}
//删除单条记录
func DeleteOneOperation(id int) error {
	return database.Where("id = ?", id).Delete(models.Operation{}).Error
}
//清空记录表
func TruncateOperation() error {
	sec := utils.Conf().Section("database")
	dbName := sec.Key("db_name").MustString("sqlite3")
	switch dbName {
	case "sqlite3":
		return database.Delete(models.Operation{}).Error
	case "mysql":
		return database.Raw("truncate table ?", models.Operation{}.TableName()).Error
	}
	return errors.New("error database engine")
}