550 lines
9.6 KiB
Go
550 lines
9.6 KiB
Go
package storm_test
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/asdine/storm"
|
|
"github.com/boltdb/bolt"
|
|
)
|
|
|
|
func ExampleDB_Save() {
|
|
dir, _ := ioutil.TempDir(os.TempDir(), "storm")
|
|
defer os.RemoveAll(dir)
|
|
|
|
type User struct {
|
|
ID int `storm:"id"`
|
|
Group string `storm:"index"`
|
|
Email string `storm:"unique"`
|
|
Name string
|
|
Age int `storm:"index"`
|
|
CreatedAt time.Time `storm:"index"`
|
|
}
|
|
|
|
// Open takes an optional list of options as the last argument.
|
|
// AutoIncrement will auto-increment integer IDs without existing values.
|
|
db, _ := storm.Open(filepath.Join(dir, "storm.db"), storm.AutoIncrement())
|
|
defer db.Close()
|
|
|
|
user := User{
|
|
Group: "staff",
|
|
Email: "john@provider.com",
|
|
Name: "John",
|
|
Age: 21,
|
|
CreatedAt: time.Now(),
|
|
}
|
|
|
|
err := db.Save(&user)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println(user.ID)
|
|
|
|
user2 := user
|
|
user2.ID = 0
|
|
|
|
// Save will fail because of the unique constraint on Email
|
|
err = db.Save(&user2)
|
|
fmt.Println(err)
|
|
|
|
// Output:
|
|
// 1
|
|
// already exists
|
|
}
|
|
|
|
func ExampleDB_One() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var user User
|
|
|
|
err := db.One("Email", "john@provider.com", &user)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// also works on unindexed fields
|
|
err = db.One("Name", "John", &user)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = db.One("Name", "Jack", &user)
|
|
fmt.Println(err)
|
|
|
|
// Output:
|
|
// not found
|
|
}
|
|
|
|
func ExampleDB_Find() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var users []User
|
|
err := db.Find("Group", "staff", &users)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Found", len(users))
|
|
|
|
// Output:
|
|
// Found 3
|
|
}
|
|
|
|
func ExampleDB_All() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var users []User
|
|
err := db.All(&users)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Found", len(users))
|
|
|
|
// Output:
|
|
// Found 3
|
|
}
|
|
|
|
func ExampleDB_AllByIndex() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var users []User
|
|
err := db.AllByIndex("CreatedAt", &users)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Found", len(users))
|
|
|
|
// Output:
|
|
// Found 3
|
|
}
|
|
|
|
func ExampleDB_Range() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var users []User
|
|
err := db.Range("Age", 21, 22, &users)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Found", len(users))
|
|
|
|
// Output:
|
|
// Found 2
|
|
}
|
|
|
|
func ExampleLimit() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var users []User
|
|
err := db.All(&users, storm.Limit(2))
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Found", len(users))
|
|
|
|
// Output:
|
|
// Found 2
|
|
}
|
|
|
|
func ExampleSkip() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var users []User
|
|
err := db.All(&users, storm.Skip(1))
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Found", len(users))
|
|
|
|
// Output:
|
|
// Found 2
|
|
}
|
|
|
|
func ExampleUseDB() {
|
|
dir, _ := ioutil.TempDir(os.TempDir(), "storm")
|
|
defer os.RemoveAll(dir)
|
|
|
|
bDB, err := bolt.Open(filepath.Join(dir, "bolt.db"), 0600, &bolt.Options{Timeout: 10 * time.Second})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
db, _ := storm.Open("", storm.UseDB(bDB))
|
|
defer db.Close()
|
|
|
|
err = db.Save(&User{ID: 10})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var user User
|
|
err = db.One("ID", 10, &user)
|
|
fmt.Println(err)
|
|
|
|
// Output:
|
|
// <nil>
|
|
}
|
|
|
|
func ExampleDB_DeleteStruct() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var user User
|
|
|
|
err := db.One("ID", 1, &user)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = db.DeleteStruct(&user)
|
|
fmt.Println(err)
|
|
|
|
// Output:
|
|
// <nil>
|
|
}
|
|
|
|
func ExampleDB_Begin() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
// both start out with a balance of 10000 cents
|
|
var account1, account2 Account
|
|
|
|
tx, err := db.Begin(true)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer tx.Rollback()
|
|
|
|
err = tx.One("ID", 1, &account1)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = tx.One("ID", 2, &account2)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
account1.Amount -= 1000
|
|
account2.Amount += 1000
|
|
|
|
err = tx.Save(&account1)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = tx.Save(&account2)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
tx.Commit()
|
|
|
|
var account1Reloaded, account2Reloaded Account
|
|
|
|
err = db.One("ID", 1, &account1Reloaded)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = db.One("ID", 2, &account2Reloaded)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Amount in account 1:", account1Reloaded.Amount)
|
|
fmt.Println("Amount in account 2:", account2Reloaded.Amount)
|
|
|
|
// Output:
|
|
// Amount in account 1: 9000
|
|
// Amount in account 2: 11000
|
|
}
|
|
|
|
func ExampleDB_From() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
// Create some sub buckets to partition the data.
|
|
privateNotes := db.From("notes", "private")
|
|
workNotes := db.From("notes", "work")
|
|
|
|
err := privateNotes.Save(&Note{ID: "private1", Text: "This is some private text."})
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = workNotes.Save(&Note{ID: "work1", Text: "Work related."})
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var privateNote, workNote, personalNote Note
|
|
|
|
err = privateNotes.One("ID", "work1", &workNote)
|
|
|
|
// Not found: Wrong bucket.
|
|
fmt.Println(err)
|
|
|
|
err = workNotes.One("ID", "work1", &workNote)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = privateNotes.One("ID", "private1", &privateNote)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println(workNote.Text)
|
|
fmt.Println(privateNote.Text)
|
|
|
|
// These can be nested further if needed:
|
|
personalNotes := privateNotes.From("personal")
|
|
err = personalNotes.Save(&Note{ID: "personal1", Text: "This is some very personal text."})
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = personalNotes.One("ID", "personal1", &personalNote)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println(personalNote.Text)
|
|
|
|
// Output:
|
|
// not found
|
|
// Work related.
|
|
// This is some private text.
|
|
// This is some very personal text.
|
|
}
|
|
|
|
func ExampleDB_Drop() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
var user User
|
|
|
|
err := db.One("Email", "john@provider.com", &user)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = db.Drop("User")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// One only works for indexed fields.
|
|
err = db.One("Email", "john@provider.com", &user)
|
|
fmt.Println(err)
|
|
|
|
// Output:
|
|
// not found
|
|
}
|
|
|
|
func ExampleNode_PrefixScan() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
// The PrefixScan method is available on both DB and Node.
|
|
// This example shows the usage on Node.
|
|
// The Node notes will be the top-level bucket.
|
|
notes := db.From("notes")
|
|
|
|
// Partition the notes in one bucket per month.
|
|
for i := 2014; i <= 2016; i++ {
|
|
for j := 1; j <= 12; j++ {
|
|
bucket := notes.From(fmt.Sprintf("%d%02d", i, j))
|
|
numNotes := 2
|
|
|
|
// Add some variation.
|
|
if j%3 == 0 {
|
|
numNotes = 3
|
|
}
|
|
|
|
for k := 0; k < numNotes; k++ {
|
|
noteID := fmt.Sprintf("%d-%d", j, k)
|
|
if err := bucket.Save(&Note{ID: noteID, Text: fmt.Sprintf("Note %s", noteID)}); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// There are now three years worth of notes. Let's look at 2016:
|
|
nodes := notes.PrefixScan("2016")
|
|
|
|
fmt.Println("Note buckets in 2016:", len(nodes))
|
|
|
|
march := notes.PrefixScan("201603")[0]
|
|
|
|
// The two below points to the same bucket:
|
|
fmt.Println("Bucket", march.Bucket()[1])
|
|
fmt.Println("Bucket", nodes[2].Bucket()[1])
|
|
|
|
count, err := march.Count(&Note{})
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Notes in March:", count)
|
|
|
|
// Output:
|
|
//Note buckets in 2016: 12
|
|
//Bucket 201603
|
|
//Bucket 201603
|
|
//Notes in March: 3
|
|
|
|
}
|
|
|
|
func ExampleNode_RangeScan() {
|
|
dir, db := prepareDB()
|
|
defer os.RemoveAll(dir)
|
|
defer db.Close()
|
|
|
|
// The RangeScan method is available on both DB and Node.
|
|
// This example shows the usage on Node.
|
|
// The Node notes will be the top-level bucket.
|
|
notes := db.From("notes")
|
|
|
|
// Partition the notes in one bucket per month.
|
|
for i := 2013; i <= 2016; i++ {
|
|
for j := 1; j <= 12; j++ {
|
|
for k := 0; k < 3; k++ {
|
|
// Must left-pad the month so it is sortable.
|
|
bucket := notes.From(fmt.Sprintf("%d%02d", i, j))
|
|
noteID := fmt.Sprintf("%d-%d", j, k)
|
|
if err := bucket.Save(&Note{ID: noteID, Text: fmt.Sprintf("Note %s", noteID)}); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// There are now four years worth of notes. Let's look at first half of 2014:
|
|
nodes := notes.RangeScan("201401", "201406")
|
|
|
|
fmt.Println("Note buckets in first half of 2014:", len(nodes))
|
|
|
|
notesCount, err := nodes[0].Count(&Note{})
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Notes in bucket:", notesCount)
|
|
|
|
// Output:
|
|
// Note buckets in first half of 2014: 6
|
|
// Notes in bucket: 3
|
|
}
|
|
|
|
type User struct {
|
|
ID int `storm:"id"`
|
|
Group string `storm:"index"`
|
|
Email string `storm:"unique"`
|
|
Name string
|
|
Age int `storm:"index"`
|
|
CreatedAt time.Time `storm:"index"`
|
|
}
|
|
|
|
type Account struct {
|
|
ID int `storm:"id"`
|
|
Amount int64 // amount in cents
|
|
}
|
|
|
|
type Note struct {
|
|
ID string `storm:"id"`
|
|
Text string
|
|
}
|
|
|
|
func prepareDB() (string, *storm.DB) {
|
|
dir, _ := ioutil.TempDir(os.TempDir(), "storm")
|
|
db, _ := storm.Open(filepath.Join(dir, "storm.db"), storm.AutoIncrement())
|
|
|
|
for i, name := range []string{"John", "Eric", "Dilbert"} {
|
|
email := strings.ToLower(name + "@provider.com")
|
|
user := User{
|
|
Group: "staff",
|
|
Email: email,
|
|
Name: name,
|
|
Age: 21 + i,
|
|
CreatedAt: time.Now(),
|
|
}
|
|
err := db.Save(&user)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
for i := int64(0); i < 10; i++ {
|
|
account := Account{Amount: 10000}
|
|
|
|
err := db.Save(&account)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
return dir, db
|
|
}
|