// Register registers a fake implementation of sqldb.Conn and returns its registered name func Register() *DB { name := fmt.Sprintf("fake-%d", rand.Int63()) db := &DB{ Name: name, data: make(map[string]*sqltypes.Result), rejectedData: make(map[string]error), queryCalled: make(map[string]int), } sqldb.Register(name, func(sqldb.ConnParams) (sqldb.Conn, error) { if db.IsConnFail() { return nil, newConnError() } return NewFakeSQLDBConn(db), nil }) return db }
// Register registers a fake implementation of sqldb.Conn and returns its registered name func Register() *DB { name := fmt.Sprintf("fake-%d", rand.Int63()) db := &DB{ data: make(map[string]*proto.QueryResult), rejectedData: make(map[string]error), queryCalled: make(map[string]int), } sqldb.Register(name, func(sqldb.ConnParams) (sqldb.Conn, error) { if db.IsConnFail() { return nil, newConnError() } return NewFakeSqlDBConn(db), nil }) sqldb.DefaultDB = name return db }
func init() { // This needs to be called before threads begin to spawn. C.vt_library_init() sqldb.Register("mysql", Connect) }