-
Notifications
You must be signed in to change notification settings - Fork 20
Expand file tree
/
Copy pathbatch_test_setup.sql
More file actions
214 lines (185 loc) · 7.37 KB
/
batch_test_setup.sql
File metadata and controls
214 lines (185 loc) · 7.37 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
-- =============================================================================
-- Sample table setup for Aurora DSQL batch operations
--
-- This script creates a test table and populates it with 25,000 rows of sample
-- data distributed across 5 categories (~5,000 rows each). Each category is
-- used by a different demo operation:
--
-- electronics → Sequential batch DELETE
-- clothing → Sequential batch UPDATE
-- food → Parallel batch DELETE
-- books → Parallel batch UPDATE
-- toys → (unused, remains in table)
--
-- Aurora DSQL limits each transaction to 3,000 row mutations, so we insert
-- in batches of 1,000 rows. Run each INSERT as a separate transaction.
-- =============================================================================
-- Drop the table if it already exists
DROP TABLE IF EXISTS batch_test;
-- Create the sample table.
-- Uses UUID primary key with gen_random_uuid() to minimize OCC contention,
-- following Aurora DSQL best practice of random keys.
CREATE TABLE batch_test (
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
category VARCHAR(50) NOT NULL,
status VARCHAR(20) NOT NULL DEFAULT 'active',
value NUMERIC(10,2),
created_at TIMESTAMPTZ DEFAULT NOW(),
updated_at TIMESTAMPTZ DEFAULT NOW()
);
-- Create an index on the category column.
CREATE INDEX ASYNC idx_batch_test_category ON batch_test (category);
-- =============================================================================
-- Populate the table with 25,000 rows of test data (25 batches of 1,000).
-- Each INSERT is 1,000 rows — run each as a separate transaction.
-- =============================================================================
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);
INSERT INTO batch_test (category, status, value)
SELECT
(ARRAY['electronics', 'clothing', 'food', 'books', 'toys'])[floor(random() * 5 + 1)],
'active',
round((random() * 1000)::numeric, 2)
FROM generate_series(1, 1000);