tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

clean up oracle foreign key bits

Browse Source
This commit is contained in:
Jim Lambert 2020-02-12 12:01:46 -05:00
parent b83dc38af2
commit 503e1356e1
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
scope.go
View File

@ -8,6 +8,7 @@ import (
"fmt"
"reflect"
"regexp"
"sort"
"strings"
"time"
)
@ -1234,7 +1235,11 @@ func (scope *Scope) addForeignKey(field string, dest string, onDelete string, on
if scope.Dialect().HasForeignKey(scope.TableName(), keyName) {
return
}
var query = `ALTER TABLE %s ADD CONSTRAINT %s FOREIGN KEY (%s) REFERENCES %s ON DELETE %s ON UPDATE %s;`
if scope.IsOracle() {
scope.Raw(fmt.Sprintf(`ALTER TABLE %s ADD CONSTRAINT %s FOREIGN KEY (%s) REFERENCES %s`, scope.QuotedTableName(), scope.quoteIfPossible(keyName), scope.quoteIfPossible(field), dest)).Exec()
return
}
var query = `ALTER TABLE %s ADD CONSTRAINT %s FOREIGN KEY (%s) REFERENCES %s ON DELETE %s ON UPDATE %s`
scope.Raw(fmt.Sprintf(query, scope.QuotedTableName(), scope.quoteIfPossible(keyName), scope.quoteIfPossible(field), dest, onDelete, onUpdate)).Exec()
}
@ -1420,3 +1425,12 @@ func (scope *Scope) hasConditions() bool {
len(scope.Search.orConditions) > 0 ||
len(scope.Search.notConditions) > 0
}
func (scope *Scope) IsOracle() bool {
oraModules := []string{"godror", "oci8", "ora"} // must be an asc sorted slice
insertAt := sort.SearchStrings(oraModules, scope.Dialect().GetName())
if insertAt < len(oraModules) && oraModules[insertAt] == scope.Dialect().GetName() {
return true
}
return false
}