今日签到 增加积分 记录积分功能

database/database.go

@@ -4,6 +4,7 @@ import (
 	"database/sql"
 	"fmt"
 	_ "github.com/go-sql-driver/mysql"
+	uuid "github.com/satori/go.uuid"
 )
 
 var (
@@ -64,3 +65,50 @@ func SelectData() string {
 
 	return stringLine
 }
+
+func SelectUserIntegrationByNickName(nickname string) int {
+	var integration_ sql.NullInt64
+	var user User
+	row := DB.QueryRow("SELECT integration FROM User where username = '" + nickname + "'")
+	err := row.Scan(&integration_)
+	if err != nil {
+		row, err := DB.Exec("insert into User(uid,account_wx,username,role,integration) "+
+			"values (?,?,?,?,?)", uuid.NewV4().String(), nickname, nickname, "user", 1)
+		if err != nil {
+			fmt.Println("exec success, ", row)
+			fmt.Println("exec failed, ", err)
+			return 9990999
+		}
+	}
+
+	row = DB.QueryRow("SELECT integration FROM User where username = '" + nickname + "'")
+	err = row.Scan(&integration_)
+	if err != nil {
+		fmt.Println(err)
+	}
+
+	user = User{
+		Integration: int(integration_.Int64),
+	}
+	return user.Integration
+}
+
+func UpDataUserIntegrationByNickName(nickname string, integration int) int {
+	var integration_ sql.NullInt64
+	var user User
+	row := DB.QueryRow("update User set integration=? where username=?", integration+1, nickname)
+	err := row.Scan(&integration_)
+	if err != nil {
+		fmt.Println(err)
+	}
+	row = DB.QueryRow("SELECT integration FROM User where username = '" + nickname + "'")
+	err = row.Scan(&integration_)
+	if err != nil {
+		fmt.Println(err)
+	}
+
+	user = User{
+		Integration: int(integration_.Int64),
+	}
+	return user.Integration
+}