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}}