function relation_generate_relations in Relation 7

Same name and namespace in other branches

1. 8.2 relation.drush.inc \relation_generate_relations()
2. 8 relation.drush.inc \relation_generate_relations()

Generates pseudorandom relations. Appropriate entities must already exist.

Parameters

$number_relations: Number of entities to generate of each entity_type.

$types: Array of relation_type to generate relations for.

$kill: Whether to delete all existing relations before creating new ones.

Return value

Array of rids of the generated relations.

2 calls to relation_generate_relations()

drush_relation_generate in ./relation.drush.inc

Drush callback to generate relations.

relation_ui_generate_form_submit in ./relation_ui.module

File

./relation.drush.inc, line 67

Drush integration for the relation module.

Code

function relation_generate_relations($number_relations = 10, $types = array(), $kill = FALSE) {
  if ($kill) {
    $query = new EntityFieldQuery();
    $results = $query
      ->entityCondition('entity_type', 'relation')
      ->execute();
    if ($results) {
      $rids = array_keys($results['relation']);
      relation_delete_multiple($rids);
    }
  }
  $relation_types = relation_get_types($types);
  $new_rids = array();
  foreach ($relation_types as $type => $relation_type) {
    $available_types = array();
    foreach ($relation_type->source_bundles as $bundle_key) {
      list($entity_type, $bundle) = explode(':', $bundle_key, 2);
      $available_types['source'][$entity_type][] = $bundle;
    }
    foreach ($relation_type->target_bundles as $bundle_key) {
      list($entity_type, $bundle) = explode(':', $bundle_key, 2);
      $available_types['target'][$entity_type][] = $bundle;
    }
    $arity = rand($relation_type->min_arity, $relation_type->min_arity);
    for ($i = $number_relations; $i > 0; $i--) {

      // start new relation
      $entity_keys = array();
      for ($r_index = 0; $r_index < $arity; $r_index++) {
        if ($relation_type->directional && $r_index > 0) {
          $direction = 'target';
        }
        else {

          // use source bundles
          $direction = 'source';
        }
        $entity_type = array_rand($available_types[$direction]);
        $query = new EntityFieldQuery();
        $query
          ->entityCondition('entity_type', $entity_type, '=')
          ->range(0, 2 * $number_relations);
        if (!in_array('*', $available_types[$direction][$entity_type])) {

          // Entities with a single bundle don't support EFQ bundle condition.
          $entity_info = entity_get_info($entity_type);
          if (count($entity_info['bundles']) > 1) {
            $query
              ->entityCondition('bundle', $available_types[$direction][$entity_type], 'IN');
          }
        }
        $results = $query
          ->execute();
        if ($results) {
          $entity_ids = array_keys(reset($results));
          $key = array_rand($entity_ids);

          // pseudorandomise until EFQ does random.
          $entity_keys[] = array(
            'entity_type' => $entity_type,
            'entity_id' => $entity_ids[$key],
            'r_index' => $r_index,
          );
        }
      }
      $relation = relation_create($type, $entity_keys);
      $relation->devel_generate = TRUE;
      $new_rids[] = relation_save($relation);
    }
  }
  relation_generate_message(t('Generated @number_relations relations.', array(
    '@number_relations' => $number_relations,
  )), 'ok');
  return $new_rids;
}