forked from unirita/cuto
-
Notifications
You must be signed in to change notification settings - Fork 0
/
connection.go
97 lines (82 loc) · 2.38 KB
/
connection.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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// Copyright 2015 unirita Inc.
// Created 2015/04/10 shanxia
package db
import (
"errors"
"fmt"
"os"
"database/sql"
"github.com/coopernurse/gorp"
_ "github.com/mattn/go-sqlite3"
)
type IConnection interface {
GetDbMap() *gorp.DbMap
GetDb() *sql.DB
Close()
}
type Connection struct {
db *sql.DB // DBコネクション
dbMap *gorp.DbMap // マッピング情報
}
var sqlite3_driver = "sqlite3"
// SQLite3のセッションを接続し、テーブルとDTOのマッピングを行う。
//
// param - dbfile sqliteファイルのパス。
//
// return - コネクション情報とエラー情報
func Open(dbfile string) (IConnection, error) {
if _, exist := os.Stat(dbfile); exist != nil {
return nil, errors.New(fmt.Sprintf("Not found dbfile[%v]", dbfile))
}
db, err := sql.Open(sqlite3_driver, dbfile)
if err != nil {
return nil, err
}
// 外部キーを有効にする。
db.Exec("PRAGMA foreign_keys=ON;")
// テーブルと構造体のマッピング。
dbmap := &gorp.DbMap{
Db: db,
Dialect: gorp.SqliteDialect{},
}
jobNetworkMapping(dbmap)
jobMapping(dbmap)
return Connection{db, dbmap}, nil
}
func jobNetworkMapping(dbmap *gorp.DbMap) {
t := dbmap.AddTableWithName(JobNetworkResult{}, "JOBNETWORK").SetKeys(true, "ID")
t.ColMap("JobnetWork").Rename("JOBNETWORK")
t.ColMap("StartDate").Rename("STARTDATE")
t.ColMap("EndDate").Rename("ENDDATE")
t.ColMap("Status").Rename("STATUS")
t.ColMap("Detail").Rename("DETAIL")
t.ColMap("CreateDate").Rename("CREATEDATE")
t.ColMap("UpdateDate").Rename("UPDATEDATE")
}
func jobMapping(dbmap *gorp.DbMap) {
t := dbmap.AddTableWithName(JobResult{}, "JOB").SetKeys(false, "ID", "JobId")
t.ColMap("JobId").Rename("JOBID")
t.ColMap("JobName").Rename("JOBNAME")
t.ColMap("StartDate").Rename("STARTDATE")
t.ColMap("EndDate").Rename("ENDDATE")
t.ColMap("Status").Rename("STATUS")
t.ColMap("Detail").Rename("DETAIL")
t.ColMap("Rc").Rename("RC")
t.ColMap("Node").Rename("NODE")
t.ColMap("Port").Rename("PORT")
t.ColMap("Variable").Rename("VARIABLE")
t.ColMap("CreateDate").Rename("CREATEDATE")
t.ColMap("UpdateDate").Rename("UPDATEDATE")
}
// SQLite3とのセッションを切断する。
func (c Connection) Close() {
c.db.Close()
}
// DBマッピング情報を返す。
func (c Connection) GetDbMap() *gorp.DbMap {
return c.dbMap
}
// DBオブジェクトを返す。
func (c Connection) GetDb() *sql.DB {
return c.db
}