From 3afa6fc5f612c7d97d5636ac3a2b6cbba24a5180 Mon Sep 17 00:00:00 2001 From: fiatjaf Date: Mon, 1 Jan 2024 10:15:06 -0300 Subject: [PATCH] update reflect/unsafe.String thing. --- helpers.go | 5 ++--- subscription_test.go | 18 +++++++++--------- 2 files changed, 11 insertions(+), 12 deletions(-) diff --git a/helpers.go b/helpers.go index 5190d46..5504673 100644 --- a/helpers.go +++ b/helpers.go @@ -1,7 +1,6 @@ package nostr import ( - "reflect" "sync" "unsafe" @@ -17,8 +16,8 @@ var namedMutexPool = make([]sync.Mutex, MAX_LOCKS) func memhash(p unsafe.Pointer, h, s uintptr) uintptr func namedLock(name string) (unlock func()) { - ss := (*reflect.StringHeader)(unsafe.Pointer(&name)) - idx := uint64(memhash(unsafe.Pointer(ss.Data), 0, uintptr(ss.Len))) % MAX_LOCKS + sptr := unsafe.StringData(name) + idx := uint64(memhash(unsafe.Pointer(sptr), 0, uintptr(len(name)))) % MAX_LOCKS namedMutexPool[idx].Lock() return namedMutexPool[idx].Unlock } diff --git a/subscription_test.go b/subscription_test.go index e127889..c8bd4c9 100644 --- a/subscription_test.go +++ b/subscription_test.go @@ -11,13 +11,13 @@ import ( const RELAY = "wss://nos.lol" // test if we can fetch a couple of random events -func TestSubscribe(t *testing.T) { +func TestSubscribeBasic(t *testing.T) { rl := mustRelayConnect(RELAY) defer rl.Close() sub, err := rl.Subscribe(context.Background(), Filters{{Kinds: []int{KindTextNote}, Limit: 2}}) if err != nil { - t.Errorf("subscription failed: %v", err) + t.Fatalf("subscription failed: %v", err) return } @@ -28,23 +28,23 @@ func TestSubscribe(t *testing.T) { select { case event := <-sub.Events: if event == nil { - t.Errorf("event is nil: %v", event) + t.Fatalf("event is nil: %v", event) } n++ case <-sub.EndOfStoredEvents: goto end case <-rl.Context().Done(): - t.Errorf("connection closed: %v", rl.Context().Err()) + t.Fatalf("connection closed: %v", rl.Context().Err()) goto end case <-timeout: - t.Errorf("timeout") + t.Fatalf("timeout") goto end } } end: if n != 2 { - t.Errorf("expected 2 events, got %d", n) + t.Fatalf("expected 2 events, got %d", n) } } @@ -58,7 +58,7 @@ func TestNestedSubscriptions(t *testing.T) { // fetch 2 replies to a note sub, err := rl.Subscribe(context.Background(), Filters{{Kinds: []int{KindTextNote}, Tags: TagMap{"e": []string{"0e34a74f8547e3b95d52a2543719b109fd0312aba144e2ef95cba043f42fe8c5"}}, Limit: 3}}) if err != nil { - t.Errorf("subscription 1 failed: %v", err) + t.Fatalf("subscription 1 failed: %v", err) return } @@ -68,7 +68,7 @@ func TestNestedSubscriptions(t *testing.T) { // now fetch author of this sub, err := rl.Subscribe(context.Background(), Filters{{Kinds: []int{KindProfileMetadata}, Authors: []string{event.PubKey}, Limit: 1}}) if err != nil { - t.Errorf("subscription 2 failed: %v", err) + t.Fatalf("subscription 2 failed: %v", err) return } @@ -95,7 +95,7 @@ func TestNestedSubscriptions(t *testing.T) { sub.Unsub() return case <-sub.Context.Done(): - t.Errorf("connection closed: %v", rl.Context().Err()) + t.Fatalf("connection closed: %v", rl.Context().Err()) return } }