sys_api.go 3.8 KB
Newer Older
1 2 3 4 5 6 7
package service

import (
	"errors"
	"gin-vue-admin/global"
	"gin-vue-admin/model"
	"gin-vue-admin/model/request"
Mr.奇淼('s avatar
Mr.奇淼( 已提交
8
	"gorm.io/gorm"
9 10
)

m0_50812349's avatar
m0_50812349 已提交
11 12 13 14 15
//@author: [piexlmax](https://github.com/piexlmax)
//@function: CreateApi
//@description: 新增基础api
//@param: api model.SysApi
//@return: err error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
16

17
func CreateApi(api model.SysApi) (err error) {
18
	if !errors.Is(global.GVA_DB.Where("path = ? AND method = ?", api.Path, api.Method).First(&model.SysApi{}).Error, gorm.ErrRecordNotFound) {
19 20
		return errors.New("存在相同api")
	}
21
	return global.GVA_DB.Create(&api).Error
22 23
}

m0_50812349's avatar
m0_50812349 已提交
24 25 26 27 28
//@author: [piexlmax](https://github.com/piexlmax)
//@function: DeleteApi
//@description: 删除基础api
//@param: api model.SysApi
//@return: err error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
29

30 31 32
func DeleteApi(api model.SysApi) (err error) {
	err = global.GVA_DB.Delete(api).Error
	ClearCasbin(1, api.Path, api.Method)
33 34 35
	return err
}

m0_50812349's avatar
m0_50812349 已提交
36 37 38 39 40
//@author: [piexlmax](https://github.com/piexlmax)
//@function: CreateApi
//@description: 自动创建api数据,
//@param: api model.SysApi
//@return: err error
41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57

func AutoCreateApi(api model.SysApi) (err error) {
	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		var fApi model.SysApi
		var txErr error
		fxErr := tx.Where("path = ? AND method = ?", api.Path, api.Method).First(&fApi).Error
		if errors.Is(fxErr, gorm.ErrRecordNotFound) {
			txErr = tx.Create(&api).Error
			if txErr != nil{
				return txErr
			}
		}
		return nil
	})
	return err
}

m0_50812349's avatar
m0_50812349 已提交
58 59 60 61 62
//@author: [piexlmax](https://github.com/piexlmax)
//@function: GetAPIInfoList
//@description: 分页获取数据,
//@param: api model.SysApi, info request.PageInfo, order string, desc bool
//@return: err error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
63

Mr.奇淼('s avatar
Mr.奇淼( 已提交
64
func GetAPIInfoList(api model.SysApi, info request.PageInfo, order string, desc bool) (err error, list interface{}, total int64) {
65 66
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
Mr.奇淼('s avatar
Mr.奇淼( 已提交
67
	db := global.GVA_DB.Model(&model.SysApi{})
68
	var apiList []model.SysApi
69

70 71
	if api.Path != "" {
		db = db.Where("path LIKE ?", "%"+api.Path+"%")
72
	}
73

74 75
	if api.Description != "" {
		db = db.Where("description LIKE ?", "%"+api.Description+"%")
76
	}
77

78 79
	if api.Method != "" {
		db = db.Where("method = ?", api.Method)
80
	}
81

82 83 84 85
	if api.ApiGroup != "" {
		db = db.Where("api_group = ?", api.ApiGroup)
	}

Mr.奇淼('s avatar
Mr.奇淼( 已提交
86
	err = db.Count(&total).Error
87

88 89 90 91
	if err != nil {
		return err, apiList, total
	} else {
		db = db.Limit(limit).Offset(offset)
92
		if order != "" {
93
			var OrderStr string
94 95
			if desc {
				OrderStr = order + " desc"
96
			} else {
97
				OrderStr = order
98
			}
Mr.奇淼('s avatar
Mr.奇淼( 已提交
99
			err = db.Order(OrderStr).Find(&apiList).Error
100
		} else {
Mr.奇淼('s avatar
Mr.奇淼( 已提交
101
			err = db.Order("api_group").Find(&apiList).Error
102 103
		}
	}
104
	return err, apiList, total
105 106
}

m0_50812349's avatar
m0_50812349 已提交
107 108 109 110
//@author: [piexlmax](https://github.com/piexlmax)
//@function: GetAllApis
//@description: 获取所有的api
//@return: err error, apis []model.SysApi
Mr.奇淼('s avatar
Mr.奇淼( 已提交
111

112 113 114 115 116
func GetAllApis() (err error, apis []model.SysApi) {
	err = global.GVA_DB.Find(&apis).Error
	return
}

m0_50812349's avatar
m0_50812349 已提交
117 118 119 120 121
//@author: [piexlmax](https://github.com/piexlmax)
//@function: GetApiById
//@description: 根据id获取api
//@param: id float64
//@return: err error, api model.SysApi
Mr.奇淼('s avatar
Mr.奇淼( 已提交
122

123 124 125 126 127
func GetApiById(id float64) (err error, api model.SysApi) {
	err = global.GVA_DB.Where("id = ?", id).First(&api).Error
	return
}

m0_50812349's avatar
m0_50812349 已提交
128 129 130 131 132
//@author: [piexlmax](https://github.com/piexlmax)
//@function: UpdateApi
//@description: 根据id更新api
//@param: api model.SysApi
//@return: err error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
133

134
func UpdateApi(api model.SysApi) (err error) {
135
	var oldA model.SysApi
136 137
	err = global.GVA_DB.Where("id = ?", api.ID).First(&oldA).Error
	if oldA.Path != api.Path || oldA.Method != api.Method {
138
		if !errors.Is(global.GVA_DB.Where("path = ? AND method = ?", api.Path, api.Method).First(&model.SysApi{}).Error, gorm.ErrRecordNotFound) {
139 140 141
			return errors.New("存在相同api路径")
		}
	}
142 143 144
	if err != nil {
		return err
	} else {
145
		err = UpdateCasbinApi(oldA.Path, api.Path, oldA.Method, api.Method)
146 147 148
		if err != nil {
			return err
		} else {
149
			err = global.GVA_DB.Save(&api).Error
150 151 152 153
		}
	}
	return err
}