Go框架三件套详解

用户7992110583125
123 阅读4分钟

基本介绍：

Gorm

Gorm 是一个已经迭代了10年+的功能强大的ORM框架，在字节内部被广泛使用并且拥有非常丰富的开源扩展。

Kitex

Kitex 是字节内部的Golang微服务RPC框架，具有高性能、强可扩展的主要特点，支持多协议并且拥有丰富的开源扩展。

Hertz

Hertz 是字节内部的 HTTP框架，参考了其他开源框架的优势，结合字节跳动内部的需求，具有高易用性、高性能、高扩展性特点。

Grom使用

  1. 首先编写gorm model

type Product struct {
    Code string

    Price vint
}
  1. 为model定义表名
func (p Product) TableName() string{
    return "product"
}
  1. 操作数据库

func main() {

    db, err := gorm.Open(
    mysql. Open( "user:pass@tcp(127.0.0.1:3306)/dbname？charset=utf8mb4&parseTime=True&loc=Local"),
    &gorm.Config{})
    is err != nil{
        panic（"failed to connect database")
    }

    // Create
    db.Create(&Product{Code: "D42",Price:100})
    // Read
    var product Product
    db.First(&product, 1)  //根据整形主键查找
    db.First (&product, "code = ?", "D42") // 含找 code 字段值为 D42 的记录
   
    // Update - 将product的Price值更新为200
    db.Model(&product).Update(column:"Price",value:200)
    // Update - 更新多个字段
    db.Model(&product).Updates(Product{price: 200, Code: "F42"}） // 仅更新非零值字段
    db.Model(&product).Updates(map[string]interface{}{"Price"：200,"Code"："F42"}
    // Delete - 删除字段
    db.Delete(&product,1)

Gorm创建数据

package main

import (
   "fmt"
   "gorm.io/driver/mysql"
   "gorm.io/gorm"
)

type Product struct {
   ID    uint   `gorm:"primaryKey"`
   Code  string `gorm:"column:code"`
   Price uint   `gorm:"column:user_id"`
}

func main() {
   dsn := "username:pwd@tcp(127.0.0.1:3306)/gorm?charset=utf8mb4"
   db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
   if err != nil {
      panic("connect error")
   }
   //p := &Product{Code: "042", Price: 100}
   //res := db.Create(p)
   products := []*Product{{Code: "041"}, {Code: "042"}, {Code: "043"}} //通过切片创建数据
   res := db.Create(products)
   fmt.Println(res.Error)
   for _, p := range products {
      fmt.Println(p.ID)
   }
}

Gorm查询数据

func main() {
   dsn := "username:pwd@tcp(127.0.0.1:3306)/gorm?charset=utf8mb4"
   db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
   if err != nil {
      panic("connect error")
   }
   P := &Product{}
   db.First(P) // 查询为空时会返回ErrRecordNotFound
   products := make([]*Product, 0)
   result := db.Where("code = 041").Find(&products)
   fmt.Println(result.RowsAffected)
   fmt.Println(result.Error)
   for _, p := range products {
      fmt.Println(p.Code, p.ID)
   }
   //查询in
   db.Where("code IN ?", []string{"041,042"}).Find(&products)
   //查询like
   db.Where("code like ?", "%4%").Find(&products)
   //多条件查询
   db.Where("code = ? AND id = ?", "041", "4")
   //直接传入结构体
   db.Where(&Product{ID: 1, Code: "041"}).Find(&products)
   //通过map查询
   db.Where(map[string]interface{}{"Code":"041","ID":4}).Find(&products)
}

Gorm修改数据

func main() {
   dsn := "username:pwd@tcp(127.0.0.1:3306)/gorm?charset=utf8mb4"
   db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
   if err != nil {
      panic("connect error")
   }
   //修改单个列
   result := db.Model(&Product{}).Where("code = ?", "041").Update("user_id", "100")
   fmt.Println(result.Error)
   fmt.Println(result.RowsAffected)

   //修改多个列
   result := db.Model(&Product{ID: 3}).Updates(Product{Code: "042", Price: 200}) //结构体里面只能指定主键，否则需要添加Where
   fmt.Println(result.Error)
   fmt.Println(result.RowsAffected)

   //更新选定字段 update prouducts SET code = `777` where id = 6
   result := db.Model(&Product{ID: 6}).Select("code").Updates(map[string]interface{}{"code": "777", "price": 18})
   fmt.Println(result.Error)
   fmt.Println(result.RowsAffected)

   //SQL表达式更新 update products SET user_id = user_id * 2 + 100 where id = 7
   result := db.Model(&Product{ID: 7}).Update("user_id", gorm.Expr("user_id * ? + ?", 2, 100))
   fmt.Println(result.Error)
   fmt.Println(result.RowsAffected)
}

GORM删除数据

type Product struct {
   ID      uint           `gorm:"primaryKey"`
   Code    string         `gorm:"column:code"`
   Price   uint           `gorm:"column:user_id"`
   Deleted gorm.DeletedAt //增加软删除
}

func main() {
   dsn := "username:pwd@tcp(127.0.0.1:3306)/gorm?charset=utf8mb4"
   db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
   if err != nil {
      panic("connect error")
   }
   //DELETE FROM products WHERE id = 1
   //db.Delete(&Product{}, 1)
   //db.Delete(&Product{}, "1")
   //DELETE FROM products WHERE id in (1,2,3)
   //result := db.Delete(&Product{}, []int{1, 2, 3})
   //DELETE FROM products WHERE code LIKE "%4%"
   result := db.Delete(&Product{}, "code like ?", "%4%")
   fmt.Println(result.RowsAffected)
}

GORM事务

  1. 手动提交
func main() {
   dsn := "username:pwd@tcp(127.0.0.1:3306)/gorm?charset=utf8mb4"
   db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
   if err != nil {
      panic("connect error")
   }
   tx := db.Begin() //开启事务，一定要使用固化链接
   result := db.Delete(&Product{}, "code like ?", "%4%")
   if result.Error != nil {
      tx.Rollback() //回滚事务
   }
   tx.Commit() //提交事务
   fmt.Println(result.RowsAffected)
}
  1. 自动提交（推荐）
func main() {
   dsn := "username:pwd@tcp(127.0.0.1:3306)/gorm?charset=utf8mb4"
   db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
   if err != nil {
      panic("connect error")
   }
   err = db.Transaction(func(tx *gorm.DB) error {
      result := db.Delete(&Product{}, "code like ?", "%7%")
      if result.Error != nil {
         return err
      }
      fmt.Println(result.RowsAffected)
      return nil
   })
   if err != nil {
      return
   }
}

Gorm HOOK

GORM 在提供了 CURD 的 Hook 能力。

Hook 是在创建、查询、更新、删除等操作之前、之后自动调用的函数。

如果任何 Hook 返回错误，GORM 将停止后续的操作并回滚事务。

func (product *Product) BeforeCreate(tx *gorm.DB) (err error) {
   if product.Price < 0 {
      return errors.New("价格必须大于0")
   }
   return
}

func (product *Product) AfterCreate(tx *gorm.DB) (err error) {
   return tx.Create(&Product{Price: 400, Code: "888"}).Error
}

Kitex

Kitex 安装

go install github.com/cloudwego/kitex/tool/cmd/kitex@ latest

go install github.com/cloudwego/thriftgo@latest

定义IDL（基于thrift语法）

  • 如果我们要进行 RPC，就需要知道对方的接口是什么，需要传什么参数，同时也需要知道返回值是什么样的。这时候，就需要通过IDL来约定双方的协议，就像在写代码的时候需要调用某个函数，我们需要知道函数签名一样。
namespace go api

struct Request {

    1: string message

}

struct Response {

    1: string message

}

service Echo {

    Response echo(1: Request req)

}

Kitex生成代码

使用 kitex -module example -service example echo.thrift 命令生成代码

Hertz

Hertz基本使用

package main

import (
   "context"
   "github.com/cloudwego/hertz/pkg/app"
   "github.com/cloudwego/hertz/pkg/app/server"
   "github.com/cloudwego/hertz/pkg/common/utils"
   "github.com/cloudwego/hertz/pkg/protocol/consts"
)

func main() {
   h := server.Default(server.WithHostPorts("127.0.0.1:8080"))
   h.GET("/ping", func(c context.Context, ct *app.RequestContext) {
      ct.JSON(consts.StatusOK, utils.H{"ping": "pong"})
   })
   h.Spin()
}

Hertz路由

func RegisterRoute(h *server.Hertz) {
   h.GET("/get", func(ct context.Context, c *app.RequestContext) {
      c.String(consts.StatusOK, "get")
   })
   h.POST("/post", func(ct context.Context, c *app.RequestContext) {
      c.String(consts.StatusOK, "post")
   })
   h.PUT("/put", func(ct context.Context, c *app.RequestContext) {
      c.String(consts.StatusOK, "put")
   })
   h.DELETE("/delete", func(ct context.Context, c *app.RequestContext) {
      c.String(consts.StatusOK, "delete")
   })
   h.PATCH("/patch", func(ct context.Context, c *app.RequestContext) {
      c.String(consts.StatusOK, "patch")
   })
   h.HEAD("/head", func(ct context.Context, c *app.RequestContext) {
      c.String(consts.StatusOK, "head")
   })
   h.OPTIONS("/options", func(ct context.Context, c *app.RequestContext) {
      c.String(consts.StatusOK, "options")
   })
}
avatar
文章
阅读
粉丝