Add RPC load tests

Author: steve-chavez

k6/RPCGETSingle.js

@@ -0,0 +1,47 @@
+import { Rate } from "k6/metrics";
+import { check, group, sleep } from 'k6';
+import http from 'k6/http';
+
+const URL = "http://" + __ENV.HOST;
+
+const RATE = (function(){
+ if(__ENV.VERSION == 'v701'){
+ switch(__ENV.HOST){
+ case 'c5xlarge': return 2400;
+ case 't3axlarge': return 1600;
+ case 't3anano': return 1500;
+ default: return 1000;
+ }
+ }
+ else switch(__ENV.HOST){
+ case 'c5xlarge': return 3000;
+ case 't3axlarge': return 2300;
+ case 't3anano': return 2100;
+ default: return 1000;
+ }
+})();
+
+export let options = {
+ discardResponseBodies: true,
+ scenarios: {
+ constant_request_rate: {
+ executor: 'constant-arrival-rate',
+ rate: RATE,
+ timeUnit: '1s',
+ duration: '30s',
+ preAllocatedVUs: 100,
+ maxVUs: 600,
+ }
+ },
+ thresholds: {
+ 'failed requests': ['rate<0.1'],
+ 'http_req_duration': ['p(95)<1000']
+ }
+};
+
+const myFailRate = new Rate('failed requests');
+
+export default function() {
+ let res = http.get(URL + "/rpc/ret_artists?select=*&artist_id=eq.3");
+ myFailRate.add(res.status !== 200);
+}

k6/RPCGETSingleEmbed.js

@@ -0,0 +1,47 @@
+import { Rate } from "k6/metrics";
+import { check, group, sleep } from 'k6';
+import http from 'k6/http';
+
+const URL = "http://" + __ENV.HOST;
+
+const RATE = (function(){
+ if(__ENV.VERSION == 'v701'){
+ switch(__ENV.HOST){
+ case 'c5xlarge': return 1100;
+ case 't3axlarge': return 850;
+ case 't3anano': return 600;
+ default: return 1000;
+ }
+ }
+ else switch(__ENV.HOST){
+ case 'c5xlarge': return 1200;
+ case 't3axlarge': return 1000;
+ case 't3anano': return 700;
+ default: return 1000;
+ }
+})();
+
+export let options = {
+ discardResponseBodies: true,
+ scenarios: {
+ constant_request_rate: {
+ executor: 'constant-arrival-rate',
+ rate: RATE,
+ timeUnit: '1s',
+ duration: '30s',
+ preAllocatedVUs: 100,
+ maxVUs: 600,
+ }
+ },
+ thresholds: {
+ 'failed requests': ['rate<0.1'],
+ 'http_req_duration': ['p(95)<1000']
+ }
+};
+
+const myFailRate = new Rate('failed requests');
+
+export default function() {
+ let res = http.get(URL + "/rpc/ret_albums?select=album_id,title,artist_id,track(*,genre(*))&artist_id=eq.127");
+ myFailRate.add(res.status !== 200);
+}

k6/RPCSimple.js

@@ -0,0 +1,49 @@
+import { Rate } from "k6/metrics";
+import { check, group, sleep } from 'k6';
+import http from 'k6/http';
+
+const URL = "http://" + __ENV.HOST;
+
+const RATE = (function(){
+ if(__ENV.VERSION == 'v701'){
+ switch(__ENV.HOST){
+ case 'c5xlarge': return 2400;
+ case 't3axlarge': return 1600;
+ case 't3anano': return 1600;
+ default: return 1500;
+ }
+ }
+ else switch(__ENV.HOST){
+ case 'c5xlarge': return 3000;
+ case 't3axlarge': return 2200;
+ case 't3anano': return 2000;
+ default: return 1000;
+ }
+})();
+
+export let options = {
+ discardResponseBodies: true,
+ scenarios: {
+ constant_request_rate: {
+ executor: 'constant-arrival-rate',
+ rate: RATE,
+ timeUnit: '1s',
+ duration: '30s',
+ preAllocatedVUs: 100,
+ maxVUs: 600,
+ }
+ },
+ thresholds: {
+ 'failed requests': ['rate<0.1'],
+ 'http_req_duration': ['p(95)<1000']
+ }
+};
+
+const myFailRate = new Rate('failed requests');
+
+export default function() {
+ //let res = http.get(URL + "/rpc/add_them1?a=1&b=2&c=3&d=4&e=5");
+ let res = http.get(URL + "/rpc/add_them2?a=1&b=2&c=3&d=4&e=5&v=0");
+ //let res = http.get(URL + "/rpc/add_them3?v=1&v=2&v=3&v=4&v=5");
+ myFailRate.add(res.status !== 200);
+}

sql/chinook.sql

@@ -18,6 +18,7 @@
  Modified for postgrest-benchmark:
  Summary of Changes:
  - Removed employee table indexes and constraints
+ - Added add_them, ret_albums and ret_artists functions
 ********************************************************************************/
 
 
@@ -23861,3 +23862,15 @@ ALTER TABLE "film_actor" ADD PRIMARY KEY ("actor_id", "film_id");
 alter table employee drop constraint pk_employee cascade;
 drop index ifk_employee_reports_to;
 drop sequence employee_employee_id_seq cascade;
+
+create or replace function ret_albums() returns setof album as $$
+ select * from album
+$$ language sql;
+
+create or replace function ret_artists() returns setof artist as $$
+ select * from artist
+$$ language sql;
+
+create or replace function add_them(a int, b int, c int, d int, e int) returns int as $$
+ select a + b + c + d + e;
+$$ language sql;