Bladeren bron

fix(go): set num access tokens in user object;

Vishal Dalwadi 1 maand geleden
bovenliggende
commit
7584235022
1 gewijzigde bestanden met toevoegingen van 9 en 0 verwijderingen
  1. 9 0
      controllers/user.go

+ 9 - 0
controllers/user.go

@@ -884,6 +884,9 @@ func getUserV1(w http.ResponseWriter, r *http.Request) {
 		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
 		return
 	}
+	user.NumAccessTokens, _ = (&schema.UserAccessToken{
+		UserName: user.UserName,
+	}).CountByUser(r.Context())
 	userRoleTemplate, err := logic.GetRole(user.PlatformRoleID)
 	if err != nil {
 		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
@@ -923,6 +926,12 @@ func getUsers(w http.ResponseWriter, r *http.Request) {
 
 	users, err := logic.GetUsers()
 
+	for i := range users {
+		users[i].NumAccessTokens, _ = (&schema.UserAccessToken{
+			UserName: users[i].UserName,
+		}).CountByUser(r.Context())
+	}
+
 	if err != nil {
 		logger.Log(0, "failed to fetch users: ", err.Error())
 		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))