first commit
This commit is contained in:
14
user/123
Normal file
14
user/123
Normal file
@@ -0,0 +1,14 @@
|
||||
#定又训练亟
|
||||
def train():
|
||||
model.train()
|
||||
for i, data in enumerate(train loader):
|
||||
#获得一个batch的数据和对应的标签
|
||||
inputs,labels = data
|
||||
inputs,labels =inputs.to(device),labels.to(device)#将数据移到GPU或CPU上
|
||||
#数据正向传播，(64，8)需要补全以下代码!!!
|
||||
#计算代价(误差)，out(batch,c),labels(batch)
|
||||
需要补全以下代码!!
|
||||
# 梯度凊θ
|
||||
optimizer.zero grad()#误差反向传播---需要补全以下代码!!
|
||||
#更新网络模型爹数
|
||||
optimizer.step()
|
||||
1
user/userforget.go
Normal file
1
user/userforget.go
Normal file
@@ -0,0 +1 @@
|
||||
package user
|
||||
90
user/userlogin.go
Normal file
90
user/userlogin.go
Normal file
@@ -0,0 +1,90 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
type Userloginstruct struct {
|
||||
Email string `json:"email"`
|
||||
Password string `json:"password"`
|
||||
}
|
||||
|
||||
type Loginresponse struct {
|
||||
Success string `json:"success"`
|
||||
Message string `json:"message"`
|
||||
Token string `json:"token"`
|
||||
}
|
||||
|
||||
func Userlogin(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Access-Control-Allow-Origin", "http://localhost:5173")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
||||
w.Header().Set("Access-Control-Allow-Credentials", "true")
|
||||
|
||||
if r.Method == "OPTIONS" {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
|
||||
body, error := io.ReadAll(r.Body)
|
||||
if error != nil {
|
||||
fmt.Println("数据读取失败")
|
||||
return
|
||||
}
|
||||
fmt.Println(string(body))
|
||||
var userinfo Userloginstruct
|
||||
error = json.Unmarshal(body, &userinfo)
|
||||
if error != nil {
|
||||
fmt.Println("数据解析失败")
|
||||
return
|
||||
}
|
||||
|
||||
mongocli := options.Client().ApplyURI("mongodb://localhost:27017")
|
||||
|
||||
mongoclient, error := mongo.Connect(context.TODO(), mongocli)
|
||||
if error != nil {
|
||||
fmt.Println("数据库连接失败")
|
||||
}
|
||||
|
||||
finduser := mongoclient.Database("user").Collection("userlist")
|
||||
|
||||
filter := bson.M{"email": userinfo.Email, "password": userinfo.Password}
|
||||
|
||||
type theuser struct {
|
||||
Email string `json:"email"`
|
||||
Password string `json:"password"`
|
||||
}
|
||||
|
||||
var oneuser theuser
|
||||
|
||||
error = finduser.FindOne(context.Background(), filter).Decode(&oneuser)
|
||||
if error != nil {
|
||||
fmt.Println("未找到该用户")
|
||||
loginres := Loginresponse{
|
||||
Success: "no",
|
||||
Message: "登录失败",
|
||||
Token: "0",
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(loginres)
|
||||
return
|
||||
} else {
|
||||
loginres := Loginresponse{
|
||||
Success: "yes",
|
||||
Message: "登录成功",
|
||||
Token: "1",
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(loginres)
|
||||
return
|
||||
}
|
||||
|
||||
}
|
||||
133
user/userregister.go
Normal file
133
user/userregister.go
Normal file
@@ -0,0 +1,133 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
|
||||
"github.com/go-redis/redis/v8"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
type Userreg struct {
|
||||
Username string `json:"username"`
|
||||
Password string `json:"password"`
|
||||
Email string `json:"email"`
|
||||
Verifycode string `json:"verifycode"`
|
||||
}
|
||||
|
||||
func Userregister(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Access-Control-Allow-Origin", "http://localhost:8080")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
||||
w.Header().Set("Access-Control-Allow-Credentials", "true")
|
||||
|
||||
if r.Method == "OPTIONS" {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
|
||||
body, error := io.ReadAll(r.Body)
|
||||
if error != nil {
|
||||
log.Println("数据读取失败")
|
||||
return
|
||||
}
|
||||
var userinfo Userreg
|
||||
error = json.Unmarshal(body, &userinfo)
|
||||
if error != nil {
|
||||
log.Println("数据读取失败")
|
||||
return
|
||||
}
|
||||
|
||||
var username = userinfo.Username
|
||||
var password = userinfo.Password
|
||||
var useremail = userinfo.Email
|
||||
var verifycode = userinfo.Verifycode
|
||||
|
||||
mongodbcli := options.Client().ApplyURI("mongodb://localhost:27017").SetMaxPoolSize(100)
|
||||
|
||||
mongodbclient, error := mongo.Connect(context.TODO(), mongodbcli)
|
||||
if error != nil {
|
||||
log.Println("数据库连接失败")
|
||||
return
|
||||
}
|
||||
findtheoneuser := bson.M{"username": username}
|
||||
|
||||
finduser := mongodbclient.Database("users").Collection("userlist")
|
||||
_, error = finduser.Find(context.Background(), findtheoneuser)
|
||||
if error == nil {
|
||||
log.Println("该用户名已存在")
|
||||
loginres := Loginresponse{
|
||||
Success: "no",
|
||||
Message: "注册失败，用户已经存在",
|
||||
Token: "2",
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(loginres)
|
||||
return
|
||||
|
||||
}
|
||||
|
||||
findtheoneemail := bson.M{"email": useremail}
|
||||
_, error = finduser.Find(context.Background(), findtheoneemail)
|
||||
if error == nil {
|
||||
log.Println("该邮箱已存在")
|
||||
loginres := Loginresponse{
|
||||
Success: "no",
|
||||
Message: "注册失败",
|
||||
Token: "3",
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(loginres)
|
||||
return
|
||||
}
|
||||
|
||||
redisdb := redis.NewClient(&redis.Options{
|
||||
Addr: "localhost:6379",
|
||||
Password: "",
|
||||
DB: 0,
|
||||
})
|
||||
|
||||
findveri, error := redisdb.HGet(context.Background(), "userverify", useremail).Result()
|
||||
if error != nil {
|
||||
log.Println("验证码验证失败")
|
||||
loginres := Loginresponse{
|
||||
Success: "no",
|
||||
Message: "注册失败，验证码错误",
|
||||
Token: "4",
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(loginres)
|
||||
return
|
||||
}
|
||||
|
||||
if findveri != verifycode {
|
||||
log.Println("验证码验证失败")
|
||||
loginres := Loginresponse{
|
||||
Success: "no",
|
||||
Message: "注册失败，验证码错误",
|
||||
Token: "4",
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(loginres)
|
||||
return
|
||||
}
|
||||
|
||||
_, error = finduser.InsertOne(context.Background(), bson.M{"username": username, "password": password, "email": useremail})
|
||||
if error != nil {
|
||||
log.Println("数据插入失败")
|
||||
loginres := Loginresponse{
|
||||
Success: "no",
|
||||
Message: "用户数据错误",
|
||||
Token: "5",
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(loginres)
|
||||
return
|
||||
}
|
||||
|
||||
}
|
||||
127
user/userregisterverify.go
Normal file
127
user/userregisterverify.go
Normal file
@@ -0,0 +1,127 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"html/template"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"net/smtp"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"math/rand"
|
||||
|
||||
"github.com/go-redis/redis/v8"
|
||||
)
|
||||
|
||||
type Userregveri struct {
|
||||
Email string `json:"email"`
|
||||
}
|
||||
|
||||
func randomCode() string {
|
||||
source := rand.NewSource(time.Now().UnixNano())
|
||||
r := rand.New(source)
|
||||
letters := []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
|
||||
b := make([]rune, 6)
|
||||
for i := range b {
|
||||
b[i] = letters[r.Intn(len(letters))]
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
|
||||
func Registerverify(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Access-Control-Allow-Origin", "http://localhost:5173")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
||||
w.Header().Set("Access-Control-Allow-Credentials", "true")
|
||||
|
||||
if r.Method == "OPTIONS" {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
bodyrec, error := io.ReadAll(r.Body)
|
||||
if error != nil {
|
||||
fmt.Println("接受数据错误")
|
||||
return
|
||||
}
|
||||
var bodyjx Userregveri
|
||||
error = json.Unmarshal(bodyrec, &bodyjx)
|
||||
if error != nil {
|
||||
fmt.Println("数据解析失败")
|
||||
return
|
||||
}
|
||||
|
||||
var useremail = bodyjx.Email
|
||||
log.Println("邮箱是", useremail)
|
||||
|
||||
redisdb := redis.NewClient(&redis.Options{
|
||||
Addr: "localhost:6379",
|
||||
Password: "",
|
||||
DB: 0,
|
||||
})
|
||||
var redisverifyhash = "userverify"
|
||||
|
||||
_, error = redisdb.Ping(context.Background()).Result()
|
||||
if error != nil {
|
||||
log.Println("redis连接失败")
|
||||
return
|
||||
}
|
||||
|
||||
result, _ := redisdb.Exists(context.Background(), redisverifyhash, useremail).Result()
|
||||
if result == 1 {
|
||||
log.Println("用户验证码已经存在")
|
||||
redisdb.HDel(context.Background(), redisverifyhash, useremail)
|
||||
}
|
||||
log.Println(result)
|
||||
codesend := randomCode()
|
||||
|
||||
error = redisdb.HSet(context.Background(), redisverifyhash, map[string]string{
|
||||
"useremail": useremail,
|
||||
"userver": codesend,
|
||||
}).Err()
|
||||
if error != nil {
|
||||
log.Println("redis键值设置失败")
|
||||
return
|
||||
}
|
||||
|
||||
_, error = redisdb.Expire(context.Background(), redisverifyhash, 3*time.Minute).Result()
|
||||
if error != nil {
|
||||
log.Println("有效时长设置失败")
|
||||
return
|
||||
}
|
||||
|
||||
from := "mayiming_person@163.com"
|
||||
password := "VChadvNigu6TdzYH"
|
||||
to := []string{useremail}
|
||||
subject := "MS用户验证码 --MS助手"
|
||||
|
||||
htmlContent := `以下为您的验证码，3分钟内有效，请勿分享给别人. 验证码：` + string(codesend) + ``
|
||||
t, err := template.New("emailBody").Parse(htmlContent)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
var bodyBuilder strings.Builder
|
||||
err = t.Execute(&bodyBuilder, nil)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
body := bodyBuilder.String()
|
||||
|
||||
msg := "From: " + from + "\r\n" +
|
||||
"To: " + to[0] + "\r\n" +
|
||||
"Subject: " + subject + "\r\n" +
|
||||
"\r\n" +
|
||||
body
|
||||
|
||||
smtpServer := "smtp.163.com:25"
|
||||
auth := smtp.PlainAuth("", from, password, "smtp.163.com")
|
||||
err = smtp.SendMail(smtpServer, auth, from, to, []byte(msg))
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
log.Println("邮箱发送成功!")
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user