/
main.go
116 lines (94 loc) · 2.49 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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package main
import (
"database/sql"
//"fmt"
"log"
"net/http"
"os"
"github.com/gin-gonic/gin"
_ "github.com/lib/pq"
)
const (
DB_USER = "postgres"
DB_PASSWORD = "madras"
DB_NAME = "banks"
)
type Bank struct {
Id int64 `db:"id" json:"id"`
Name string `db:"name" json:"name"`
}
type Branch struct {
Ifsc string `db:"ifsc" json:"ifsc"`
BankId int64 `db:"bank_id" json:"bank_id"`
Branch string `db:"branch" json:"branch"`
Address string `db:"address" json:"address"`
City string `db:"city" json:"city"`
District string `db:"district" json:"district"`
State string `db:"state" json:"state"`
}
func main() {
//dbinfo := fmt.Sprintf("user=%s password=%s dbname=%s sslmode=disable", DB_USER, DB_PASSWORD, DB_NAME)
db, err := sql.Open("postgres", os.Getenv("DATABASE_URL"))
checkErr(err)
defer db.Close()
port := os.Getenv("PORT")
if port == "" {
log.Fatal("$PORT must be set")
}
router := gin.New()
router.Use(gin.Logger())
router.LoadHTMLGlob("templates/*.html")
router.Static("/static", "static")
router.GET("/", func(c *gin.Context) {
c.HTML(http.StatusOK, "index.html", nil)
})
router.GET("/bank", func(c *gin.Context) {
banks := []Bank{}
rows, err := db.Query("SELECT * FROM banks")
checkErr(err)
defer rows.Close()
for rows.Next() {
bank := Bank{}
err = rows.Scan(&bank.Name, &bank.Id)
checkErr(err)
banks = append(banks, bank)
}
c.JSON(200, banks)
})
router.GET("/city/:bank", func(c *gin.Context) {
bank := c.Params.ByName("bank")
//find the cities of a bank
cities := []string{}
rows, err := db.Query("SELECT DISTINCT city FROM branches WHERE bank_id=$1", bank)
checkErr(err)
defer rows.Close()
for rows.Next() {
var city string
err = rows.Scan(&city)
checkErr(err)
cities = append(cities, city)
}
c.JSON(200, cities)
})
router.GET("/branch/:bank/:city", func(c *gin.Context) {
bank := c.Params.ByName("bank")
city := c.Params.ByName("city")
branches := []Branch{}
rows, err := db.Query("SELECT ifsc, bank_id, branch, address, city, district, state FROM bank_branches WHERE bank_id=$1 AND city=$2", bank, city)
checkErr(err)
defer rows.Close()
for rows.Next() {
branch := Branch{}
err = rows.Scan(&branch.Ifsc, &branch.BankId, &branch.Branch, &branch.Address, &branch.City, &branch.District, &branch.State)
checkErr(err)
branches = append(branches, branch)
}
c.JSON(200, branches)
})
router.Run(":" + port)
}
func checkErr(err error) {
if err != nil {
panic(err)
}
}