test version early test event

internal/cli/event.go

@@ -0,0 +1,15 @@
+package cli
+
+import (
+	"log"
+
+	"github.com/rarimo/geo-points-svc/internal/config"
+	"github.com/rarimo/geo-points-svc/internal/service/event"
+)
+
+func eventStart(cfg config.Config, data int) {
+	err := event.Run(cfg, data)
+	if err != nil {
+		log.Fatalf("Error starting event: %v", err)
+	}
+}

internal/cli/main.go

@@ -29,6 +29,10 @@ func Run(args []string) bool {
 		migrateCmd     = app.Command("migrate", "migrate command")
 		migrateUpCmd   = migrateCmd.Command("up", "migrate db up")
 		migrateDownCmd = migrateCmd.Command("down", "migrate db down")
+
+		event    = app.Command("event", "claim event command")
+		eventCmd = event.Command("start", "claim event command")
+		data     = eventCmd.Arg("data", "data after ...").Required().Int()
 	)
 
 	cmd, err := app.Parse(args[1:])
@@ -48,6 +52,8 @@ func Run(args []string) bool {
 		err = MigrateUp(cfg)
 	case migrateDownCmd.FullCommand():
 		err = MigrateDown(cfg)
+	case eventCmd.FullCommand():
+		eventStart(cfg, *data)
 	default:
 		log.Errorf("unknown command %s", cmd)
 		return false

internal/data/balances.go

@@ -54,6 +54,8 @@ type BalancesQ interface {
 	FilterByInternalAID(aid string) BalancesQ
 	FilterByExternalAID(aid string) BalancesQ
 	FilterBySharedHash(hash string) BalancesQ
+	FilterByCreatedBefore(time int) BalancesQ
+	FilterVerified() BalancesQ
 }
 
 type WithoutPassportEventBalance struct {

internal/data/evtypes/models/extra.go

@@ -21,6 +21,7 @@ const (
 	TypeInternalPassportScan = "internal_passport_scan"
 	TypeExternalPassportScan = "external_passport_scan"
 	TypePollParticipation    = "poll_participation"
+	TypeEarlyTest            = "early_test"
 )
 
 const (

internal/data/pg/balances.go

@@ -222,10 +222,18 @@ func (q *balances) FilterBySharedHash(hash string) data.BalancesQ {
 	return q.applyCondition(squirrel.Eq{"shared_hash": hash})
 }
 
+func (q *balances) FilterByCreatedBefore(data int) data.BalancesQ {
+	return q.applyCondition(squirrel.Lt{"created_at": data})
+}
+
 func (q *balances) applyCondition(cond squirrel.Sqlizer) data.BalancesQ {
 	q.selector = q.selector.Where(cond)
 	q.updater = q.updater.Where(cond)
 	q.rank = q.rank.Where(cond)
 	q.counter = q.counter.Where(cond)
 	return q
 }
+
+func (q *balances) FilterVerified() data.BalancesQ {
+	return q.applyCondition(squirrel.Gt{"verified": true})
+}

internal/service/event/main.go

@@ -0,0 +1,55 @@
+package event
+
+import (
+	"fmt"
+
+	"github.com/rarimo/geo-points-svc/internal/config"
+	"github.com/rarimo/geo-points-svc/internal/data"
+	"github.com/rarimo/geo-points-svc/internal/data/evtypes/models"
+	"github.com/rarimo/geo-points-svc/internal/data/pg"
+)
+
+func Run(cfg config.Config, date int) error {
+	log := cfg.Log()
+	db := cfg.DB()
+
+	balances := pg.NewBalances(db)
+	events := pg.NewEvents(db)
+
+	users, err := balances.FilterByCreatedBefore(date).FilterVerified().Select()
+	filtEvents := events.FilterByType(models.TypeEarlyTest)
+
+	if err != nil {
+		log.WithError(err).Error("failed to filter by updated before")
+		return err
+	}
+
+	if users == nil {
+		log.Infof("no users found")
+		return nil
+	}
+
+	for _, user := range users {
+		userNull := user.Nullifier
+		eve, err := filtEvents.FilterByNullifier(userNull).Get()
+
+		if err != nil {
+			log.WithError(err).Error("failed to filter by nullifier")
+			return err
+		}
+		if eve != nil {
+			break
+		}
+
+		err = events.Insert(data.Event{
+			Nullifier: user.Nullifier,
+			Type:      models.TypeEarlyTest,
+			Status:    data.EventFulfilled,
+		})
+		if err != nil {
+			return fmt.Errorf("failed to insert `early_test` event: %w", err)
+		}
+	}
+
+	return nil
+}

internal/service/workers/leveljustice/main.go

@@ -37,5 +37,4 @@ func Run(cfg config.Config, sig chan struct{}) {
 	}
 
 	sig <- struct{}{}
-
 }