Gin实现依赖注入

前言

依赖注入的好处和特点这里不讲述了，本篇文章主要介绍gin框架如何实现依赖注入，将项目解耦。

项目结构



├── cmd          程序入口

├── common   通用模块代码

├── config       配置文件

├── controller API控制器

├── docs         数据库文件

├── models     数据表实体

├── page        页面数据返回实体

├── repository 数据访问层

├── router       路由

├── service     业务逻辑层

├── vue-admin Vue前端页面代码

相信很多Java或者.NET的码友对这个项目结构还是比较熟悉的，现在我们就用这个项目结构在gin框架中实现依赖注入。这里主要介绍controller、service和repository。

数据访问层repository

首先定义IStartRepo接口，里面包括Speak方法

//IStartRepo 定义IStartRepo接口

type IStartRepo interface {

	Speak(message string) string

}

实现该接口，这里就不访问数据库了，直接返回数据

//StartRepo 注入数据库

type StartRepo struct {

	Source datasource.IDb `inject:""`

}

//Speak 实现Speak方法

func (s *StartRepo) Speak(message string) string {

	//使用注入的IDb访问数据库

	//s.Source.DB().Where("name = ?", "jinzhu").First(&user)

	return fmt.Sprintf("[Repository] speak: %s", message)

}

这样我们就简单完成了数据库访问层的接口封装

业务逻辑层service

定义IStartService接口

//IStartService 定义IStartService接口

type IStartService interface {

	Say(message string) string

}

实现IStartService接口方法，注入IStartRepo数据访问层

//StartService 注入IStartRepo

type StartService struct {

	Repo repository.IStartRepo `inject:""`

}

//Say 实现Say方法

func (s *StartService) Say(message string) string {

	return s.Repo.Speak(message)

}

控制器controller

注入IStartService业务逻辑层，并调用Say方法

//Index 注入IStartService

type Index struct {

	Service service.IStartService `inject:""`

}

//GetName 调用IStartService的Say方法

func (i *Index) GetName(ctx *gin.Context) {

	var message = ctx.Param("msg")

	ctx.JSON(200, i.Service.Say(message))

}

注入gin中

到此三个层此代码已写好，然后我们使用"github.com/facebookgo/inject"Facebook的工具包将它们注入到gin中。

func Configure(app *gin.Engine) {

	//controller declare

	var index controller.Index

	//inject declare

	db := datasource.Db{}

	//Injection

	var injector inject.Graph

	err := injector.Provide(

		&inject.Object{Value: &index},

		&inject.Object{Value: &db},

		&inject.Object{Value: &repository.StartRepo{}},

		&inject.Object{Value: &service.StartService{}},

	)

	if err != nil {

		log.Fatal("inject fatal: ", err)

	}

	if err := injector.Populate(); err != nil {

		log.Fatal("inject fatal: ", err)

	}

	//database connect

	err = db.Connect()

	if err != nil {

		log.Fatal("db fatal:", err)

	}

	v1 := app.Group("/")

	{

		v1.GET("/get/:msg", index.GetName)

	}

}

有关更多的github.com/facebookgo/inject使用方法请参考文档

本demo源码地址:https://github.com/Bingjian-Zhu/gin-inject

巴特西