sqlite/example/kvstore/main.go at main · solod-dev/sqlite · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
// A simple key-value store backed by an SQLite database.
//
// Usage:
//
//	make example name=kvstore
//	./build/kvstore -db build/kvstore.db -op set -key name -val alice
//	./build/kvstore -db build/kvstore.db -op get -key name
//	./build/kvstore -db build/kvstore.db -op del -key name
package main

import (
	"solod.dev/so/flag"
	"solod.dev/so/fmt"
	"solod.dev/so/mem"
	"solod.dev/so/os"
)

var (
	dbFlag  string
	opFlag  string
	keyFlag string
	valFlag string
)

func main() {
	parseFlags()

	store, err := NewStore(dbFlag)
	check(err)
	defer store.Close()

	switch opFlag {
	case "set":
		err = store.SetString(keyFlag, valFlag)
		check(err)
	case "get":
		val, err := store.GetString(mem.System, keyFlag)
		check(err)
		if err == ErrNotFound {
			fmt.Println("(none)")
		} else {
			fmt.Println(val)
		}
		mem.FreeString(mem.System, val)
	case "del":
		err = store.Delete(keyFlag)
		check(err)
	default:
		flag.Usage()
		os.Exit(1)
	}
}

func parseFlags() {
	flag.StringVar(&dbFlag, "db", "kvstore.db", "database file")
	flag.StringVar(&opFlag, "op", "", "operation: get, set, or del")
	flag.StringVar(&keyFlag, "key", "", "key name")
	flag.StringVar(&valFlag, "val", "", "value (for set operation)")
	flag.Parse()
}

func check(err error) {
	if err != nil && err != ErrNotFound {
		fmt.Fprintf(os.Stderr, "Error: %s\n", err.Error())
		os.Exit(1)
	}
}