test(slo): xorm: add retry on create and drop table

Author: ImpressionableRaccoon

tests/slo/internal/generator/row.go

@@ -5,7 +5,7 @@ import "time"
 type RowID = uint64
 
 type Row struct {
-	Hash             uint64     `gorm:"column:hash;primarykey;autoIncrement:false" xorg:"pk 'hash'"`
+	Hash             uint64     `gorm:"column:hash;primarykey;autoIncrement:false" xorm:"pk 'hash'"`
 	ID               RowID      `gorm:"column:id;primarykey;autoIncrement:false" xorm:"pk 'id'"`
 	PayloadStr       *string    `gorm:"column:payload_str" xorm:"'payload_str'"`
 	PayloadDouble    *float64   `gorm:"column:payload_double" xorm:"'payload_double'"`

tests/slo/xorm/storage.go

@@ -109,6 +109,7 @@ func NewStorage(ctx context.Context, cfg *config.Config, poolSize int) (_ *Stora
 	s.x.SetTableMapper(newMapper(cfg.Table, "entry"))
 
 	s.x.SetLogLevel(log.LOG_DEBUG)
+	s.x.ShowSQL(true)
 
 	tableParams := map[string]string{
 		"AUTO_PARTITIONING_BY_SIZE":              "ENABLED",
@@ -134,7 +135,7 @@ func (s *Storage) Read(ctx context.Context, id generator.RowID) (row generator.R
 	row.ID = id
 
 	err = retry.Do(ydb.WithTxControl(ctx, readTx), s.x.DB().DB,
-		func(ctx context.Context, cc *sql.Conn) (err error) {
+		func(ctx context.Context, _ *sql.Conn) (err error) {
 			has, err := s.x.Context(ctx).Get(&row)
 			if err != nil {
 				return fmt.Errorf("get entry error: %w", err)
@@ -173,7 +174,7 @@ func (s *Storage) Write(ctx context.Context, row generator.Row) (attempts int, e
 	defer cancel()
 
 	err = retry.Do(ydb.WithTxControl(ctx, writeTx), s.x.DB().DB,
-		func(ctx context.Context, cc *sql.Conn) (err error) {
+		func(ctx context.Context, _ *sql.Conn) (err error) {
 			if err = ctx.Err(); err != nil {
 				return err
 			}
@@ -208,7 +209,9 @@ func (s *Storage) createTable(ctx context.Context) error {
 	ctx, cancel := context.WithTimeout(ctx, time.Duration(s.cfg.WriteTimeout)*time.Millisecond)
 	defer cancel()
 
-	return s.x.Context(ctx).CreateTable(generator.Row{})
+	return retry.Do(ctx, s.x.DB().DB, func(ctx context.Context, _ *sql.Conn) error {
+		return s.x.Context(ctx).CreateTable(generator.Row{})
+	})
 }
 
 func (s *Storage) dropTable(ctx context.Context) error {
@@ -219,7 +222,9 @@ func (s *Storage) dropTable(ctx context.Context) error {
 	ctx, cancel := context.WithTimeout(ctx, time.Duration(s.cfg.WriteTimeout)*time.Millisecond)
 	defer cancel()
 
-	return s.x.Context(ctx).DropTable(generator.Row{})
+	return retry.Do(ctx, s.x.DB().DB, func(ctx context.Context, _ *sql.Conn) error {
+		return s.x.Context(ctx).DropTable(generator.Row{})
+	})
 }
 
 func (s *Storage) close(ctx context.Context) error {