mirror of
https://github.com/openziti/zrok.git
synced 2025-02-16 18:20:51 +01:00
store updates (#10)
This commit is contained in:
parent
abad64c808
commit
e63e831374
@ -28,6 +28,14 @@ func (self *Store) CreateAccount(a *Account, tx *sqlx.Tx) (int, error) {
|
|||||||
return int(id), nil
|
return int(id), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (self *Store) GetAccount(id int, tx *sqlx.Tx) (*Account, error) {
|
||||||
|
a := &Account{}
|
||||||
|
if err := tx.QueryRowx("select * from accounts where id = ?", id).StructScan(a); err != nil {
|
||||||
|
return nil, errors.Wrap(err, "error selecting account by id")
|
||||||
|
}
|
||||||
|
return a, nil
|
||||||
|
}
|
||||||
|
|
||||||
func (self *Store) FindAccountWithToken(token string, tx *sqlx.Tx) (*Account, error) {
|
func (self *Store) FindAccountWithToken(token string, tx *sqlx.Tx) (*Account, error) {
|
||||||
a := &Account{}
|
a := &Account{}
|
||||||
if err := tx.QueryRowx("select * from accounts where token = ?", token).StructScan(a); err != nil {
|
if err := tx.QueryRowx("select * from accounts where token = ?", token).StructScan(a); err != nil {
|
||||||
|
64
controller/store/service.go
Normal file
64
controller/store/service.go
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
package store
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/jmoiron/sqlx"
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Service struct {
|
||||||
|
Model
|
||||||
|
AccountId int
|
||||||
|
ZitiId string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Store) CreateService(accountId int, svc *Service, tx *sqlx.Tx) (int, error) {
|
||||||
|
stmt, err := tx.Prepare("insert into services (account_id, ziti_id) values (?, ?)")
|
||||||
|
if err != nil {
|
||||||
|
return 0, errors.Wrap(err, "error preparing services insert statement")
|
||||||
|
}
|
||||||
|
res, err := stmt.Exec(accountId, svc.ZitiId)
|
||||||
|
if err != nil {
|
||||||
|
return 0, errors.Wrap(err, "error executing services insert statement")
|
||||||
|
}
|
||||||
|
id, err := res.LastInsertId()
|
||||||
|
if err != nil {
|
||||||
|
return 0, errors.Wrap(err, "error retrieving last services insert id")
|
||||||
|
}
|
||||||
|
return int(id), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Store) GetService(id int, tx *sqlx.Tx) (*Service, error) {
|
||||||
|
svc := &Service{}
|
||||||
|
if err := tx.QueryRowx("select * from services where id = ?", id).StructScan(svc); err != nil {
|
||||||
|
return nil, errors.Wrap(err, "error selecting service by id")
|
||||||
|
}
|
||||||
|
return svc, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Store) FindServicesForAccount(accountId int, tx *sqlx.Tx) ([]*Service, error) {
|
||||||
|
rows, err := tx.Queryx("select services.* from services where account_id = ?", accountId)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "error selecting services by account id")
|
||||||
|
}
|
||||||
|
var svcs []*Service
|
||||||
|
for rows.Next() {
|
||||||
|
svc := &Service{}
|
||||||
|
if err := rows.StructScan(svc); err != nil {
|
||||||
|
return nil, errors.Wrap(err, "error scanning service")
|
||||||
|
}
|
||||||
|
svcs = append(svcs, svc)
|
||||||
|
}
|
||||||
|
return svcs, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Store) DeleteService(id int, tx *sqlx.Tx) error {
|
||||||
|
stmt, err := tx.Prepare("delete from services where id = ?")
|
||||||
|
if err != nil {
|
||||||
|
return errors.Wrap(err, "error preparing services delete statement")
|
||||||
|
}
|
||||||
|
_, err = stmt.Exec(id)
|
||||||
|
if err != nil {
|
||||||
|
return errors.Wrap(err, "error executing services delete statement")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
@ -21,9 +21,10 @@ create table accounts (
|
|||||||
--
|
--
|
||||||
create table services (
|
create table services (
|
||||||
id integer primary key,
|
id integer primary key,
|
||||||
name string not null unique,
|
account_id integer constraint fk_accounts_service references accounts on delete cascade,
|
||||||
|
ziti_id string not null unique,
|
||||||
created_at datetime not null default(strftime('%Y-%m-%d %H:%M:%f', 'now')),
|
created_at datetime not null default(strftime('%Y-%m-%d %H:%M:%f', 'now')),
|
||||||
updated_at datetime not null default(strftime('%Y-%m-%d %H:%M:%f', 'now')),
|
updated_at datetime not null default(strftime('%Y-%m-%d %H:%M:%f', 'now')),
|
||||||
|
|
||||||
constraint chk_name check (name <> '')
|
constraint chk_name check (ziti_id <> '')
|
||||||
);
|
);
|
Loading…
Reference in New Issue
Block a user