initial commit

Author: xiam

.gitignore

@@ -0,0 +1,3 @@
+certs
+private
+cockroach-data

Makefile

@@ -0,0 +1,21 @@
+clean:
+	rm -rf cockroach-data certs private
+
+secure-node: clean
+	mkdir -p certs private && \
+	cockroach cert create-ca --certs-dir=certs --ca-key=private/ca.key && \
+	cockroach cert create-node localhost --certs-dir=certs --ca-key=private/ca.key && \
+	cockroach cert create-client root --certs-dir=certs --ca-key=private/ca.key && \
+	cockroach cert create-client maxroach --certs-dir=certs --ca-key=private/ca.key && \
+	cockroach start --certs-dir=certs
+
+insecure-node: clean
+	cockroach start --insecure
+
+bank-database-insecure:
+	cockroach sql --insecure -e 'CREATE USER IF NOT EXISTS maxroach; CREATE DATABASE IF NOT EXISTS bank; GRANT ALL ON DATABASE bank TO maxroach;'
+	cockroach sql --insecure --database bank -e 'CREATE TABLE IF NOT EXISTS accounts (ID SERIAL PRIMARY KEY, balance INT);'
+
+bank-database-secure:
+	cockroach sql --certs-dir=certs -e 'CREATE USER IF NOT EXISTS maxroach; CREATE DATABASE IF NOT EXISTS bank; GRANT ALL ON DATABASE bank TO maxroach;'
+	cockroach sql --certs-dir=certs --database bank -e 'CREATE TABLE IF NOT EXISTS accounts (ID SERIAL PRIMARY KEY, balance INT);'

README.md

@@ -0,0 +1,59 @@
+# upper/db + CockroachDB example
+
+## Run the insecure node example
+
+Create and run an insecure node:
+
+```
+make insecure-node
+```
+
+Open a second terminal and create the bank database:
+
+```
+make bank-database-insecure
+```
+
+Run the `main.go` example:
+
+```
+go run main.go
+```
+
+## Run the secure node example
+
+Create and run a secure node:
+
+```
+make secure-node
+```
+
+Open a second terminal and create the bank database:
+
+```
+make bank-database-secure
+```
+
+Edit the `main.go` file and add options for secure connection:
+
+```
+var settings = cockroachdb.ConnectionURL{
+	Host:     "localhost",
+	Database: "bank",
+	User:     "maxroach",
+	Options: map[string]string{
+		// Insecure node.
+		// "sslmode": "disable",
+		// Secure node.
+		"sslrootcert": "certs/ca.crt",
+		"sslkey":      "certs/client.maxroach.key",
+		"sslcert":     "certs/client.maxroach.crt",
+	},
+}
+```
+
+Run the `main.go` example:
+
+```
+go run main.go
+```

main.go

@@ -0,0 +1,160 @@
+package main
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/upper/db/v4"
+	"github.com/upper/db/v4/adapter/cockroachdb"
+)
+
+// The settings variable stores connection details.
+var settings = cockroachdb.ConnectionURL{
+	Host:     "localhost",
+	Database: "bank",
+	User:     "maxroach",
+	Options: map[string]string{
+		// Insecure node.
+		"sslmode": "disable",
+		// Secure node.
+		// "sslrootcert": "certs/ca.crt",
+		// "sslkey":      "certs/client.maxroach.key",
+		// "sslcert":     "certs/client.maxroach.crt",
+	},
+}
+
+// Accounts is a handy way to represent a collection.
+func Accounts(sess db.Session) db.Store {
+	return sess.Collection("accounts")
+}
+
+// Account is used to represent a single record in the "accounts" table.
+type Account struct {
+	ID      uint64 `db:"id,omitempty"`
+	Balance int64  `db:"balance"`
+}
+
+// Collection is required in order to create a relation between the Account
+// struct and the "accounts" table.
+func (a *Account) Store(sess db.Session) db.Store {
+	return Accounts(sess)
+}
+
+// crdbForceRetry can be used to simulate a transaction error and
+// demonstrate upper/db's ability to retry the transaction automatically.
+//
+// By default, upper/db will retry the transaction five times, if you want
+// to modify this number use: sess.SetMaxTransactionRetries(n).
+//
+// This is only used for demonstration purposes.
+func crdbForceRetry(sess db.Session) error {
+	var err error
+
+	_, err = sess.SQL().Exec(`SELECT NOW()`)
+	if err != nil {
+		return err
+	}
+
+	_, err = sess.SQL().Exec(`SELECT crdb_internal.force_retry('1ms'::INTERVAL)`)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func main() {
+	// Connect to the local CockroachDB node.
+	sess, err := cockroachdb.Open(settings)
+	if err != nil {
+		log.Fatal("cockroachdb.Open: ", err)
+	}
+	defer sess.Close()
+
+	// Adjust this number to fit your specific needs (set to 5, by default)
+	// sess.SetMaxTransactionRetries(10)
+
+	// Delete all the previous items in the "accounts" table.
+	err = Accounts(sess).Truncate()
+	if err != nil {
+		log.Fatal("Truncate: ", err)
+	}
+
+	// Create a new account with a balance of 1000.
+	account1 := Account{Balance: 1000}
+	err = Accounts(sess).InsertReturning(&account1)
+	if err != nil {
+		log.Fatal("sess.Save: ", err)
+	}
+
+	// Create a new account with a balance of 250.
+	account2 := Account{Balance: 250}
+	err = Accounts(sess).InsertReturning(&account2)
+	if err != nil {
+		log.Fatal("sess.Save: ", err)
+	}
+
+	// Printing records
+	printRecords(sess)
+
+	// Change the balance of the first account.
+	account1.Balance = 500
+	err = sess.Save(&account1)
+	if err != nil {
+		log.Fatal("sess.Save: ", err)
+	}
+
+	// Change the balance of the second account.
+	account2.Balance = 999
+	err = sess.Save(&account2)
+	if err != nil {
+		log.Fatal("sess.Save: ", err)
+	}
+
+	// Printing records
+	printRecords(sess)
+
+	// Delete the first record.
+	err = sess.Delete(&account1)
+	if err != nil {
+		log.Fatal("Delete: ", err)
+	}
+
+	// Add a couple of new records within a transaction.
+	err = sess.Tx(func(tx db.Session) error {
+		var err error
+
+		// Increases the possibility of transaction failure
+		_ = crdbForceRetry(tx)
+
+		err = tx.Save(&Account{Balance: 887})
+		if err != nil {
+			return err
+		}
+
+		err = tx.Save(&Account{Balance: 342})
+		if err != nil {
+			return err
+		}
+
+		return nil
+	})
+	if err != nil {
+		log.Fatal("Could not commit transaction: ", err)
+	}
+
+	// Printing records
+	printRecords(sess)
+}
+
+func printRecords(sess db.Session) {
+	accounts := []Account{}
+	err := Accounts(sess).Find().All(&accounts)
+	if err != nil {
+		log.Fatal("Find: ", err)
+	}
+	log.Printf("Balances:")
+	for i := range accounts {
+		fmt.Printf("\taccounts[%d]: %d\n", accounts[i].ID, accounts[i].Balance)
+	}
+}