golang beego结合wire依赖注入及自动路由

2024-04-30 10:20

本文主要是介绍golang beego结合wire依赖注入及自动路由,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

1 安装wire

1.1 通过命令直接安装

go install github.com/google/wire/cmd/wire@latest

1.2 通过go get方式安装

go get github.com/google/wire/cmd/wire

进入目录编译

cd C:\Users\leell\go\pkg\mod\github.com\google\wire@v0.6.0\cmd\wire
go build

然后将wire.exe移动到 C:\Users\leell\go\bin目录下

具体目录通过go env进行查看

C:\Users\leell\go>go env
set GO111MODULE=on
set GOARCH=amd64
set GOBIN=C:\Users\leell\go\bin

2 配置业务代码

我没有按照beego的默认目录,而是通过自定义app包名进行业务模块化

2.1 分类模块

app/category/category_controller.go

type Controller struct {controller.BaseControllerservice *Service
}func NewController(service *Service) *Controller {return &Controller{service: service}
}

app/category/category_service.go

type Service struct {dao *Dao
}func NewService(dao *Dao) *Service {return &Service{dao}
}

app/category/category_dao.go 

type Dao struct {ormer orm.Ormer
}func NewDao(ormer orm.Ormer) *Dao {return &Dao{ormer}
}

app/category/fx.go

package categoryimport ("github.com/beego/beego/v2/server/web""github.com/google/wire"
)type FxRouter struct {controller *Controller
}func NewAppRouter(controller *Controller) *FxRouter {return &FxRouter{controller}
}
func (r *FxRouter) Router(namespace *web.Namespace) {namespace.Get("/category", r.controller.GetAll)namespace.Get("/category/:id", r.controller.GetOne)namespace.Post("/category", r.controller.Post)namespace.Put("/category/:id", r.controller.Put)namespace.Delete("/category/:id", r.controller.Delete)
}var Set = wire.NewSet(NewController, NewService, NewDao, NewAppRouter)

app/fx.go:将分类模块引入app/fx.go中并且等待注册路由

package appimport ("github.com/beego/beego/v2/server/web""github.com/google/wire""zhiqu/app/category""zhiqu/app/login""zhiqu/app/user"
)type FxRouter struct {categoryFxRouter *category.FxRouterloginFxRouter    *login.FxRouteruserFxRouter     *user.FxRouter
}func NewAppRouter(categoryFxRouter *category.FxRouter,loginFxRouter *login.FxRouter,userFxRouter *user.FxRouter) *FxRouter {return &FxRouter{categoryFxRouter, loginFxRouter, userFxRouter}
}
func (r *FxRouter) Router(namespace *web.Namespace) {r.categoryFxRouter.Router(namespace)r.loginFxRouter.Router(namespace)r.userFxRouter.Router(namespace)
}var Set = wire.NewSet(NewAppRouter, category.Set, login.Set, user.Set)

2.2 项目wire定义总提供者provider

infrastructure/database/pgsql/db.go

package pgsqlimport ("fmt""github.com/beego/beego/v2/client/orm"beego "github.com/beego/beego/v2/server/web"_ "github.com/lib/pq"
)func init() {pgsqlUser, _ := beego.AppConfig.String("pgsqlUser")pgsqlPass, _ := beego.AppConfig.String("pgsqlPass")pgsqlDbName, _ := beego.AppConfig.String("pgsqlDbName")pgsqlHost, _ := beego.AppConfig.String("pgsqlHost")pgsqlPort, _ := beego.AppConfig.String("pgsqlPort")pgsqlSslMode, _ := beego.AppConfig.String("pgsqlSslMode")// 注册驱动err := orm.RegisterDriver("postgres", orm.DRPostgres)if err != nil {panic(err)}// 设置默认数据库err = orm.RegisterDataBase("default", "postgres", fmt.Sprintf("user=%s password=%s dbname=%s host=%s port=%s sslmode=%s", pgsqlUser, pgsqlPass, pgsqlDbName, pgsqlHost, pgsqlPort, pgsqlSslMode))if err != nil {panic(err)}
}
func NewPgsqlOrm() orm.Ormer {ormer := orm.NewOrm()return ormer
}

provider/provider.go

var Set = wire.NewSet(NewBeanFactory, pgsql.NewPgsqlOrm, app.Set)

2.3 给项目一个bean工厂

这点有点像spring的beanfactory,这里主要是:

(1)解决beego路由一开始执行init能够得到所有controller的实例

(2)go项目中存在recycle的循环依赖,这里就可以用beanFactory做一个中间过度

package providerimport ("github.com/beego/beego/v2/core/logs""sync""zhiqu/app"
)var (AppBeanFactory *BeanFactoryonce           sync.OnceBeeLog         *logs.BeeLogger
)type BeanFactory struct {AppFxRouter *app.FxRouter
}func init() {BeeLog = logs.NewLogger()err := BeeLog.SetLogger(logs.AdapterConsole)if err != nil {return}
}
func NewBeanFactory(appFxRouter *app.FxRouter) *BeanFactory {once.Do(func() {AppBeanFactory = &BeanFactory{AppFxRouter: appFxRouter,}})return AppBeanFactory
}

3 wire配置代码及生成gen代码

 wire/wire.go

//go:build wireinject
// +build wireinjectpackage wireimport ("zhiqu/provider"
)
import "github.com/google/wire"// wire.go 初始化模块
func NewApp() (*provider.BeanFactory, error) {panic(wire.Build(provider.Set))
}

执行wire命令生成gen代码得到wire/wire_gen.go

wire gen wire/wire.go

生成最终代码文件内容

wire/wire_gen.go

// Code generated by Wire. DO NOT EDIT.//go:generate go run -mod=mod github.com/google/wire/cmd/wire
//go:build !wireinject
// +build !wireinjectpackage wireimport ("zhiqu/app/category""zhiqu/provider"
)// Injectors from wire.go:// wire.go 初始化模块
func NewApp() (*provider.BeanFactory, error) {categoryDao := category.NewDao()categoryService := category.NewService(categoryDao)categoryController := category.NewController(categoryService)beanFactory := provider.NewBeanFactory(categoryController)return beanFactory, nil
}

4 路由配置

routers/router.go

// Package routers 路由配置
package routersimport ("github.com/beego/beego/v2/server/web""zhiqu/wire"
)func init() {beanFactory, err := wire.NewApp()if err != nil {panic(err)}//这是一种可实现的依赖注入全自动方式ns := web.NewNamespace("/v1", func(namespace *web.Namespace) {beanFactory.AppFxRouter.Router(namespace)})web.SetStaticPath("/swagger", "swagger")web.AddNamespace(ns)
}

可以看出在路由文件中添加了wire生成的gen代码的获取的beanfactory工厂,然后将所有的单例go对象交由wire进行管理

注意:这里采用的是beego的web方式而不是beego方式,这里都是手动配置路由,不要去通过命令生成commentsRouter.go这样的路由文件

为啥不采用beego注册路由?

因为我们要想做到依赖注入,得确保controller、service、dao这些是单例,而beego注册的路由内部是现实每次收到请求都会通过反射技术重新构建一个controler进行处理。

正式因为最开始通过官方生成的项目配置路由注册时beego方式,而不能实现依赖注入,最终选择了gin+gorm+go.uber.org/fx的架构组合

5 查看最终结果

不过我不太推荐这种方式的依赖注入,虽然是在编译期实现了代码的生成,减少了启动通过反射来进行依赖注入的方式的时间,但是对开发人员来说多了一道工序。并且启动时间也要不了多久,而且编译器和运行期的依赖注入对接口运行是没有影响。

给大家推荐一块依赖注入框架:go.uber.org/fx

源代码:GitHub - leellun/beego-wire

这篇关于golang beego结合wire依赖注入及自动路由的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/948606

相关文章

Python结合PyWebView库打造跨平台桌面应用

《Python结合PyWebView库打造跨平台桌面应用》随着Web技术的发展,将HTML/CSS/JavaScript与Python结合构建桌面应用成为可能,本文将系统讲解如何使用PyWebView... 目录一、技术原理与优势分析1.1 架构原理1.2 核心优势二、开发环境搭建2.1 安装依赖2.2 验

IDEA自动生成注释模板的配置教程

《IDEA自动生成注释模板的配置教程》本文介绍了如何在IntelliJIDEA中配置类和方法的注释模板,包括自动生成项目名称、包名、日期和时间等内容,以及如何定制参数和返回值的注释格式,需要的朋友可以... 目录项目场景配置方法类注释模板定义类开头的注释步骤类注释效果方法注释模板定义方法开头的注释步骤方法注

Spring Boot循环依赖原理、解决方案与最佳实践(全解析)

《SpringBoot循环依赖原理、解决方案与最佳实践(全解析)》循环依赖指两个或多个Bean相互直接或间接引用,形成闭环依赖关系,:本文主要介绍SpringBoot循环依赖原理、解决方案与最... 目录一、循环依赖的本质与危害1.1 什么是循环依赖?1.2 核心危害二、Spring的三级缓存机制2.1 三

pytorch自动求梯度autograd的实现

《pytorch自动求梯度autograd的实现》autograd是一个自动微分引擎,它可以自动计算张量的梯度,本文主要介绍了pytorch自动求梯度autograd的实现,具有一定的参考价值,感兴趣... autograd是pytorch构建神经网络的核心。在 PyTorch 中,结合以下代码例子,当你

Python如何自动生成环境依赖包requirements

《Python如何自动生成环境依赖包requirements》:本文主要介绍Python如何自动生成环境依赖包requirements问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑... 目录生成当前 python 环境 安装的所有依赖包1、命令2、常见问题只生成当前 项目 的所有依赖包1、

PyInstaller打包selenium-wire过程中常见问题和解决指南

《PyInstaller打包selenium-wire过程中常见问题和解决指南》常用的打包工具PyInstaller能将Python项目打包成单个可执行文件,但也会因为兼容性问题和路径管理而出现各种运... 目录前言1. 背景2. 可能遇到的问题概述3. PyInstaller 打包步骤及参数配置4. 依赖

Spring Boot项目中结合MyBatis实现MySQL的自动主从切换功能

《SpringBoot项目中结合MyBatis实现MySQL的自动主从切换功能》:本文主要介绍SpringBoot项目中结合MyBatis实现MySQL的自动主从切换功能,本文分步骤给大家介绍的... 目录原理解析1. mysql主从复制(Master-Slave Replication)2. 读写分离3.

golang获取当前时间、时间戳和时间字符串及它们之间的相互转换方法

《golang获取当前时间、时间戳和时间字符串及它们之间的相互转换方法》:本文主要介绍golang获取当前时间、时间戳和时间字符串及它们之间的相互转换,本文通过实例代码给大家介绍的非常详细,感兴趣... 目录1、获取当前时间2、获取当前时间戳3、获取当前时间的字符串格式4、它们之间的相互转化上篇文章给大家介

微信公众号脚本-获取热搜自动新建草稿并发布文章

《微信公众号脚本-获取热搜自动新建草稿并发布文章》本来想写一个自动化发布微信公众号的小绿书的脚本,但是微信公众号官网没有小绿书的接口,那就写一个获取热搜微信普通文章的脚本吧,:本文主要介绍微信公众... 目录介绍思路前期准备环境要求获取接口token获取热搜获取热搜数据下载热搜图片给图片加上标题文字上传图片

springboot循环依赖问题案例代码及解决办法

《springboot循环依赖问题案例代码及解决办法》在SpringBoot中,如果两个或多个Bean之间存在循环依赖(即BeanA依赖BeanB,而BeanB又依赖BeanA),会导致Spring的... 目录1. 什么是循环依赖?2. 循环依赖的场景案例3. 解决循环依赖的常见方法方法 1:使用 @La