Compare commits
12 Commits
05226985c7
...
b332088638
Author | SHA1 | Date | |
---|---|---|---|
b332088638 | |||
3556c5b805 | |||
ee89092ea4 | |||
3cc29f8383 | |||
2dc7275f15 | |||
5042f1048e | |||
3538818769 | |||
491d2c1b22 | |||
d83f1c6f4c | |||
cd4f6fc6c7 | |||
cb8e6d23f3 | |||
93d9a14858 |
31
concurrency/concurrency.go
Normal file
31
concurrency/concurrency.go
Normal file
@ -0,0 +1,31 @@
|
||||
package concurrency
|
||||
|
||||
type WebsiteChecker func(string) bool
|
||||
|
||||
type result struct {
|
||||
string
|
||||
bool
|
||||
}
|
||||
|
||||
func CheckWebsites(wc WebsiteChecker, urls []string) map[string]bool {
|
||||
results := make(map[string]bool)
|
||||
resultChannel := make(chan result)
|
||||
|
||||
for _, url := range urls {
|
||||
go func(u string) {
|
||||
resultChannel <- result{u, wc(u)}
|
||||
}(url)
|
||||
}
|
||||
// XXX: it seems that this works on Go 1.23
|
||||
// for _, u := range urls {
|
||||
// go func() {
|
||||
// resultChannel <- result{u, wc(u)}
|
||||
// }()
|
||||
// }
|
||||
for i := 0; i < len(urls); i++ {
|
||||
r := <-resultChannel
|
||||
results[r.string] = r.bool
|
||||
}
|
||||
|
||||
return results
|
||||
}
|
47
concurrency/concurrency_test.go
Normal file
47
concurrency/concurrency_test.go
Normal file
@ -0,0 +1,47 @@
|
||||
package concurrency
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func mockWebsiteChecker(url string) bool {
|
||||
return url != "waat://furhurterwe.geds"
|
||||
}
|
||||
|
||||
func TestCheckWebsites(t *testing.T) {
|
||||
websites := []string{
|
||||
"http://google.com",
|
||||
"http://blog.abc.com",
|
||||
"waat://furhurterwe.geds",
|
||||
}
|
||||
|
||||
want := map[string]bool{
|
||||
"http://google.com": true,
|
||||
"http://blog.abc.com": true,
|
||||
"waat://furhurterwe.geds": false,
|
||||
}
|
||||
|
||||
got := CheckWebsites(mockWebsiteChecker, websites)
|
||||
if !reflect.DeepEqual(got, want) {
|
||||
t.Fatalf("wanted %v, got %v", want, got)
|
||||
}
|
||||
}
|
||||
|
||||
func slowStubWebsiteChecker(_ string) bool {
|
||||
time.Sleep(20 * time.Millisecond)
|
||||
return true
|
||||
}
|
||||
|
||||
func BenchmarkCheckWebsites(b *testing.B) {
|
||||
urls := make([]string, 100)
|
||||
for i := 0; i < len(urls); i++ {
|
||||
urls[i] = "a url"
|
||||
}
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
CheckWebsites(slowStubWebsiteChecker, urls)
|
||||
}
|
||||
}
|
45
racer/racer.go
Normal file
45
racer/racer.go
Normal file
@ -0,0 +1,45 @@
|
||||
package racer
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
var ErrRacerTimeout = errors.New("timeout")
|
||||
|
||||
const tenSecondTimeout = 10 * time.Second
|
||||
|
||||
func ConfigurableRacer(a, b string, timeout time.Duration) (string, error) {
|
||||
select {
|
||||
// wait until channel closed
|
||||
case <-ping(a):
|
||||
return a, nil
|
||||
case <-ping(b):
|
||||
return b, nil
|
||||
case <-time.After(timeout):
|
||||
return "", ErrRacerTimeout
|
||||
}
|
||||
}
|
||||
|
||||
func Racer(a, b string) (string, error) {
|
||||
return ConfigurableRacer(a, b, tenSecondTimeout)
|
||||
}
|
||||
|
||||
func ping(url string) chan struct{} {
|
||||
ch := make(chan struct{})
|
||||
go func() {
|
||||
http.Get(url)
|
||||
// if got, close channel
|
||||
close(ch)
|
||||
}()
|
||||
return ch
|
||||
}
|
||||
|
||||
// func measureResponseTime(url string) time.Duration {
|
||||
// start := time.Now()
|
||||
// http.Get(url)
|
||||
// duration := time.Since(start)
|
||||
// fmt.Println(duration)
|
||||
// return duration
|
||||
// }
|
55
racer/racer_test.go
Normal file
55
racer/racer_test.go
Normal file
@ -0,0 +1,55 @@
|
||||
package racer
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
type SpyRacerTimeout struct{}
|
||||
|
||||
func (s *SpyRacerTimeout) Timeout() <-chan time.Time {
|
||||
return time.After(1 * time.Millisecond)
|
||||
}
|
||||
|
||||
func TestRacer(t *testing.T) {
|
||||
t.Run("compares speeds of servers, returning the url of the fasted one", func(t *testing.T) {
|
||||
slowServer := makeDelayedServer(20 * time.Millisecond)
|
||||
defer slowServer.Close()
|
||||
|
||||
fastServer := makeDelayedServer(0 * time.Millisecond)
|
||||
defer fastServer.Close()
|
||||
|
||||
slowURL := slowServer.URL
|
||||
fastURL := fastServer.URL
|
||||
|
||||
want := fastURL
|
||||
got, _ := Racer(slowURL, fastURL)
|
||||
|
||||
if got != want {
|
||||
t.Errorf("got %q, want %q", got, want)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("returns an error if a server doesn't respond within 10s", func(t *testing.T) {
|
||||
serverA := makeDelayedServer(25 * time.Millisecond)
|
||||
defer serverA.Close()
|
||||
|
||||
serverB := makeDelayedServer(25 * time.Millisecond)
|
||||
defer serverB.Close()
|
||||
|
||||
_, err := ConfigurableRacer(serverA.URL, serverB.URL, 20*time.Millisecond)
|
||||
if err == nil {
|
||||
t.Error("expected an error but didn't got one")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func makeDelayedServer(delay time.Duration) *httptest.Server {
|
||||
return httptest.NewServer(http.HandlerFunc(
|
||||
func(w http.ResponseWriter, r *http.Request) {
|
||||
time.Sleep(delay)
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}))
|
||||
}
|
11
walk/walk.go
Normal file
11
walk/walk.go
Normal file
@ -0,0 +1,11 @@
|
||||
package walk
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
)
|
||||
|
||||
func Walk(x interface{}, fn func(string)) {
|
||||
val := reflect.ValueOf(x)
|
||||
field := val.Field(0)
|
||||
fn(field.String())
|
||||
}
|
27
walk/walk_test.go
Normal file
27
walk/walk_test.go
Normal file
@ -0,0 +1,27 @@
|
||||
package walk
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestWalk(t *testing.T) {
|
||||
t.Run("walk function test", func(t *testing.T) {
|
||||
expected := "Chris"
|
||||
var got []string
|
||||
|
||||
x := struct {
|
||||
Name string
|
||||
}{expected}
|
||||
|
||||
Walk(x, func(input string) {
|
||||
got = append(got, input)
|
||||
})
|
||||
|
||||
if len(got) != 1 {
|
||||
t.Errorf("wrong number of function calls, got %d want %d", len(got), 1)
|
||||
}
|
||||
if got[0] != expected {
|
||||
t.Errorf("got %q want %q", got[0], expected)
|
||||
}
|
||||
})
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user