Function count_degrees_direction

src/store/store.c:2066–2107

}

static int count_degrees_direction(cbm_store_t *s, const int64_t *node_ids, int id_count,
                                   const char *in_clause, bool has_type, const char *edge_type,
                                   bool inbound, int *out_counts) {
    char sql[ST_SQL_BUF];
    const char *id_col = inbound ? "target_id" : "source_id";
    if (has_type) {
        snprintf(sql, sizeof(sql),
                 "SELECT %s, COUNT(*) FROM edges "
                 "WHERE %s IN (%s) AND type = ? GROUP BY %s",
                 id_col, id_col, in_clause, id_col);
    } else {
        snprintf(sql, sizeof(sql),
                 "SELECT %s, COUNT(*) FROM edges "
                 "WHERE %s IN (%s) GROUP BY %s",
                 id_col, id_col, in_clause, id_col);
    }

    sqlite3_stmt *stmt = NULL;
    if (sqlite3_prepare_v2(s->db, sql, CBM_NOT_FOUND, &stmt, NULL) != SQLITE_OK) {
        return CBM_STORE_ERR;
    }

    for (int i = 0; i < id_count; i++) {
        sqlite3_bind_int64(stmt, i + SKIP_ONE, node_ids[i]);
    }
    if (has_type) {
        bind_text(stmt, id_count + SKIP_ONE, edge_type);
    }

    while (sqlite3_step(stmt) == SQLITE_ROW) {
        int64_t nid = sqlite3_column_int64(stmt, 0);
        int cnt = sqlite3_column_int(stmt, SKIP_ONE);
        for (int i = 0; i < id_count; i++) {
            if (node_ids[i] == nid) {
                out_counts[i] = cnt;
                break;
            }
        }
    }
    sqlite3_finalize(stmt);
    return CBM_STORE_OK;
}

int cbm_store_batch_count_degrees(cbm_store_t *s, const int64_t *node_ids, int id_count,
                                  const char *edge_type, int *out_in, int *out_out) {