grassrootseconomics/cic-custodial
8
0
Fork 0
mirror of https://github.com/grassrootseconomics/cic-custodial.git synced 2025-11-04 10:48:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: moving locking to tasker

Browse Source
This commit is contained in:
Mohamed Sohail 2022-10-25 16:04:43 +00:00
parent 1392c03ab0
commit b8029be6ee
Signed by: kamikazechaser
GPG Key ID: 7DD45520C01CD85D
6 changed files with 50 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/service/init.go
View File

@ -66,7 +66,7 @@ func initChainProvider() *chain.Provider {
func initTaskerClient() *tasker_client.TaskerClient {
return tasker_client.NewTaskerClient(tasker_client.Opts{
RedisDSN: ko.MustString("tasker.dsn"),
RedisDSN: ko.MustString("redis.dsn"),
})
}
@ -93,10 +93,10 @@ func initNoncestore() noncestore.Noncestore {
switch provider := ko.MustString("noncestore.provider"); provider {
case "redis":
redisNoncestore, err := redis_noncestore.NewRedisNoncestore(redis_noncestore.Opts{
RedisDSN: ko.MustString("noncestore.dsn"),
RedisDSN: ko.MustString("redis.dsn"),
RedisDB: 2,
MinIdleConns: 8,
PoolSize: 15,
MinIdleConns: 5,
PoolSize: 10,
ChainProvider: chainProvider,
})
if err != nil {

5
cmd/service/main.go
View File

@ -62,9 +62,12 @@ func main() {
taskerServer = tasker_server.NewTaskerServer(tasker_server.Opts{
ActionsProvider: actionsProvider,
TaskerClient: taskerClient,
RedisDSN: ko.MustString("tasker.dsn"),
RedisDSN: ko.MustString("redis.dsn"),
Concurrency: 15,
Logger: lo,
RedisLockDB: 1,
RedisLockMinIdleConns: 3,
RedisLockPoolSize: 6,
})
var wg sync.WaitGroup

7
config.toml
View File

@ -6,10 +6,10 @@ address = ":5000"
endpoint = "https://rpc.sarafu.network"
[admin]
public = "0x28Daa9bA426383fA0296A669Af13f3EeF57A2877"
key = "c7f7e36c62a1ecfc4fba508421919c6449e7dced0ea846ce00fe3e93a9aaa608"
public = "0x80097c773B3E83472FC7952c5206a7DB35d42bEF"
key = "a6af6c597c614e3c8ee4b7638ab7c3f737aece3773a5413ca8caf4338e6b06d1"
[tasker]
[redis]
dsn = "127.0.0.1:6379"
[keystore]
@ -18,4 +18,3 @@ dsn = "postgres://postgres:postgres@localhost:5432/cic_custodial"
[noncestore]
provider = "redis"
dsn = "127.0.0.1:6379"

45
internal/noncestore/providers/redis/redis.go
View File

@ -2,9 +2,7 @@ package redis
import (
"context"
"time"
"github.com/bsm/redislock"
"github.com/go-redis/redis/v8"
"github.com/grassrootseconomics/cic-custodial/internal/noncestore"
"github.com/grassrootseconomics/cic-go-sdk/chain"
@ -12,11 +10,6 @@ import (
"github.com/lmittmann/w3/module/eth"
)
const (
mutexLockTTL = 200 * time.Millisecond
mutexKeyPrefix = "lock_"
)
// Opts represents the Redis nonce store specific params
type Opts struct {
RedisDSN string
@ -28,7 +21,6 @@ type Opts struct {
// RedisNoncestore implements `noncestore.Noncestore`
type RedisNoncestore struct {
redisLockProvider *redislock.Client
chainProvider *chain.Provider
redis *redis.Client
}
@ -46,19 +38,12 @@ func NewRedisNoncestore(o Opts) (noncestore.Noncestore, error) {
}
return &RedisNoncestore{
redisLockProvider: redislock.New(redisClient),
redis: redisClient,
chainProvider: o.ChainProvider,
}, nil
}
func (ns *RedisNoncestore) Peek(ctx context.Context, publicKey string) (uint64, error) {
lock, err := ns.redisLockProvider.Obtain(ctx, mutexKeyPrefix+publicKey, mutexLockTTL, nil)
if err != nil {
return 0, err
}
defer lock.Release(ctx)
nonce, err := ns.redis.Get(ctx, publicKey).Uint64()
if err != nil {
return 0, err
@ -72,13 +57,7 @@ func (ns *RedisNoncestore) Acquire(ctx context.Context, publicKey string) (uint6
nonce uint64
)
lock, err := ns.redisLockProvider.Obtain(ctx, mutexKeyPrefix+publicKey, mutexLockTTL, nil)
if err != nil {
return 0, err
}
defer lock.Release(ctx)
nonce, err = ns.redis.Get(ctx, publicKey).Uint64()
nonce, err := ns.redis.Get(ctx, publicKey).Uint64()
if err == redis.Nil {
networkNonce, err := ns.SyncNetworkNonce(ctx, publicKey)
if err != nil {
@ -99,12 +78,6 @@ func (ns *RedisNoncestore) Acquire(ctx context.Context, publicKey string) (uint6
}
func (ns *RedisNoncestore) Return(ctx context.Context, publicKey string) (uint64, error) {
lock, err := ns.redisLockProvider.Obtain(ctx, mutexKeyPrefix+publicKey, mutexLockTTL, nil)
if err != nil {
return 0, err
}
defer lock.Release(ctx)
nonce, err := ns.redis.Get(ctx, publicKey).Uint64()
if err != nil {
return 0, err
@ -125,13 +98,7 @@ func (ns *RedisNoncestore) SyncNetworkNonce(ctx context.Context, publicKey strin
networkNonce uint64
)
lock, err := ns.redisLockProvider.Obtain(ctx, mutexKeyPrefix+publicKey, mutexLockTTL, nil)
if err != nil {
return 0, err
}
defer lock.Release(ctx)
err = ns.chainProvider.EthClient.CallCtx(
err := ns.chainProvider.EthClient.CallCtx(
ctx,
eth.Nonce(w3.A(publicKey), nil).Returns(&networkNonce),
)
@ -148,13 +115,7 @@ func (ns *RedisNoncestore) SyncNetworkNonce(ctx context.Context, publicKey strin
}
func (ns *RedisNoncestore) SetNewAccountNonce(ctx context.Context, publicKey string) error {
lock, err := ns.redisLockProvider.Obtain(ctx, mutexKeyPrefix+publicKey, mutexLockTTL, nil)
if err != nil {
return err
}
defer lock.Release(ctx)
err = ns.redis.Set(ctx, publicKey, 0, 0).Err()
err := ns.redis.Set(ctx, publicKey, 0, 0).Err()
if err != nil {
return err
}

6
internal/tasker/server/registration.go
View File

@ -39,6 +39,12 @@ func (tp *TaskerProcessor) giftGasProcessor(ctx context.Context, t *asynq.Task)
return fmt.Errorf("json.Unmarshal failed: %v: %w", err, asynq.SkipRetry)
}
lock, err := tp.LockProvider.Obtain(ctx, tp.ActionsProvider.SystemPublicKey, LockTTL, nil)
if err != nil {
return err
}
defer lock.Release(ctx)
signedTx, err := tp.ActionsProvider.SignGiftGasTx(ctx, p.PublicKey)
if err != nil {
return err

17
internal/tasker/server/server.go
View File

@ -4,21 +4,30 @@ import (
"time"
"github.com/bsm/redislock"
"github.com/go-redis/redis/v8"
"github.com/grassrootseconomics/cic-custodial/internal/actions"
tasker_client "github.com/grassrootseconomics/cic-custodial/internal/tasker/client"
"github.com/hibiken/asynq"
"github.com/zerodha/logf"
)
const (
LockTTL = 1 * time.Second
)
type Opts struct {
ActionsProvider *actions.ActionsProvider
TaskerClient *tasker_client.TaskerClient
RedisDSN string
RedisLockDB int
RedisLockMinIdleConns int
RedisLockPoolSize int
Concurrency int
Logger logf.Logger
}
type TaskerProcessor struct {
LockProvider *redislock.Client
ActionsProvider *actions.ActionsProvider
TaskerClient *tasker_client.TaskerClient
}
@ -29,9 +38,17 @@ type TaskerServer struct {
}
func NewTaskerServer(o Opts) *TaskerServer {
redisLockClient := redis.NewClient(&redis.Options{
Addr: o.RedisDSN,
DB: o.RedisLockDB,
MinIdleConns: o.RedisLockMinIdleConns,
PoolSize: o.RedisLockPoolSize,
})
taskerProcessor := &TaskerProcessor{
ActionsProvider: o.ActionsProvider,
TaskerClient: o.TaskerClient,
LockProvider: redislock.New(redisLockClient),
}
asynqServer := asynq.NewServer(