From 9cd6b0fca87679e570fbdd4942e3068feb3439fa Mon Sep 17 00:00:00 2001
From: Alexander Kiryukhin <a.kiryukhin@mail.ru>
Date: Thu, 7 Apr 2022 21:36:40 +0300
Subject: initial

---
 filter.go | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 filter.go

(limited to 'filter.go')

diff --git a/filter.go b/filter.go
new file mode 100644
index 0000000..20c23a9
--- /dev/null
+++ b/filter.go
@@ -0,0 +1,34 @@
+package collection
+
+import "sync"
+
+func FilterSync[T any](collection []T, filter func(item T, idx int) bool) []T {
+	var result []T
+	for i, v := range collection {
+		if filter(v, i) {
+			result = append(result, v)
+		}
+	}
+	return result
+}
+
+func Filter[T any](collection []T, filter func(item T, idx int) bool) []T {
+	var result []T
+	mu := sync.Mutex{}
+	wg := sync.WaitGroup{}
+	wg.Add(len(collection))
+	for i, v := range collection {
+		func(v T, i int) {
+			go func() {
+				defer wg.Done()
+				if filter(v, i) {
+					mu.Lock()
+					result = append(result, v)
+					mu.Unlock()
+				}
+			}()
+		}(v, i)
+	}
+	wg.Wait()
+	return result
+}
-- 
cgit v1.2.3