-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
102 lines (81 loc) · 1.89 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
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
98
99
100
101
102
package main
import (
"database/sql"
"net/http"
"os"
"strings"
"github.com/gin-contrib/gzip"
"github.com/gin-gonic/gin"
"github.com/joho/godotenv"
_ "github.com/nakagami/firebirdsql"
"github.com/pquerna/ffjson/ffjson"
)
type SqlQuery struct {
Sql string `json:"sql"`
Params []interface{} `json:"params"`
}
func connectToDb() *sql.DB {
err := godotenv.Load()
if err != nil {
panic("Error loading .env file")
}
USERNAME := os.Getenv("FIREBIRD_REST_USERNAME")
PASSWORD := os.Getenv("FIREBIRD_REST_PASSWORD")
HOST := os.Getenv("FIREBIRD_REST_HOST")
PORT := os.Getenv("FIREBIRD_REST_PORT")
DATABASE := os.Getenv("FIREBIRD_REST_DATABASE")
CONNECTION_OPTIONS := os.Getenv("FIREBIRD_REST_CONNECTION_OPTIONS")
connectionString := strings.Join([]string{
USERNAME, ":", PASSWORD, "@", HOST, ":", PORT, DATABASE,
CONNECTION_OPTIONS,
}, "")
var db *sql.DB
db, err = sql.Open("firebirdsql", connectionString)
if err != nil {
panic(err)
}
return db
}
func rawExecuteSql(
db *sql.DB,
query SqlQuery,
) map[string][]interface{} {
rows, err := db.Query(query.Sql, query.Params...)
if err != nil {
panic(err)
}
row_json, err := SQLToJSON(rows)
if err != nil {
panic(err)
}
return row_json
}
func executeSql(db *sql.DB) gin.HandlerFunc {
fn := func(c *gin.Context) {
var params SqlQuery
if err := c.BindJSON(¶ms); err != nil {
panic(err)
}
row_json := rawExecuteSql(db, params)
data, err := ffjson.Marshal(row_json)
if err != nil {
panic(err)
}
c.String(200, string(data))
}
return gin.HandlerFunc(fn)
}
func sayHello(c *gin.Context) {
c.String(http.StatusOK, "hello world")
}
func main() {
gin.SetMode(gin.ReleaseMode)
db := connectToDb()
db.SetMaxOpenConns(25)
db.SetMaxIdleConns(25)
router := gin.New()
router.Use(gzip.Gzip(gzip.DefaultCompression))
router.GET("/", sayHello)
router.POST("/", executeSql(db))
router.Run("0.0.0.0:8080")
}