master
修改密码漏洞修复完成 8 months ago
parent bdaf4bb678
commit 603a1b1e16

@ -5,8 +5,10 @@
</component>
<component name="ChangeListManager">
<list default="true" id="9f8c3ce4-c378-4191-b7a4-3d5acbc0ebbc" name="Changes" comment="rrddf">
<change afterPath="$PROJECT_DIR$/models/Userdyw.go" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/main.go" beforeDir="false" afterPath="$PROJECT_DIR$/main.go" afterDir="false" />
<change beforePath="$PROJECT_DIR$/models/User.go" beforeDir="false" afterPath="$PROJECT_DIR$/models/User.go" afterDir="false" />
</list>
<option name="SHOW_DIALOG" value="false" />
<option name="HIGHLIGHT_CONFLICTS" value="true" />
@ -67,6 +69,14 @@
</key>
</component>
<component name="RunManager">
<configuration default="true" type="GoApplicationRunConfiguration" factoryName="Go Application">
<module name="go_mqtt" />
<working_directory value="$PROJECT_DIR$" />
<kind value="FILE" />
<directory value="$PROJECT_DIR$" />
<filePath value="$PROJECT_DIR$" />
<method v="2" />
</configuration>
<configuration name="go build main.go" type="GoApplicationRunConfiguration" factoryName="Go Application" temporary="true" nameIsGenerated="true">
<module name="go_mqtt" />
<working_directory value="$PROJECT_DIR$" />
@ -76,6 +86,15 @@
<filePath value="$PROJECT_DIR$/main.go" />
<method v="2" />
</configuration>
<configuration default="true" type="GoTestRunConfiguration" factoryName="Go Test">
<module name="go_mqtt" />
<working_directory value="$PROJECT_DIR$" />
<kind value="DIRECTORY" />
<directory value="$PROJECT_DIR$" />
<filePath value="$PROJECT_DIR$" />
<framework value="gotest" />
<method v="2" />
</configuration>
<recent_temporary>
<list>
<item itemvalue="Go Build.go build main.go" />

@ -526,16 +526,16 @@ MIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQCYwvInDUICXbmWcOAR5hm86mz7WlKL
decrptCode := RSA_Decrypts(decodeByte, privatePath)
fmt.Println("解密后的字符串:")
fmt.Println(string(decrptCode))
fmt.Println("-----user--------")
//var user models.User
//json.Unmarshal(decrptCode, &user)
//fmt.Println(user)
//
//if models.GetUser(&user) {
// models.UpdateUser(&user)
//}else{
// models.SaveUser(&user)
//}
fmt.Println("-----userdyw--------")
var user models.Userdyw
json.Unmarshal(decrptCode, &user)
fmt.Println(user)
if models.GetUserdyw(&user) {
models.UpdateUserdyw(&user)
}else{
models.SaveUserdyw(&user)
}
}else{
result := strings.Split(message, " ")

@ -17,7 +17,7 @@ type User struct {
Appversion string `gorm:"size:255"`
Name string `gorm:"size:255"`
}
//Userdyw
func init() {
fmt.Println("User init()")
mydb.DB.AutoMigrate(&User{})

@ -0,0 +1,77 @@
package models
import (
"errors"
"fmt"
"go_mqtt/mydb"
"gorm.io/gorm"
)
type Userdyw struct {
Username string `gorm:"primaryKey;size:255"`
Password string `gorm:"size:255"`
Date string `gorm:"size:255"`
Imeinum string `gorm:"size:50"`
Sysversion string `gorm:"size:100"`
Phonemodel string `gorm:"size:255"`
Appversion string `gorm:"size:255"`
Name string `gorm:"size:255"`
}
//
func init() {
fmt.Println("Userdyw init()")
mydb.DB.AutoMigrate(&Userdyw{})
}
// TableName 会将 User 的表名重写为 `user`
func (Userdyw) TableName() string {
return "userdyw"
}
func SaveUserdyw(user *Userdyw) {
result := mydb.DB.Create(&user)
if result.Error != nil {
fmt.Println("Failed to create User:", result.Error)
} else {
fmt.Println("User created successfully!")
}
}
func UpdateUserdyw(item *Userdyw) {
fmt.Println("用户存在,更新操作")
var user Userdyw
mydb.DB.First(&user, item.Username)
user.Password=item.Password
user.Date=item.Date
user.Imeinum=item.Imeinum
user.Sysversion=item.Sysversion
user.Phonemodel=item.Phonemodel
user.Appversion=item.Appversion
user.Name=item.Name
mydb.DB.Save(&user)
}
func GetUserdyw(item *Userdyw) bool {
var user Userdyw
//mydb.DB.First(&user, item.Username)
result := mydb.DB.First(&user, item.Username).Error
fmt.Println("-----user---->>")
fmt.Println(user)
if errors.Is(result, gorm.ErrRecordNotFound) {
// 记录不存在
fmt.Println("用户不存在")
} else if result == nil {
// 记录存在
fmt.Println("用户存在")
return true
} else {
// 其他错误
fmt.Println("查询出错")
}
return false
}
Loading…
Cancel
Save