sosedoff/pgweb
1
1
Fork 0
mirror of https://github.com/sosedoff/pgweb.git synced 2024-12-15 03:36:33 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

DRY up api module

Browse Source
This commit is contained in:
Dan Sosedoff 2015-05-02 20:32:16 -05:00
parent cb3c3e0e2e
commit 04fe0023b7
2 changed files with 21 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
pkg/api/api.go
View File

@ -67,13 +67,7 @@ func Connect(c *gin.Context) {
func GetDatabases(c *gin.Context) {
names, err := DbClient.Databases()
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, names)
serveResult(names, err, c)
}
func RunQuery(c *gin.Context) {
@ -84,7 +78,7 @@ func RunQuery(c *gin.Context) {
return
}
GetHandleQuery(query, c)
HandleQuery(query, c)
}
func ExplainQuery(c *gin.Context) {
@ -95,40 +89,22 @@ func ExplainQuery(c *gin.Context) {
return
}
GetHandleQuery(fmt.Sprintf("EXPLAIN ANALYZE %s", query), c)
HandleQuery(fmt.Sprintf("EXPLAIN ANALYZE %s", query), c)
}
func GetSchemas(c *gin.Context) {
names, err := DbClient.Schemas()
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, names)
serveResult(names, err, c)
}
func GetTables(c *gin.Context) {
names, err := DbClient.Tables()
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, names)
serveResult(names, err, c)
}
func GetTable(c *gin.Context) {
res, err := DbClient.Table(c.Params.ByName("table"))
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, res)
serveResult(res, err, c)
}
func GetTableRows(c *gin.Context) {
@ -158,13 +134,7 @@ func GetTableRows(c *gin.Context) {
}
res, err := DbClient.TableRows(c.Params.ByName("table"), opts)
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, res)
serveResult(res, err, c)
}
func GetTableInfo(c *gin.Context) {
@ -195,26 +165,15 @@ func GetConnectionInfo(c *gin.Context) {
func GetActivity(c *gin.Context) {
res, err := DbClient.Activity()
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, res)
serveResult(res, err, c)
}
func GetTableIndexes(c *gin.Context) {
res, err := DbClient.TableIndexes(c.Params.ByName("table"))
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, res)
serveResult(res, err, c)
}
func GetHandleQuery(query string, c *gin.Context) {
func HandleQuery(query string, c *gin.Context) {
result, err := DbClient.Query(query)
if err != nil {
@ -236,11 +195,5 @@ func GetHandleQuery(query string, c *gin.Context) {
func GetBookmarks(c *gin.Context) {
bookmarks, err := bookmarks.ReadAll(bookmarks.Path())
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, bookmarks)
serveResult(bookmarks, err, c)
}

11
pkg/api/helpers.go
View File

@ -79,8 +79,17 @@ func serveStaticAsset(path string, c *gin.Context) {
data, err := data.Asset("static" + path)
if err != nil {
c.String(400, err.Error())
c.Abort()
return
}
c.Data(200, assetContentType(path), data)
}
func serveResult(result interface{}, err error, c *gin.Context) {
if err != nil {
c.JSON(400, NewError(err))
return
}
c.JSON(200, result)
}