ads mode support consistent hash

Signed-off-by: 1640528278@qq.com <1640528278@qq.com>

Author: bfforever

pkg/bpf/ads/loader.go

@@ -32,8 +32,8 @@ import (
 
 	"kmesh.net/kmesh/daemon/options"
 	"kmesh.net/kmesh/pkg/bpf/restart"
-	"kmesh.net/kmesh/pkg/logger"
 	"kmesh.net/kmesh/pkg/consistenthash/maglev"
+	"kmesh.net/kmesh/pkg/logger"
 )
 
 var log = logger.NewLoggerScope("bpf_ads")

pkg/cache/v2/cluster.go

@@ -29,8 +29,8 @@ import (
 	core_v2 "kmesh.net/kmesh/api/v2/core"
 	bpfads "kmesh.net/kmesh/pkg/bpf/ads"
 	maps_v2 "kmesh.net/kmesh/pkg/cache/v2/maps"
-	"kmesh.net/kmesh/pkg/utils"
 	"kmesh.net/kmesh/pkg/consistenthash/maglev"
+	"kmesh.net/kmesh/pkg/utils"
 )
 
 type ClusterStatsKey struct {
@@ -181,8 +181,8 @@ func (cache *ClusterCache) Flush() {
 			err := maps_v2.ClusterUpdate(name, cluster)
 			if cluster.GetLbPolicy() == cluster_v2.Cluster_MAGLEV {
 				// create consistent lb here and update table to bpf map
-				if err := maglev.CreateLB(cluster);err != nil {
-					log.Errorf("maglev lb update %v cluster failed: %v",name, err)
+				if err := maglev.CreateLB(cluster); err != nil {
+					log.Errorf("maglev lb update %v cluster failed: %v", name, err)
 				}
 			}
 			if err == nil {

pkg/consistenthash/maglev/maglev.go

@@ -23,7 +23,7 @@ import (
 	"unsafe"
 
 	"github.com/cilium/ebpf"
-	
+
 	cluster_v2 "kmesh.net/kmesh/api/v2/cluster"
 	"kmesh.net/kmesh/api/v2/endpoint"
 	"kmesh.net/kmesh/pkg/utils/hash"

pkg/consistenthash/maglev/maglev_test.go

@@ -24,7 +24,7 @@ import (
 
 	"github.com/cilium/ebpf"
 	"github.com/stretchr/testify/suite"
-	
+
 	cluster_v2 "kmesh.net/kmesh/api/v2/cluster"
 	"kmesh.net/kmesh/api/v2/core"
 	"kmesh.net/kmesh/api/v2/endpoint"

pkg/controller/ads/cache.go

@@ -474,10 +474,10 @@ func newApiRouteAction(action *config_route_v3.RouteAction) *route_v2.RouteActio
 		return nil
 	}
 	// current only support http header based hash policy
-	apiHashPolicys := make([]*route_v2.RouteAction_HashPolicy,0)
+	apiHashPolicys := make([]*route_v2.RouteAction_HashPolicy, 0)
 	for _, hp := range action.GetHashPolicy() {
 		apiHashPolicy := &route_v2.RouteAction_HashPolicy{
-				PolicySpecifier: nil,
+			PolicySpecifier: nil,
 		}
 		switch hp.GetPolicySpecifier().(type) {
 		case *config_route_v3.RouteAction_HashPolicy_Header_:

pkg/utils/hash/murmur3.go

@@ -13,15 +13,13 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
- 
-package hash
 
+package hash
 
 import (
 	"unsafe"
 )
 
-
 // Hash128 calculates a 128 bits hash for the given data. It returns different
 // results when running on big-endian and little-endian machines.
 //
@@ -33,20 +31,16 @@ func Hash128(data []byte, seed uint32) (uint64, uint64) {
 		c2 = uint64(0x4cf5ad432745937f)
 	)
 
-
 	nblocks := len(data) / 16
 
-
 	h1 := uint64(seed)
 	h2 := uint64(seed)
 
-
 	for i := 0; i < nblocks; i++ {
 		tmp := (*[2]uint64)(unsafe.Pointer(&data[i*16]))
 		k1 := tmp[0]
 		k2 := tmp[1]
 
-
 		k1 *= c1
 		k1 = rotl64(k1, 31)
 		k1 *= c2
@@ -63,14 +57,11 @@ func Hash128(data []byte, seed uint32) (uint64, uint64) {
 		h2 = h2*5 + 0x38495ab5
 	}
 
-
 	k1 := uint64(0)
 	k2 := uint64(0)
 
-
 	tail := data[nblocks*16:]
 
-
 	switch len(tail) & 15 {
 	case 15:
 		k2 ^= uint64(tail[14]) << 48
@@ -126,7 +117,6 @@ func Hash128(data []byte, seed uint32) (uint64, uint64) {
 		h1 ^= k1
 	}
 
-
 	h1 ^= uint64(len(data))
 	h2 ^= uint64(len(data))
 	h1 += h2
@@ -136,23 +126,19 @@ func Hash128(data []byte, seed uint32) (uint64, uint64) {
 	h1 += h2
 	h2 += h1
 
-
 	return h1, h2
 }
 
-
 func rotl64(x uint64, r int8) uint64 {
 	return (x << r) | (x >> (64 - r))
 }
 
-
 func fmix64(k uint64) uint64 {
 	k ^= k >> 33
 	k *= 0xff51afd7ed558ccd
 	k ^= k >> 33
 	k *= 0xc4ceb9fe1a85ec53
 	k ^= k >> 33
 
-
 	return k
-}
+}