bigdata_archives/unity/unity.go

package unity

import (
	"bigdata_archives/global"
	"errors"
)

type PageParams struct {
	Page int    `json:"page" form:"page"`
	Size int    `json:"size" form:"size"`
	Desc string `json:"desc" form:"desc"`
}
type QueryPageParams struct {
	Page  int    `json:"page" form:"page"`
	Size  int    `json:"size" form:"size"`
	Query string `json:"query" form:"query"`
	Desc  string `json:"desc" form:"desc"`
}
type TimePageParams struct {
	Page      int    `json:"page" form:"page"`
	Size      int    `json:"size" form:"size"`
	StartTime string `json:"startTime" form:"startTime"`
	EndTime   string `json:"endTime" form:"endTime"`
	Desc      string `json:"desc" form:"desc"`
}

// Paginate 使用给定的DB连接执行分页查询
func Paginate[T any](params PageParams, model T) (result []T, total int64, current int, err error) {
	var count int64
	if err = global.DBLink.Model(model).Count(&count).Error; err != nil {
		return nil, 0, 0, err
	}

	// 计算总页数
	//	totalPages := int(math.Ceil(float64(count) / float64(params.Size)))
	// 计算偏移量并设置分页大小
	offset := (params.Page - 1) * params.Size
	// 执行实际分页查询
	if err = global.DBLink.Offset(offset).Limit(params.Size).Order(params.Desc).Find(&result).Error; err != nil {
		return nil, 0, 0, err
	}
	return result, count, params.Page, nil
}

func PaginateWithCondition[T any](params QueryPageParams, model T, queryCond interface{}) (result []T, total int64, err error) {
	var count int64
	if queryCond != nil {
		if err = global.DBLink.Model(model).Where(queryCond, params.Query).Count(&count).Error; err != nil {
			return nil, 0, err
		}
		// 计算查询的偏移量，并设置每次查询的记录数量
		offset := (params.Page - 1) * params.Size
		// 执行分页查询，包括偏移量设置、限制查询数量、排序及应用额外查询条件
		if err = global.DBLink.Offset(offset).Limit(params.Size).Order(params.Desc).Where(queryCond, params.Query).Find(&result).Error; err != nil {
			return nil, 0, err
		}
	} else {
		if err = global.DBLink.Model(model).Count(&count).Error; err != nil {
			return nil, 0, err
		}
		// 计算查询的偏移量，并设置每次查询的记录数量
		offset := (params.Page - 1) * params.Size
		// 执行分页查询，包括偏移量设置、限制查询数量、排序及应用额外查询条件
		if err = global.DBLink.Offset(offset).Limit(params.Size).Order(params.Desc).Find(&result).Error; err != nil {
			return nil, 0, err
		}
	}
	return result, count, nil
}

// PaginateWithTime 给定时间范围查询
func PaginateWithTime[T any](params TimePageParams, tableName string, model T) (result []T, total int64, err error) {
	table := global.DBLink.Table(tableName)

	// 确保 total 初始化为 0
	total = 0

	if params.StartTime != "" && params.EndTime != "" {
		table = table.Where("created_at BETWEEN ? AND ?", params.StartTime, params.EndTime)
	}

	if err = table.Count(&total).Error; err != nil {
		return nil, 0, err
	}

	offset := (params.Page - 1) * params.Size

	if err := table.Offset(offset).Limit(params.Size).Order(params.Desc).Find(&result).Error; err != nil {
		return nil, 0, err
	}

	return result, total, nil
}

func AddGenericItem[T any](item T, mapper func(T) interface{}) error {
	dbModel := mapper(item)
	if err := global.DBLink.Create(dbModel).Error; err != nil {
		return err
	}
	return nil
}

// GetById 根据id查询
func GetById[T any](id int, model T) (T, error) {
	tx := global.DBLink.Where("id = ?", id).First(&model)
	if tx.RowsAffected <= 0 {
		return model, errors.New("查询失败,当前id不存在")
	} else {
		return model, nil
	}
}

// UpdateById 根据id进行修改
func UpdateById[T any](id uint, model T) (T, error) {
	tx := global.DBLink.Where("id = ?", id).Updates(&model)
	if tx.RowsAffected > 0 {
		return model, nil
	} else {
		return model, errors.New("修改失败当前id不存在")
	}
}

// Add 添加
func Add[T any](model T) (T, error) {
	tx := global.DBLink.Create(&model)
	if tx.RowsAffected > 0 {
		return model, nil
	} else {
		return model, errors.New("添加失败")
	}
}

// DeleteById 根据id删除
func DeleteById[T any](id uint, model T) (T, error) {
	tx := global.DBLink.Where("id = ?", id).Delete(&model)
	if tx.RowsAffected > 0 {
		return model, nil
	} else {
		return model, errors.New("删除失败当前id不存在")
	}
}

// QueryAll 获取所有信息
func QueryAll(table string) ([]any, error) {
	var result []any
	tx := global.DBLink.Table(table).Find(&result)
	if tx.Error != nil {
		return result, errors.New("查询失败")
	}
	return result, nil
}