reorganize project structure

This commit is contained in:
Sergey Chubaryan 2025-01-17 12:45:24 +03:00
parent b2702d8375
commit 84b2e000b0
53 changed files with 138 additions and 386 deletions

View File

@ -1,17 +1,16 @@
package src
package main
import (
"backend/src/args_parser"
"backend/src/cache"
"backend/src/client_notifier"
"backend/src/config"
"backend/src/core/models"
"backend/src/core/repos"
"backend/src/core/services"
"backend/src/core/utils"
"backend/src/integrations"
"backend/src/logger"
"backend/src/server"
"backend/cmd/backend/args_parser"
"backend/cmd/backend/config"
"backend/cmd/backend/server"
"backend/internal/core/models"
"backend/internal/core/repos"
"backend/internal/core/services"
"backend/internal/core/utils"
"backend/internal/integrations"
"backend/pkg/cache"
"backend/pkg/logger"
"context"
"crypto/rsa"
"crypto/x509"
@ -187,8 +186,6 @@ func (a *App) Run(p RunParams) {
// go shortlinkService.ShortlinkRoutine(ctx)
}
clientNotifier := client_notifier.NewBasicNotifier()
// Start profiling
if args.GetProfilePath() != "" {
pprofFile, err := os.Create(args.GetProfilePath())
@ -211,7 +208,6 @@ func (a *App) Run(p RunParams) {
server.NewServerOpts{
DebugMode: debugMode,
Logger: logger,
Notifier: clientNotifier,
ShortlinkService: shortlinkService,
UserService: userService,
Tracer: tracer,

View File

@ -1,5 +1,5 @@
port: 8080
postgres_url: "postgres://postgres:postgres@localhost:5432/postgres"
jwt_signing_key: "./misc/jwt_signing_key"
jwt_signing_key: "./jwt_signing_key"
kafka_url: "localhost:9092"
kafka_topic: "backend_events"

View File

@ -1,7 +1,6 @@
package main
import (
"backend/src"
"context"
"os"
"strings"
@ -14,9 +13,9 @@ func main() {
env[kv[0]] = kv[1]
}
app := &src.App{}
app := &App{}
app.Run(
src.RunParams{
RunParams{
Ctx: context.Background(),
OsArgs: os.Args,
EnvVars: env,

View File

@ -1,8 +1,8 @@
package handlers
import (
"backend/src/core/services"
"backend/src/logger"
"backend/internal/core/services"
"backend/pkg/logger"
"encoding/json"
"fmt"
"net/url"

View File

@ -1,8 +1,8 @@
package handlers
import (
"backend/src/core/services"
"backend/src/logger"
"backend/internal/core/services"
"backend/pkg/logger"
"encoding/json"
"github.com/gin-gonic/gin"

View File

@ -1,8 +1,8 @@
package handlers
import (
"backend/src/core/services"
"backend/src/logger"
"backend/internal/core/services"
"backend/pkg/logger"
"encoding/json"
"github.com/gin-gonic/gin"

View File

@ -1,7 +1,7 @@
package middleware
import (
"backend/src/core/services"
"backend/internal/core/services"
"fmt"
"github.com/gin-gonic/gin"

View File

@ -3,8 +3,8 @@ package middleware
// Modified recovery from gin, use own logger
import (
"backend/src/integrations"
"backend/src/logger"
"backend/internal/integrations"
"backend/pkg/logger"
"bytes"
"errors"
"fmt"

View File

@ -1,8 +1,8 @@
package middleware
import (
"backend/src/integrations"
log "backend/src/logger"
"backend/internal/integrations"
log "backend/pkg/logger"
"time"
"github.com/gin-gonic/gin"

View File

@ -1,13 +1,12 @@
package server
import (
"backend/src/client_notifier"
"backend/src/core/services"
"backend/src/integrations"
"backend/src/logger"
"backend/src/server/handlers"
"backend/src/server/middleware"
"backend/src/server/utils"
"backend/cmd/backend/server/handlers"
"backend/cmd/backend/server/middleware"
"backend/cmd/backend/server/utils"
"backend/internal/core/services"
"backend/internal/integrations"
"backend/pkg/logger"
"context"
"fmt"
"net"
@ -24,7 +23,6 @@ type Server struct {
type NewServerOpts struct {
DebugMode bool
Logger logger.Logger
Notifier client_notifier.ClientNotifier
UserService services.UserService
ShortlinkService services.ShortlinkService
Tracer trace.Tracer
@ -48,8 +46,6 @@ func New(opts NewServerOpts) *Server {
r.Use(middleware.NewRequestLogMiddleware(opts.Logger, opts.Tracer, prometheus))
r.Use(middleware.NewTracingMiddleware(opts.Tracer))
r.GET("/pooling", handlers.NewLongPoolingHandler(opts.Logger, opts.Notifier))
linkGroup := r.Group("/s")
linkGroup.POST("/new", handlers.NewShortlinkCreateHandler(opts.Logger, opts.ShortlinkService))
linkGroup.GET("/:linkId", handlers.NewShortlinkResolveHandler(opts.Logger, opts.ShortlinkService))

View File

@ -1,7 +1,7 @@
package utils
import (
"backend/src/core/models"
"backend/internal/core/models"
"github.com/gin-gonic/gin"
)

View File

@ -1,9 +0,0 @@
kafka:
brokers:
- localhost:9092
topic: backend_events
smtp:
server: smtp.yandex.ru
port: 587
email: ""
password: ""

View File

@ -1,13 +0,0 @@
module helper
go 1.22.5
require gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df
require (
github.com/klauspost/compress v1.15.9 // indirect
github.com/pierrec/lz4/v4 v4.1.15 // indirect
github.com/segmentio/kafka-go v0.4.47 // indirect
gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc // indirect
gopkg.in/yaml.v3 v3.0.1
)

View File

@ -1,64 +0,0 @@
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/klauspost/compress v1.15.9 h1:wKRjX6JRtDdrE9qwa4b/Cip7ACOshUI4smpCQanqjSY=
github.com/klauspost/compress v1.15.9/go.mod h1:PhcZ0MbTNciWF3rruxRgKxI5NkcHHrHUDtV4Yw2GlzU=
github.com/pierrec/lz4/v4 v4.1.15 h1:MO0/ucJhngq7299dKLwIMtgTfbkoSPF6AoMYDd8Q4q0=
github.com/pierrec/lz4/v4 v4.1.15/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/segmentio/kafka-go v0.4.47 h1:IqziR4pA3vrZq7YdRxaT3w1/5fvIH5qpCwstUanQQB0=
github.com/segmentio/kafka-go v0.4.47/go.mod h1:HjF6XbOKh0Pjlkr5GVZxt6CsjjwnmhVOfURM5KMd8qg=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
github.com/xdg-go/scram v1.1.2/go.mod h1:RT/sEzTbU5y00aCK8UOx6R7YryM0iF1N2MOmC3kKLN4=
github.com/xdg-go/stringprep v1.0.4/go.mod h1:mPGuuIYwz7CmR2bT9j4GbQqutWS1zV24gijq1dTyGkM=
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
golang.org/x/crypto v0.14.0/go.mod h1:MVFd36DqK4CsrnJYDkBA3VC4m2GkXAM0PvzMCn4JQf4=
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.13.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo=
golang.org/x/term v0.13.0/go.mod h1:LTmsnFJwVN6bCy1rVCoS+qHT1HhALEFxKncY3WNNh4U=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc h1:2gGKlE2+asNV9m7xrywl36YYNnBG5ZQ0r/BOOxqPpmk=
gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc/go.mod h1:m7x9LTH6d71AHyAX77c9yqWCCa3UKHcVEj9y7hAtKDk=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df h1:n7WqCuqOuCbNr617RXOY0AWRXxgwEyPp2z+p0+hgMuE=
gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df/go.mod h1:LRQQ+SO6ZHR7tOkpBDuZnXENFzX8qRjMDMyPD6BRkCw=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

View File

@ -1,113 +0,0 @@
package main
import (
"context"
"encoding/json"
"io"
"log"
"os"
"strings"
"github.com/segmentio/kafka-go"
"gopkg.in/gomail.v2"
"gopkg.in/yaml.v3"
)
const MSG_TEXT = `
<html>
<head>
</head>
<body>
<p>This message was sent because you forgot a password</p>
<p>To change a password, use <a href="{{Link}}"/>this</a> link</p>
</body>
</html>
`
func SendEmailForgotPassword(dialer *gomail.Dialer, from, to, token string) error {
link := "localhost:8080/restore-password?token=" + token
msgText := strings.ReplaceAll(MSG_TEXT, "{{Link}}", link)
m := gomail.NewMessage()
m.SetHeader("From", m.FormatAddress(from, "Pet Backend"))
m.SetHeader("To", to)
m.SetHeader("Subject", "Hello!")
m.SetBody("text/html", msgText)
return dialer.DialAndSend(m)
}
type Config struct {
Kafka struct {
Brokers []string `yaml:"brokers"`
Topic string `yaml:"topic"`
}
SMTP struct {
Server string `yaml:"server"`
Port int `yaml:"port"`
Email string `yaml:"email"`
Password string `yaml:"password"`
} `yaml:"smtp"`
}
func main() {
ctx := context.Background()
configFile, err := os.ReadFile("config.yaml")
if err != nil {
log.Fatal(err.Error())
}
config := &Config{}
if err := yaml.Unmarshal(configFile, config); err != nil {
log.Fatal(err.Error())
}
dialer := gomail.NewDialer(config.SMTP.Server, config.SMTP.Port, config.SMTP.Email, config.SMTP.Password)
log.Println("starting reader...")
r := kafka.NewReader(kafka.ReaderConfig{
Brokers: config.Kafka.Brokers,
Topic: config.Kafka.Topic,
GroupID: "consumer-group-id",
})
log.Println("reader started")
for {
msg, err := r.FetchMessage(ctx)
if err == io.EOF {
log.Fatal("EOF")
return
}
if err != nil {
log.Fatal(err.Error())
return
}
log.Printf("offset: %d, partition: %d, key: %s, value: %s\n", msg.Offset, msg.Partition, string(msg.Key), string(msg.Value))
if err := r.CommitMessages(ctx, msg); err != nil {
log.Fatalf("failed to commit: %s\n", err.Error())
continue
}
value := struct {
Email string `json:"email"`
Token string `json:"token"`
}{}
if err := json.Unmarshal(msg.Value, &value); err != nil {
log.Fatalf("failed to unmarshal: %s\n", err.Error())
continue
}
if err := SendEmailForgotPassword(dialer, config.SMTP.Email, value.Email, value.Token); err != nil {
log.Fatalf("failed to send email: %s\n", err.Error())
continue
}
}
}

View File

@ -1,8 +1,8 @@
package repos
import (
"backend/src/core/models"
"backend/src/integrations"
"backend/internal/core/models"
"backend/internal/integrations"
"context"
"database/sql"
)

View File

@ -1,7 +1,7 @@
package repos
import (
"backend/src/integrations"
"backend/internal/integrations"
"context"
"encoding/json"
)

View File

@ -1,7 +1,7 @@
package repos
import (
"backend/src/integrations"
"backend/internal/integrations"
"context"
"database/sql"
"errors"

View File

@ -1,8 +1,8 @@
package repos
import (
"backend/src/core/models"
"backend/src/integrations"
"backend/internal/core/models"
"backend/internal/integrations"
"context"
"database/sql"
"errors"

View File

@ -1,9 +1,9 @@
package services
import (
"backend/src/cache"
"backend/src/charsets"
"backend/src/core/repos"
"backend/internal/core/repos"
"backend/pkg/cache"
"backend/pkg/charsets"
"context"
"fmt"
"math/rand"

View File

@ -1,10 +1,10 @@
package services
import (
"backend/src/cache"
"backend/src/core/models"
"backend/src/core/repos"
"backend/src/core/utils"
"backend/internal/core/models"
"backend/internal/core/repos"
"backend/internal/core/utils"
"backend/pkg/cache"
"context"
"fmt"
"time"

View File

@ -1,9 +1,10 @@
package utils
import (
"backend/src/charsets"
"fmt"
"backend/pkg/charsets"
"golang.org/x/crypto/bcrypt"
)

View File

@ -13,13 +13,3 @@ func (e Expiration) Get() time.Time {
}
return e.ExpiresAt
}
type Cache[K comparable, V any] interface {
Get(key K) (V, bool)
GetEx(key K, exp Expiration) (V, bool)
Set(key K, value V, exp Expiration)
Del(key K)
CheckExpired(batchSize int)
}

11
pkg/cache/interface.go vendored Normal file
View File

@ -0,0 +1,11 @@
package cache
type Cache[K comparable, V any] interface {
Get(key K) (V, bool)
GetEx(key K, exp Expiration) (V, bool)
Set(key K, value V, exp Expiration)
Del(key K)
CheckExpired(batchSize int)
}

43
pkg/containers/queue.go Normal file
View File

@ -0,0 +1,43 @@
package containers
func NewQueue[T interface{}]() Queue[T] {
return Queue[T]{
front: NewStack[T](),
back: NewStack[T](),
}
}
type Queue[T interface{}] struct {
front, back *Stack[T]
}
func (m *Queue[T]) Push(val T) {
m.front.Push(val)
}
func (m *Queue[T]) swapStacks() {
if m.back.Empty() {
for !m.front.Empty() {
val, _ := m.front.Pop()
m.back.Push(val)
}
}
}
func (m *Queue[T]) Pop() T {
m.swapStacks()
val, _ := m.back.Pop()
return val
}
func (m *Queue[T]) Peek() T {
m.swapStacks()
val, _ := m.back.Peek()
return val
}
func (m *Queue[T]) Empty() bool {
return m.front.Empty() && m.back.Empty()
}

37
pkg/containers/stack.go Normal file
View File

@ -0,0 +1,37 @@
package containers
func NewStack[T interface{}]() *Stack[T] {
return &Stack[T]{[]T{}}
}
type Stack[T interface{}] struct {
arr []T
}
func (s *Stack[T]) Empty() bool {
return len(s.arr) <= 0
}
func (s *Stack[T]) Push(val T) {
s.arr = append(s.arr, val)
}
func (s *Stack[T]) Peek() (T, bool) {
if len(s.arr) <= 0 {
var t T
return t, false
}
return s.arr[len(s.arr)-1], true
}
func (s *Stack[T]) Pop() (T, bool) {
if len(s.arr) <= 0 {
var t T
return t, false
}
maxIndex := len(s.arr) - 1
element := s.arr[maxIndex]
s.arr = s.arr[:maxIndex]
return element, true
}

View File

@ -1,12 +0,0 @@
package client_notifier
type Event struct {
Type EventType
Data []byte
}
type EventType string
const (
EventTypeEmailConfirmed EventType = "event_email_confirmed"
)

View File

@ -1,57 +0,0 @@
package client_notifier
import "sync"
type ClientNotifier interface {
RegisterClient(id string) <-chan Event
UnregisterClient(id string)
NotifyClient(id string, e Event)
}
type client struct {
id string
eventChan chan Event
}
func NewBasicNotifier() ClientNotifier {
return &basicNotifier{
m: &sync.RWMutex{},
clients: map[string]client{},
}
}
type basicNotifier struct {
m *sync.RWMutex
clients map[string]client
}
func (p *basicNotifier) RegisterClient(id string) <-chan Event {
p.m.Lock()
defer p.m.Unlock()
eventChan := make(chan Event)
p.clients[id] = client{
id: id,
eventChan: eventChan,
}
return eventChan
}
func (p *basicNotifier) UnregisterClient(id string) {
p.m.Lock()
defer p.m.Unlock()
delete(p.clients, id)
}
func (p *basicNotifier) NotifyClient(id string, e Event) {
p.m.RLock()
defer p.m.RUnlock()
client, ok := p.clients[id]
if !ok {
return
}
client.eventChan <- e
}

View File

@ -1,25 +0,0 @@
package leader_elector
import (
"context"
"database/sql"
)
func Lock(ctx context.Context, db *sql.DB, lockName, id string) error {
query := `
update locks (id)
set id = $1
where name == lockName and timestamp < $1 returning id
on conflict
insert into locks(id, name) values($1, $2);`
row := db.QueryRowContext(ctx, query, id)
result := ""
err := row.Scan(&result)
if err != nil {
return err
}
return nil
}

View File

@ -1,28 +0,0 @@
package handlers
import (
"backend/src/client_notifier"
"backend/src/logger"
"backend/src/server/utils"
"github.com/gin-gonic/gin"
)
func NewLongPoolingHandler(logger logger.Logger, notifier client_notifier.ClientNotifier) gin.HandlerFunc {
return func(c *gin.Context) {
user := utils.GetUserFromRequest(c)
if user == nil {
c.Data(403, "plain/text", []byte("Unauthorized"))
return
}
eventChan := notifier.RegisterClient(user.Id)
select {
case <-c.Done():
notifier.UnregisterClient(user.Id)
case event := <-eventChan:
c.Data(200, "application/json", event.Data)
}
}
}