-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.go
62 lines (53 loc) · 1.36 KB
/
db.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
package lexie
import (
"errors"
"fmt"
"os"
"sync"
"github.com/pschlump/ansi"
)
// ------------------------------------------------------------------------------------------------------------------------------------------
// Debug Print - controllable with flags.
// ------------------------------------------------------------------------------------------------------------------------------------------
var dbOn map[string]bool
var dbOnLock sync.Mutex
func init() {
dbOn = make(map[string]bool)
dbOn["debug"] = true
}
func DbPrintf(db string, format string, args ...interface{}) {
dbOnLock.Lock()
defer dbOnLock.Unlock()
if x, o := dbOn[db]; o && x {
fmt.Printf(format, args...)
}
}
var (
Red = ansi.ColorCode("red")
Yellow = ansi.ColorCode("yellow")
Green = ansi.ColorCode("green")
Reset = ansi.ColorCode("reset")
)
func DbOn(db string) (ok bool) {
ok = false
dbOnLock.Lock()
defer dbOnLock.Unlock()
if x, o := dbOn[db]; o {
ok = x
}
return
}
func Fopen(fn string, mode string) (file *os.File, err error) {
file = nil
if mode == "r" {
file, err = os.Open(fn) // For read access.
} else if mode == "w" {
file, err = os.OpenFile(fn, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
} else if mode == "a" {
file, err = os.OpenFile(fn, os.O_RDWR|os.O_APPEND, 0660)
} else {
err = errors.New("Invalid Mode")
}
return
}
/* vim: set noai ts=4 sw=4: */