候选

2023-06-24 02:13:50 +08:00 · 2023-06-24 02:13:50 +08:00 · c77e3b9db7
commit c77e3b9db7
parent 7880432d78
3 changed files with 248 additions and 0 deletions
--- a/go.mod
+++ b/go.mod
@ -0,0 +1,20 @@
+module github.com/InvisibleFuture/to_entangle
+
+go 1.18
+
+require gorm.io/gorm v1.25.2-0.20230530020048-26663ab9bf55
+
+require (
+	github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db // indirect
+	github.com/mattn/go-sqlite3 v1.14.17 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
+	github.com/modern-go/reflect2 v1.0.2 // indirect
+)
+
+require (
+	github.com/jinzhu/inflection v1.0.0 // indirect
+	github.com/jinzhu/now v1.1.5 // indirect
+	github.com/json-iterator/go v1.1.12
+	github.com/syndtr/goleveldb v1.0.0
+	gorm.io/driver/sqlite v1.5.2
+)
--- a/go.sum
+++ b/go.sum
@ -0,0 +1,42 @@
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
+github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
+github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db h1:woRePGFeVFfLKN/pOkfl+p/TAqKOfFu+7KPlMVpok/w=
+github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
+github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
+github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
+github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
+github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
+github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
+github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
+github.com/mattn/go-sqlite3 v1.14.17 h1:mCRHCLDUBXgpKAqIKsaAaAsrAlbkeomtRFKXh2L6YIM=
+github.com/mattn/go-sqlite3 v1.14.17/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
+github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc=
+github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
+github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
+github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/ginkgo v1.7.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/syndtr/goleveldb v1.0.0 h1:fBdIW9lB4Iz0n9khmH8w27SJ3QEJ7+IgjPEwGSZiFdE=
+github.com/syndtr/goleveldb v1.0.0/go.mod h1:ZVVdQEZoIme9iO1Ch2Jdy24qqXrMMOU6lpPAyBWyWuQ=
+golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
+gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
+gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gorm.io/driver/sqlite v1.5.2 h1:TpQ+/dqCY4uCigCFyrfnrJnrW9zjpelWVoEVNy5qJkc=
+gorm.io/driver/sqlite v1.5.2/go.mod h1:qxAuCol+2r6PannQDpOP1FP6ag3mKi4esLnB/jHed+4=
+gorm.io/gorm v1.25.1 h1:nsSALe5Pr+cM3V1qwwQ7rOkw+6UeLrX5O4v3llhHa64=
+gorm.io/gorm v1.25.1/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=
+gorm.io/gorm v1.25.2-0.20230530020048-26663ab9bf55 h1:sC1Xj4TYrLqg1n3AN10w871An7wJM0gzgcm8jkIkECQ=
+gorm.io/gorm v1.25.2-0.20230530020048-26663ab9bf55/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=
--- a/to_entangle.go
+++ b/to_entangle.go
@ -0,0 +1,186 @@
+package toentangle
+
+import (
+	"sync"
+
+	jsoniter "github.com/json-iterator/go"
+	leveldb "github.com/syndtr/goleveldb/leveldb"
+)
+
+// 1. 存入一对数据, 使其双向绑定
+// 2. 移除一对数据, 使其双向解绑
+// 3. 移除一个数据, 使其全部解绑
+// 4. 获取一个数据的全部绑定数据
+
+type ToEntangle struct {
+	lock sync.Mutex
+	db   *leveldb.DB
+}
+
+// Get 获取一个数据的全部绑定数据
+func (t *ToEntangle) Get(a string) (arr []string, err error) {
+	// 获取 a 的数据, 如果不存在, 则直接返回空数组
+	data, err := t.db.Get([]byte(a), nil)
+	if err != nil {
+		if err == leveldb.ErrNotFound {
+			return []string{}, nil
+		}
+		return nil, err
+	}
+
+	// data 为json字符串的数组, 解码后返回
+	err = jsoniter.Unmarshal(data, &arr)
+	if err != nil {
+		return nil, err
+	}
+	return arr, nil
+}
+
+// Add 添加一对数据, 使其双向绑定
+func (t *ToEntangle) Add(a string, b string) (err error) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	var set_data = func(a string, b string) (err error) {
+		// 获取 a 的数据
+		data, err := t.db.Get([]byte(a), nil)
+		if err != nil {
+			// 如果不存在, 则创建并直接添加 b
+			if err == leveldb.ErrNotFound {
+				data, _ = jsoniter.Marshal([]string{b})
+				return t.db.Put([]byte(a), data, nil)
+			}
+			return err
+		}
+
+		// data 为json字符串的数组, 解码后添加 b
+		var arr []string
+		err = jsoniter.Unmarshal(data, &arr)
+		if err != nil {
+			return err
+		}
+
+		// 转换为 map, 添加 b, 以去重
+		m := make(map[string]bool)
+		for _, v := range arr {
+			m[v] = true
+		}
+		m[b] = true
+
+		// 转换回数组
+		arr = make([]string, 0, len(m))
+		for k := range m {
+			arr = append(arr, k)
+		}
+
+		// 重新编码后存入
+		data, _ = jsoniter.Marshal(arr)
+		return t.db.Put([]byte(a), data, nil)
+	}
+
+	err = set_data(a, b)
+	if err != nil {
+		return err
+	}
+	err = set_data(b, a)
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+// 单向解绑
+func (t *ToEntangle) remove_item(a string, b string) (err error) {
+	// 获取 a 的数据, 如果不存在, 则直接返回
+	data, err := t.db.Get([]byte(a), nil)
+	if err != nil {
+		if err == leveldb.ErrNotFound {
+			return nil
+		}
+		return err
+	}
+
+	// data 为json字符串的数组, 解码后移除 b
+	var arr []string
+	err = jsoniter.Unmarshal(data, &arr)
+	if err != nil {
+		return err
+	}
+
+	// 如果只有一个, 则直接移除条目
+	if len(arr) == 1 {
+		return t.db.Delete([]byte(a), nil)
+	}
+
+	// 直接遍历移除
+	for i, v := range arr {
+		if v == b {
+			arr = append(arr[:i], arr[i+1:]...)
+			break
+		}
+	}
+
+	// 重新编码后存入
+	data, _ = jsoniter.Marshal(arr)
+	return t.db.Put([]byte(a), data, nil)
+}
+
+// Remove 移除一对数据, 使其双向解绑
+func (t *ToEntangle) Remove(a string, b string) (err error) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	if err = t.remove_item(a, b); err != nil {
+		return err
+	}
+
+	if err = t.remove_item(b, a); err != nil {
+		return err
+	}
+	return err
+}
+
+// RemoveAll 移除一个数据, 使其全部解绑
+func (t *ToEntangle) RemoveAll(a string) (err error) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	// 获取 a 的数据
+	data, err := t.db.Get([]byte(a), nil)
+	if err != nil {
+		// 如果不存在, 则直接返回
+		if err == leveldb.ErrNotFound {
+			return nil
+		}
+		return err
+	}
+
+	// data 为json字符串的数组, 解码后移除 b
+	var arr []string
+	err = jsoniter.Unmarshal(data, &arr)
+	if err != nil {
+		return err
+	}
+
+	// 遍历移除
+	for _, v := range arr {
+		if err = t.remove_item(a, v); err != nil {
+			return err
+		}
+		if err = t.remove_item(v, a); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+// New 创建一个 ToEntangle
+func New(path string) *ToEntangle {
+	db, err := leveldb.OpenFile(path, nil)
+	if err != nil {
+		panic(err)
+	}
+	return &ToEntangle{
+		db: db,
+	}
+}