README.md

@@ -48,10 +48,11 @@ Create native client with `connect` method. Arguments:
 - dial timeout in seconds (0 for the default value)
 - stream timeout in seconds (0 for the default value)
 - generate object header on the client side (for big object - split locally too)
+- max size for generated object header on the client side (for big object - the size that the object is splitted into)
 
 ```js
 import native from 'k6/x/frostfs/native';
-const frostfs_cli = native.connect("s01.frostfs.devenv:8080", "", 0, 0, false)
+const frostfs_cli = native.connect("s01.frostfs.devenv:8080", "", 0, 0, false, 0)
 ```
 
 ### Methods

examples/native.js

@@ -3,11 +3,11 @@ import { fail } from "k6";
 import { uuidv4 } from '../scenarios/libs/k6-utils-1.4.0.js';
 
 const payload = open('../go.sum', 'b');
-const frostfs_cli = native.connect("s01.frostfs.devenv:8080", "1dd37fba80fec4e6a6f13fd708d8dcb3b29def768017052f6c930fa1c5d90bbb", 0, 0, false)
+const frostfs_cli = native.connect("s01.frostfs.devenv:8080", "1dd37fba80fec4e6a6f13fd708d8dcb3b29def768017052f6c930fa1c5d90bbb", 0, 0, false, 0)
 
 export const options = {
     stages: [
-        {duration: '30s', target: 10},
+        { duration: '30s', target: 10 },
     ],
 };
 
@@ -24,7 +24,7 @@ export function setup() {
         fail(res.error)
     }
     console.info("created container", res.container_id)
-    return {container_id: res.container_id}
+    return { container_id: res.container_id }
 }
 
 export default function (data) {

examples/native_onsite.js

@@ -3,7 +3,7 @@ import { uuidv4 } from '../scenarios/libs/k6-utils-1.4.0.js';
 
 const payload = open('../go.sum', 'b');
 const container = "AjSxSNNXbJUDPqqKYm1VbFVDGCakbpUNH8aGjPmGAH3B"
-const frostfs_cli = native.connect("s01.frostfs.devenv:8080", "", 0, 0, false)
+const frostfs_cli = native.connect("s01.frostfs.devenv:8080", "", 0, 0, false, 0)
 const frostfs_obj = frostfs_cli.onsite(container, payload)
 
 export const options = {

internal/native/client.go

@@ -35,6 +35,7 @@ type (
 		tok            session.Object
 		cli            *client.Client
 		prepareLocally bool
+		maxObjSize     uint64
 	}
 
 	PutResponse struct {
@@ -71,6 +72,7 @@ type (
 		hdr            object.Object
 		payload        []byte
 		prepareLocally bool
+		maxObjSize     uint64
 	}
 )
 
@@ -103,7 +105,7 @@ func (c *Client) Put(containerID string, headers map[string]string, payload data
 	o.SetOwnerID(owner)
 	o.SetAttributes(attrs...)
 
-	resp, err := put(c.vu, c.cli, c.prepareLocally, &tok, &o, payload, chunkSize)
+	resp, err := put(c.vu, c.cli, c.prepareLocally, &tok, &o, payload, chunkSize, c.maxObjSize)
 	if err != nil {
 		return PutResponse{Success: false, Error: err.Error()}
 	}
@@ -373,6 +375,7 @@ func (c *Client) Onsite(containerID string, payload datagen.Payload) PreparedObj
 		hdr:            *obj,
 		payload:        data,
 		prepareLocally: c.prepareLocally,
+		maxObjSize:     c.maxObjSize,
 	}
 }
 
@@ -398,7 +401,7 @@ func (p PreparedObject) Put(headers map[string]string) PutResponse {
 		return PutResponse{Success: false, Error: err.Error()}
 	}
 
-	_, err = put(p.vu, p.cli, p.prepareLocally, nil, &obj, datagen.NewFixedPayload(p.payload), 0)
+	_, err = put(p.vu, p.cli, p.prepareLocally, nil, &obj, datagen.NewFixedPayload(p.payload), 0, p.maxObjSize)
 	if err != nil {
 		return PutResponse{Success: false, Error: err.Error()}
 	}
@@ -413,7 +416,7 @@ func (s epochSource) CurrentEpoch() uint64 {
 }
 
 func put(vu modules.VU, cli *client.Client, prepareLocally bool, tok *session.Object,
-	hdr *object.Object, payload datagen.Payload, chunkSize int,
+	hdr *object.Object, payload datagen.Payload, chunkSize int, maxObjSize uint64,
 ) (*client.ResObjectPut, error) {
 	bufSize := defaultBufferSize
 	if chunkSize > 0 {
@@ -441,6 +444,9 @@ func put(vu modules.VU, cli *client.Client, prepareLocally bool, tok *session.Ob
 		prm.MaxSize = res.Info().MaxObjectSize()
 		prm.EpochSource = epochSource(res.Info().CurrentEpoch())
 		prm.WithoutHomomorphHash = true
+		if maxObjSize > 0 {
+			prm.MaxSize = maxObjSize
+		}
 	}
 
 	objectWriter, err := cli.ObjectPutInit(vu.Context(), prm)

internal/native/native.go

@@ -52,13 +52,17 @@ func (n *Native) Exports() modules.Exports {
 	return modules.Exports{Default: n}
 }
 
-func (n *Native) Connect(endpoint, hexPrivateKey string, dialTimeout, streamTimeout int, prepareLocally bool) (*Client, error) {
+func (n *Native) Connect(endpoint, hexPrivateKey string, dialTimeout, streamTimeout int, prepareLocally bool, maxObjSize int) (*Client, error) {
 	var (
 		cli client.Client
 		pk  *keys.PrivateKey
 		err error
 	)
 
+	if maxObjSize < 0 {
+		return nil, fmt.Errorf("max object size value must be positive")
+	}
+
 	pk, err = keys.NewPrivateKey()
 	if len(hexPrivateKey) != 0 {
 		pk, err = keys.NewPrivateKeyFromHex(hexPrivateKey)
@@ -114,6 +118,16 @@ func (n *Native) Connect(endpoint, hexPrivateKey string, dialTimeout, streamTime
 	tok.SetAuthKey(&key)
 	tok.SetExp(exp)
 
+	if prepareLocally && maxObjSize > 0 {
+		res, err := cli.NetworkInfo(n.vu.Context(), client.PrmNetworkInfo{})
+		if err != nil {
+			return nil, err
+		}
+		if uint64(maxObjSize) > res.Info().MaxObjectSize() {
+			return nil, fmt.Errorf("max object size must be not greater than %d bytes", res.Info().MaxObjectSize())
+		}
+	}
+
 	// register metrics
 
 	objPutSuccess, _ = stats.Registry.NewMetric("frostfs_obj_put_success", metrics.Counter)
@@ -140,5 +154,6 @@ func (n *Native) Connect(endpoint, hexPrivateKey string, dialTimeout, streamTime
 		tok:            tok,
 		cli:            &cli,
 		prepareLocally: prepareLocally,
+		maxObjSize:     uint64(maxObjSize),
 	}, nil
 }

scenarios/grpc.js

@@ -1,25 +1,25 @@
-import {sleep} from 'k6';
+import { sleep } from 'k6';
-import {SharedArray} from 'k6/data';
+import { SharedArray } from 'k6/data';
 import exec from 'k6/execution';
 import logging from 'k6/x/frostfs/logging';
 import native from 'k6/x/frostfs/native';
 import registry from 'k6/x/frostfs/registry';
 import stats from 'k6/x/frostfs/stats';
 
-import {newGenerator} from './libs/datagen.js';
+import { newGenerator } from './libs/datagen.js';
-import {parseEnv} from './libs/env-parser.js';
+import { parseEnv } from './libs/env-parser.js';
-import {textSummary} from './libs/k6-summary-0.0.2.js';
+import { textSummary } from './libs/k6-summary-0.0.2.js';
-import {uuidv4} from './libs/k6-utils-1.4.0.js';
+import { uuidv4 } from './libs/k6-utils-1.4.0.js';
 
 parseEnv();
 
 const obj_list = new SharedArray(
   'obj_list',
-    function() { return JSON.parse(open(__ENV.PREGEN_JSON)).objects; });
+  function () { return JSON.parse(open(__ENV.PREGEN_JSON)).objects; });
 
 const container_list = new SharedArray(
   'container_list',
-    function() { return JSON.parse(open(__ENV.PREGEN_JSON)).containers; });
+  function () { return JSON.parse(open(__ENV.PREGEN_JSON)).containers; });
 
 const read_size = JSON.parse(open(__ENV.PREGEN_JSON)).obj_size;
 const summary_json = __ENV.SUMMARY_JSON || '/tmp/summary.json';
@@ -31,8 +31,8 @@ const grpc_endpoint =
 const grpc_client = native.connect(
   grpc_endpoint, '', __ENV.DIAL_TIMEOUT ? parseInt(__ENV.DIAL_TIMEOUT) : 5,
   __ENV.STREAM_TIMEOUT ? parseInt(__ENV.STREAM_TIMEOUT) : 60,
-    __ENV.PREPARE_LOCALLY ? __ENV.PREPARE_LOCALLY.toLowerCase() === 'true'
+  __ENV.PREPARE_LOCALLY ? __ENV.PREPARE_LOCALLY.toLowerCase() === 'true' : false,
-                          : false);
+  1024 * parseInt(__ENV.MAX_OBJECT_SIZE || '0'));
 const log = logging.new().withField('endpoint', grpc_endpoint);
 
 const registry_enabled = !!__ENV.REGISTRY_FILE;
@@ -51,8 +51,8 @@ if (registry_enabled) {
   obj_to_read_selector = registry.getLoopedSelector(
     __ENV.REGISTRY_FILE, 'obj_to_read',
     __ENV.SELECTION_SIZE ? parseInt(__ENV.SELECTION_SIZE) : 0, {
-        status : 'created',
+    status: 'created',
-        age : read_age,
+    age: read_age,
   })
 }
 
@@ -63,11 +63,11 @@ const write_grpc_chunk_size = 1024 * parseInt(__ENV.GRPC_CHUNK_SIZE || '0')
 const generator = newGenerator(write_vu_count > 0);
 if (write_vu_count > 0) {
   scenarios.write = {
-    executor : 'constant-vus',
+    executor: 'constant-vus',
-    vus : write_vu_count,
+    vus: write_vu_count,
-    duration : `${duration}s`,
+    duration: `${duration}s`,
-    exec : 'obj_write',
+    exec: 'obj_write',
-    gracefulStop : '5s',
+    gracefulStop: '5s',
   };
 }
 
@@ -83,19 +83,19 @@ if (registry_enabled && delete_age) {
   obj_to_delete_selector =
     constructor(__ENV.REGISTRY_FILE, 'obj_to_delete',
       __ENV.SELECTION_SIZE ? parseInt(__ENV.SELECTION_SIZE) : 0, {
-                    status : 'created',
+      status: 'created',
-                    age : delete_age,
+      age: delete_age,
     });
 }
 
 const read_vu_count = parseInt(__ENV.READERS || '0');
 if (read_vu_count > 0) {
   scenarios.read = {
-    executor : 'constant-vus',
+    executor: 'constant-vus',
-    vus : read_vu_count,
+    vus: read_vu_count,
-    duration : `${duration}s`,
+    duration: `${duration}s`,
-    exec : 'obj_read',
+    exec: 'obj_read',
-    gracefulStop : '5s',
+    gracefulStop: '5s',
   };
 }
 
@@ -107,17 +107,17 @@ if (delete_vu_count > 0) {
   }
 
   scenarios.delete = {
-    executor : 'constant-vus',
+    executor: 'constant-vus',
-    vus : delete_vu_count,
+    vus: delete_vu_count,
-    duration : `${duration}s`,
+    duration: `${duration}s`,
-    exec : 'obj_delete',
+    exec: 'obj_delete',
-    gracefulStop : '5s',
+    gracefulStop: '5s',
   };
 }
 
 export const options = {
   scenarios,
-  setupTimeout : '5s',
+  setupTimeout: '5s',
 };
 
 export function setup() {
@@ -147,8 +147,8 @@ export function teardown(data) {
 
 export function handleSummary(data) {
   return {
-    'stdout' : textSummary(data, {indent : ' ', enableColors : false}),
+    'stdout': textSummary(data, { indent: ' ', enableColors: false }),
-    [summary_json] : JSON.stringify(data),
+    [summary_json]: JSON.stringify(data),
   };
 }
 
@@ -157,7 +157,7 @@ export function obj_write() {
     sleep(__ENV.SLEEP_WRITE);
   }
 
-  const headers = {unique_header : uuidv4()};
+  const headers = { unique_header: uuidv4() };
   const container =
     container_list[Math.floor(Math.random() * container_list.length)];
 
@@ -186,7 +186,7 @@ export function obj_read() {
     }
     const resp = grpc_client.get(obj.c_id, obj.o_id)
     if (!resp.success) {
-      log.withFields({cid : obj.c_id, oid : obj.o_id}).error(resp.error);
+      log.withFields({ cid: obj.c_id, oid: obj.o_id }).error(resp.error);
     }
     return
   }
@@ -194,7 +194,7 @@ export function obj_read() {
   const obj = obj_list[Math.floor(Math.random() * obj_list.length)];
   const resp = grpc_client.get(obj.container, obj.object)
   if (!resp.success) {
-    log.withFields({cid : obj.container, oid : obj.object}).error(resp.error);
+    log.withFields({ cid: obj.container, oid: obj.object }).error(resp.error);
   }
 }
 
@@ -214,7 +214,7 @@ export function obj_delete() {
   const resp = grpc_client.delete(obj.c_id, obj.o_id);
   if (!resp.success) {
     // Log errors except (2052 - object already deleted)
-    log.withFields({cid : obj.c_id, oid : obj.o_id}).error(resp.error);
+    log.withFields({ cid: obj.c_id, oid: obj.o_id }).error(resp.error);
     return;
   }
 

scenarios/grpc_car.js

@@ -1,22 +1,22 @@
-import {sleep} from 'k6';
+import { sleep } from 'k6';
-import {SharedArray} from 'k6/data';
+import { SharedArray } from 'k6/data';
 import logging from 'k6/x/frostfs/logging';
 import native from 'k6/x/frostfs/native';
 import registry from 'k6/x/frostfs/registry';
 import stats from 'k6/x/frostfs/stats';
 
-import {newGenerator} from './libs/datagen.js';
+import { newGenerator } from './libs/datagen.js';
-import {parseEnv} from './libs/env-parser.js';
+import { parseEnv } from './libs/env-parser.js';
-import {textSummary} from './libs/k6-summary-0.0.2.js';
+import { textSummary } from './libs/k6-summary-0.0.2.js';
-import {uuidv4} from './libs/k6-utils-1.4.0.js';
+import { uuidv4 } from './libs/k6-utils-1.4.0.js';
 
 parseEnv();
 
-const obj_list = new SharedArray('obj_list', function() {
+const obj_list = new SharedArray('obj_list', function () {
   return JSON.parse(open(__ENV.PREGEN_JSON)).objects;
 });
 
-const container_list = new SharedArray('container_list', function() {
+const container_list = new SharedArray('container_list', function () {
   return JSON.parse(open(__ENV.PREGEN_JSON)).containers;
 });
 
@@ -30,8 +30,8 @@ const grpc_endpoint =
 const grpc_client = native.connect(
   grpc_endpoint, '', __ENV.DIAL_TIMEOUT ? parseInt(__ENV.DIAL_TIMEOUT) : 5,
   __ENV.STREAM_TIMEOUT ? parseInt(__ENV.STREAM_TIMEOUT) : 60,
-    __ENV.PREPARE_LOCALLY ? __ENV.PREPARE_LOCALLY.toLowerCase() === 'true' :
+  __ENV.PREPARE_LOCALLY ? __ENV.PREPARE_LOCALLY.toLowerCase() === 'true' : false,
-                            false);
+  1024 * parseInt(__ENV.MAX_OBJECT_SIZE || '0'));
 const log = logging.new().withField('endpoint', grpc_endpoint);
 
 const registry_enabled = !!__ENV.REGISTRY_FILE;
@@ -166,7 +166,7 @@ export function teardown(data) {
 
 export function handleSummary(data) {
   return {
-    'stdout': textSummary(data, {indent: ' ', enableColors: false}),
+    'stdout': textSummary(data, { indent: ' ', enableColors: false }),
     [summary_json]: JSON.stringify(data),
   };
 }
@@ -176,7 +176,7 @@ export function obj_write() {
     sleep(__ENV.SLEEP_WRITE);
   }
 
-  const headers = {unique_header: uuidv4()};
+  const headers = { unique_header: uuidv4() };
   const container =
     container_list[Math.floor(Math.random() * container_list.length)];
 
@@ -205,7 +205,7 @@ export function obj_read() {
     }
     const resp = grpc_client.get(obj.c_id, obj.o_id)
     if (!resp.success) {
-      log.withFields({cid: obj.c_id, oid: obj.o_id}).error(resp.error);
+      log.withFields({ cid: obj.c_id, oid: obj.o_id }).error(resp.error);
     }
     return
   }
@@ -213,7 +213,7 @@ export function obj_read() {
   const obj = obj_list[Math.floor(Math.random() * obj_list.length)];
   const resp = grpc_client.get(obj.container, obj.object)
   if (!resp.success) {
-    log.withFields({cid: obj.container, oid: obj.object}).error(resp.error);
+    log.withFields({ cid: obj.container, oid: obj.object }).error(resp.error);
   }
 }
 
@@ -230,7 +230,7 @@ export function obj_delete() {
   const resp = grpc_client.delete(obj.c_id, obj.o_id);
   if (!resp.success) {
     // Log errors except (2052 - object already deleted)
-    log.withFields({cid: obj.c_id, oid: obj.o_id}).error(resp.error);
+    log.withFields({ cid: obj.c_id, oid: obj.o_id }).error(resp.error);
     return;
   }
 

scenarios/verify.js

@@ -1,13 +1,13 @@
-import {sleep} from 'k6';
+import { sleep } from 'k6';
-import {Counter} from 'k6/metrics';
+import { Counter } from 'k6/metrics';
 import logging from 'k6/x/frostfs/logging';
 import native from 'k6/x/frostfs/native';
 import registry from 'k6/x/frostfs/registry';
 import s3 from 'k6/x/frostfs/s3';
 import stats from 'k6/x/frostfs/stats';
 
-import {parseEnv} from './libs/env-parser.js';
+import { parseEnv } from './libs/env-parser.js';
-import {textSummary} from './libs/k6-summary-0.0.2.js';
+import { textSummary } from './libs/k6-summary-0.0.2.js';
 
 parseEnv();
 
@@ -44,16 +44,15 @@ if (__ENV.GRPC_ENDPOINTS) {
   grpc_client = native.connect(
     grpcEndpoint, '', __ENV.DIAL_TIMEOUT ? parseInt(__ENV.DIAL_TIMEOUT) : 0,
     __ENV.STREAM_TIMEOUT ? parseInt(__ENV.STREAM_TIMEOUT) : 0,
-      __ENV.PREPARE_LOCALLY ? __ENV.PREPARE_LOCALLY.toLowerCase() === 'true' :
+    __ENV.PREPARE_LOCALLY ? __ENV.PREPARE_LOCALLY.toLowerCase() === 'true' : false,
-                              false,
+    1024 * parseInt(__ENV.MAX_OBJECT_SIZE || '0'));
-      '');
 }
 
 // Connect to random S3 endpoint
 let s3_client = undefined;
 if (__ENV.S3_ENDPOINTS) {
   const no_verify_ssl = __ENV.NO_VERIFY_SSL || 'true';
-  const connection_args = {no_verify_ssl: no_verify_ssl};
+  const connection_args = { no_verify_ssl: no_verify_ssl };
   const s3_endpoints = __ENV.S3_ENDPOINTS.split(',');
   const s3_endpoint =
     s3_endpoints[Math.floor(Math.random() * s3_endpoints.length)];
@@ -68,7 +67,7 @@ const obj_to_verify_selector = registry.getSelector(
   __ENV.REGISTRY_FILE, 'obj_to_verify',
   __ENV.SELECTION_SIZE ? parseInt(__ENV.SELECTION_SIZE) : 0, {
   status: 'created',
-    });
+});
 const obj_to_verify_count = obj_to_verify_selector.count();
 // Execute at least one iteration (executor shared-iterations can't run 0
 // iterations)
@@ -98,14 +97,14 @@ export function setup() {
   for (const [status, counter] of Object.entries(obj_counters)) {
     const obj_selector = registry.getSelector(
       __ENV.REGISTRY_FILE, status,
-        __ENV.SELECTION_SIZE ? parseInt(__ENV.SELECTION_SIZE) : 0, {status});
+      __ENV.SELECTION_SIZE ? parseInt(__ENV.SELECTION_SIZE) : 0, { status });
     counter.add(obj_selector.count());
   }
 }
 
 export function handleSummary(data) {
   return {
-    'stdout': textSummary(data, {indent: ' ', enableColors: false}),
+    'stdout': textSummary(data, { indent: ' ', enableColors: false }),
     [summary_json]: JSON.stringify(data),
   };
 }
@@ -138,10 +137,10 @@ function verify_object_with_retries(obj, attempts) {
     // ReferenceError: Cannot access a variable before initialization.
     let lg = log;
     if (obj.c_id && obj.o_id) {
-      lg = lg.withFields({cid: obj.c_id, oid: obj.o_id});
+      lg = lg.withFields({ cid: obj.c_id, oid: obj.o_id });
       result = grpc_client.verifyHash(obj.c_id, obj.o_id, obj.payload_hash);
     } else if (obj.s3_bucket && obj.s3_key) {
-      lg = lg.withFields({bucket: obj.s3_bucket, key: obj.s3_key});
+      lg = lg.withFields({ bucket: obj.s3_bucket, key: obj.s3_key });
       result =
         s3_client.verifyHash(obj.s3_bucket, obj.s3_key, obj.payload_hash);
     } else {