uisqlfilewriter.java

国外的一套开源CRM

            }

            resultString += (String.valueOf(uiDisplayAttribGVL.size()) +
            " display attributes<BR>\n");
            displayTypeGVI = null;

            outputFile.close();

            return resultString;
        } catch (IOException e) {
            Debug.logError(
                "[UISQLFileWriter.writeUiDisplayTypeSql]: IO Exception - Error was:",module);
            Debug.logError(e.getMessage(), module);

            return e.getMessage();
        } catch (SecurityException se) {
            Debug.logError(
                "[UISQLFileWriter.writeUiDisplayTypeSql]: Security Exception - Error was:", module);
            Debug.logError(se.getMessage(), module);

            return se.getMessage();
        }
    }

    /**
     * DOCUMENT ME!
     *
     * @param delegator 
     * @param entityId 
     * @param filePath 
     *
     * @return 
     */
    public String writeUiEntityFile(GenericDelegator delegator,
        String entityId, String filePath) {
        HashMap findHashMap = new HashMap();
        ArrayList findOrder = null;
        ModelEntity modelEntity = null;
        GenericValue genericValue = null;
        String entityName = "";
        String resultString = "";

        try {
            // Open the file, and replace any existing contents.
            FileWriter tempFile = new FileWriter(filePath, false);
            BufferedWriter outputFile = new BufferedWriter(tempFile);

            ////////////////////
            // Write out the SQL for inserting the UI entity.
            ////////////////////
            List uiEntityGVL = null;
            findHashMap = new HashMap();
            findHashMap.put("entityId", entityId);
            findOrder = new ArrayList();
            findOrder.add("entityId");
            uiEntityGVL = writeSqlInserts("UiEntity", findHashMap, findOrder,
                    delegator, outputFile);

            if (uiEntityGVL.size() <= 0) {
                return "No entity found with ID \"" + entityId + "\"";
            } else {
                resultString += (String.valueOf(uiEntityGVL.size()) +
                " UI entities<BR>\n");
            }

            ////////////////////
            // Write out the SQL for inserting the UI attributes.
            ////////////////////
            // Loop through the UI entities.  (Should be only one).
            Iterator uiEntityGVI = uiEntityGVL.iterator();
            List uiAttributeGVL = new LinkedList();

            while (uiEntityGVI.hasNext()) {
                GenericValue uiEntityGV = (GenericValue) uiEntityGVI.next();

                findHashMap = new HashMap();
                findHashMap.put("entityId", uiEntityGV.getString("entityId"));
                findOrder = new ArrayList();
                findOrder.add("attributeId");
                uiAttributeGVL.addAll(writeSqlInserts("UiAttribute",
                        findHashMap, findOrder, delegator, outputFile));
            }

            resultString += (String.valueOf(uiAttributeGVL.size()) +
            " UI attributes<BR>\n");
            uiEntityGVI = null;

            outputFile.close();

            return resultString;
        } catch (IOException e) {
            Debug.logError(
                "[UISQLFileWriter.writeUiEntitySql]: IO Exception - Error was:", module);
            Debug.logError(e.getMessage(), module);

            return e.getMessage();
        } catch (SecurityException se) {
            Debug.logError(
                "[UISQLFileWriter.writeUiEntitySql]: Security Exception - Error was:", module);
            Debug.logError(se.getMessage(), module);

            return se.getMessage();
        }
    }

    /**
     * DOCUMENT ME!
     *
     * @param delegator 
     * @param codeTypeId 
     * @param filePath 
     *
     * @return 
     */
    public String writeCodeTypeFile(GenericDelegator delegator,
        String codeTypeId, String filePath) {
        HashMap findHashMap = new HashMap();
        ArrayList findOrder = null;
        ModelEntity modelEntity = null;
        GenericValue genericValue = null;
        String entityName = "";
        String resultString = "";

        try {
            // Open the file, and replace any existing contents.
            FileWriter tempFile = new FileWriter(filePath, false);
            BufferedWriter outputFile = new BufferedWriter(tempFile);

            ////////////////////
            // Write out the SQL for inserting the Code Type.
            ////////////////////
            List codeTypeGVL = null;
            findHashMap = new HashMap();
            findHashMap.put("codeTypeId", codeTypeId);
            findOrder = new ArrayList();
            findOrder.add("codeTypeId");
            codeTypeGVL = writeSqlInserts("CodeType", findHashMap, findOrder,
                    delegator, outputFile);

            if (codeTypeGVL.size() <= 0) {
                return "No code type found with ID \"" + codeTypeId + "\"";
            } else {
                resultString += (String.valueOf(codeTypeGVL.size()) +
                " Code Types<BR>\n");
            }

            ////////////////////
            // Write out the SQL for inserting the Codes.
            ////////////////////
            // Loop through the UI Code Types.  (Should be only one).
            Iterator codeTypeGVI = codeTypeGVL.iterator();
            List codeGVL = new LinkedList();

            while (codeTypeGVI.hasNext()) {
                GenericValue codeGV = (GenericValue) codeTypeGVI.next();

                findHashMap = new HashMap();
                findHashMap.put("codeTypeId", codeGV.getString("codeTypeId"));
                findOrder = new ArrayList();
                findOrder.add("codeId");
                codeGVL.addAll(writeSqlInserts("Code", findHashMap, findOrder,
                        delegator, outputFile));
            }

            resultString += (String.valueOf(codeGVL.size()) + " Codes<BR>\n");
            codeTypeGVI = null;

            outputFile.close();

            return resultString;
        } catch (IOException e) {
            Debug.logError(
                "[UISQLFileWriter.writeUiEntitySql]: IO Exception - Error was:", module);
            Debug.logError(e.getMessage(), module);

            return e.getMessage();
        } catch (SecurityException se) {
            Debug.logError(
                "[UISQLFileWriter.writeUiEntitySql]: Security Exception - Error was:", module);
            Debug.logError(se.getMessage(), module);

            return se.getMessage();
        }
    }

    /**
     * DOCUMENT ME!
     *
     * @param outputFile 
     * @param outputLine 
     */
    protected void writeLine(BufferedWriter outputFile, String outputLine) {
        try {
            outputFile.write(outputLine);
            outputFile.newLine();
            outputFile.flush();

            return;
        } catch (IOException e) {
            Debug.logError(
                "[UISQLFileWriter.writeUiScreenSql]: IO Exception - Error was:", module);
            Debug.logError(e.getMessage(), module);
        } catch (SecurityException se) {
            Debug.logError(
                "[UISQLFileWriter.writeUiScreenSql]: Security Exception - Error was:", module);
            Debug.logError(se.getMessage(), module);
        }
    }

    /**
     * DOCUMENT ME!
     *
     * @param entityName 
     * @param findHashMap 
     * @param findOrder 
     * @param delegator 
     * @param outputFile 
     *
     * @return 
     */
    public List writeSqlInserts(String entityName, HashMap findHashMap,
        ArrayList findOrder, GenericDelegator delegator,
        BufferedWriter outputFile) {
        String fieldName = "";
        String fieldValue = "";
        int fieldCount = 0;
        String outputString = "";
        List genericValueList = null;

        try {
            ModelEntity modelEntity = delegator.getModelEntity(entityName);
            genericValueList = delegator.findByLike(entityName, findHashMap,
                    findOrder);

            Iterator genericValueIterator = genericValueList.iterator();

            while (genericValueIterator.hasNext()) {
                writeLine(outputFile, "-- Insert " + entityName);

                GenericValue genericValue = (GenericValue) genericValueIterator.next();
                writeLine(outputFile,
                    "INSERT INTO " + ModelUtil.javaNameToDbName(entityName) +
                    " (");

                Iterator screenFieldIterator = modelEntity.getFieldsIterator();
                fieldCount = 0;

                while (screenFieldIterator.hasNext()) {
                    outputString = "";
                    fieldCount++;

                    ModelField screenField = (ModelField) screenFieldIterator.next();
                    fieldName = screenField.getName();

                    if (fieldCount == 1) {
                        outputString += "  ";
                    } else {
                        outputString += " ,";
                    }

                    outputString += ModelUtil.javaNameToDbName(fieldName);
                    writeLine(outputFile, outputString);
                }

                writeLine(outputFile, ") VALUES (");
                screenFieldIterator = modelEntity.getFieldsIterator();
                fieldCount = 0;

                while (screenFieldIterator.hasNext()) {
                    outputString = "";
                    fieldCount++;

                    ModelField screenField = (ModelField) screenFieldIterator.next();
                    fieldName = screenField.getName();
                    fieldValue = genericValue.getString(fieldName);

                    if (fieldValue == null) {
                        fieldValue = "";
                    }

                    if (fieldCount == 1) {
                        outputString += "  ";
                    } else {
                        outputString += " ,";
                    }

                    // Replace all single quotes with two single qoutes.
                    fieldValue = StringHelper.replaceAll(fieldValue, "'", "''");

                    // Put single quotes around everything for now.
                    outputString += ("'" + fieldValue + "'");
                    writeLine(outputFile, outputString);
                }

                writeLine(outputFile, ");");
                writeLine(outputFile, "");
            }

            return genericValueList;
        } catch (GenericEntityException gee) {
            Debug.logError(
                "[UISQLFileWriter.writeSqlInserts]: Generic Entity Exception - Error was:", module);
            Debug.logError(gee.getMessage(), module);

            return genericValueList;
        }
    }
}