github.com/pingcap/tiflow@v0.0.0-20240520035814-5bf52d54e205/engine/pkg/dbutil/util.go (about) 1 // Copyright 2022 PingCAP, Inc. 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // See the License for the specific language governing permissions and 12 // limitations under the License. 13 14 package dbutil 15 16 import ( 17 "database/sql" 18 "strings" 19 20 "github.com/pingcap/log" 21 "github.com/pingcap/tiflow/pkg/errors" 22 "go.uber.org/zap" 23 ) 24 25 // NewSQLDB return sql.DB for specified driver and dsn 26 func NewSQLDB(driver string, dsn string, dbConf *DBConfig) (*sql.DB, error) { 27 db, err := sql.Open(driver, dsn) 28 if err != nil { 29 log.Error("open dsn fail", zap.String("dsn", dsn), zap.Any("config", dbConf), zap.Error(err)) 30 return nil, errors.ErrMetaOpFail.Wrap(err) 31 } 32 33 if dbConf != nil { 34 db.SetConnMaxIdleTime(dbConf.ConnMaxIdleTime) 35 db.SetConnMaxLifetime(dbConf.ConnMaxLifeTime) 36 db.SetMaxIdleConns(dbConf.MaxIdleConns) 37 db.SetMaxOpenConns(dbConf.MaxOpenConns) 38 } 39 return db, nil 40 } 41 42 // GetSQLStrictMode return SQL strict mode for metastore 43 // Use strict mode to avoid unexpected silent processing of meta data 44 func GetSQLStrictMode() string { 45 needEnable := []string{ 46 "STRICT_TRANS_TABLES", 47 "STRICT_ALL_TABLES", 48 "ERROR_FOR_DIVISION_BY_ZERO", 49 } 50 51 return strings.Join(needEnable, ",") 52 }