util.go

package utils

import (
	"database/sql"
	"fmt"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/orm"

	//切记：导入驱动包
	_ "github.com/go-sql-driver/mysql"
	"log"
)

var db *sql.DB

func InitMysql() {

	fmt.Println("InitMysql....")
	driverName := beego.AppConfig.String("driverName")

	//注册数据库驱动
	orm.RegisterDriver(driverName, orm.DRMySQL)

	//数据库连接
	user := beego.AppConfig.String("mysqluser")
	pwd := beego.AppConfig.String("mysqlpwd")
	host := beego.AppConfig.String("host")
	port := beego.AppConfig.String("port")
	dbname := beego.AppConfig.String("dbname")

	//dbConn := "root:yu271400@tcp(127.0.0.1:3306)/myblog?charset=utf8"
	dbConn := user + ":" + pwd + "@tcp(" + host + ":" + port + ")/" + dbname + "?charset=utf8"

	db1, err := sql.Open(driverName, dbConn)
	if err != nil {
		fmt.Println(err.Error())
	} else {
		db = db1

		//创建用户表
		CreateTableWithUser()
		//创建文章表
		CreateTableWithArticle()
		//创建相册表
		CreateTableWithAlbum()
	}
}

//操作数据库
func ModifyDB(sql string, args ...interface{}) (int64, error) {
	result, err := db.Exec(sql, args...)
	if err != nil {
		log.Println(err)
		return 0, err
	}
	count, err := result.RowsAffected()
	if err != nil {
		log.Println(err)
		return 0, err
	}
	return count, nil
}

//创建用户表
func CreateTableWithUser() {
	sql := `CREATE TABLE IF NOT EXISTS users(
		id INT(4) PRIMARY KEY AUTO_INCREMENT NOT NULL,
		username VARCHAR(64),
		password VARCHAR(64),
		status INT(4),
		createtime INT(10)
		);`
	ModifyDB(sql)
}

//查询
func QueryRowDB(sql string) *sql.Row {
	return db.QueryRow(sql)
}

func QueryDB(sql string) (*sql.Rows, error) {
	return db.Query(sql)
}

/**
 * 将文章详情的内容，转换成HTMl语句
 */
//func SwitchMarkdownToHtml(content string) template.HTML {
//
//	markdown := blackfriday.MarkdownCommon([]byte(content))
//
//	//获取到html文档
//	doc, _ := goquery.NewDocumentFromReader(bytes.NewReader(markdown))
//
//	/**
//	对document进程查询，选择器和css的语法一样
//	第一个参数：i是查询到的第几个元素
//	第二个参数：selection就是查询到的元素
//	*/
//	doc.Find("code").Each(func(i int, selection *goquery.Selection) {
//		light, _ := syntaxhighlight.AsHTML([]byte(selection.Text()))
//		selection.SetHtml(string(light))
//		fmt.Println(selection.Html())
//		fmt.Println("light:", string(light))
//		fmt.Println("\n\n\n")
//	})
//	htmlString, _ := doc.Html()
//	return template.HTML(htmlString)
//}