Strona główna Odkrywaj Pomoc
Zarejestruj się Zaloguj się
wangbin
/
log-server
Obserwuj 1
Polub 0
Forkuj 0
Pliki Problemy 0 Oczekujące zmiany 0 Wiki
Drzewo: 618b26b1ef
Gałęzie Tagi
log-server
/
source
/
system
/
casbin.go

casbin.go 13 KB
Historia Czysty

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259 
  1. package system
    2. 

  2. 

  3. import (
    4. 

  4. 	"context"
    5. 

  5. 	adapter "github.com/casbin/gorm-adapter/v3"
    6. 

  6. 	"github.com/pkg/errors"
    7. 

  7. 	"gorm.io/gorm"
    8. 

  8. 	"log-server/service/system"
    9. 

  9. )
    10. 

  10. 

  11. const initOrderCasbin = initOrderApi + 1
    12. 

  12. 

  13. type initCasbin struct{}
    14. 

  14. 

  15. // auto run
    16. 

  16. func init() {
    17. 

  17. 	system.RegisterInit(initOrderCasbin, &initCasbin{})
    18. 

  18. }
    19. 

  19. 

  20. func (i *initCasbin) MigrateTable(ctx context.Context) (context.Context, error) {
    21. 

  21. 	db, ok := ctx.Value("db").(*gorm.DB)
    22. 

  22. 	if !ok {
    23. 

  23. 		return ctx, system.ErrMissingDBContext
    24. 

  24. 	}
    25. 

  25. 	return ctx, db.AutoMigrate(&adapter.CasbinRule{})
    26. 

  26. }
    27. 

  27. 

  28. func (i *initCasbin) TableCreated(ctx context.Context) bool {
    29. 

  29. 	db, ok := ctx.Value("db").(*gorm.DB)
    30. 

  30. 	if !ok {
    31. 

  31. 		return false
    32. 

  32. 	}
    33. 

  33. 	return db.Migrator().HasTable(&adapter.CasbinRule{})
    34. 

  34. }
    35. 

  35. 

  36. func (i initCasbin) InitializerName() string {
    37. 

  37. 	var entity adapter.CasbinRule
    38. 

  38. 	return entity.TableName()
    39. 

  39. }
    40. 

  40. 

  41. func (i *initCasbin) InitializeData(ctx context.Context) (context.Context, error) {
    42. 

  42. 	db, ok := ctx.Value("db").(*gorm.DB)
    43. 

  43. 	if !ok {
    44. 

  44. 		return ctx, system.ErrMissingDBContext
    45. 

  45. 	}
    46. 

  46. 	entities := []adapter.CasbinRule{
    47. 

  47. 		{PType: "p", V0: "888", V1: "/base/login", V2: "POST"},
    48. 

  48. 		{PType: "p", V0: "888", V1: "/user/admin_register", V2: "POST"},
    49. 

  49. 

  50. 		{PType: "p", V0: "888", V1: "/api/createApi", V2: "POST"},
    51. 

  51. 		{PType: "p", V0: "888", V1: "/api/getApiList", V2: "POST"},
    52. 

  52. 		{PType: "p", V0: "888", V1: "/api/getApiById", V2: "POST"},
    53. 

  53. 		{PType: "p", V0: "888", V1: "/api/deleteApi", V2: "POST"},
    54. 

  54. 		{PType: "p", V0: "888", V1: "/api/updateApi", V2: "POST"},
    55. 

  55. 		{PType: "p", V0: "888", V1: "/api/getAllApis", V2: "POST"},
    56. 

  56. 		{PType: "p", V0: "888", V1: "/api/deleteApisByIds", V2: "DELETE"},
    57. 

  57. 

  58. 		{PType: "p", V0: "888", V1: "/authority/copyAuthority", V2: "POST"},
    59. 

  59. 		{PType: "p", V0: "888", V1: "/authority/updateAuthority", V2: "PUT"},
    60. 

  60. 		{PType: "p", V0: "888", V1: "/authority/createAuthority", V2: "POST"},
    61. 

  61. 		{PType: "p", V0: "888", V1: "/authority/deleteAuthority", V2: "POST"},
    62. 

  62. 		{PType: "p", V0: "888", V1: "/authority/getAuthorityList", V2: "POST"},
    63. 

  63. 		{PType: "p", V0: "888", V1: "/authority/setDataAuthority", V2: "POST"},
    64. 

  64. 

  65. 		{PType: "p", V0: "888", V1: "/menu/getMenu", V2: "POST"},
    66. 

  66. 		{PType: "p", V0: "888", V1: "/menu/getMenuList", V2: "POST"},
    67. 

  67. 		{PType: "p", V0: "888", V1: "/menu/addBaseMenu", V2: "POST"},
    68. 

  68. 		{PType: "p", V0: "888", V1: "/menu/getBaseMenuTree", V2: "POST"},
    69. 

  69. 		{PType: "p", V0: "888", V1: "/menu/addMenuAuthority", V2: "POST"},
    70. 

  70. 		{PType: "p", V0: "888", V1: "/menu/getMenuAuthority", V2: "POST"},
    71. 

  71. 		{PType: "p", V0: "888", V1: "/menu/deleteBaseMenu", V2: "POST"},
    72. 

  72. 		{PType: "p", V0: "888", V1: "/menu/updateBaseMenu", V2: "POST"},
    73. 

  73. 		{PType: "p", V0: "888", V1: "/menu/getBaseMenuById", V2: "POST"},
    74. 

  74. 

  75. 		{PType: "p", V0: "888", V1: "/user/getUserInfo", V2: "GET"},
    76. 

  76. 		{PType: "p", V0: "888", V1: "/user/setUserInfo", V2: "PUT"},
    77. 

  77. 		{PType: "p", V0: "888", V1: "/user/setSelfInfo", V2: "PUT"},
    78. 

  78. 		{PType: "p", V0: "888", V1: "/user/getUserList", V2: "POST"},
    79. 

  79. 		{PType: "p", V0: "888", V1: "/user/deleteUser", V2: "DELETE"},
    80. 

  80. 		{PType: "p", V0: "888", V1: "/user/changePassword", V2: "POST"},
    81. 

  81. 		{PType: "p", V0: "888", V1: "/user/setUserAuthority", V2: "POST"},
    82. 

  82. 		{PType: "p", V0: "888", V1: "/user/setUserAuthorities", V2: "POST"},
    83. 

  83. 		{PType: "p", V0: "888", V1: "/user/resetPassword", V2: "POST"},
    84. 

  84. 

  85. 		{PType: "p", V0: "888", V1: "/fileUploadAndDownload/findFile", V2: "GET"},
    86. 

  86. 		{PType: "p", V0: "888", V1: "/fileUploadAndDownload/breakpointContinueFinish", V2: "POST"},
    87. 

  87. 		{PType: "p", V0: "888", V1: "/fileUploadAndDownload/breakpointContinue", V2: "POST"},
    88. 

  88. 		{PType: "p", V0: "888", V1: "/fileUploadAndDownload/removeChunk", V2: "POST"},
    89. 

  89. 

  90. 		{PType: "p", V0: "888", V1: "/fileUploadAndDownload/upload", V2: "POST"},
    91. 

  91. 		{PType: "p", V0: "888", V1: "/fileUploadAndDownload/deleteFile", V2: "POST"},
    92. 

  92. 		{PType: "p", V0: "888", V1: "/fileUploadAndDownload/editFileName", V2: "POST"},
    93. 

  93. 		{PType: "p", V0: "888", V1: "/fileUploadAndDownload/getFileList", V2: "POST"},
    94. 

  94. 

  95. 		{PType: "p", V0: "888", V1: "/casbin/updateCasbin", V2: "POST"},
    96. 

  96. 		{PType: "p", V0: "888", V1: "/casbin/getPolicyPathByAuthorityId", V2: "POST"},
    97. 

  97. 

  98. 		{PType: "p", V0: "888", V1: "/jwt/jsonInBlacklist", V2: "POST"},
    99. 

  99. 

  100. 		{PType: "p", V0: "888", V1: "/system/getSystemConfig", V2: "POST"},
    101. 

  101. 		{PType: "p", V0: "888", V1: "/system/setSystemConfig", V2: "POST"},
    102. 

  102. 		{PType: "p", V0: "888", V1: "/system/getServerInfo", V2: "POST"},
    103. 

  103. 

  104. 		{PType: "p", V0: "888", V1: "/customer/customer", V2: "GET"},
    105. 

  105. 		{PType: "p", V0: "888", V1: "/customer/customer", V2: "PUT"},
    106. 

  106. 		{PType: "p", V0: "888", V1: "/customer/customer", V2: "POST"},
    107. 

  107. 		{PType: "p", V0: "888", V1: "/customer/customer", V2: "DELETE"},
    108. 

  108. 		{PType: "p", V0: "888", V1: "/customer/customerList", V2: "GET"},
    109. 

  109. 

  110. 		{PType: "p", V0: "888", V1: "/autoCode/getDB", V2: "GET"},
    111. 

  111. 		{PType: "p", V0: "888", V1: "/autoCode/getMeta", V2: "POST"},
    112. 

  112. 		{PType: "p", V0: "888", V1: "/autoCode/preview", V2: "POST"},
    113. 

  113. 		{PType: "p", V0: "888", V1: "/autoCode/getTables", V2: "GET"},
    114. 

  114. 		{PType: "p", V0: "888", V1: "/autoCode/getColumn", V2: "GET"},
    115. 

  115. 		{PType: "p", V0: "888", V1: "/autoCode/rollback", V2: "POST"},
    116. 

  116. 		{PType: "p", V0: "888", V1: "/autoCode/createTemp", V2: "POST"},
    117. 

  117. 		{PType: "p", V0: "888", V1: "/autoCode/delSysHistory", V2: "POST"},
    118. 

  118. 		{PType: "p", V0: "888", V1: "/autoCode/getSysHistory", V2: "POST"},
    119. 

  119. 		{PType: "p", V0: "888", V1: "/autoCode/createPackage", V2: "POST"},
    120. 

  120. 		{PType: "p", V0: "888", V1: "/autoCode/getPackage", V2: "POST"},
    121. 

  121. 		{PType: "p", V0: "888", V1: "/autoCode/delPackage", V2: "POST"},
    122. 

  122. 		{PType: "p", V0: "888", V1: "/autoCode/createPlug", V2: "POST"},
    123. 

  123. 

  124. 		{PType: "p", V0: "888", V1: "/sysDictionaryDetail/findSysDictionaryDetail", V2: "GET"},
    125. 

  125. 		{PType: "p", V0: "888", V1: "/sysDictionaryDetail/updateSysDictionaryDetail", V2: "PUT"},
    126. 

  126. 		{PType: "p", V0: "888", V1: "/sysDictionaryDetail/createSysDictionaryDetail", V2: "POST"},
    127. 

  127. 		{PType: "p", V0: "888", V1: "/sysDictionaryDetail/getSysDictionaryDetailList", V2: "GET"},
    128. 

  128. 		{PType: "p", V0: "888", V1: "/sysDictionaryDetail/deleteSysDictionaryDetail", V2: "DELETE"},
    129. 

  129. 

  130. 		{PType: "p", V0: "888", V1: "/sysDictionary/findSysDictionary", V2: "GET"},
    131. 

  131. 		{PType: "p", V0: "888", V1: "/sysDictionary/updateSysDictionary", V2: "PUT"},
    132. 

  132. 		{PType: "p", V0: "888", V1: "/sysDictionary/getSysDictionaryList", V2: "GET"},
    133. 

  133. 		{PType: "p", V0: "888", V1: "/sysDictionary/createSysDictionary", V2: "POST"},
    134. 

  134. 		{PType: "p", V0: "888", V1: "/sysDictionary/deleteSysDictionary", V2: "DELETE"},
    135. 

  135. 

  136. 		{PType: "p", V0: "888", V1: "/sysOperationRecord/findSysOperationRecord", V2: "GET"},
    137. 

  137. 		{PType: "p", V0: "888", V1: "/sysOperationRecord/updateSysOperationRecord", V2: "PUT"},
    138. 

  138. 		{PType: "p", V0: "888", V1: "/sysOperationRecord/createSysOperationRecord", V2: "POST"},
    139. 

  139. 		{PType: "p", V0: "888", V1: "/sysOperationRecord/getSysOperationRecordList", V2: "GET"},
    140. 

  140. 		{PType: "p", V0: "888", V1: "/sysOperationRecord/deleteSysOperationRecord", V2: "DELETE"},
    141. 

  141. 		{PType: "p", V0: "888", V1: "/sysOperationRecord/deleteSysOperationRecordByIds", V2: "DELETE"},
    142. 

  142. 

  143. 		{PType: "p", V0: "888", V1: "/email/emailTest", V2: "POST"},
    144. 

  144. 

  145. 		{PType: "p", V0: "888", V1: "/simpleUploader/upload", V2: "POST"},
    146. 

  146. 		{PType: "p", V0: "888", V1: "/simpleUploader/checkFileMd5", V2: "GET"},
    147. 

  147. 		{PType: "p", V0: "888", V1: "/simpleUploader/mergeFileMd5", V2: "GET"},
    148. 

  148. 

  149. 		{PType: "p", V0: "888", V1: "/excel/importExcel", V2: "POST"},
    150. 

  150. 		{PType: "p", V0: "888", V1: "/excel/loadExcel", V2: "GET"},
    151. 

  151. 		{PType: "p", V0: "888", V1: "/excel/exportExcel", V2: "POST"},
    152. 

  152. 		{PType: "p", V0: "888", V1: "/excel/downloadTemplate", V2: "GET"},
    153. 

  153. 

  154. 		{PType: "p", V0: "888", V1: "/authorityBtn/setAuthorityBtn", V2: "POST"},
    155. 

  155. 		{PType: "p", V0: "888", V1: "/authorityBtn/getAuthorityBtn", V2: "POST"},
    156. 

  156. 		{PType: "p", V0: "888", V1: "/authorityBtn/canRemoveAuthorityBtn", V2: "POST"},
    157. 

  157. 

  158. 		{PType: "p", V0: "8881", V1: "/base/login", V2: "POST"},
    159. 

  159. 		{PType: "p", V0: "8881", V1: "/user/admin_register", V2: "POST"},
    160. 

  160. 		{PType: "p", V0: "8881", V1: "/api/createApi", V2: "POST"},
    161. 

  161. 		{PType: "p", V0: "8881", V1: "/api/getApiList", V2: "POST"},
    162. 

  162. 		{PType: "p", V0: "8881", V1: "/api/getApiById", V2: "POST"},
    163. 

  163. 		{PType: "p", V0: "8881", V1: "/api/deleteApi", V2: "POST"},
    164. 

  164. 		{PType: "p", V0: "8881", V1: "/api/updateApi", V2: "POST"},
    165. 

  165. 		{PType: "p", V0: "8881", V1: "/api/getAllApis", V2: "POST"},
    166. 

  166. 		{PType: "p", V0: "8881", V1: "/authority/createAuthority", V2: "POST"},
    167. 

  167. 		{PType: "p", V0: "8881", V1: "/authority/deleteAuthority", V2: "POST"},
    168. 

  168. 		{PType: "p", V0: "8881", V1: "/authority/getAuthorityList", V2: "POST"},
    169. 

  169. 		{PType: "p", V0: "8881", V1: "/authority/setDataAuthority", V2: "POST"},
    170. 

  170. 		{PType: "p", V0: "8881", V1: "/menu/getMenu", V2: "POST"},
    171. 

  171. 		{PType: "p", V0: "8881", V1: "/menu/getMenuList", V2: "POST"},
    172. 

  172. 		{PType: "p", V0: "8881", V1: "/menu/addBaseMenu", V2: "POST"},
    173. 

  173. 		{PType: "p", V0: "8881", V1: "/menu/getBaseMenuTree", V2: "POST"},
    174. 

  174. 		{PType: "p", V0: "8881", V1: "/menu/addMenuAuthority", V2: "POST"},
    175. 

  175. 		{PType: "p", V0: "8881", V1: "/menu/getMenuAuthority", V2: "POST"},
    176. 

  176. 		{PType: "p", V0: "8881", V1: "/menu/deleteBaseMenu", V2: "POST"},
    177. 

  177. 		{PType: "p", V0: "8881", V1: "/menu/updateBaseMenu", V2: "POST"},
    178. 

  178. 		{PType: "p", V0: "8881", V1: "/menu/getBaseMenuById", V2: "POST"},
    179. 

  179. 		{PType: "p", V0: "8881", V1: "/user/changePassword", V2: "POST"},
    180. 

  180. 		{PType: "p", V0: "8881", V1: "/user/getUserList", V2: "POST"},
    181. 

  181. 		{PType: "p", V0: "8881", V1: "/user/setUserAuthority", V2: "POST"},
    182. 

  182. 		{PType: "p", V0: "8881", V1: "/fileUploadAndDownload/upload", V2: "POST"},
    183. 

  183. 		{PType: "p", V0: "8881", V1: "/fileUploadAndDownload/getFileList", V2: "POST"},
    184. 

  184. 		{PType: "p", V0: "8881", V1: "/fileUploadAndDownload/deleteFile", V2: "POST"},
    185. 

  185. 		{PType: "p", V0: "8881", V1: "/fileUploadAndDownload/editFileName", V2: "POST"},
    186. 

  186. 		{PType: "p", V0: "8881", V1: "/casbin/updateCasbin", V2: "POST"},
    187. 

  187. 		{PType: "p", V0: "8881", V1: "/casbin/getPolicyPathByAuthorityId", V2: "POST"},
    188. 

  188. 		{PType: "p", V0: "8881", V1: "/jwt/jsonInBlacklist", V2: "POST"},
    189. 

  189. 		{PType: "p", V0: "8881", V1: "/system/getSystemConfig", V2: "POST"},
    190. 

  190. 		{PType: "p", V0: "8881", V1: "/system/setSystemConfig", V2: "POST"},
    191. 

  191. 		{PType: "p", V0: "8881", V1: "/customer/customer", V2: "POST"},
    192. 

  192. 		{PType: "p", V0: "8881", V1: "/customer/customer", V2: "PUT"},
    193. 

  193. 		{PType: "p", V0: "8881", V1: "/customer/customer", V2: "DELETE"},
    194. 

  194. 		{PType: "p", V0: "8881", V1: "/customer/customer", V2: "GET"},
    195. 

  195. 		{PType: "p", V0: "8881", V1: "/customer/customerList", V2: "GET"},
    196. 

  196. 		{PType: "p", V0: "8881", V1: "/user/getUserInfo", V2: "GET"},
    197. 

  197. 

  198. 		{PType: "p", V0: "9528", V1: "/base/login", V2: "POST"},
    199. 

  199. 		{PType: "p", V0: "9528", V1: "/user/admin_register", V2: "POST"},
    200. 

  200. 		{PType: "p", V0: "9528", V1: "/api/createApi", V2: "POST"},
    201. 

  201. 		{PType: "p", V0: "9528", V1: "/api/getApiList", V2: "POST"},
    202. 

  202. 		{PType: "p", V0: "9528", V1: "/api/getApiById", V2: "POST"},
    203. 

  203. 		{PType: "p", V0: "9528", V1: "/api/deleteApi", V2: "POST"},
    204. 

  204. 		{PType: "p", V0: "9528", V1: "/api/updateApi", V2: "POST"},
    205. 

  205. 		{PType: "p", V0: "9528", V1: "/api/getAllApis", V2: "POST"},
    206. 

  206. 

  207. 		{PType: "p", V0: "9528", V1: "/authority/createAuthority", V2: "POST"},
    208. 

  208. 		{PType: "p", V0: "9528", V1: "/authority/deleteAuthority", V2: "POST"},
    209. 

  209. 		{PType: "p", V0: "9528", V1: "/authority/getAuthorityList", V2: "POST"},
    210. 

  210. 		{PType: "p", V0: "9528", V1: "/authority/setDataAuthority", V2: "POST"},
    211. 

  211. 

  212. 		{PType: "p", V0: "9528", V1: "/menu/getMenu", V2: "POST"},
    213. 

  213. 		{PType: "p", V0: "9528", V1: "/menu/getMenuList", V2: "POST"},
    214. 

  214. 		{PType: "p", V0: "9528", V1: "/menu/addBaseMenu", V2: "POST"},
    215. 

  215. 		{PType: "p", V0: "9528", V1: "/menu/getBaseMenuTree", V2: "POST"},
    216. 

  216. 		{PType: "p", V0: "9528", V1: "/menu/addMenuAuthority", V2: "POST"},
    217. 

  217. 		{PType: "p", V0: "9528", V1: "/menu/getMenuAuthority", V2: "POST"},
    218. 

  218. 		{PType: "p", V0: "9528", V1: "/menu/deleteBaseMenu", V2: "POST"},
    219. 

  219. 		{PType: "p", V0: "9528", V1: "/menu/updateBaseMenu", V2: "POST"},
    220. 

  220. 		{PType: "p", V0: "9528", V1: "/menu/getBaseMenuById", V2: "POST"},
    221. 

  221. 		{PType: "p", V0: "9528", V1: "/user/changePassword", V2: "POST"},
    222. 

  222. 		{PType: "p", V0: "9528", V1: "/user/getUserList", V2: "POST"},
    223. 

  223. 		{PType: "p", V0: "9528", V1: "/user/setUserAuthority", V2: "POST"},
    224. 

  224. 		{PType: "p", V0: "9528", V1: "/fileUploadAndDownload/upload", V2: "POST"},
    225. 

  225. 		{PType: "p", V0: "9528", V1: "/fileUploadAndDownload/getFileList", V2: "POST"},
    226. 

  226. 		{PType: "p", V0: "9528", V1: "/fileUploadAndDownload/deleteFile", V2: "POST"},
    227. 

  227. 		{PType: "p", V0: "9528", V1: "/fileUploadAndDownload/editFileName", V2: "POST"},
    228. 

  228. 		{PType: "p", V0: "9528", V1: "/casbin/updateCasbin", V2: "POST"},
    229. 

  229. 		{PType: "p", V0: "9528", V1: "/casbin/getPolicyPathByAuthorityId", V2: "POST"},
    230. 

  230. 		{PType: "p", V0: "9528", V1: "/jwt/jsonInBlacklist", V2: "POST"},
    231. 

  231. 		{PType: "p", V0: "9528", V1: "/system/getSystemConfig", V2: "POST"},
    232. 

  232. 		{PType: "p", V0: "9528", V1: "/system/setSystemConfig", V2: "POST"},
    233. 

  233. 		{PType: "p", V0: "9528", V1: "/customer/customer", V2: "PUT"},
    234. 

  234. 		{PType: "p", V0: "9528", V1: "/customer/customer", V2: "GET"},
    235. 

  235. 		{PType: "p", V0: "9528", V1: "/customer/customer", V2: "POST"},
    236. 

  236. 		{PType: "p", V0: "9528", V1: "/customer/customer", V2: "DELETE"},
    237. 

  237. 		{PType: "p", V0: "9528", V1: "/customer/customerList", V2: "GET"},
    238. 

  238. 		{PType: "p", V0: "9528", V1: "/autoCode/createTemp", V2: "POST"},
    239. 

  239. 		{PType: "p", V0: "9528", V1: "/user/getUserInfo", V2: "GET"},
    240. 

  240. 	}
    241. 

  241. 	if err := db.Create(&entities).Error; err != nil {
    242. 

  242. 		return ctx, errors.Wrap(err, "Casbin 表 ("+i.InitializerName()+") 数据初始化失败!")
    243. 

  243. 	}
    244. 

  244. 	next := context.WithValue(ctx, i.InitializerName(), entities)
    245. 

  245. 	return next, nil
    246. 

  246. }
    247. 

  247. 

  248. func (i *initCasbin) DataInserted(ctx context.Context) bool {
    249. 

  249. 	db, ok := ctx.Value("db").(*gorm.DB)
    250. 

  250. 	if !ok {
    251. 

  251. 		return false
    252. 

  252. 	}
    253. 

  253. 	if errors.Is(db.Where(adapter.CasbinRule{PType: "p", V0: "9528", V1: "GET", V2: "/user/getUserInfo"}).
    254. 

  254. 		First(&adapter.CasbinRule{}).Error, gorm.ErrRecordNotFound) { // 判断是否存在数据
    255. 

  255. 		return false
    256. 

  256. 	}
    257. 

  257. 	return true
    258. 

  258. }
    259. 

  259.