超简易Go语言实现的留言板代码_go

超简易Go语言实现的留言板代码
package main
    
import (
    // "fmt"
    "io"
    "log"
    "net/http"
    "text/template"
    "time"
    "database/sql"
    "github.com/ziutek/mymysql/godrv"
)
    
// 留言结构
type Liuyan struct {
    Id int
    Name string
    Content string
    Time int
}
    
// 显示留言时间
func (l Liuyan) ShowTime() string {
    t := time.Unix(int64(l.Time), 0)
    return t.Format("2006-01-02 15:04:05")
}
    
func main() {
    godrv.Register("SET NAMES utf8")
    
    // 连接数据库
    db, err := sql.Open("mymysql", "tcp:127.0.0.1:3306*go/root/123456")
    if err != nil {
        panic(err)
    }
    defer db.Close()
    
    // 准备模板
    tpl, err := template.New("liuyanbook").Parse(html)
    if err != nil {
        panic(err)
    }
    
    // 显示留言页面 /
    requestList := func(w http.ResponseWriter, req *http.Request) {
        // 查询数据
        rows, err := db.Query("select * from liuyan")
        if err != nil {
            log.Fatal(err)
        }
        defer rows.Close()
    
        // 获取数据
        lys := []Liuyan{}
        for rows.Next() {
            ly := Liuyan{}
            err := rows.Scan(&ly.Id, &ly.Name, &ly.Content, &ly.Time)
            if nil != err {
                log.Fatal(err)
            }
            lys = append(lys, ly)
        }
    
        // 显示数据
        err = tpl.ExecuteTemplate(w, "list", lys)
        if err != nil {
            log.Fatal(err)
        }
    }
    
    // 留言页面 /liuyan
    requestLiuyan := func(w http.ResponseWriter, req *http.Request) {
        err := req.ParseForm()
        if err != nil{
            log.Fatal(err)
        }
    
        if "POST" == req.Method {
            if len(req.Form["name"]) < 1 {
                io.WriteString(w, "参数错误!\n")
                return
            }
            if len(req.Form["content"]) < 1 {
                io.WriteString(w, "参数错误!\n")
                return
            }
    
            name := template.HTMLEscapeString(req.Form.Get("name"))
            content := template.HTMLEscapeString(req.Form.Get("content"))
    
            // sql语句
            sql, err := db.Prepare("insert into liuyan(name, content, time) values(?, ?, ?)")
            if err != nil {
                log.Fatal(err)
            }
            defer sql.Close()
    
            // sql参数,并执行
            _, err = sql.Exec(name, content, time.Now().Unix())
            if err != nil {
                log.Fatal(err)
            }
    
            // 跳转
            w.Header().Add("Location", "/")
            w.WriteHeader(302)
    
            // 提示信息
            io.WriteString(w, "提交成功!\n")
    
            return
        }
    
        err = tpl.ExecuteTemplate(w, "liuyan", nil)
        if err != nil {
            log.Fatal(err)
        }
    }
    
    http.HandleFunc("/", requestList)
    http.HandleFunc("/liuyan", requestLiuyan)
    err = http.ListenAndServe(":12345", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}
    
// 网页模板
var html string = `{{define "list"}}{{/* 留言列表页面 */}}<!DOCTYPE html>
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
</head>
<body>
    <p><a href="/liuyan">给我留言</a></p>
    <table>
{{range .}}
    <tr>
        <td>{{.Id}}</td><td>{{.Name}}</td><td>{{.Content}}</td><td>{{.ShowTime}}</td>
    </tr>
{{end}}
    </table>
</body>
</html>{{end}}
{{define "liuyan"}}{{/* 发布留言页面 */}}<!DOCTYPE html>
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
</head>
<body>
    <form method="post">
        姓名:<input type="text" name="name" /><br>
        内容:<input type="text" name="content" /><br>
        <input type="submit" value="提交" />
    </form>
</body>
</html>{{end}}