go语言gorm实战——基础技术总结

水深无声 2023-07-09 06:22 28阅读 0赞

前言：  
一个go项目过后，有许多的精华部分、容易违犯go语言常规的部分，总结、记录，以供后续的学习、参考和大家探讨。

### 1 CRUD ###

#### 1.1 单个查询、批量查询 ####

// leimin
    // 查询指定ID主键查询和版本的数据
    func (this *Project) Find(db *gorm.DB) (b bool, err error) { 
    
    	db = db.First(this)
    	if db.RecordNotFound() { 
    		return false, nil
    	}
    	return true, db.Error
    }
    
    // leimin
    // 批量查询所有匹配数据
    func (this *Project) FindAll(db *gorm.DB) (u []Project, err error) { 
    
    	db = db.Where("user_id = ?", this.UserID).Find(&u)
    	return u, db.Error
    }

#### 1.2 模糊查询 ####

func (this *Project) FindListByLikeName(db *gorm.DB) ([]Project, error) { 
    	var result []Project
    	projects := db.Where("name like ?", "%"+this.Name+"%").Find(&result)
    	if projects.RecordNotFound() { 
    		return nil, nil
    	}
    	return result, db.Error
    }

#### 1.3 多条件查询 ####

//获取表单列表
    //libing.niu
    func (this *Form) FindList(db *gorm.DB) ([]Form, error) { 
    
    	var forms []Form
    	db.Where("major=? AND app_id=?", this.Major, this.AppID).Find(&forms)
    	return forms, db.Error
    
    }

#### 1.4 不等于查询 ####

func (this *User) FuzzySearchByName(db *gorm.DB) (users []User, err error) { 
    	db.Where("name LIKE ? AND id <> ?", "%"+this.Name+"%", this.ID).Find(&users)
    	return users, db.Error
    }

#### 1.5 倒叙，获取查询第一个元素 ####

func (this *UserProject) FindByAppIdAndMajorAndUserId(db *gorm.DB) (bool, error) { 
    
    	db = db.Order("created_at desc").First(this, "app_id = ? and major = ? and user_id = ? ", this.AppID, this.Major, this.UserID)
    	if db.RecordNotFound() { 
    		return false, nil
    	}
    	return true, db.Error
    }

#### 1.6 单个新增，批量新增 ####

// leimin
    // 新增数据
    func (this *Project) Create(db *gorm.DB) (err error) { 
    
    	return db.Create(this).Error
    }
    
    // 批量新增

#### 1.7 单个删除、批量删除 ####

// leimin
    //通过结构体删除
    func (this *UserProject) Delete(db *gorm.DB) (err error) { 
    	db = db.Delete(this)
    	return db.Error
    }
    // 批量删除
    //leimin
    func (this *UserProject) DeleteAll(db *gorm.DB) (err error) { 
    
    	return db.Where("app_id = ? AND major = ?", this.AppID, this.Major).
    		Delete(UserProject{ }).Error
    }

#### 1.8 单个修改、批量修改 ####

// leimin
    // 修改数据属性
    func (this *Project) Update(db *gorm.DB) (err error) { 
    
    	return db.Save(this).Error
    }
    
    //批量修改
    //author: weihong.zhu
    func (this *Project) UpdateMinor(db *gorm.DB) error { 
    	return db.Model(this).Where("major = ?", this.Major).Update("minor", this.Minor).Error
    }

### 2 异常处理 ###

#### 2.1 model层异常 ####

db.Error

#### 2.2 service 层异常 ####

db, err := orm.CreateDB()
    	if err != nil { 
    		logrus.Errorf("UpdateProject service, orm.CreateDB error :%v\n", err.Error())
    		return nil, errors.New("数据库连接异常")
    	}
    	// 修改完成后，再到数据库获取最新数据，用于封装返回参数
    	found, err := project.Find(db)
    	if err != nil { 
    		logrus.Errorf("UpdateProject service, project.Find error :%v\n", err.Error())
    		return nil, errors.New("数据库查询异常")
    	}
    	if !found { 
    		return nil, errors.New("无数据")
    	}

#### 2.3 controller 层异常 ####

// author:leimin
    // title：创建应用
    // detail：从project表新增一条数据
    // return：新增后的数据
    func ProjectCreate(c *gin.Context) { 
    
    	userId := c.GetHeader(USER_ID)
    	name := c.PostForm(NAME)
    	appType := c.PostForm(TYPE)
    
    	if userId == "" || name == "" || appType == "" { 
    		c.JSON(http.StatusOK, response.NewBaseResponseWithMessage(response.ResponseCodeBadRequest, "请求参数异常"))
    		return
    	}
    	//todo 保证返回的值都有值
    	pic, suffix, err := getIcon(c)
    	if err != nil { 
    		c.JSON(http.StatusOK, response.NewBaseResponseWithMessage(response.ResponseCodeBadRequest, err.Error()))
    		return
    	}
    	res, err := service.AddProject(name, pic, appType, userId, suffix)
    	if err != nil { 
    		logrus.Error(err)
    		c.JSON(http.StatusOK, response.NewBaseResponseWithMessage(response.ResponseCodeInternalServerError, err.Error()))
    		return
    	}
    
    	//返回id,name,version,type,creator
    	c.JSON(http.StatusOK, response.NewBaseResponseOkWithData(*res))
    }

### 3 非空校验 ###

字符串、指针、数组、数字

ups, err := userProject.FindByUserId(db)
    	if err != nil { 
    		logrus.Errorf("projectList service, userProject.FindByUserId error on row 24:%v\n", err.Error())
    		return nil, errors.New("数据库查询异常")
    	}
    	if len(ups) == 0 { 
    		return nil, nil
    	}
    	if str == "" {  // 字符串如此判断
    		return nil, nil
    	}

### 4 返回参数 ###

// leimin
    // 返回参数、其类型提前定义，方法中直接使用
    func UpdateProject(appId string, majorCode int, name string, icon []byte, userId string, appType string, fileType string) (re *view.Project, err error) { ...}
    // 同类型合并
    // 基础参数放前边
    func UpdateProject(icon []byte, majorCode int, appId ,  name , userId , appType , fileType string) (re *view.Project, err error) { ...}

### 5 去重 ###

// 对象切片去重
    func deduplication(list []model.UserProject) (mp map[string]model.UserProject) { 
    
    	var dedupMap = make(map[string]model.UserProject)
    	for _, v := range list { 
    		if _, ok := dedupMap[v.AppID+strconv.Itoa(v.Major)]; !ok { 
    			dedupMap[v.AppID+strconv.Itoa(v.Major)] = v
    		}
    	}
    
    	return dedupMap
    }

### 6 获取db、获取事务、回滚、提交 ###

// leimin
    func AddProject(icon []byte, name , appType , userId , fileType string) (re *view.Project, err error) { 
    
    	vp := view.Project{ }
    
    	db, err := orm.CreateTx()
    	if err != nil { 
    		logrus.Errorf("projectCreate service, orm.CreateTx error:%v\n", err.Error())
    		return nil, errors.New("数据库连接异常")
    	}
    
    	project := model.Project{ 
    		Model: model.Model{ 
    			ID: utils.UUID(),
    		},
    		Major:    1,
    		Minor:    1,
    		Name:     name,
    		Icon:     icon,
    		IconType: fileType,
    		Type:     appType,
    		UserID:   userId,
    	}
    
    	err = project.Create(db)
    	if err != nil { 
    		db.Rollback()
    		logrus.Errorf("projectCreate service, project.Create error:%v\n", err.Error())
    		return nil, errors.New("数据库操作异常")
    	}
    
    	userProject := model.UserProject{ 
    		Model: model.Model{ 
    			ID: utils.UUID(),
    		},
    		AppID:  project.ID,
    		Major:  project.Major,
    		UserID: project.UserID,
    		Type:   "creator",
    	}
    
    	err = userProject.Create(db)
    	if err != nil { 
    		db.Rollback()
    		logrus.Errorf("projectCreate service, userProject.Create error:%v\n", err.Error())
    		return nil, errors.New("数据库操作异常")
    	}
    	db.Commit()
    
    	vp.ID = project.ID
    	vp.Name = project.Name
    	vp.Type = project.Type
    	vp.Created = true
    	vp.Version = strconv.Itoa(project.Major) + "." + strconv.Itoa(project.Minor)
    	return &vp, err
    }