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.BaseController
	service *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 category

import (
	"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 app

import (
	"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.FxRouter
	loginFxRouter    *login.FxRouter
	userFxRouter     *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 pgsql

import (
	"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 provider

import (
	"github.com/beego/beego/v2/core/logs"
	"sync"
	"zhiqu/app"
)

var (
	AppBeanFactory *BeanFactory
	once           sync.Once
	BeeLog         *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 wireinject

package wire

import (
	"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 !wireinject

package wire

import (
	"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 routers

import (
	"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

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:/a/583859.html

如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!

相关文章

C++中list的使用

文章目录 一、 list简介二、 构造函数1. 默认构造函数2. 拷贝构造3. 迭代器区间初始化4. 插入n个值为x的数据5. 代码示例 三、 容量和元素访问1. empty()2. size()3. max_size()3. back()4. front()5. 代码示例 四、 增删查改1. push_back()2. push_front()3. emplace_back()4.…

设计模式之装饰者模式DecoratorPattern(四)

一、模板模式介绍 模板方法模式(Template Method Pattern),又叫模板模式(Template Pattern), 在一个抽象类公开定义了执行它的方法的模板。它的子类可以更需要重写方法实现,但可以成为典型类中…

d16(149-153)-勇敢开始Java,咖啡拯救人生

跳过了p151 四小时的讲题我不敢听:) Stream Stream流,是JDK8后新增的API,可以用于操作集合或者数组的数据 优势:大量结合了Lambda的语法风格,该方式更强大更简单,代码简洁,可读性好 常用方法 …

Mycat(三)读写分离双主双从

文章目录 搭建双主双从双主机配置双从机配置双从配置两个主机互相复制停止从服务复制功能重新配置主从 修改 Mycat 的集群配置实现多种主从双主双从集群角色划分增加两个数据源修改集群配置文件读写分离配置扩展(1)读写分离(一主一从,无备)(m是主,s是从)…

Grafana页面嵌入自建Web应用页面

目录 一、应用场景 二、实现方式 1、修改Grafana配置文件 2、获取监控页面url 3、隐藏左侧和顶部菜单 一、应用场景 需要将Grafana监控页面嵌入自建Web应用页面,使Grafana监控页面成为自建Web应用的一部分。 二、实现方式 总体思路:修改Grafana配…

刷机维修进阶教程-----红米note7 修复基带 更改参数 nv损坏故障 实例步骤操作解析

在前面的博文中我有说过。不管刷更改参数还是修复基带,尽可能的情况下备份一些主要分区,上期讲了小米6x 小米5 小米6这些机型更改参数的具体步骤。今天的教程以红米note7为例解析下改参数和修复nv损坏的具体步骤,两者操作实际没有什么冲突。有兴趣的友友建议多看下我关于…

10G MAC层设计系列-(2)MAC RX模块

一、概述 MAC RX模块的需要进行解码、对齐、CRC校验。 因为在空闲的时候10G PCS/PMA会一直向外吐空闲符(x07)所以需要根据开始符、结束符将有效数据从码流中截取,也就是解码。 因为开始字符的所在位置有两种形式,而结束字符的位…

大数据学习笔记14-Hive基础2

一、数据字段类型 数据类型 :LanguageManual Types - Apache Hive - Apache Software Foundation 基本数据类型 数值相关类型 整数 tinyint smallint int bigint 小数 float double decimal 精度最高 日期类型 date 日期 timestamps 日期时间 字符串类型 s…

UE C++ 链表

目录 概要单链表双向链表头插入尾插入中间插入删除查找 小结 概要 链表 简单说明,链表有单链表,双向链表,循环链表(本篇文章以UE c代码说明)。链表的操作,插入,删除,查找。插入,删除效率高&…

练习题(2024/4/29)

在深度优先遍历中:有三个顺序,前中后序遍历 这里前中后,其实指的就是中间节点的遍历顺序,只要记住 前中后序指的就是中间节点的位置就可以了。 如图 1二叉树的前序遍历 给你二叉树的根节点 root ,返回它节点值的 前…

【项目】仿muduo库One Thread One Loop式主从Reactor模型实现高并发服务器(Http测试板块)

【项目】仿muduo库One Thread One Loop式主从Reactor模型实现高并发服务器(Http测试板块) 一、使用Http网页界面1、main.cc原码和index.html原码2、运行结果(1)测试结果1:用index.html内部的代码(2&#xf…

《HelloGitHub》第 97 期

兴趣是最好的老师,HelloGitHub 让你对编程感兴趣! 简介 HelloGitHub 分享 GitHub 上有趣、入门级的开源项目。 github.com/521xueweihan/HelloGitHub 这里有实战项目、入门教程、黑科技、开源书籍、大厂开源项目等,涵盖多种编程语言 Python、…

win下vscode的vim切换模式的中英文切换

问题描述 在vscode中安装vim插件后,如果insert模式下完成输入后,在中文输入方式下按esc会发生无效输入,需要手动切换到英文。 解决方法 下载完成vscode并在其中配置vim插件下载github—im-select.exe插件(注意很多博文中的gitcod…

Microsoft Threat Modeling Tool 使用(二)

主界面 翻译 详细描述 选择了 “SDL TM Knowledge Base (Core)” 模板并打开了一个新的威胁模型。这个界面主要用于绘制数据流图(Data Flow Diagram, DFD),它帮助您可视化系统的组成部分和它们之间的交互。以下是界面中各个部分的功能介绍&a…

软件设计师-重点的行为型设计模式

一、命令模式(Command): 意图:(上午题) 将一个请求封装为一个对象,从而使得可以用不同的请求对客户进行参数化;对请求排队或记录请求日志,以及支持可撤销的操作。 结构…

Django-基础篇

Django是一个开放源代码的Web应用框架,由Python语言编写。它遵循MVC(Model-View-Controller)的软件设计模式,使开发者能够以高效、可扩展和安全的方式构建Web应用程序。 Django具有以下特点和优势: 强大的功能&#x…

[技术小技巧] 可视化分析:在jupyter中使用d3可视化树形结构

首先在python中定义一个字符串,记录d3.js绘制属性图的js脚本代码模版。其中{{data}}就是将来要被替换的内容。 d3_code_template """ // 创建树状结构数据 var treeData {{data}};// 创建d3树布局 var margin { top: 20, right: 90, bottom: 30,…

行业推荐:数据防泄漏软件首先解决方案

随着信息时代的快速发展,数据安全已成为企业经营的关键之一。然而,数据泄漏事件时有发生,不仅可能导致巨大的经济损失,更会损害企业的声誉和客户信任。 为了帮助企业有效地保护数据安全,Ping32 数据防泄漏系统应运而生…

【跟我学RISC-V】认识RISC-V指令集并搭建实验环境

写在前面 现在计算机的体系架构正是发展得如火如荼的时候,占领桌面端市场的x86架构、占领移动端市场的arm架构、在服务器市场仍有一定地位的mips架构、国产自研的指令集loongarch架构、还有我现在要讲到的新型开源开放的RISC-V指令集架构。 我先说一说我的学习经历…

Facebook的语言学:社交媒体如何影响我们的沟通方式

1. 引言 社交媒体已经成为人们日常生活中不可或缺的一部分,而Facebook作为其中最具影响力的平台之一,不仅改变了人们之间的社交方式,也对我们的语言学产生了深远的影响。本文将深入探讨Facebook的语言学特点,以及它如何塑造和改变…