utils.go

package console_sqlite3

import (
	SQL "database/sql"
	"fmt"
)

func getCountSQL(db *SQL.DB,sql string, args ...interface{}) int {
	var count int
	countSQL := fmt.Sprintf("SELECT COUNT(*) FROM (%s) A", sql)
	err := db.QueryRow(countSQL, args...).Scan(&count)
	if err != nil {
		return 0
	}
	return count
}

func getPageSQL(db *SQL.DB,sql string, orderBy, orderType string, page, pageSize int, args ...interface{}) (*SQL.Rows, error) {
	pageSQL := fmt.Sprintf("%s ORDER BY %s %s LIMIT ?,?", sql, orderBy, orderType)
	args = append(args, (page-1)*pageSize, pageSize)
	return db.Query(pageSQL, args...)
}