本地存储

 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package db

import (
	"database/sql"
	"fmt"
	_ "github.com/mattn/go-sqlite3"
	"log"
	"os"
)

var db *sql.DB

var err error

func init() {
	if isExist("./keybase.db") {
		db, err = sql.Open("sqlite3", "./keybase.db")
		if err != nil {
			log.Fatal(err)
		}
	}else{
		db, err = sql.Open("sqlite3", "./keybase.db")
		if err != nil {
			log.Fatal(err)
		}

		sqlStmt := `
		create table keywords (id integer PRIMARY KEY autoincrement,keyword text, url text);
		`
		_, err = db.Exec(sqlStmt)
		if err != nil {
			log.Printf("%q: %s\n", err, sqlStmt)
			return
		}

	}

}

func isExist(path string)(bool){
	_, err := os.Stat(path)
	if err != nil{
		if os.IsExist(err){
			return true
		}
		if os.IsNotExist(err){
			return false
		}
		fmt.Println(err)
		return false
	}
	return true
}

func AddKeyword(keyword,url string) {

	tx, err := db.Begin()
	if err != nil {
		log.Fatal(err)
	}
	stmt, err := tx.Prepare("insert into keywords(keyword, url) values(?, ?)")
	if err != nil {
		log.Fatal(err)
	}
	defer stmt.Close()
	_, err = stmt.Exec(keyword,  url)

	tx.Commit()

}
func QueryKey(keyword string) string {
	stmt, err := db.Prepare("select url from keywords where keyword = ?")
	if err != nil {
		log.Fatal(err)
		return ""
	}
	defer stmt.Close()
	var url string
	err = stmt.QueryRow(keyword).Scan(&url)
	if err != nil {
		log.Fatal(err)
		return ""
	}
	fmt.Println(url)
	return url
}

异常处理

异常捕获

 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

func Try(fun func(), handler func(interface{})) {

	defer func() {

		if err := recover(); err != nil {

			handler(err)

		}

	}()

	fun()

}



func AddKeyword(keyword,url string) {
	try.Try(func() {
		tx, err := db.Begin()
		if err != nil {
			log.Fatal(err)
		}
		stmt, err := tx.Prepare("insert into keywords(keyword, url) values(?, ?)")
		if err != nil {
			log.Fatal(err)
		}
		defer stmt.Close()
		_, err = stmt.Exec(keyword,  url)

		tx.Commit()

	}, func(i interface{}) {
		fmt.Println(i)
	})
	

}