package repo

import (
	"fmt"

	"github.com/zhilv666/go-gin-scaffold/pkg/config"
	"gorm.io/gorm"
)

func columnName(name string) string {
	if config.GetConfig().Database.Driver == "postgres" {
		return fmt.Sprintf(`"%s"`, name)
	}
	return fmt.Sprintf("`%s`", name)
}

func addStorageOrder(db *gorm.DB) *gorm.DB {
	return DB().Order(fmt.Sprintf("%s, %s", columnName("order"), columnName("id")))
}