Exemple #1
0
  /*
   * for each avro field, search for nested schema.
   * if field is nested, create tree and recursive
   * else fetch field as a element
   */
  private static void fetchToTree(Field field, Tree parent) {
    //		System.out.println(" Field: " + field.name());
    if (field.schema().getType().toString().equalsIgnoreCase("RECORD")) {
      //			if(!multipleData)
      //				multipleData = true;
      //

      Tree child = new Tree(parent);
      child.setName(field.name());

      List<Field> list = field.schema().getFields();
      Iterator<Field> it = list.iterator();

      while (it.hasNext()) {
        Field fieldOfField = it.next();
        fetchToTree(fieldOfField, child);
      }
      parent.getTrees().add(child);
    } else if (field.schema().getType().getName().equalsIgnoreCase("ARRAY")) {
      if (field.schema().getElementType().getType().name().toString().equalsIgnoreCase("RECORD")) {
        if (!multipleData) multipleData = true;

        Schema arraySchema = field.schema().getElementType();
        Tree childParent = new Tree(parent);
        childParent.setName(field.name()); // employee
        //					parent.getTrees().add(childParent);
        //					Tree child = new Tree(childParent);
        //					child.setName(arraySchema.getName());//employeerecord
        List<Field> list = arraySchema.getFields();
        Iterator<Field> it = list.iterator();

        while (it.hasNext()) {
          Field fieldOfField = it.next();
          fetchToTree(fieldOfField, childParent);
        }
        parent.getTrees().add(childParent);
      }
    } else {
      Element elementNew = new Element(parent);
      elementNew.setName(field.name());
      parent.getElements().add(elementNew);
    }
    //		return parent;
  }
  /** @generated NOT */
  private void createTree(Tree treeN, TreeNode treeNode) {
    TreeNode treeNodeNew = DataMapperFactory.eINSTANCE.createTreeNode();
    // treeNodeNew.setName(treeN.getCount() + "," + treeN.getName());
    treeNodeNew.setName(treeN.getName());
    treeNodeNew.setLevel(treeN.getCount());
    if (treeN.getSchemaType() != null) {
      switch (treeN.getSchemaType()) {
        case ARRAY:
          treeNodeNew.setSchemaDataType(SchemaDataType.ARRAY);
          break;
        case BOOLEAN:
          treeNodeNew.setSchemaDataType(SchemaDataType.BOOLEAN);
          break;
        case BYTES:
          treeNodeNew.setSchemaDataType(SchemaDataType.BYTES);
          break;
        case DOUBLE:
          treeNodeNew.setSchemaDataType(SchemaDataType.DOUBLE);
          break;
        case ENUM:
          treeNodeNew.setSchemaDataType(SchemaDataType.ENUM);
          break;
        case FIXED:
          treeNodeNew.setSchemaDataType(SchemaDataType.FIXED);
          break;
        case FLOAT:
          treeNodeNew.setSchemaDataType(SchemaDataType.FLOAT);
          break;
        case INT:
          treeNodeNew.setSchemaDataType(SchemaDataType.INT);
          break;
        case LONG:
          treeNodeNew.setSchemaDataType(SchemaDataType.LONG);
          break;
        case MAP:
          treeNodeNew.setSchemaDataType(SchemaDataType.MAP);
          break;
        case NULL:
          treeNodeNew.setSchemaDataType(SchemaDataType.NULL);
          break;
        case RECORD:
          treeNodeNew.setSchemaDataType(SchemaDataType.RECORD);
          break;
        case STRING:
          treeNodeNew.setSchemaDataType(SchemaDataType.STRING);
          break;
        case UNION:
          treeNodeNew.setSchemaDataType(SchemaDataType.UNION);
          break;
        default:
          break;
      }
    }
    treeNode.getNode().add(treeNodeNew);

    if (!(treeN.getTrees().isEmpty())) {
      for (Tree treeNew : treeN.getTrees()) {
        createTree(treeNew, treeNodeNew);
      }
    }

    if (!(treeN.getElements().isEmpty())) {
      for (dataMapper.diagram.tree.model.Element element : treeN.getElements()) {
        createElement(element, treeNodeNew);
      }
    }
    if (!(treeN.getAttributes().isEmpty())) {
      for (dataMapper.diagram.tree.model.Attribute attribute : treeN.getAttributes()) {
        createAttribute(attribute, treeNodeNew);
      }
    }
  }