nosqlbench/driver-cockroachdb/src/main/resources/activities/cockroachdb-basic.yaml

# java -jar nb.jar run driver=cockroachdb workload=cockroachdb-basic tags=phase:rampup cycles=10 serverName=localhost
# java -jar nb.jar run driver=cockroachdb workload=cockroachdb-basic tags=phase:main cycles=10 serverName=localhost
description: An example of a basic cockroach insert

scenarios:
  default:
    - run driver=cockroachdb tags==phase:main threads=auto cycles===<<main-cycles:1000000>>
      serverName=localhost connectionpool=hikari
      errors=SQLTransient.*:warn,count,retry;.*0800.*:warn,count,retry;.*40001:count,retry;stop
  rampup:
    - run driver=cockroachdb tags==phase:rampup threads=auto cycles===<<rampup-cycles:1000000>>
      serverName=localhost connectionpool=hikari
      errors=SQLTransient.*:warn,count,retry;.*0800.*:warn,count,retry;.*40001:count,retry;stop
  schema:
    - run driver=cockroachdb tags==phase:schema threads===1 serverName=localhost

bindings:
  seq_key: Mod(<<keyCount:1000000>>L); ToInt()
  seq_value: Mod(<<valueCount:1000000000>>L); <<valueSizeDist:Hash()>>; ToString() -> String
  rw_key: <<keyDist:Uniform(0,1000000)->long>>; ToInt()
  rw_value: <<valDist:Uniform(0,1000000000)->int>>; <<valueSizeDist:Hash()>>; ToString() -> String

blocks:
  - name: schema
    tags:
      phase: schema
    params:
    statements:
      - create-database: |
          CREATE DATABASE <<database:bank>>;
        tags:
          name: create-database
      - create-table: |
          CREATE TABLE IF NOT EXISTS <<database:bank>>.<<table:banktransaction>> (
            code STRING PRIMARY KEY,
            amount INTEGER
          );
        tags:
          name: create-table
  - name: rampup
    tags:
      phase: rampup
    params:
    statements:
      - rampup-insert: |
          INSERT INTO "<<database:bank>>"."<<table:banktransaction>>"
          (code, amount) VALUES ('{seq_key}', {seq_value})
          ON CONFLICT (code) DO NOTHING;
        params:
        tags:
          name: rampup-insert
  - name: main-read
    tags:
      phase: main
      type: read
    params:
      ratio: <<read_ratio:1>>
    statements:
      - main-find: |
          SELECT code, amount FROM "<<database:bank>>"."<<table:banktransaction>>"
          WHERE code = '{rw_key}' AND amount = {rw_value};
        params:
        tags:
          name: main-find
  - name: main-write
    tags:
      phase: main
      type: write
    params:
      ratio: <<write_ratio:1>>
    statements:
      - main-insert: |
          UPDATE "<<database:bank>>"."<<table:banktransaction>>" SET amount = {seq_value} WHERE code = '{seq_key}';
        params:
        tags:
          name: main-insert