diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..dace29f --- /dev/null +++ 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 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..a0d62b5 --- /dev/null +++ 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= diff --git a/to_entangle.go b/to_entangle.go new file mode 100644 index 0000000..8d8a559 --- /dev/null +++ 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, + } +}