From 91357afb1ff764395ad4ff3060521673535d7779 Mon Sep 17 00:00:00 2001
From: bel <bel>
Date: Sat, 14 Dec 2024 21:32:39 -0700
Subject: [PATCH] locking

---
 cmd/server/.db.go.swp | Bin 0 -> 12288 bytes
 cmd/server/db.go      |  21 +++++++++++++++++++++
 2 files changed, 21 insertions(+)
 create mode 100644 cmd/server/.db.go.swp

diff --git a/cmd/server/.db.go.swp b/cmd/server/.db.go.swp
new file mode 100644
index 0000000000000000000000000000000000000000..2bd5208c853cb238788864adb2c30b4de6066cc0
GIT binary patch
literal 12288
zcmeI2UuYaf9LJ{x|C}~e6n%Ai+Hx1KyGebpBvjfY*AmR1Ho4k>Vmh}sxh%cgz0A%v
zxs>QbUkZX~X;Baq!9pK=Q#7;>im0eqK}Bg56vT%r_@Ylf)$eb2?_ch0YD<N}9DK4j
zvopW>{l4?t-y~%7eNT;!^PN^d!{=VcR-b;hxXN!&T;9%DUY4kL-Ose6gm&iF9rqjz
z+WUMvP>x-6wG_5r)wWYE*nw0>rLqfotLU$5#tbkD7zM6NfgbkQj{ZKH`fzd!zyH32
z*JZ^_G71<4i~>dhqkvJsC}0#Y3K#`$O$Bsi1A88>Z;ahP7GF24xyDU%F$x$3i~>dh
zqkvJsC}0#Y3K#{90!9I&fKlKUR6rCMJ8>IhFWmv>@&Et&-~WHz!`PqTH}Dhq3Va4W
z1gqeE@CrB$PJ!b<0|_R<I2Z#7@a<;CJ_VnEbKop^6|8_|Py@4I0t|wkU@Le8T<Kx#
z2k<5M0GtD-!3l62JOf6-<6sBa1peN{*e~E4Z~>eLXTcfp7B~e?f|o%a<iHc4AK1VG
zm+xlm68H$b3Elv&gBh?J*dPfW0-He(`1LNvE`smDcR&LPCO`^o16#qz3C7+AuYniA
z3t$=4z>{DAYzMvILGS?B2!6hkv9G}e@CEoBya(O|?|>RO0CoclJPNL0U;hBVgCD^q
za1opVCjssIGB6*bfKk9GU=+C73aqe%lDeuGeW^D*o=j%A>suWi*~z_ZuIf2FRmk&^
zp)}to^?sprE#($?pq1+t(>xV@$#AKZuhJZ+c+-%R;htM!3G_rOd(;lcvV=QFbNIk6
z?pXU_YuNX+T+%5=FQo^=TU&O~aG1QRf#xVI7h4P#O4ejWdhpx|oOxN68IG@x?|JF=
zm<p~a(UKhZXn8Ak1CjNTm64(Dp(%ecu(C>}()dk-;v?!o!V9!o0?emqd`DPwzT%;w
z2?{(EFHw+lGVUEN`GLd&t>K5JW6RCR$w(u3$2w4zs@9@shA(8eP{qKqEb=#6!?CV?
zfu^*Ov98$~DHmwfZdzuzljoF4k)%^3MU*daEY76JKMn_lSBvkqCleNKtUwmqgNkFq
zAhl>s9rQ}RbA+UeJI11BTRMj--8_H8otlkyg>pQ*B%S|#Pb3!Y9^!uzzHa7*H75~n
z)E_4wZfy9RMBBS|ySwf7Ay?0b{q<aolk8t|E!?;G(GH8IhZ1;fCF)Kb?s$|F=|L(0
z6;UhsqQG<6nH;sXTb6+q<q99!lgs8t$FmvgE{d|8tyWP-X&`kCPh~NH=k^SZW%=k{
zJ~5f&*_qMl+%%U*rKba)!l-DG_Ss~wir1L0*<Ph2YErctY<O~YM`+#FX&%~*NE<fR
ziH2Maji89dPX~?wDzzOC$l*q*qTAhyRm+xi)I`i_ltiRSvuMTjba8;wXIoJ79Bb;(
zcopRz8w?3>6nJ!#uGEfBkO3Z_Bve%B*uvrs2OZ&2QsN*%X%b#|s>KHiC0-6`fH%%5
z^D~@s!kU(_Ug+)bi;L)lTpStd{$$h-Fmk7P?sgtbi(x#yt?|Gkn#AL2EDojK)U}F4
z7}Tsfn#wI=nB_^lB0pcH<FafQOENE%T(%)6UYt)N;v@x-#FbW}tLxLxg3ux_0!bsO
bjg{#@O)|G!@s&pYvWjqyh@#|W;d<;JfeZDM

literal 0
HcmV?d00001

diff --git a/cmd/server/db.go b/cmd/server/db.go
index bd8f05a..c8f60fa 100644
--- a/cmd/server/db.go
+++ b/cmd/server/db.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"database/sql"
 	"io"
+	"sync"
 	"time"
 
 	_ "github.com/glebarez/sqlite"
@@ -12,6 +13,7 @@ import (
 type DB struct {
 	scheme string
 	conn   string
+	rw     *sync.RWMutex
 }
 
 func NewDB(ctx context.Context, scheme, conn string) (DB, error) {
@@ -21,6 +23,7 @@ func NewDB(ctx context.Context, scheme, conn string) (DB, error) {
 	db := DB{
 		scheme: scheme,
 		conn:   conn,
+		rw:     &sync.RWMutex{},
 	}
 
 	sql, err := db.dial(ctx)
@@ -53,7 +56,19 @@ func NewDB(ctx context.Context, scheme, conn string) (DB, error) {
 	return db, err
 }
 
+func (db DB) WithLock(cb func() error) error {
+	db.rw.Lock()
+	defer db.rw.Unlock()
+	return cb()
+}
+
 func (db DB) Exec(ctx context.Context, q string, args ...any) error {
+	db.rw.RLock()
+	defer db.rw.RUnlock()
+	return db.exec(ctx, q, args...)
+}
+
+func (db DB) exec(ctx context.Context, q string, args ...any) error {
 	c, err := db.dial(ctx)
 	if err != nil {
 		return err
@@ -65,6 +80,12 @@ func (db DB) Exec(ctx context.Context, q string, args ...any) error {
 }
 
 func (db DB) Query(ctx context.Context, cb func(*sql.Rows) error, q string, args ...any) error {
+	db.rw.RLock()
+	defer db.rw.RUnlock()
+	return db.query(ctx, cb, q, args...)
+}
+
+func (db DB) query(ctx context.Context, cb func(*sql.Rows) error, q string, args ...any) error {
 	c, err := db.dial(ctx)
 	if err != nil {
 		return err