Senior
// Meg kell változtatni a wrapper függvényt, amely egy adott időkorláttal fog működni (például 1 másodperc). // Ha a "hosszú" függvény ezen idő alatt lefut, nagyszerű, visszaadjuk az eredményt. // Ha nem, hibát adunk vissza. A működés eredménye ebben az esetben nem számít. // // Ezenkívül mérni kell, hogy mennyi ideig futott ez a függvény (csak ki kell írni a logba). // A wrapper függvény aláírása módosítható.
sobes.tech MI
Válasz az MI-től
Ниже пример функции-обёртки в Go, которая запускает переданную функцию с таймаутом 1 секунду, измеряет время выполнения и возвращает ошибку, если время вышло:
package main
import (
"errors"
"log"
"time"
)
// TimeoutWrapper запускает функцию fn с таймаутом timeout
func TimeoutWrapper(fn func() (interface{}, error), timeout time.Duration) (interface{}, error) {
start := time.Now()
resultChan := make(chan interface{})
errChan := make(chan error)
go func() {
res, err := fn()
if err != nil {
errChan <- err
return
}
resultChan <- res
}()
select {
case res := <-resultChan:
elapsed := time.Since(start)
log.Printf("Function completed in %v", elapsed)
return res, nil
case err := <-errChan:
elapsed := time.Since(start)
log.Printf("Function errored in %v", elapsed)
return nil, err
case <-time.After(timeout):
elapsed := time.Since(start)
log.Printf("Function timed out after %v", elapsed)
return nil, errors.New("function timed out")
}
}
// Пример "длинной" функции
func longFunc() (interface{}, error) {
time.Sleep(2 * time.Second) // имитация долгой работы
return "done", nil
}
func main() {
res, err := TimeoutWrapper(longFunc, 1*time.Second)
if err != nil {
log.Println("Error:", err)
} else {
log.Println("Result:", res)
}
}
В этом примере функция TimeoutWrapper запускает переданную функцию в горутине, ждёт её результат или ошибку, либо таймаут. Время выполнения логируется в любом случае.