fix(sharding): added unit tests for ID

yaitoo · Feb 24, 2024 · d331ffd · d331ffd
1 parent 160dd57
commit d331ffd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/sharding/id_test.go b/sharding/id_test.go
@@ -31,10 +31,10 @@ func TestID(t *testing.T) {
 		{
 			name:        "build_max_values_should_work",
 			timeNow:     time.UnixMilli(TimeEnd),
-			workerID:    3,
-			databaseID:  1023,
+			workerID:    MaxWorkerID,
+			databaseID:  MaxDatabaseID,
 			tableRotate: Daily,
-			sequence:    1023,
+			sequence:    MaxSequence,
 		},
 		{
 			name:        "build_should_work",

diff --git a/sharding/option.go b/sharding/option.go
@@ -6,15 +6,15 @@ type Option func(g *Generator)
 
 func WithWorker(i int8) Option {
 	return func(g *Generator) {
-		if i > 0 && i < 4 {
+		if i >= 0 && i <= MaxWorkerID {
 			g.workerID = i
 		}
 	}
 }
 
 func WithDatabase(num int16) Option {
 	return func(g *Generator) {
-		if num > 0 && num < 1024 {
+		if num >= 0 && num <= MaxDatabaseID {
 			g.databaseNum = num
 		}
 	}