-
Notifications
You must be signed in to change notification settings - Fork 59
/
main.go
67 lines (56 loc) · 1.32 KB
/
main.go
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
/*
Command lmdb_copy is a clone of mdb_copy that copies an LMDB environment. A
consistent copy is made even if the source database is in use.
Command line flags mirror the flags for the original program. For information
about, run lmdb_copy with the -h flag.
lmdb_copy -h
*/
package main
import (
"flag"
"log"
"os"
"github.com/bmatsuo/lmdb-go/internal/lmdbcmd"
"github.com/bmatsuo/lmdb-go/lmdb"
)
func main() {
opt := &Options{}
flag.BoolVar(&opt.Compact, "c", false, "Compact while copying.")
flag.Parse()
lmdbcmd.PrintVersion()
if flag.NArg() > 2 {
log.Fatalf("too many arguments specified")
}
if flag.NArg() == 0 {
log.Fatalf("at least one argument must be specified")
}
var srcpath, dstpath string
srcpath = flag.Arg(0)
if flag.NArg() > 1 {
dstpath = flag.Arg(1)
}
copyEnv(srcpath, dstpath, opt)
}
// Options contain the command line options for an lmdb_copy command.
type Options struct {
Compact bool
}
func copyEnv(srcpath, dstpath string, opt *Options) error {
env, err := lmdb.NewEnv()
if err != nil {
return err
}
err = env.Open(srcpath, lmdbcmd.OpenFlag(), 0644)
if err != nil {
return err
}
var flags uint
if opt != nil && opt.Compact {
flags |= lmdb.CopyCompact
}
if dstpath != "" {
return env.CopyFlag(dstpath, flags)
}
fd := os.Stdout.Fd()
return env.CopyFDFlag(fd, flags)
}