ugrom.go

package main

import (
	"fmt"
	"html/template"
	"net/http"
	"strconv"

	"github.com/gorilla/mux"
	"github.com/jinzhu/gorm"

	_ "github.com/go-sql-driver/mysql"
)

//Item Структура данных
type Item struct {
	Id          int //'sql:"AUTO_INCREMENT" gorm:"primary_key"'
	Title       string
	Description string
	Updated     string //sql:"null"'
}

//Handler Структура
type Handler struct {
	DB   *gorm.DB
	Tmpl *template.Template
}

//List функция вывода данных
func (h *Handler) List(w http.ResponseWriter, r *http.Request) {
	items := []*Item{}
	db := h.DB.Find(&items)
	err := db.Error
	__err_panic(err)
	err = h.Tmpl.ExecuteTemplate(w, "index.html", struct {
		Items []*Item
	}{
		Items: items,
	})
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
}

//Add фукнция добавления данных
func (h *Handler) Add(w http.ResponseWriter, r *http.Request) {
	newItem := &Item{
		Title:       r.FormValue("title"),
		Description: r.FormValue("description"),
	}
	db := h.DB.Create(&newItem)
	err := db.Error
	__err_panic(err)
	affected := db.RowsAffected
	fmt.Println("Insert - RowsAffected", affected, "LastInsertId: ", newItem.Id)
	http.Redirect(w, r, "/", http.StatusFound)
}

//Edit функция редактирования данных
func (h *Handler) Edit(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	id, err := strconv.Atoi(vars["id"])
	__err_panic(err)

	post := &Item{}
	db := h.DB.Find(post, id)
	err = db.Error
	if err == gorm.ErrRecordNotFound {
		fmt.Println("Запись не найдена", id)
	} else {
		__err_panic(err)
	}
	err = h.Tmpl.ExecuteTemplate(w, "edit.html", post)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
}

//Update Фукнция редактирования  данных
func (h *Handler) Update(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	id, err := strconv.Atoi(vars["id"])
	__err_panic(err)
	post := &Item{}
	h.DB.Find(post, id)
	post.Title = r.FormValue("title")
	post.Description = r.FormValue("description")
	post.Updated = "User"
	db := h.DB.Save(post)
	err = db.Error
	__err_panic(err)
	affected := db.RowsAffected
	fmt.Println("Update - RowsAffected", affected)
	http.Redirect(w, r, "/", http.StatusFound)
}

//Delete Функция удаления данных
func (h *Handler) Delete(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	id, err := strconv.Atoi(vars["id"])
	__err_panic(err)
	db := h.DB.Delete(&Item{Id: id})
	err = db.Error
	__err_panic(err)
	affected := db.RowsAffected
	fmt.Println("Delete - RowsAffected", affected)
	w.Header().Set("Content-type", "application/json")
	resp := strconv.Itoa(int(affected))
	w.Write([]byte(resp))
}

func main() {
	// основные настройки к базе
	dsn := "usergo@tcp(localhost:3306)"
	// указываем кодировку
	dsn += "&charset=utf8"
	// отказываемся от prapared statements
	// параметры подставляются сразу
	dsn += "&interpolateParams=true"
	db, err := gorm.Open("mysql", dsn)
	db.DB()
	db.DB().Ping()
	// Подключение к БД
	if err != nil {
		panic(err)
	}
	handlers := &Handler{
		DB:   db,
		Tmpl: template.Must(template.ParseGlob("../templates/*")),
	}

	r := mux.NewRouter()
	r.HandleFunc("/", handlers.List).Methods("GET")
	r.HandleFunc("/items", handlers.List).Methods("GET")
	r.HandleFunc("/items/new", handlers.Add).Methods("POST")
	r.HandleFunc("/items/{id}", handlers.Edit).Methods("GET")
	r.HandleFunc("/items/{id}", handlers.Update).Methods("POST")
	r.HandleFunc("/items/{id}", handlers.Delete).Methods("DELETE")
	fmt.Println("starting server at :8080")
	http.ListenAndServe(":8080", r)
}

//Простой вызов ошибки, без обработки
func __err_panic(err error) {
	if err != nil {
		panic(err)
	}
}