test(slo): add options for xorm table

Author: ImpressionableRaccoon

examples/go.mod

@@ -77,4 +77,4 @@ require (
 
 replace github.com/ydb-platform/ydb-go-sdk/v3 => ../
 
-replace xorm.io/xorm => github.com/ydb-platform/xorm v0.0.2
+replace xorm.io/xorm => github.com/ydb-platform/xorm v0.0.3

examples/go.sum

@@ -909,8 +909,8 @@ github.com/yandex-cloud/go-genproto v0.0.0-20220815090733-4c139c0154e2 h1:blq4r0
 github.com/yandex-cloud/go-genproto v0.0.0-20220815090733-4c139c0154e2/go.mod h1:HEUYX/p8966tMUHHT+TsS0hF/Ca/NYwqprC5WXSDMfE=
 github.com/ydb-platform/gorm-driver v0.0.5 h1:q6Cg/iSFw4TAmSyMh25YM0GRmr6LVM2gnFD+9aCTGlo=
 github.com/ydb-platform/gorm-driver v0.0.5/go.mod h1:fkCvWZlA3PzL5MiMc7yFOzxUOzLpY1uT8yZo+e4SV4Y=
-github.com/ydb-platform/xorm v0.0.2 h1:VsRAoN2bqqqVuUNthEB7cd/m77PAdtNc0cl9Q9zwmVc=
-github.com/ydb-platform/xorm v0.0.2/go.mod h1:hFsU7EUF0o3S+l5c0eyP2yPVjJ0d4gsFdqCsyazzwBc=
+github.com/ydb-platform/xorm v0.0.3 h1:MXk42lANB6r/MMLg/XdJfyXJycGUDlCeLiMlLGDKVPw=
+github.com/ydb-platform/xorm v0.0.3/go.mod h1:hFsU7EUF0o3S+l5c0eyP2yPVjJ0d4gsFdqCsyazzwBc=
 github.com/ydb-platform/ydb-go-genproto v0.0.0-20230528143953-42c825ace222 h1:8ddsk8HKBkVPH8w3k81si6SeCVJIAtw8dnw+s3h0ciE=
 github.com/ydb-platform/ydb-go-genproto v0.0.0-20230528143953-42c825ace222/go.mod h1:Er+FePu1dNUieD+XTMDduGpQuCPssK5Q4BjF+IIXJ3I=
 github.com/ydb-platform/ydb-go-sdk-auth-environ v0.1.2 h1:EYSI1kulnHb0H0zt3yOw4cRj4ABMSMGwNe43D+fX7e4=

tests/slo/go.mod

@@ -52,4 +52,4 @@ require (
 
 replace github.com/ydb-platform/ydb-go-sdk/v3 => ../../.
 
-replace xorm.io/xorm => github.com/ydb-platform/xorm v0.0.2
+replace xorm.io/xorm => github.com/ydb-platform/xorm v0.0.3

tests/slo/go.sum

@@ -887,8 +887,8 @@ github.com/yandex-cloud/go-genproto v0.0.0-20230403093326-123923969dc6 h1:BkuaOC
 github.com/yandex-cloud/go-genproto v0.0.0-20230403093326-123923969dc6/go.mod h1:HEUYX/p8966tMUHHT+TsS0hF/Ca/NYwqprC5WXSDMfE=
 github.com/ydb-platform/gorm-driver v0.0.6-0.20230529184330-9104dd08ecae h1:YC6wzJqGCdFFGQe7vBcRas3IRncp3UOgBCzcmQsX9fU=
 github.com/ydb-platform/gorm-driver v0.0.6-0.20230529184330-9104dd08ecae/go.mod h1:rt/AeXUY1Mh4FtlTGqZ4IjXJu5BFYXpJlgT6UmKqxgs=
-github.com/ydb-platform/xorm v0.0.2 h1:VsRAoN2bqqqVuUNthEB7cd/m77PAdtNc0cl9Q9zwmVc=
-github.com/ydb-platform/xorm v0.0.2/go.mod h1:hFsU7EUF0o3S+l5c0eyP2yPVjJ0d4gsFdqCsyazzwBc=
+github.com/ydb-platform/xorm v0.0.3 h1:MXk42lANB6r/MMLg/XdJfyXJycGUDlCeLiMlLGDKVPw=
+github.com/ydb-platform/xorm v0.0.3/go.mod h1:hFsU7EUF0o3S+l5c0eyP2yPVjJ0d4gsFdqCsyazzwBc=
 github.com/ydb-platform/ydb-go-genproto v0.0.0-20230528143953-42c825ace222 h1:8ddsk8HKBkVPH8w3k81si6SeCVJIAtw8dnw+s3h0ciE=
 github.com/ydb-platform/ydb-go-genproto v0.0.0-20230528143953-42c825ace222/go.mod h1:Er+FePu1dNUieD+XTMDduGpQuCPssK5Q4BjF+IIXJ3I=
 github.com/ydb-platform/ydb-go-sdk-auth-environ v0.1.3 h1:+DrFgi0hjjLarFcWHwI6WPk28hcWr6N7ga+OHAxtemI=

tests/slo/xorm/storage.go

@@ -6,6 +6,7 @@ import (
 	"errors"
 	"fmt"
 	"path"
+	"strconv"
 	"time"
 
 	env "github.com/ydb-platform/ydb-go-sdk-auth-environ"
@@ -109,6 +110,16 @@ func NewStorage(ctx context.Context, cfg *config.Config, poolSize int) (_ *Stora
 
 	s.x.SetLogLevel(log.LOG_DEBUG)
 
+	tableParams := map[string]string{
+		"AUTO_PARTITIONING_BY_SIZE":              "ENABLED",
+		"AUTO_PARTITIONING_BY_LOAD":              "ENABLED",
+		"AUTO_PARTITIONING_PARTITION_SIZE_MB":    strconv.FormatUint(s.cfg.PartitionSize, 10),
+		"AUTO_PARTITIONING_MIN_PARTITIONS_COUNT": strconv.FormatUint(s.cfg.MinPartitionsCount, 10),
+		"AUTO_PARTITIONING_MAX_PARTITIONS_COUNT": strconv.FormatUint(s.cfg.MaxPartitionsCount, 10),
+		"UNIFORM_PARTITIONS":                     strconv.FormatUint(s.cfg.MinPartitionsCount, 10),
+	}
+	s.x.Dialect().SetParams(tableParams)
+
 	return s, nil
 }