feat: 数据库管理优化1

internal/data/database_server.go

@@ -89,6 +89,10 @@ func (r databaseServerRepo) Update(req *request.DatabaseServerUpdate) error {
 	return app.Orm.Save(server).Error
 }
 
+func (r databaseServerRepo) UpdateRemark(req *request.DatabaseServerUpdateRemark) error {
+	return app.Orm.Model(&biz.DatabaseServer{}).Where("id = ?", req.ID).Update("remark", req.Remark).Error
+}
+
 func (r databaseServerRepo) Delete(id uint) error {
 	// 删除服务器下的所有用户
 	if err := NewDatabaseUserRepo().DeleteByServerID(id); err != nil {

internal/data/database_user.go

@@ -88,12 +88,17 @@ func (r databaseUserRepo) Create(req *request.DatabaseUserCreate) error {
 	user := &biz.DatabaseUser{
 		ServerID: req.ServerID,
 		Username: req.Username,
-		Password: req.Password,
 		Host:     req.Host,
-		Remark:   req.Remark,
 	}
 
-	return app.Orm.Create(user).Error
+	if err = app.Orm.FirstOrInit(user, user).Error; err != nil {
+		return err
+	}
+
+	user.Password = req.Password
+	user.Remark = req.Remark
+
+	return app.Orm.Save(user).Error
 }
 
 func (r databaseUserRepo) Update(req *request.DatabaseUserUpdate) error {
@@ -146,6 +151,17 @@ func (r databaseUserRepo) Update(req *request.DatabaseUserUpdate) error {
 	return app.Orm.Save(user).Error
 }
 
+func (r databaseUserRepo) UpdateRemark(req *request.DatabaseUserUpdateRemark) error {
+	user, err := r.Get(req.ID)
+	if err != nil {
+		return err
+	}
+
+	user.Remark = req.Remark
+
+	return app.Orm.Save(user).Error
+}
+
 func (r databaseUserRepo) Delete(id uint) error {
 	user, err := r.Get(id)
 	if err != nil {