示例#1
0
文件: db_user.go 项目: Tangmz/Blog
func DbInit() (*sql.DB, error) {

	logErr := logger.GetErr()

	db_info, err := DbConfig()
	if err != nil {
		log.Println("Db config error ", err)
		return nil, err
	} else {
		log.Println("Db config suc")
	}

	dbCmd := db_info.User + ":" + db_info.Cert + "@/" + db_info.Database + "?charset=utf8"
	//	fmt.Println(dbCmd)

	db, err := sql.Open("mysql", dbCmd)
	if err != nil {
		logErr.Println("Db open failed ", err)
		return db, err
	} else {
		logSuc := logger.GetSuc()
		logSuc.Println("Db open success")
	}
	return db, nil
}
示例#2
0
文件: db_user.go 项目: Tangmz/Blog
func DbConfig() (Db, error) {

	dbInfo := Db{}
	logErr := logger.GetErr()
	logSuc := logger.GetSuc()

	f, err := os.Open("db.xml")
	defer f.Close()

	if err != nil {
		logErr.Println("db open error ", err)
		return dbInfo, err
	}

	xmlData, err := ioutil.ReadAll(f)
	if err != nil {
		logErr.Println("read xml error  ", err)
		return dbInfo, err
	}

	err = xml.Unmarshal(xmlData, &dbInfo)
	if err != nil {
		logErr.Println("xml Unmarshal error ", err)
		return dbInfo, err
	}
	logSuc.Println("xml read succeed \n", dbInfo)

	return dbInfo, nil
}
示例#3
0
文件: db_user.go 项目: Tangmz/Blog
func GetDb() *sql.DB {
	db, err := DbInit()
	for i := 0; i < 20; i++ {
		if err != nil {
			logErr := logger.GetErr()
			logErr.Println("datebase open failed the ", i, "th times")
		} else {
			logSuc := logger.GetSuc()
			logSuc.Println("database open success the ", i, "th times")
			break
		}
		db, err = DbInit()
	}
	return db
}
示例#4
0
文件: baseop.go 项目: Tangmz/Blog
func register_User(user_info RegisterStruct) (string, error) {
	db := database.GetDb()
	defer db.Close()
	logErr := logger.GetErr()
	logInfo := logger.GetInfo()

	times := 0
	rows, err := db.Query("select reg_times from userInfo where user_name = ? and if_use = 1", user_info.Username)
	if err != nil {
		logErr.Println(err)
		return "", err
	}

	for rows.Next() {
		err = rows.Scan(&times)
		if err != nil {
			logErr.Println(err)
			return "", err
		}
	}
	if times != 0 {
		res := "This username is using."
		logInfo.Println(res)
		return res, nil
	}
	times++

	stmt, err := db.Prepare("insert into userInfo(reg_times,user_name,user_cert) values(?,?,?)")
	if err != nil {
		logErr.Println(err)
		return "", err
	}
	_, err = stmt.Exec(times, user_info.Username, user_info.Psw1)
	if err != nil {
		logErr.Println(err)
		return "", err
	}

	return "", nil
}
示例#5
0
文件: baseop.go 项目: Tangmz/Blog
	"fmt"
	"github.com/gorilla/securecookie"
	"html/template"
	"logger"
	"net/http"
)

func Init() {
	ResultTypeInit()
	UserOnline = make(map[string]string)
}

var res_type_info ResultType
var logInfo = logger.GetInfo()
var logWar = logger.GetWar()
var logErr = logger.GetErr()
var logDeb = logger.GetSuc()

func RegisterHandler(w http.ResponseWriter, r *http.Request) {

	fmt.Println(r.Method)

	if r.Method == "GET" {
		reg_Info := Reg_Type_Info{
			IsHide:   "none",
			Info:     "",
			BackShow: false,
		}

		file, err := template.ParseFiles("../www/html/register.html")
		if err != nil {