005-Go 操作PostgreSQL数据库

package main

import(
    "fmt"
    "database/sql"
    _ "github.com/lib/pq"
    "github.com/kataras/iris"
    "github.com/kataras/iris/middleware/logger"
    "github.com/kataras/iris/middleware/recover"
)

const(
    host = "127.0.0.1"
    port = 5432
    user = "test"
    password = "test"
    dbname = "test"
)

type User struct{
    Uno        string `json:"uno"`
    Uname    string    `json:"uname"`
}

func main(){
    app := iris.New()
    
    app.Logger().SetLevel("debug")

    app.Use(recover.New())
    app.Use(logger.New())

    app.Get("/users", func(ctx iris.Context){
        psqlInfo := fmt.Sprintf("host=%s port=%d user=%s "+
            "password=%s dbname=%s sslmode=disable",host, port, user, password, dbname)

        db, err := sql.Open("postgres", psqlInfo)
        if err != nil{
            panic(err)
        }
        defer db.Close()

        err = db.Ping()
        if err != nil{
            panic(err)
        }

        fmt.Println("Successfully connected!")

        var users []User

        rows,err := db.Query("select uno,uname from t_user")
        if err != nil{
            panic(err.Error)
        }
        defer rows.Close()

        for rows.Next(){
            rows.Columns()

            var user User
            err = rows.Scan(&user.Uno,&user.Uname)

            if err != nil {
                panic(err.Error)
            }

            users = append(users, user)
        }

        ctx.JSON(users)
    })

    app.Get("/user", func(ctx iris.Context){
        psqlInfo := fmt.Sprintf("host=%s port=%d user=%s "+
            "password=%s dbname=%s sslmode=disable",host, port, user, password, dbname)

        db, err := sql.Open("postgres", psqlInfo)
        if err != nil{
            panic(err)
        }
        defer db.Close()

        err = db.Ping()
        if err != nil{
            panic(err)
        }

        fmt.Println("Successfully connected!")

        sqlQuery := `select uno,uname from t_user where uno=$1`

        var user User

        row := db.QueryRow(sqlQuery, ctx.URLParam("uno"))

        err = row.Scan(&user.Uno,&user.Uname)

        switch err{
        case sql.ErrNoRows:
            fmt.Println("No rows were returned!")
        case nil:
            fmt.Println(user)
            ctx.JSON(user)
        default:
            panic(err) 
        }

    })

    app.Run(iris.Addr(":8080"), iris.WithoutServerError(iris.ErrServerClosed))
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值