Sentinel-Go 集成 Nacos 實(shí)現(xiàn)外部動態(tài)數(shù)據(jù)源

package main
import (  "fmt"  "log"  "math/rand"  "time"
  sentinel "github.com/alibaba/sentinel-golang/api"  "github.com/alibaba/sentinel-golang/core/base"  "github.com/alibaba/sentinel-golang/core/flow"  "github.com/alibaba/sentinel-golang/util")
func main() {  // We should initialize Sentinel first.  err := sentinel.InitDefault()  if err != nil {    log.Fatalf("Unexpected error: %+v", err)  }
  _, err = flow.LoadRules([]*flow.FlowRule{    {      Resource:        "some-test",      MetricType:      flow.QPS,      Count:           10,      ControlBehavior: flow.Reject,    },  })  if err != nil {    log.Fatalf("Unexpected error: %+v", err)    return  }
  ch := make(chan struct{})
  for i := 0; i < 10; i++ {    go func() {      for {        e, b := sentinel.Entry("some-test", sentinel.WithTrafficType(base.Inbound))        if b != nil {          // Blocked. We could get the block reason from the BlockError.          time.Sleep(time.Duration(rand.Uint64()%10) * time.Millisecond)        } else {          // Passed, wrap the logic here.          fmt.Println(util.CurrentTimeMillis(), "passed")          time.Sleep(time.Duration(rand.Uint64()%10) * time.Millisecond)
          // Be sure the entry is exited finally.          e.Exit()        }
      }    }()  }  <-ch}

git clone https://github.com/alibaba/nacos.gitcd nacos/mvn -Prelease-nacos -Dmaven.test.skip=true clean install -U  ls -al distribution/target/// change the $version to your actual pathcd distribution/target/nacos-server-$version/nacos/bin

unzip nacos-server-$version.zip 或者 tar -xvf nacos-server-$version.tar.gz  cd nacos/bin

[    {        "resource": "some-test",        "metricType": 1,        "count": 100.0,        "controlBehavior":0    }]

module sentinel-go-nacos-example
go 1.13
require (  github.com/alibaba/sentinel-golang v0.6.0  github.com/nacos-group/nacos-sdk-go v1.0.0)

package main
import (  "fmt"  "math/rand"  "sync/atomic"  "time"
  sentinel "github.com/alibaba/sentinel-golang/api"  "github.com/alibaba/sentinel-golang/core/base"  "github.com/alibaba/sentinel-golang/ext/datasource/nacos"  "github.com/alibaba/sentinel-golang/util"  "github.com/nacos-group/nacos-sdk-go/clients"
  "github.com/alibaba/sentinel-golang/ext/datasource"  "github.com/nacos-group/nacos-sdk-go/common/constant")
type Counter struct {  pass  *int64  block *int64  total *int64}
func main() {  //流量計(jì)數(shù)器,為了流控打印日志更直觀,和集成nacos數(shù)據(jù)源無關(guān)。  counter := Counter{pass: new(int64), block: new(int64), total: new(int64)}
  //nacos server地址  sc := []constant.ServerConfig{    {      ContextPath: "/nacos",      Port:        8848,      IpAddr:      "127.0.0.1",    },  }  //nacos client 相關(guān)參數(shù)配置,具體配置可參考https://github.com/nacos-group/nacos-sdk-go  cc := constant.ClientConfig{    TimeoutMs: 5000,  }  //生成nacos config client(配置中心客戶端)  client, err := clients.CreateConfigClient(map[string]interface{}{    "serverConfigs": sc,    "clientConfig":  cc,  })  if err != nil {    fmt.Printf("Fail to create client, err: %+v", err)    return  }  //注冊流控規(guī)則Handler  h := datasource.NewFlowRulesHandler(datasource.FlowRuleJsonArrayParser)  //創(chuàng)建NacosDataSource數(shù)據(jù)源  //sentinel-go 對應(yīng)在nacos中創(chuàng)建配置文件的group  //flow 對應(yīng)在nacos中創(chuàng)建配置文件的dataId  nds, err := nacos.NewNacosDataSource(client, "sentinel-go", "flow", h)  if err != nil {    fmt.Printf("Fail to create nacos data source client, err: %+v", err)    return  }  //nacos數(shù)據(jù)源初始化  err = nds.Initialize()  if err != nil {    fmt.Printf("Fail to initialize nacos data source client, err: %+v", err)    return  }  //啟動統(tǒng)計(jì)  go timerTask(&counter)
  //模擬流量  ch := make(chan struct{})  for i := 0; i < 10; i++ {    go func() {      for {        atomic.AddInt64(counter.total, 1)        //some-test 對應(yīng)在nacos 流控配置文件中的resource        e, b := sentinel.Entry("some-test", sentinel.WithTrafficType(base.Inbound))        if b != nil {          atomic.AddInt64(counter.block, 1)          // Blocked. We could get the block reason from the BlockError.          time.Sleep(time.Duration(rand.Uint64()%10) * time.Millisecond)        } else {          atomic.AddInt64(counter.pass, 1)          time.Sleep(time.Duration(rand.Uint64()%10) * time.Millisecond)
          // Be sure the entry is exited finally.          e.Exit()        }
      }    }()  }  <-ch}
//statistic printfunc timerTask(counter *Counter) {  fmt.Println("begin to statistic!!!")  var (    oldTotal, oldPass, oldBlock int64  )  for {    time.Sleep(1 * time.Second)    globalTotal := atomic.LoadInt64(counter.total)    oneSecondTotal := globalTotal - oldTotal    oldTotal = globalTotal
    globalPass := atomic.LoadInt64(counter.pass)    oneSecondPass := globalPass - oldPass    oldPass = globalPass
    globalBlock := atomic.LoadInt64(counter.block)    oneSecondBlock := globalBlock - oldBlock    oldBlock = globalBlock    fmt.Println(util.CurrentTimeMillis()/1000, "total:", oneSecondTotal, " pass:", oneSecondPass, " block:", oneSecondBlock)  }}

  h := datasource.NewFlowRulesHandler(datasource.FlowRulesJsonConverter)  nds, err := nacos.NewNacosDataSource(client, "sentinel-go", "flow", h)  if err != nil {    fmt.Printf("Fail to create nacos data source client, err: %+v", err)    return  }  err = nds.Initialize()  if err != nil {    fmt.Printf("Fail to initialize nacos data source client, err: %+v", err)    return  }