/**
* Retrieve entry attributes from its primary key
*/
int listmgr_get_by_pk( lmgr_t * p_mgr, PK_ARG_T pk, attr_set_t * p_info )
{
int rc;
char *first_table = NULL;
GString *req, *from;
/* attribute count is up to 1 per bit (8 per byte).
* x2 for bullet proofing */
char *result_tab[2*8*sizeof(p_info->attr_mask)];
result_handle_t result;
bool checkmain = true;
int main_count = 0,
annex_count = 0,
name_count = 0;
attr_mask_t gen = gen_fields(p_info->attr_mask);
if (p_info == NULL)
return 0;
/* init entry info */
memset(&p_info->attr_values, 0, sizeof(entry_info_t));
req = g_string_new("SELECT ");
from = g_string_new(" FROM ");
/* retrieve source info for generated fields (only about std fields)*/
add_source_fields_for_gen(&p_info->attr_mask.std);
/* don't get fields that are not in main, names, annex, stripe...
* This allows the caller to set all bits 'on' to get everything.
* Note: this also clear generated fields. They will be restored after.
*/
supported_bits_only(&p_info->attr_mask);
/* get info from main table (if asked) */
main_count = attrmask2fieldlist(req, p_info->attr_mask, T_MAIN, "", "", 0);
if (main_count < 0)
{
rc = -main_count;
goto free_str;
}
else if (main_count > 0)
{
checkmain = false;
first_table = MAIN_TABLE;
g_string_append(from, MAIN_TABLE);
}
annex_count = attrmask2fieldlist(req, p_info->attr_mask, T_ANNEX, "", "",
first_table != NULL ? AOF_LEADING_SEP : 0);
if (annex_count < 0)
{
rc = -annex_count;
goto free_str;
}
else if (annex_count > 0)
{
if (first_table != NULL)
g_string_append_printf(from, " LEFT JOIN "ANNEX_TABLE" ON %s.id="
ANNEX_TABLE".id", first_table);
else
{
first_table = ANNEX_TABLE;
g_string_append(from, ANNEX_TABLE);
}
}
name_count = attrmask2fieldlist(req, p_info->attr_mask, T_DNAMES, "", "",
first_table != NULL ? AOF_LEADING_SEP : 0);
if (name_count < 0)
{
rc = -name_count;
goto free_str;
}
else if (name_count > 0)
{
if (first_table)
/* it's OK to JOIN with NAMES table here even if there are multiple paths,
* as we only take one result record. The important thing is to return
* consistent values for parent_id, name and fullpath. */
g_string_append_printf(from, " LEFT JOIN "DNAMES_TABLE" ON %s.id="
DNAMES_TABLE".id", first_table);
else
{
first_table = DNAMES_TABLE;
g_string_append(from, DNAMES_TABLE);
}
}
if (first_table != NULL)
{
int shift = 0;
g_string_append_printf(req, "%s WHERE %s.id="DPK, from->str,
first_table, pk);
rc = db_exec_sql(&p_mgr->conn, req->str, &result);
if (rc)
goto free_str;
rc = db_next_record(&p_mgr->conn, &result, result_tab,
main_count + annex_count + name_count);
/* END_OF_LIST means it does not exist */
if (rc == DB_END_OF_LIST)
{
clean_std_table_bits(&p_info->attr_mask);
/* not found, but did not check MAIN yet */
if (checkmain)
goto next_table;
rc = DB_NOT_EXISTS;
}
if (rc)
goto free_res;
/* set info from result */
if (main_count)
{
rc = result2attrset(T_MAIN, result_tab + shift, main_count, p_info);
shift += main_count;
if (rc)
goto free_res;
}
if (annex_count)
{
rc = result2attrset(T_ANNEX, result_tab + shift, annex_count,
p_info);
shift += annex_count;
if (rc)
goto free_res;
}
if (name_count)
{
rc = result2attrset(T_DNAMES, result_tab + shift, name_count,
p_info);
shift += name_count;
if (rc)
goto free_res;
}
next_table:
db_result_free(&p_mgr->conn, &result);
}
/* remove stripe info if it is not a file */
if (stripe_fields(p_info->attr_mask) && ATTR_MASK_TEST(p_info, type)
&& strcmp(ATTR(p_info, type), STR_TYPE_FILE) != 0)
{
p_info->attr_mask = attr_mask_and_not(&p_info->attr_mask, &stripe_attr_set);
}
/* get stripe info if asked */
#ifdef _LUSTRE
if (stripe_fields(p_info->attr_mask))
{
rc = get_stripe_info(p_mgr, pk, &ATTR(p_info, stripe_info),
ATTR_MASK_TEST(p_info, stripe_items)?
&ATTR(p_info, stripe_items) : NULL);
if (rc == DB_ATTR_MISSING || rc == DB_NOT_EXISTS)
{
/* stripe info is in std mask */
p_info->attr_mask.std &= ~ATTR_MASK_stripe_info;
if (ATTR_MASK_TEST(p_info, stripe_items))
p_info->attr_mask.std &= ~ATTR_MASK_stripe_items;
}
else if (rc)
goto free_str;
else
checkmain = false; /* entry exists */
}
#else
/* POSIX: always clean stripe bits */
p_info->attr_mask = attr_mask_and_not(&p_info->attr_mask, &stripe_attr_set);
#endif
/* special field dircount */
if (dirattr_fields(p_info->attr_mask))
{
if (listmgr_get_dirattrs(p_mgr, pk, p_info))
{
DisplayLog(LVL_MAJOR, LISTMGR_TAG, "listmgr_get_dirattrs failed for "DPK, pk);
p_info->attr_mask = attr_mask_and_not(&p_info->attr_mask, &dir_attr_set);
}
}
if (checkmain)
{
/* verify it exists in main table */
g_string_printf(req, "SELECT id FROM " MAIN_TABLE " WHERE id="DPK, pk);
/* execute the request */
rc = db_exec_sql(&p_mgr->conn, req->str, &result);
if (rc)
goto free_str;
rc = db_next_record(&p_mgr->conn, &result, result_tab, 1);
db_result_free(&p_mgr->conn, &result);
if (rc)
{
rc = DB_NOT_EXISTS;
goto free_str;
}
}
/* restore generated fields in attr mask */
p_info->attr_mask = attr_mask_or(&p_info->attr_mask, &gen);
/* generate them */
generate_fields(p_info);
/* update operation stats */
p_mgr->nbop[OPIDX_GET]++;
rc = DB_SUCCESS;
goto free_str;
free_res:
db_result_free(&p_mgr->conn, &result);
free_str:
g_string_free(req, TRUE);
g_string_free(from, TRUE);
return rc;
} /* listmgr_get_by_pk */
/**
* Retrieve entry attributes from its primary key
*/
int listmgr_get_by_pk( lmgr_t * p_mgr, PK_ARG_T pk, attr_set_t * p_info )
{
int rc;
char fieldlist[4096] = "";
char *first_table = NULL;
char from[1024] = "";
char query[4096] = "";
/* we assume there is not more than 128 fields */
char *result_tab[128];
result_handle_t result;
int checkmain = 1;
int main_count = 0, annex_count = 0, name_count = 0;
if (p_info == NULL)
return 0;
/* init entry info */
memset( &p_info->attr_values, 0, sizeof( entry_info_t ) );
fieldlist[0] = '\0';
/* retrieve source info for generated fields */
add_source_fields_for_gen( &p_info->attr_mask );
/* get info from main table (if asked) */
main_count = attrmask2fieldlist(fieldlist, p_info->attr_mask, T_MAIN, FALSE,
FALSE, "", "");
if (main_count < 0)
return -main_count;
else if (main_count > 0)
{
checkmain = 0;
first_table = MAIN_TABLE;
sprintf(from, MAIN_TABLE);
}
annex_count = attrmask2fieldlist(fieldlist + strlen(fieldlist),
p_info->attr_mask, T_ANNEX,
first_table != NULL, FALSE, "", "");
if (annex_count < 0)
return -annex_count;
else if (annex_count > 0)
{
if (first_table)
sprintf(from + strlen(from), " LEFT JOIN "ANNEX_TABLE" ON %s.id="
ANNEX_TABLE".id", first_table);
else
{
first_table = ANNEX_TABLE;
sprintf(from, ANNEX_TABLE);
}
}
name_count = attrmask2fieldlist(fieldlist + strlen(fieldlist),
p_info->attr_mask, T_DNAMES,
first_table != NULL, FALSE, "", "");
if (name_count < 0)
return -name_count;
else if (name_count > 0)
{
if (first_table)
/* it's OK to JOIN with NAMES table here even if there are multiple paths,
* as we only take one result record. The important thing is to return
* consistent values for parent_id, name and fullpath. */
sprintf(from + strlen(from), " LEFT JOIN "DNAMES_TABLE" ON %s.id="
DNAMES_TABLE".id", first_table);
else
{
first_table = DNAMES_TABLE;
sprintf(from, DNAMES_TABLE);
}
}
if (first_table != NULL)
{
int shift = 0;
sprintf(query, "SELECT %s FROM %s WHERE %s.id="DPK, fieldlist, from,
first_table, pk);
rc = db_exec_sql(&p_mgr->conn, query, &result);
if (rc)
return rc;
rc = db_next_record(&p_mgr->conn, &result, result_tab,
main_count + annex_count + name_count);
/* END_OF_LIST means it does not exist */
if (rc == DB_END_OF_LIST)
rc = DB_NOT_EXISTS;
if (rc)
goto free_res;
/* set info from result */
if (main_count)
{
rc = result2attrset(T_MAIN, result_tab + shift, main_count, p_info);
shift += main_count;
if (rc)
goto free_res;
}
if (annex_count)
{
rc = result2attrset(T_ANNEX, result_tab + shift, annex_count,
p_info);
shift += annex_count;
if (rc)
goto free_res;
}
if (name_count)
{
rc = result2attrset(T_DNAMES, result_tab + shift, name_count,
p_info);
shift += name_count;
if (rc)
goto free_res;
}
db_result_free(&p_mgr->conn, &result);
}
/* remove stripe info if it is not a file */
if (stripe_fields(p_info->attr_mask) && ATTR_MASK_TEST(p_info, type)
&& strcmp(ATTR(p_info, type), STR_TYPE_FILE) != 0)
{
p_info->attr_mask &= ~stripe_attr_set;
}
/* get stripe info if asked */
#ifdef _LUSTRE
if (stripe_fields( p_info->attr_mask ))
{
rc = get_stripe_info( p_mgr, pk, &ATTR( p_info, stripe_info ),
ATTR_MASK_TEST( p_info, stripe_items ) ? &ATTR( p_info,
stripe_items ) :
NULL );
if ( rc == DB_ATTR_MISSING || rc == DB_NOT_EXISTS )
{
p_info->attr_mask &= ~ATTR_MASK_stripe_info;
if ( ATTR_MASK_TEST( p_info, stripe_items ) )
p_info->attr_mask &= ~ATTR_MASK_stripe_items;
}
else if ( rc )
return rc;
else
checkmain = 0; /* entry exists */
}
#else
/* always clean them */
p_info->attr_mask &= ~(ATTR_MASK_stripe_info | ATTR_MASK_stripe_items);
#endif
/* special field dircount */
if (dirattr_fields( p_info->attr_mask ))
{
if (listmgr_get_dirattrs(p_mgr, pk, p_info))
{
DisplayLog( LVL_MAJOR, LISTMGR_TAG, "listmgr_get_dirattrs failed for "DPK, pk );
p_info->attr_mask &= ~dir_attr_set;
}
}
if (checkmain)
{
/* verify it exists in main table */
sprintf( query, "SELECT id FROM " MAIN_TABLE " WHERE id="DPK, pk );
/* execute the request */
rc = db_exec_sql( &p_mgr->conn, query, &result );
if ( rc )
return rc;
rc = db_next_record( &p_mgr->conn, &result, result_tab, 1 );
db_result_free( &p_mgr->conn, &result );
if (rc)
return DB_NOT_EXISTS;
}
/* compute generated fields if asked */
generate_fields( p_info );
p_mgr->nbop[OPIDX_GET]++;
return DB_SUCCESS;
free_res:
db_result_free( &p_mgr->conn, &result );
return rc;
} /* listmgr_get_by_pk */
