Ejemplo n.º 1
0
  /**
   * @param args
   *     <p>read *.avro file convert into java tree view. return tree
   */
  public Tree generateInputTree() {
    GenericDatumReader<GenericData> genericReader = new GenericDatumReader<GenericData>();
    //		DatumReader inputDatumReader = new SpecificDatumReader(UserInfo.class);
    DataFileReader<GenericData> dataFileReader;
    Tree root = new Tree(); // root tree for Tree data struture
    try {

      ///
      // home/lali/old/svn/trunck/eclipse/data-mapper/org.wso2.developerstudio.visualdatamapper.diagram/resource/inputs.avro
      String path =
          DataMapperCreationWizardPage
              .avroFilePath; // path for avro file selected in Create Datamapper Diagram swizard
      dataFileReader = new DataFileReader<GenericData>(new File(path), genericReader);

      Schema schm = dataFileReader.getSchema();
      multipleData = false;
      root.setName(schm.getName());

      List<Field> list = dataFileReader.getSchema().getFields();
      Iterator<Field> it = list.iterator();

      while (it.hasNext()) {
        Field field = it.next();
        fetchToTree(field, root);
      }

    } catch (IOException e) {

      e.printStackTrace();
    }

    return root;
  }
Ejemplo n.º 2
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;
  }
Ejemplo n.º 3
0
  /** @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);
      }
    }
  }