qsql/api_page.go at master · gwaylib/qsql · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package qsql

import (
	"context"

	"github.com/gwaylib/errors"
)

type PageSql struct {
	countBD *SqlBuilder
	queryBD *SqlBuilder
}

func NewPageSql(countBD, queryBD *SqlBuilder) *PageSql {
	return &PageSql{
		countBD: countBD,
		queryBD: queryBD,
	}
}

func (p *PageSql) QueryCount(db *DB) (int64, error) {
	count := int64(0)
	if err := queryElem(db, context.TODO(), &count, p.countBD.String(), p.countBD.Args()...); err != nil {
		return 0, errors.As(err)
	}
	return count, nil
}

func (p *PageSql) QueryPageArr(db *DB) ([]string, [][]interface{}, error) {
	titles, data, err := queryPageArr(db, context.TODO(), p.queryBD.String(), p.queryBD.Args()...)
	if err != nil {
		return nil, nil, errors.As(err)
	}
	return titles, data, nil
}

func (p *PageSql) QueryPageMap(db *DB) ([]string, []map[string]interface{}, error) {
	titles, data, err := queryPageMap(db, context.TODO(), p.queryBD.String(), p.queryBD.Args()...)
	if err != nil {
		return nil, nil, errors.As(err)
	}
	return titles, data, nil
}