cli

README

Rockset CLI

Usage

The Rockset cli can be used as an alternative to the console, and is built as a UNIX tool to allow it to be used in pipes.

Query

There are three ways to query a collection, either using the first argument as the SQL

$ rockset query 'SELECT COUNT(*) FROM _events'
+--------+
| ?COUNT |
+--------+
|   1016 |
+--------+
Elapsed time: 26 ms

Or using interactive mode

$ rockset query
[R]> SELECT COUNT(*)
>>> FROM _events;
+--------+
| ?COUNT |
+--------+
|   1016 |
+--------+
Elapsed time: 26 ms
^D

And reading the SQL from stdin

$ rockset query < query.sql
+--------+
| ?COUNT |
+--------+
|   1016 |
+--------+
Elapsed time: 26 ms

Cloning a collection

A common workflow is to want to clone a collection, but change a few settings, e.g. the retention. This can be done using two commands

rockset get collection --output - movies | rockset create collection --input - --retention 12h movies2

Configuration

The Rockset CLI requires having access to either an API key or a bearer token, together with an apiserver, and in the case of a bearer token also the organization.

These are called an authentication context.

$ rockset list contexts
Available Authentication Contexts:
apikeys:
   dev (https://api.usw2a1.dev.rockset.com)
   test (https://api.use1a1.rockset.com)
bearer tokens:
-> usw2a1 (https://api.usw2a1.rockset.com)
$ rockset use context prod
using prod
$ rockset list contexts
Available Authentication Contexts:
apikeys:
   dev (https://api.usw2a1.dev.rockset.com)
-> test (https://api.use1a1.rockset.com)
bearer tokens:
   usw2a1 (https://api.usw2a1.rockset.com)

Configuration File

The configuration file is stored in ~/.config/rockset/config.yaml

---
current: dev
keys:
  dev:
    apikey: ...
    apiserver: api.usw2a1.dev.rockset.com
tokens:
  prod:
    token: ...
    org: rockset-test
    apiserver: api.usw2a1.dev.rockset.com
    expiration: 2023-10-17T08:13:48.785337-07:00

Building

go build -o rockset

Testing

go test ./...

Integration testing

Requires the environment variable ROCKSET_APIKEY to be set

go test ./...

Create recordings

We use vhs to record terminal sessions

vhs vhs/demo.tape