updated foreign check switch

viant · Sep 17, 2019 · b973413 · b973413
1 parent fc74519
commit b973413
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/service.go b/service.go
@@ -508,9 +508,13 @@ func (s *service) prepareWithRequest(request *PrepareRequest, response *PrepareR
 func (s *service) enableForeignKeyCheck(datastore string, connection dsc.Connection) error {
 	dialect := GetDatastoreDialect(datastore, s.registry)
 	manager := s.registry.Get(datastore)
-	adminManager, err := s.getAdminManager(datastore)
-	if err != nil {
-		return err
+	adminManager := manager
+	var err error
+	if ! dialect.IsKeyCheckSwitchSessionLevel() {
+		adminManager, err = s.getAdminManager(datastore)
+		if err != nil {
+			return err
+		}
 	}
 	if manager != adminManager {
 		defer connection.Close()