From fe75bce121487b1a63930be45c33b5c27ddcfde6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E5=BE=B7=E6=BB=A1?= <go_developer@163.com>
Date: Wed, 24 Feb 2021 14:03:43 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=B8=A6segment=E4=B8=94?=
 =?UTF-8?q?=E5=B9=B6=E5=8F=91=E5=AE=89=E5=85=A8=E7=9A=84map?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 easymap/error.go   |   9 +++
 easymap/segment.go | 138 ++++++++++++++++++++++++++++++++++++++++++++-
 go.mod             |   5 +-
 go.sum             |   2 +
 4 files changed, 152 insertions(+), 2 deletions(-)

diff --git a/easymap/error.go b/easymap/error.go
index 493ec68..f7d8b65 100644
--- a/easymap/error.go
+++ b/easymap/error.go
@@ -26,3 +26,12 @@ func keyNotFound(key interface{}) error {
 func convertFail(err error) error {
 	return errors.Wrapf(err, "数据类型转换失败")
 }
+
+// segmentError ...
+//
+// Author : go_developer@163.com<张德满>
+//
+// Date : 1:44 下午 2021/2/24
+func segmentError() error {
+	return errors.New("segment需要大于0")
+}
diff --git a/easymap/segment.go b/easymap/segment.go
index 1a234ec..cb80d2d 100644
--- a/easymap/segment.go
+++ b/easymap/segment.go
@@ -7,5 +7,141 @@
 // Date : 2021-02-23 10:47 下午
 package easymap
 
-type segment struct {
+import (
+	"fmt"
+
+	"github.com/spaolacci/murmur3"
+)
+
+func NewSegment(segmentCnt int, withLock bool) (EasyMap, error) {
+	if segmentCnt <= 0 {
+		return nil, segmentError()
+	}
+	em := &segment{
+		segment: segmentCnt,
+	}
+	em.dataTable = make([]EasyMap, segmentCnt)
+	for i := 0; i < segmentCnt; i++ {
+		em.dataTable[0] = NewNormal(withLock)
+	}
+	return em, nil
+}
+
+type segment struct {
+	dataTable []EasyMap
+	segment   int
+}
+
+func (s *segment) Get(key interface{}) (interface{}, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].Get(key)
+}
+
+func (s *segment) GetWithReceiver(key interface{}, dest interface{}) error {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetWithReceiver(key, dest)
+}
+
+func (s *segment) GetUint(key interface{}) (uint, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetUint(key)
+}
+
+func (s *segment) GetUint8(key interface{}) (uint8, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetUint8(key)
+}
+
+func (s *segment) GetUint16(key interface{}) (uint16, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetUint16(key)
+}
+
+func (s *segment) GetUint32(key interface{}) (uint32, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetUint32(key)
+}
+
+func (s *segment) GetUint64(key interface{}) (uint64, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetUint64(key)
+}
+
+func (s *segment) GetInt(key interface{}) (int, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetInt(key)
+}
+
+func (s *segment) GetInt8(key interface{}) (int8, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetInt8(key)
+}
+
+func (s *segment) GetInt16(key interface{}) (int16, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetInt16(key)
+}
+
+func (s *segment) GetInt32(key interface{}) (int32, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetInt32(key)
+}
+
+func (s *segment) GetInt64(key interface{}) (int64, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetInt64(key)
+}
+
+func (s *segment) GetFloat32(key interface{}) (float32, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetFloat32(key)
+}
+
+func (s *segment) GetFloat64(key interface{}) (float64, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetFloat64(key)
+}
+
+func (s *segment) GetBool(key interface{}) (bool, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetBool(key)
+}
+
+func (s *segment) GetString(key interface{}) (string, error) {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].GetString(key)
+}
+
+func (s *segment) Set(key interface{}, value interface{}) {
+	segmentIndex := s.getSegment(key)
+	s.dataTable[segmentIndex].Set(key, value)
+}
+
+func (s *segment) Del(key interface{}) {
+	segmentIndex := s.getSegment(key)
+	s.dataTable[segmentIndex].Del(key)
+}
+
+func (s *segment) Exist(key interface{}) bool {
+	segmentIndex := s.getSegment(key)
+	return s.dataTable[segmentIndex].Exist(key)
+}
+
+func (s *segment) GetAll() map[interface{}]interface{} {
+	result := make(map[interface{}]interface{})
+	for i := 0; i < s.segment; i++ {
+		for k, v := range s.dataTable[i].GetAll() {
+			result[k] = v
+		}
+	}
+	return result
+}
+
+// getSegment 根据key获取segment
+//
+// Author : go_developer@163.com<张德满>
+//
+// Date : 1:51 下午 2021/2/24
+func (s *segment) getSegment(key interface{}) int {
+	return int(murmur3.Sum64([]byte(fmt.Sprintf("%v", key))) % uint64(s.segment))
 }
diff --git a/go.mod b/go.mod
index 5f16773..f55ae00 100644
--- a/go.mod
+++ b/go.mod
@@ -2,4 +2,7 @@ module github.com/go-developer/gopkg
 
 go 1.15
 
-require github.com/pkg/errors v0.9.1
+require (
+	github.com/pkg/errors v0.9.1
+	github.com/spaolacci/murmur3 v1.1.0
+)
diff --git a/go.sum b/go.sum
index 7c401c3..752863c 100644
--- a/go.sum
+++ b/go.sum
@@ -1,2 +1,4 @@
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI=
+github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=