Skip to content

Commit 7b86a6b

Browse files
committed
fix: allow owner mapping with dbname option
1 parent 3b6244f commit 7b86a6b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

copy/copy_validator.go

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -290,14 +290,16 @@ func (v *OwnerMappingValidator) Validate() error {
290290
func (v *OwnerMappingValidator) validateOwnerMappingFile() error {
291291
ownerMappingFile := utils.MustGetFlagString(option.OWNER_MAPPING_FILE)
292292
if len(ownerMappingFile) > 0 {
293+
dbname := len(utils.MustGetFlagStringSlice(option.DBNAME))
293294
schema := len(utils.MustGetFlagStringSlice(option.SCHEMA))
294295
inclTabFile := len(utils.MustGetFlagString(option.INCLUDE_TABLE_FILE))
295296
schemaMapfile := len(utils.MustGetFlagString(option.SCHEMA_MAPPING_FILE))
296297

297298
// Owner mapping file can only be used with specific options
298-
if schema == 0 && inclTabFile == 0 && schemaMapfile == 0 {
299+
if dbname == 0 && schema == 0 && inclTabFile == 0 && schemaMapfile == 0 {
299300
return &ValidationError{
300-
"Option[s] \"--owner-mapping-file\" only supports with option \"--schema or --include-table-file or --schema-mapping-file\"",
301+
"Option[s] \"--owner-mapping-file\" only supports with option \"--dbname or --schema " +
302+
"or --include-table-file or --schema-mapping-file\"",
301303
}
302304
}
303305
}

0 commit comments

Comments
 (0)