mirror of
https://github.com/openziti/zrok.git
synced 2024-11-23 00:23:48 +01:00
94 lines
3.1 KiB
Go
94 lines
3.1 KiB
Go
package store
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type ShareLimitJournal struct {
|
|
Model
|
|
ShareId int
|
|
RxBytes int64
|
|
TxBytes int64
|
|
Action LimitAction
|
|
}
|
|
|
|
func (str *Store) CreateShareLimitJournal(j *ShareLimitJournal, trx *sqlx.Tx) (int, error) {
|
|
stmt, err := trx.Prepare("insert into share_limit_journal (share_id, rx_bytes, tx_bytes, action) values ($1, $2, $3, $4) returning id")
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "error preparing share_limit_journal insert statement")
|
|
}
|
|
var id int
|
|
if err := stmt.QueryRow(j.ShareId, j.RxBytes, j.TxBytes, j.Action).Scan(&id); err != nil {
|
|
return 0, errors.Wrap(err, "error executing share_limit_journal insert statement")
|
|
}
|
|
return id, nil
|
|
}
|
|
|
|
func (str *Store) IsShareLimitJournalEmpty(shrId int, trx *sqlx.Tx) (bool, error) {
|
|
count := 0
|
|
if err := trx.QueryRowx("select count(0) from share_limit_journal where share_id = $1", shrId).Scan(&count); err != nil {
|
|
return false, err
|
|
}
|
|
return count == 0, nil
|
|
}
|
|
|
|
func (str *Store) FindLatestShareLimitJournal(shrId int, trx *sqlx.Tx) (*ShareLimitJournal, error) {
|
|
j := &ShareLimitJournal{}
|
|
if err := trx.QueryRowx("select * from share_limit_journal where share_id = $1 order by created_at desc limit 1", shrId).StructScan(j); err != nil {
|
|
return nil, errors.Wrap(err, "error finding share_limit_journal by share_id")
|
|
}
|
|
return j, nil
|
|
}
|
|
|
|
func (str *Store) FindSelectedLatestShareLimitjournal(shrIds []int, trx *sqlx.Tx) ([]*ShareLimitJournal, error) {
|
|
if len(shrIds) < 1 {
|
|
return nil, nil
|
|
}
|
|
in := "("
|
|
for i := range shrIds {
|
|
if i > 0 {
|
|
in += ", "
|
|
}
|
|
in += fmt.Sprintf("%d", shrIds[i])
|
|
}
|
|
in += ")"
|
|
rows, err := trx.Queryx("select id, share_id, rx_bytes, tx_bytes, action, created_at, updated_at from share_limit_journal where id in (select max(id) as id from share_limit_journal group by share_id) and share_id in " + in)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error selecting all latest share_limit_journal")
|
|
}
|
|
var sljs []*ShareLimitJournal
|
|
for rows.Next() {
|
|
slj := &ShareLimitJournal{}
|
|
if err := rows.StructScan(slj); err != nil {
|
|
return nil, errors.Wrap(err, "error scanning share_limit_journal")
|
|
}
|
|
sljs = append(sljs, slj)
|
|
}
|
|
return sljs, nil
|
|
}
|
|
|
|
func (str *Store) FindAllLatestShareLimitJournal(trx *sqlx.Tx) ([]*ShareLimitJournal, error) {
|
|
rows, err := trx.Queryx("select id, share_id, rx_bytes, tx_bytes, action, created_at, updated_at from share_limit_journal where id in (select max(id) as id from share_limit_journal group by share_id)")
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error selecting all latest share_limit_journal")
|
|
}
|
|
var sljs []*ShareLimitJournal
|
|
for rows.Next() {
|
|
slj := &ShareLimitJournal{}
|
|
if err := rows.StructScan(slj); err != nil {
|
|
return nil, errors.Wrap(err, "error scanning share_limit_journal")
|
|
}
|
|
sljs = append(sljs, slj)
|
|
}
|
|
return sljs, nil
|
|
}
|
|
|
|
func (str *Store) DeleteShareLimitJournalForShare(shrId int, trx *sqlx.Tx) error {
|
|
if _, err := trx.Exec("delete from share_limit_journal where share_id = $1", shrId); err != nil {
|
|
return errors.Wrapf(err, "error deleting share_limit_journal for '#%d'", shrId)
|
|
}
|
|
return nil
|
|
}
|