Ecosystem/xmr-remote-nodes
1
0
Fork 0
mirror of https://github.com/ditatompel/xmr-remote-nodes.git synced 2024-11-16 17:07:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

refactor: Use proberRepo struct instead of interface
Some checks failed
Build / build (push) Has been cancelled
Details
Test / test (push) Has been cancelled
Details

Browse source 
o need to use interface when calling `NewProber()`.
This commit is contained in:
Christian Ditaputratama 2024-07-07 03:25:30 +07:00
parent 4d1a2da49c
commit 73308d2a32
No known key found for this signature in database
GPG key ID: 31D3D06D77950979
1 changed files with 8 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
internal/monero/prober.go
View file

@ -12,15 +12,7 @@ import (
const ProberAPIKey = "X-Prober-Api-Key" // HTTP header key
type ProberRepository interface {
Add(name string) (Prober, error)
Edit(id int, name string) error
Probers(QueryProbers) ([]Prober, error)
CheckAPI(key string) (Prober, error)
Delete(id int) error
}
type ProberRepo struct {
type proberRepo struct {
db *database.DB
}
@ -34,12 +26,12 @@ type Prober struct {
// Initializes a new ProberRepository
//
// NOTE: This "prober" is different with "probe" which is used to fetch a new job
func NewProber() ProberRepository {
return &ProberRepo{db: database.GetDB()}
func NewProber() *proberRepo {
return &proberRepo{db: database.GetDB()}
}
// Add a new prober machine
func (r *ProberRepo) Add(name string) (Prober, error) {
func (r *proberRepo) Add(name string) (Prober, error) {
apiKey := uuid.New()
query := `
INSERT INTO tbl_prober (
@ -59,7 +51,7 @@ func (r *ProberRepo) Add(name string) (Prober, error) {
}
// Edit an existing prober
func (r *ProberRepo) Edit(id int, name string) error {
func (r *proberRepo) Edit(id int, name string) error {
query := `UPDATE tbl_prober SET name = ? WHERE id = ?`
res, err := r.db.Exec(query, name, id)
if err != nil {
@ -76,7 +68,7 @@ func (r *ProberRepo) Edit(id int, name string) error {
}
// Delete an existing prober
func (r *ProberRepo) Delete(id int) error {
func (r *proberRepo) Delete(id int) error {
res, err := r.db.Exec(`DELETE FROM tbl_prober WHERE id = ?`, id)
if err != nil {
return err
@ -120,7 +112,7 @@ func (q QueryProbers) toSQL() (args []interface{}, where, sortBy, sortDirection
return args, where, sortBy, sortDirection
}
func (r *ProberRepo) Probers(q QueryProbers) ([]Prober, error) {
func (r *proberRepo) Probers(q QueryProbers) ([]Prober, error) {
args, where, sortBy, sortDirection := q.toSQL()
var probers []Prober
@ -153,7 +145,7 @@ func (r *ProberRepo) Probers(q QueryProbers) ([]Prober, error) {
return probers, nil
}
func (r *ProberRepo) CheckAPI(key string) (Prober, error) {
func (r *proberRepo) CheckAPI(key string) (Prober, error) {
var p Prober
query := `
SELECT