Changeset 300


Ignore:
Timestamp:
Jan 26, 2010, 9:44:50 PM (14 years ago)
Author:
gav
Message:

Improvements to CsvService importAssetTree method, still in progress.
Added Department column, second header line with name and description as well as improved out of bounds checking.

Location:
trunk/grails-app
Files:
2 edited

Legend:

Unmodified
Added
Removed

  • trunk/grails-app/i18n/messages.properties

    r290 r300  
    11asset.tree.import.success=Asset tree imported.
    22asset.tree.import.failure=Could not create asset tree from supplied file, failed on line {0}.
    3 asset.tree.import.file.over.max.size=Supplied file is greater than max size of {0} Mbytes.
     3asset.tree.import.file.over.max.size=Supplied file is greater than max size of {0} {1}.
    44asset.tree.import.file.not.supplied=No file supplied.
    5 asset.tree.import.no.header=The supplied file does not have the correct headers, please see the template file.
     5asset.tree.import.no.header=The supplied file does not have the correct header lines, please see the template file.
    66
    77asset.copy.method.required=Please select a copy method for sub items.

  • trunk/grails-app/services/CsvService.groovy

    r290 r300  
    1919            def result = [:]
    2020
    21             def megaByteMultiplier = 1000 * 1000
    22             def fileMaxSize = 10 * megaByteMultiplier //Mb
     21            def kByteMultiplier = 1000
     22            def fileMaxSize = 500 * kByteMultiplier
    2323
    2424            def multiPartFile = request.getFile('file')
    … …  
    3838
    3939            if (multiPartFile.getSize() > fileMaxSize)
    40                 return fail(code: "asset.tree.import.file.over.max.size", args: [fileMaxSize/megaByteMultiplier])
    41 
     40                return fail(code: "asset.tree.import.file.over.max.size", args: [fileMaxSize/kByteMultiplier, "kB"])
     41
     42            def columnIndex = 0
     43            def numberOfColumns = 0
     44            def maxNumberOfColumns = 20
     45
     46            // Get first line.
    4247            def line = reader.readNext()
    4348            def lineNumber = 1
    4449
    45             def header = ["Site", "Section", "Asset", "Sub Asset", "Functional Assembly", "Sub Assembly Group"]
    46 
    47             if(line != header)
     50            // Check for header line 1.
     51            if(line != templateHeaderLine1) {
     52                log.error "Failed to find header line 1. "
     53                log.error "Required: " + templateHeaderLine1.toString()
     54                log.error "Supplied: " + line.toString()
    4855                return fail(code: "asset.tree.import.no.header")
     56            }
     57
     58            // Get second line.
     59            line = reader.readNext()
     60            lineNumber ++
     61
     62            // Check for header line 2.
     63            if(line != templateHeaderLine2) {
     64                log.error "Failed to find header line 2. "
     65                log.error "Required: " + templateHeaderLine2.toString()
     66                log.error "Supplied: " + line.toString()
     67                return fail(code: "asset.tree.import.no.header")
     68            }
    4969
    5070            log.info "Import checks passed, start processing asset file."
    … …  
    5575
    5676            def siteInstance
     77            def departmentInstance
    5778            def sectionInstance
    5879            def assetInstance
     80            def assetSubItemInstance
     81            def parentItem
    5982
    6083            while(line) {
    61                 def lineSize = line.size()
    62     //             log.info lineNumber+ "(" + lineSize + ")" + " : " + line
    63 
    64                 if(line[0]) {
    65                         if( !Site.findByName(line[0]) )
    66                             siteInstance = new Site(name: line[0])
    67                             if(!siteInstance.save())
     84                columnIndex = 0
     85                numberOfColumns = Math.min( line.size(), maxNumberOfColumns )
     86
     87                if( (columnIndex+2) > numberOfColumns ) {
     88                    line = reader.readNext()
     89                    lineNumber ++
     90                    continue
     91                }
     92
     93                if(line[columnIndex]) {
     94                    siteInstance = Site.findByName(line[columnIndex])
     95                    if(!siteInstance) {
     96                        log.info "Creating site: " + line[columnIndex]
     97                        siteInstance = new Site(name: line[columnIndex],
     98                                                                    description: line[++columnIndex])
     99                        if(!siteInstance.save()) {
     100                            log.error "Failed to create site on line: " + line[--columnIndex] + "(" + lineNumber + ")"
     101                            return fail(code: "asset.tree.import.failure", args: [lineNumber])
     102                        }
     103                    }
     104                    else log.info "Existing site: " + siteInstance
     105
     106                    columnIndex++
     107                    if( (columnIndex+2) > numberOfColumns ) {
     108                        line = reader.readNext()
     109                        lineNumber ++
     110                        continue
     111                    }
     112
     113                    if(line[columnIndex]) {
     114                        departmentInstance = Department.findByName(line[columnIndex])
     115                        if(!departmentInstance) {
     116                            departmentInstance = new Department(name: line[columnIndex],
     117                                                                                                    description: line[++columnIndex],
     118                                                                                                    site: siteInstance)
     119                            if(!departmentInstance.save()) {
     120                                log.error "Failed to create department on line: " + line[--columnIndex] + "(" + lineNumber + ")"
    68121                                return fail(code: "asset.tree.import.failure", args: [lineNumber])
    69                 }
    70                 else continue
    71 
    72                 if(line[1]) {
    73                         if( !Section.findByName(line[1]) )
    74                            sectionInstance =  new Section(name: line[1],
    75                                                                                 site: siteInstance)
    76                             if(!sectionInstance.save())
    77                                 return fail(code: "asset.tree.import.failure", args: [lineNumber])
    78                 }
    79                 else continue
    80 
    81                 if(line[2]) {
    82                         if( !Asset.findByName(line[2]) )
    83                             assetInstance = new Asset(name: line[2],
    84                                                                         section: sectionInstance)
    85                             if(!sectionInstance.save())
    86                                 return fail(code: "asset.tree.import.failure", args: [lineNumber])
    87                 }
    88                 else continue
     122                            }
     123                        }
     124
     125                        columnIndex++
     126                        if( (columnIndex+2) > numberOfColumns ) {
     127                            line = reader.readNext()
     128                            lineNumber ++
     129                            continue
     130                        }
     131
     132                        if(line[columnIndex]) {
     133                            sectionInstance = Section.findByName(line[columnIndex])
     134                            if(!sectionInstance) {
     135                                sectionInstance =  new Section(name: line[columnIndex],
     136                                                                                        description: line[++columnIndex],
     137                                                                                        site: siteInstance,
     138                                                                                        department: departmentInstance)
     139                                if(!sectionInstance.save()) {
     140                                    log.error "Failed to create section on line: " + line[--columnIndex] + "(" + lineNumber + ")"
     141                                    return fail(code: "asset.tree.import.failure", args: [lineNumber])
     142                                }
     143                            }
     144
     145                            columnIndex++
     146                            if( (columnIndex+2) > numberOfColumns ) {
     147                                line = reader.readNext()
     148                                lineNumber ++
     149                                continue
     150                            }
     151
     152                            if(line[columnIndex]) {
     153                                assetInstance = Asset.findByName(line[columnIndex])
     154                                if(!assetInstance) {
     155                                    assetInstance = new Asset(name: line[columnIndex],
     156                                                                                    description: line[++columnIndex],
     157                                                                                    section: sectionInstance)
     158                                    if(!assetInstance.save()) {
     159                                        log.error "Failed to create asset on line: " + line[--columnIndex] + "(" + lineNumber + ")"
     160                                        return fail(code: "asset.tree.import.failure", args: [lineNumber])
     161                                    }
     162                                }
     163
     164                                columnIndex++
     165                                if( (columnIndex+2) > numberOfColumns ) {
     166                                    line = reader.readNext()
     167                                    lineNumber ++
     168                                    continue
     169                                }
     170
     171                                if(line[columnIndex]) {
     172                                    assetSubItemInstance = AssetSubItem.findByName(line[columnIndex])
     173                                    if(!assetSubItemInstance) {
     174                                        assetSubItemInstance = new AssetSubItem(name: line[columnIndex],
     175                                                                                                                    description: line[++columnIndex])
     176                                        if(!assetInstance.save()) {
     177                                            log.error "Failed to create assetSubItem on line: " + line[--columnIndex] + "(" + lineNumber + ")"
     178                                            return fail(code: "asset.tree.import.failure", args: [lineNumber])
     179                                        }
     180                                    }
     181
     182                                    assetInstance.addToAssetSubItems(assetSubItemInstance)
     183
     184                                    columnIndex++
     185                                    if( (columnIndex+2) > numberOfColumns ) {
     186                                        line = reader.readNext()
     187                                        lineNumber ++
     188                                        continue
     189                                    }
     190
     191                                    while( (columnIndex+2) < numberOfColumns ) {
     192
     193                                        if(line[columnIndex]) {
     194                                            parentItem = assetSubItemInstance
     195                                            assetSubItemInstance = new AssetSubItem(name: line[columnIndex],
     196                                                                                                                        description: line[++columnIndex],
     197                                                                                                                        parentItem: parentItem)
     198                                            if(!assetInstance.save()) {
     199                                                log.error "Failed to create assetSubItem on line: " + line[--columnIndex] + "(" + lineNumber + ")"
     200                                                return fail(code: "asset.tree.import.failure", args: [lineNumber])
     201                                            }
     202                                        }
     203                                        else break
     204
     205                                        columnIndex++
     206                                    } // while()
     207
     208                                } // AssetSubItem L1
     209                            } // Asset
     210                        } // Section
     211                    } // Department
     212                } // Site
    89213
    90214                line = reader.readNext()
    … …  
    109233        CSVWriter writer = new CSVWriter(sw)
    110234
    111         //Header
    112         def header = ["Site", "Section", "Asset", "Sub Asset", "Functional Assembly", "Sub Assembly Group"]
    113         def blankLine = []
    114         def noteLine = ["Note: the header is required, start by replacing this line."]
    115 
    116         writer.writeNext(header as String[])
    117         writer.writeNext(blankLine as String[])
    118         writer.writeNext(noteLine as String[])
     235        writer.writeNext(templateHeaderLine1 as String[])
     236        writer.writeNext(templateHeaderLine2 as String[])
     237        writer.writeNext()
     238        writer.writeNext("Note: the header lines are required, start by replacing this line.")
    119239
    120240        writer.close()
    … …  
    212332    } // end buildAssetTree
    213333
     334    private getTemplateHeaderLine1() {
     335            ["Site", "", "Department", "", "Section", "","Asset", "", "Sub Asset", "", "Functional Assembly", "", "Sub Assembly Group", "", "SubItem", ""]
     336    }
     337
     338    private getTemplateHeaderLine2() {
     339            (["Name", "Description"])*8
     340    }
     341
    214342} // end class
Note: See TracChangeset for help on using the changeset viewer.