I think that's more up to CockroachDB. They offer a postgres compatible interface so this should be usable with CockroachDB already.

I don't know CockroachDB, but if there is a Go sql driver (compatible with database/sql") and it uses standard SQL, you can write an adapter, see : https://github.com/samonzeweb/godb/tree/master/adapters