etcd-io
/
dbtester
mirror of https://github.com/etcd-io/dbtester.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

dbtester: set max etcd quota size

This commit is contained in:
Gyu-Ho Lee 2017-02-23 10:14:18 -08:00
parent 580c46f4ef
commit 414759ce63
No known key found for this signature in database
GPG Key ID: 1DDD39C7EB70C24C
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
config_dbtester.go
View File

@ -296,6 +296,8 @@ func ReadConfig(fpath string, analyze bool) (*Config, error) {
return &cfg, nil return &cfg, nil
} }
const maxEtcdQuotaSize = 8000000000
// ToRequest converts configuration to 'dbtesterpb.Request'. // ToRequest converts configuration to 'dbtesterpb.Request'.
func (cfg *Config) ToRequest(databaseID string, op dbtesterpb.Operation, idx int) (req *dbtesterpb.Request, err error) { func (cfg *Config) ToRequest(databaseID string, op dbtesterpb.Operation, idx int) (req *dbtesterpb.Request, err error) {
gcfg, ok := cfg.DatabaseIDToConfigClientMachineAgentControl[databaseID] gcfg, ok := cfg.DatabaseIDToConfigClientMachineAgentControl[databaseID]
@ -326,16 +328,28 @@ func (cfg *Config) ToRequest(databaseID string, op dbtesterpb.Operation, idx int
SnapshotCount: gcfg.Flag_Etcd_V2_3.SnapshotCount, SnapshotCount: gcfg.Flag_Etcd_V2_3.SnapshotCount,
} }
case dbtesterpb.DatabaseID_etcd__v3_1: case dbtesterpb.DatabaseID_etcd__v3_1:
if gcfg.Flag_Etcd_V3_1.QuotaSizeBytes > maxEtcdQuotaSize {
err = fmt.Errorf("maximum etcd quota is 8 GB (%d), got %d", maxEtcdQuotaSize, gcfg.Flag_Etcd_V3_1.QuotaSizeBytes)
return
}
req.Flag_Etcd_V3_1 = &dbtesterpb.Flag_Etcd_V3_1{ req.Flag_Etcd_V3_1 = &dbtesterpb.Flag_Etcd_V3_1{
SnapshotCount: gcfg.Flag_Etcd_V3_1.SnapshotCount, SnapshotCount: gcfg.Flag_Etcd_V3_1.SnapshotCount,
QuotaSizeBytes: gcfg.Flag_Etcd_V3_1.QuotaSizeBytes, QuotaSizeBytes: gcfg.Flag_Etcd_V3_1.QuotaSizeBytes,
} }
case dbtesterpb.DatabaseID_etcd__v3_2: case dbtesterpb.DatabaseID_etcd__v3_2:
if gcfg.Flag_Etcd_V3_2.QuotaSizeBytes > maxEtcdQuotaSize {
err = fmt.Errorf("maximum etcd quota is 8 GB (%d), got %d", maxEtcdQuotaSize, gcfg.Flag_Etcd_V3_2.QuotaSizeBytes)
return
}
req.Flag_Etcd_V3_2 = &dbtesterpb.Flag_Etcd_V3_2{ req.Flag_Etcd_V3_2 = &dbtesterpb.Flag_Etcd_V3_2{
SnapshotCount: gcfg.Flag_Etcd_V3_2.SnapshotCount, SnapshotCount: gcfg.Flag_Etcd_V3_2.SnapshotCount,
QuotaSizeBytes: gcfg.Flag_Etcd_V3_2.QuotaSizeBytes, QuotaSizeBytes: gcfg.Flag_Etcd_V3_2.QuotaSizeBytes,
} }
case dbtesterpb.DatabaseID_etcd__tip: case dbtesterpb.DatabaseID_etcd__tip:
if gcfg.Flag_Etcd_Tip.QuotaSizeBytes > maxEtcdQuotaSize {
err = fmt.Errorf("maximum etcd quota is 8 GB (%d), got %d", maxEtcdQuotaSize, gcfg.Flag_Etcd_Tip.QuotaSizeBytes)
return
}
req.Flag_Etcd_Tip = &dbtesterpb.Flag_Etcd_Tip{ req.Flag_Etcd_Tip = &dbtesterpb.Flag_Etcd_Tip{
SnapshotCount: gcfg.Flag_Etcd_Tip.SnapshotCount, SnapshotCount: gcfg.Flag_Etcd_Tip.SnapshotCount,
QuotaSizeBytes: gcfg.Flag_Etcd_Tip.QuotaSizeBytes, QuotaSizeBytes: gcfg.Flag_Etcd_Tip.QuotaSizeBytes,