/** * 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 */