Skip to content

Commit

Permalink
Return error when failed to get user from database
Browse files Browse the repository at this point in the history
  • Loading branch information
杨赫然 committed Sep 18, 2024
1 parent f27ab08 commit 9e3242d
Show file tree
Hide file tree
Showing 5 changed files with 37 additions and 16 deletions.
6 changes: 3 additions & 3 deletions common/rpc-service.c
Original file line number Diff line number Diff line change
Expand Up @@ -4726,7 +4726,7 @@ ccnet_rpc_get_emailuser (const char *email, GError **error)
CcnetUserManager *user_mgr = seaf->user_mgr;
CcnetEmailUser *emailuser = NULL;

emailuser = ccnet_user_manager_get_emailuser (user_mgr, email);
emailuser = ccnet_user_manager_get_emailuser (user_mgr, email, error);

return (GObject *)emailuser;
}
Expand All @@ -4742,7 +4742,7 @@ ccnet_rpc_get_emailuser_with_import (const char *email, GError **error)
CcnetUserManager *user_mgr = seaf->user_mgr;
CcnetEmailUser *emailuser = NULL;

emailuser = ccnet_user_manager_get_emailuser_with_import (user_mgr, email);
emailuser = ccnet_user_manager_get_emailuser_with_import (user_mgr, email, error);

return (GObject *)emailuser;
}
Expand Down Expand Up @@ -5448,7 +5448,7 @@ ccnet_rpc_get_org_emailusers (const char *url_prefix, int start , int limit,
while (ptr) {
char *email = ptr->data;
CcnetEmailUser *emailuser = ccnet_user_manager_get_emailuser (user_mgr,
email);
email, NULL);
if (emailuser != NULL) {
ret = g_list_prepend (ret, emailuser);
}
Expand Down
38 changes: 29 additions & 9 deletions common/user-mgr.c
Original file line number Diff line number Diff line change
Expand Up @@ -1147,26 +1147,41 @@ get_ldap_emailuser_cb (CcnetDBRow *row, void *data)
static CcnetEmailUser*
get_emailuser (CcnetUserManager *manager,
const char *email,
gboolean import)
gboolean import,
GError **error)
{
CcnetDB *db = manager->priv->db;
char *sql;
CcnetEmailUser *emailuser = NULL;
char *email_down;
int rc;

sql = "SELECT e.id, e.email, is_staff, is_active, ctime, passwd, reference_id, role "
" FROM EmailUser e LEFT JOIN UserRole ON e.email = UserRole.email "
" WHERE e.email=?";
if (seaf_db_statement_foreach_row (db, sql, get_emailuser_cb, &emailuser,
1, "string", email) > 0) {
rc = seaf_db_statement_foreach_row (db, sql, get_emailuser_cb, &emailuser,
1, "string", email);
if (rc > 0) {
return emailuser;
} else if (rc < 0) {
if (error) {
g_set_error (error, CCNET_DOMAIN, 500, "Database error");
}
return NULL;
}

email_down = g_ascii_strdown (email, strlen(email));
if (seaf_db_statement_foreach_row (db, sql, get_emailuser_cb, &emailuser,
1, "string", email_down) > 0) {
rc = seaf_db_statement_foreach_row (db, sql, get_emailuser_cb, &emailuser,
1, "string", email_down);
if (rc > 0) {
g_free (email_down);
return emailuser;
} else if (rc < 0) {
if (error) {
g_set_error (error, CCNET_DOMAIN, 500, "Database error");
}
g_free (email_down);
return NULL;
}

#ifdef HAVE_LDAP
Expand All @@ -1179,6 +1194,9 @@ get_emailuser (CcnetUserManager *manager,
get_ldap_emailuser_cb,
&emailuser, 1, "string", email_down);
if (ret < 0) {
if (error) {
g_set_error (error, CCNET_DOMAIN, CCNET_ERR_INTERNAL, "Database error");
}
ccnet_warning ("get ldapuser from db failed.\n");
g_free (email_down);
return NULL;
Expand Down Expand Up @@ -1236,16 +1254,18 @@ get_emailuser (CcnetUserManager *manager,

CcnetEmailUser*
ccnet_user_manager_get_emailuser (CcnetUserManager *manager,
const char *email)
const char *email,
GError **error)
{
return get_emailuser (manager, email, FALSE);
return get_emailuser (manager, email, FALSE, error);
}

CcnetEmailUser*
ccnet_user_manager_get_emailuser_with_import (CcnetUserManager *manager,
const char *email)
const char *email,
GError **error)
{
return get_emailuser (manager, email, TRUE);
return get_emailuser (manager, email, TRUE, error);
}

CcnetEmailUser*
Expand Down
5 changes: 3 additions & 2 deletions common/user-mgr.h
Original file line number Diff line number Diff line change
Expand Up @@ -84,11 +84,12 @@ ccnet_user_manager_validate_emailuser (CcnetUserManager *manager,
const char *passwd);

CcnetEmailUser*
ccnet_user_manager_get_emailuser (CcnetUserManager *manager, const char *email);
ccnet_user_manager_get_emailuser (CcnetUserManager *manager, const char *email, GError **error);

CcnetEmailUser*
ccnet_user_manager_get_emailuser_with_import (CcnetUserManager *manager,
const char *email);
const char *email,
GError **error);
CcnetEmailUser*
ccnet_user_manager_get_emailuser_by_id (CcnetUserManager *manager, int id);

Expand Down
2 changes: 1 addition & 1 deletion fuse/getattr.c
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ static int getattr_user(SeafileSession *seaf, const char *user, struct stat *stb
{
CcnetEmailUser *emailuser;

emailuser = ccnet_user_manager_get_emailuser (seaf->user_mgr, user);
emailuser = ccnet_user_manager_get_emailuser (seaf->user_mgr, user, NULL);
if (!emailuser) {
return -ENOENT;
}
Expand Down
2 changes: 1 addition & 1 deletion fuse/readdir.c
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ static int readdir_user(SeafileSession *seaf, const char *user,
GList *list = NULL, *p;
GString *name;

emailuser = ccnet_user_manager_get_emailuser (seaf->user_mgr, user);
emailuser = ccnet_user_manager_get_emailuser (seaf->user_mgr, user, NULL);
if (!emailuser) {
return -ENOENT;
}
Expand Down

0 comments on commit 9e3242d

Please sign in to comment.