Function CheckDbConnection

backend/core/runner/db.go:162–197

(dbUrl string, d time.Duration)

}

func CheckDbConnection(dbUrl string, d time.Duration) errors.Error {
	ctx := context.Background()

	result := make(chan errors.Error, 1)
	done := make(chan struct{}, 1)
	go func() {
		defer func() {
			if err := recover(); err != nil {
				result <- errors.Default.New(fmt.Sprintf("panic when checking db connection: %v", err))
			}
		}()
		db, err := MakeDbConnection(dbUrl, &gorm.Config{})
		if err != nil {
			result <- errors.Convert(err)
			return
		}
		if d > 0 {
			var cancel context.CancelFunc
			ctx, cancel = context.WithTimeout(context.Background(), d)
			defer cancel()
		}
		if err := db.WithContext(ctx).Exec("SELECT 1").Error; err != nil {
			done <- struct{}{}
		} else {
			result <- errors.Convert(err)
		}
	}()
	select {
	case <-time.After(d):
		return errors.Default.New("timeout")
	case <-done:
		return nil
	case err := <-result:
		return err
	}
}