diff --git a/.idea/workspace.xml b/.idea/workspace.xml index d6a601b..9627097 100755 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -5,8 +5,10 @@ + + + + + + + + + + @@ -76,6 +86,15 @@ + + + + + + + + + diff --git a/main.go b/main.go index d13fe8c..43fccc3 100755 --- a/main.go +++ b/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, " ") diff --git a/models/User.go b/models/User.go index 38e55b0..d076bb2 100644 --- a/models/User.go +++ b/models/User.go @@ -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{}) diff --git a/models/Userdyw.go b/models/Userdyw.go new file mode 100644 index 0000000..7a1c519 --- /dev/null +++ b/models/Userdyw.go @@ -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 +} +