(filename string)
| 81 | } |
| 82 | |
| 83 | func (i *importer) Imports(filename string) [][]ImportSpec { |
| 84 | dbFileName := "db.go" |
| 85 | if i.Options.OutputDbFileName != "" { |
| 86 | dbFileName = i.Options.OutputDbFileName |
| 87 | } |
| 88 | modelsFileName := "models.go" |
| 89 | if i.Options.OutputModelsFileName != "" { |
| 90 | modelsFileName = i.Options.OutputModelsFileName |
| 91 | } |
| 92 | querierFileName := "querier.go" |
| 93 | if i.Options.OutputQuerierFileName != "" { |
| 94 | querierFileName = i.Options.OutputQuerierFileName |
| 95 | } |
| 96 | copyfromFileName := "copyfrom.go" |
| 97 | if i.Options.OutputCopyfromFileName != "" { |
| 98 | copyfromFileName = i.Options.OutputCopyfromFileName |
| 99 | } |
| 100 | batchFileName := "batch.go" |
| 101 | if i.Options.OutputBatchFileName != "" { |
| 102 | batchFileName = i.Options.OutputBatchFileName |
| 103 | } |
| 104 | |
| 105 | switch filename { |
| 106 | case dbFileName: |
| 107 | return mergeImports(i.dbImports()) |
| 108 | case modelsFileName: |
| 109 | return mergeImports(i.modelImports()) |
| 110 | case querierFileName: |
| 111 | return mergeImports(i.interfaceImports()) |
| 112 | case copyfromFileName: |
| 113 | return mergeImports(i.copyfromImports()) |
| 114 | case batchFileName: |
| 115 | return mergeImports(i.batchImports()) |
| 116 | default: |
| 117 | return mergeImports(i.queryImports(filename)) |
| 118 | } |
| 119 | } |
| 120 | |
| 121 | func (i *importer) dbImports() fileImports { |
| 122 | var pkg []ImportSpec |
no test coverage detected