From ea05b2f23649dc97fddd7bae218b9ac237950a54 Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Wed, 22 Jul 2020 22:18:47 +0800 Subject: [PATCH] =?UTF-8?q?=E9=98=B6=E6=AE=B5=E6=80=A7=E6=8F=90=E4=BA=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- build/.gitignore | 1 - build/configs/.gitignore | 2 + build/configs/api.template.yaml | 4 + go.mod | 9 +- go.sum | 110 +++++++++- internal/configs/api_config.go | 28 +++ internal/configs/api_config_test.go | 11 + internal/configs/filter_config.go | 4 + internal/configs/server_config.go | 1 + internal/nodes/listener.go | 58 +++++- internal/nodes/listener_test.go | 18 ++ internal/nodes/node.go | 15 +- internal/nodes/rpc_client.go | 50 +++++ internal/nodes/rpc_client_test.go | 30 +++ internal/rpc/node/service.pb.go | 299 ++++++++++++++++++++++++++++ internal/rpc/node/service.proto | 19 ++ internal/utils/rlimit_darwin.go | 29 +++ internal/utils/rlimit_linux.go | 29 +++ internal/utils/rlimit_others.go | 13 ++ 19 files changed, 716 insertions(+), 14 deletions(-) create mode 100644 build/configs/.gitignore create mode 100644 build/configs/api.template.yaml create mode 100644 internal/configs/api_config.go create mode 100644 internal/configs/api_config_test.go create mode 100644 internal/configs/filter_config.go create mode 100644 internal/nodes/listener_test.go create mode 100644 internal/nodes/rpc_client.go create mode 100644 internal/nodes/rpc_client_test.go create mode 100644 internal/rpc/node/service.pb.go create mode 100644 internal/rpc/node/service.proto create mode 100644 internal/utils/rlimit_darwin.go create mode 100644 internal/utils/rlimit_linux.go create mode 100644 internal/utils/rlimit_others.go diff --git a/build/.gitignore b/build/.gitignore index 8390922..36f971e 100644 --- a/build/.gitignore +++ b/build/.gitignore @@ -1,2 +1 @@ bin/* -configs/node.yaml \ No newline at end of file diff --git a/build/configs/.gitignore b/build/configs/.gitignore new file mode 100644 index 0000000..11e11e6 --- /dev/null +++ b/build/configs/.gitignore @@ -0,0 +1,2 @@ +node.yaml +api.yaml \ No newline at end of file diff --git a/build/configs/api.template.yaml b/build/configs/api.template.yaml new file mode 100644 index 0000000..a5f96fd --- /dev/null +++ b/build/configs/api.template.yaml @@ -0,0 +1,4 @@ +rpc: + endpoints: [ "127.0.0.1:8003" ] + nodeId: "" + secret: "" \ No newline at end of file diff --git a/go.mod b/go.mod index 7e59267..925d7bd 100644 --- a/go.mod +++ b/go.mod @@ -4,9 +4,8 @@ go 1.14 require ( github.com/go-yaml/yaml v2.1.0+incompatible - github.com/iwind/TeaGo v0.0.0-20200720020412-96dbe21b81d4 - github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e // indirect - github.com/pquerna/ffjson v0.0.0-20190930134022-aa0246cd15f7 // indirect - gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f // indirect - gopkg.in/yaml.v2 v2.3.0 // indirect + github.com/golang/protobuf v1.4.2 + github.com/iwind/TeaGo v0.0.0-20200722010955-47dd648dc761 + google.golang.org/grpc v1.30.0 + google.golang.org/protobuf v1.25.0 ) diff --git a/go.sum b/go.sum index 22b7f7b..210f3a5 100644 --- a/go.sum +++ b/go.sum @@ -1,16 +1,122 @@ +cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU= +github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= +github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc= +github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= +github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= +github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98= +github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c= +github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= +github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ= +github.com/go-redis/redis v6.15.8+incompatible/go.mod h1:NAIEuMOZ/fxfXJIrKDQDz8wamY7mA7PouImQ2Jvg6kA= +github.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg= github.com/go-yaml/yaml v2.1.0+incompatible h1:RYi2hDdss1u4YE7GwixGzWwVo47T8UQwnTLB6vQiq+o= github.com/go-yaml/yaml v2.1.0+incompatible/go.mod h1:w2MrLa16VYP0jy6N7M5kHaCkaLENm+P+Tv+MfurjSw0= -github.com/iwind/TeaGo v0.0.0-20200720020412-96dbe21b81d4 h1:893FbgV8PRV/rCqEAtpifgAmsIYTDrZruWHMWIw4+x8= -github.com/iwind/TeaGo v0.0.0-20200720020412-96dbe21b81d4/go.mod h1:taNzU+Tt7Axz5t1v8pEV7xRuioNJxbMeMAbDVQpxq20= +github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q= +github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw= +github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8= +github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA= +github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs= +github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w= +github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0= +github.com/golang/protobuf v1.4.1 h1:ZFgWrT+bLgsYPirOnRfKLYJLvssAegOj/hgyMFdJZe0= +github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8= +github.com/golang/protobuf v1.4.2 h1:+Z5KGCizgyZCbGh1KZqA0fcLLkwbsjIzS4aV2v7wJX0= +github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= +github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M= +github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.0 h1:/QaMHBdZ26BB3SSst0Iwl10Epc+xhTquomWX0oZEB6w= +github.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU= +github.com/iwind/TeaGo v0.0.0-20200722010955-47dd648dc761 h1:70Iaf6iVF4CiH1P5Au3hqile/2Ea0XIkR4SPpdiaKI0= +github.com/iwind/TeaGo v0.0.0-20200722010955-47dd648dc761/go.mod h1:zjM7k+b+Jthhf0T0fKwuF0iy4TWb5SsU1gmKR2l+OmE= github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE= github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e h1:fD57ERR4JtEqsWbfPhv4DMiApHyliiK5xCTNVSPiaAs= github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno= +github.com/nxadm/tail v1.4.4/go.mod h1:kenIhsEOeOJmVchQTgglprH7qJGnHDVpk1VPCcaMI8A= +github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= +github.com/onsi/ginkgo v1.12.1/go.mod h1:zj2OWP4+oCPe1qIXoGWkgMRwljMUYCdkwsT2108oapk= +github.com/onsi/ginkgo v1.14.0/go.mod h1:iSB4RoI2tjJc9BBv4NKIKWKya62Rps+oPG/Lv9klQyY= +github.com/onsi/gomega v1.7.1/go.mod h1:XdKZgCCFLUoM/7CFJVPcG8C1xQ1AJ0vpAezJrB7JYyY= +github.com/onsi/gomega v1.10.1/go.mod h1:iN09h71vgCQne3DLsj+A5owkum+a2tYe+TOCB1ybHNo= github.com/pquerna/ffjson v0.0.0-20190930134022-aa0246cd15f7 h1:xoIK0ctDddBMnc74udxJYBqlo9Ylnsp1waqjLsnef20= github.com/pquerna/ffjson v0.0.0-20190930134022-aa0246cd15f7/go.mod h1:YARuvh7BUWHNhzDq2OM5tzR2RiCcN2D7sapiKyCel/M= +github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= +golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE= +golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU= +golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= +golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190311183353-d8887717615a h1:oWX7TPOiFAMXLq8o0ikBYfCJVlRHBcsciT5bXOrH628= +golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20200520004742-59133d7f0dd7 h1:AeiKBIuRw3UomYXSbLy0Mc2dDLfdtbT/IVn4keq83P0= +golang.org/x/net v0.0.0-20200520004742-59133d7f0dd7/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A= +golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a h1:1BGLXjeY4akVXGgbC9HugT3Jv3hCI0z56oJR5vAMgBU= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190904154756-749cb33beabd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191005200804-aed5e4c7ecf9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191120155948-bd437916bb0e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200519105757-fe76b779f299 h1:DYfZAGf2WMFjMxbgTjaC+2HC7NkNAQs+6Q8b9WEB/F4= +golang.org/x/sys v0.0.0-20200519105757-fe76b779f299/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/text v0.3.0 h1:g61tztE5qeGQ89tm6NTjjM9VPIm088od1l6aSorWRWg= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.2 h1:tW2bmiBqwgJj/UpqtC8EpXEZVYOwU0yG4iWbprSVAcs= +golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY= +golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM= +google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4= +google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc= +google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc= +google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013 h1:+kGHl1aib/qcwaRi1CbqBZ1rk19r85MNUf8HaBghugY= +google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo= +google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c= +google.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg= +google.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY= +google.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk= +google.golang.org/grpc v1.30.0 h1:M5a8xTlYTxwMn5ZFkwhRabsygDY5G8TYLyQDBxJNAxE= +google.golang.org/grpc v1.30.0/go.mod h1:N36X2cJ7JwdamYAgDz+s+rVMFjt3numwzf/HckM8pak= +google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8= +google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0= +google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM= +google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE= +google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo= +google.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.25.0 h1:Ejskq+SyPohKW+1uil0JJMtmHCgJPJ/qWTxr8qp+R4c= +google.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f h1:BLraFXnmrev5lT+xlilqcH8XK9/i0At2xKjWk4p6zsU= gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f/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.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= gopkg.in/yaml.v2 v2.3.0 h1:clyUAQHOM3G0M3f5vQj7LuJrETvjVot3Z5el9nffUtU= gopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= +honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= diff --git a/internal/configs/api_config.go b/internal/configs/api_config.go new file mode 100644 index 0000000..9f6d85d --- /dev/null +++ b/internal/configs/api_config.go @@ -0,0 +1,28 @@ +package configs + +import ( + "github.com/go-yaml/yaml" + "github.com/iwind/TeaGo/Tea" + "io/ioutil" +) + +type APIConfig struct { + RPC struct { + Endpoints []string `yaml:"endpoints"` + } `yaml:"rpc"` +} + +func LoadAPIConfig() (*APIConfig, error) { + data, err := ioutil.ReadFile(Tea.ConfigFile("api.yaml")) + if err != nil { + return nil, err + } + + config := &APIConfig{} + err = yaml.Unmarshal(data, config) + if err != nil { + return nil, err + } + + return config, nil +} diff --git a/internal/configs/api_config_test.go b/internal/configs/api_config_test.go new file mode 100644 index 0000000..795550e --- /dev/null +++ b/internal/configs/api_config_test.go @@ -0,0 +1,11 @@ +package configs + +import "testing" + +func TestLoadAPIConfig(t *testing.T) { + config, err := LoadAPIConfig() + if err != nil { + t.Fatal(err) + } + t.Log(config) +} diff --git a/internal/configs/filter_config.go b/internal/configs/filter_config.go new file mode 100644 index 0000000..d90f7fc --- /dev/null +++ b/internal/configs/filter_config.go @@ -0,0 +1,4 @@ +package configs + +type FilterConfig struct { +} diff --git a/internal/configs/server_config.go b/internal/configs/server_config.go index 9068817..9990f10 100644 --- a/internal/configs/server_config.go +++ b/internal/configs/server_config.go @@ -4,6 +4,7 @@ type ServerConfig struct { Id string `yaml:"id"` // ID IsOn bool `yaml:"isOn"` // 是否开启 Components []*ComponentConfig `yaml:"components"` // 组件 + Filters []*FilterConfig `yaml:"filters"` // 过滤器 Name string `yaml:"name"` // 名称 Description string `yaml:"description"` // 描述 ServerNames []string `yaml:"serverNames"` // 域名 diff --git a/internal/nodes/listener.go b/internal/nodes/listener.go index 1264dbb..03f3eaa 100644 --- a/internal/nodes/listener.go +++ b/internal/nodes/listener.go @@ -1,13 +1,20 @@ package nodes import ( + "context" "errors" "github.com/TeaOSLab/EdgeNode/internal/configs" + "github.com/iwind/TeaGo/logs" + "net" + "net/http" "sync" ) type Listener struct { - group *configs.ServerGroup + group *configs.ServerGroup + isListening bool + listener interface{} // 监听器 + locker sync.RWMutex } @@ -34,13 +41,13 @@ func (this *Listener) Listen() error { } protocol := this.group.Protocol() switch protocol { - case configs.ProtocolHTTP: + case configs.ProtocolHTTP, configs.ProtocolHTTP4, configs.ProtocolHTTP6: return this.listenHTTP() - case configs.ProtocolHTTPS: + case configs.ProtocolHTTPS, configs.ProtocolHTTPS4, configs.ProtocolHTTPS6: return this.ListenHTTPS() - case configs.ProtocolTCP: + case configs.ProtocolTCP, configs.ProtocolTCP4, configs.ProtocolTCP6: return this.listenTCP() - case configs.ProtocolTLS: + case configs.ProtocolTLS, configs.ProtocolTLS4, configs.ProtocolTLS6: return this.listenTLS() case configs.ProtocolUnix: return this.listenUnix() @@ -57,25 +64,66 @@ func (this *Listener) Close() error { } func (this *Listener) listenHTTP() error { + listener, err := this.createListener() + if err != nil { + return err + } + + mux := http.NewServeMux() + mux.HandleFunc("/", func(writer http.ResponseWriter, request *http.Request) { + _, _ = writer.Write([]byte("Hello, World")) + }) + server := &http.Server{ + Addr: this.group.Addr(), + Handler: mux, + } + + go func() { + err = server.Serve(listener) + if err != nil { + logs.Println("[LISTENER]" + err.Error()) + } + }() return nil } func (this *Listener) ListenHTTPS() error { + // TODO 需要实现 return nil } func (this *Listener) listenTCP() error { + // TODO 需要实现 return nil } func (this *Listener) listenTLS() error { + // TODO 需要实现 return nil } func (this *Listener) listenUnix() error { + // TODO 需要实现 return nil } func (this *Listener) listenUDP() error { + // TODO 需要实现 return nil } + +func (this *Listener) createListener() (net.Listener, error) { + listenConfig := net.ListenConfig{ + Control: nil, + KeepAlive: 0, + } + + switch this.group.Protocol() { + case configs.ProtocolHTTP4, configs.ProtocolHTTPS4, configs.ProtocolTLS4: + return listenConfig.Listen(context.Background(), "tcp4", this.group.Addr()) + case configs.ProtocolHTTP6, configs.ProtocolHTTPS6, configs.ProtocolTLS6: + return listenConfig.Listen(context.Background(), "tcp6", this.group.Addr()) + } + + return listenConfig.Listen(context.Background(), "tcp", this.group.Addr()) +} diff --git a/internal/nodes/listener_test.go b/internal/nodes/listener_test.go new file mode 100644 index 0000000..76c9cfb --- /dev/null +++ b/internal/nodes/listener_test.go @@ -0,0 +1,18 @@ +package nodes + +import ( + "github.com/TeaOSLab/EdgeNode/internal/configs" + "testing" +) + +func TestListener_Listen(t *testing.T) { + listener := NewListener() + + group := configs.NewServerGroup("http://:1234") + + listener.Reload(group) + err := listener.Listen() + if err != nil { + t.Fatal(err) + } +} diff --git a/internal/nodes/node.go b/internal/nodes/node.go index d069fbb..f0b797f 100644 --- a/internal/nodes/node.go +++ b/internal/nodes/node.go @@ -2,10 +2,12 @@ package nodes import ( "github.com/TeaOSLab/EdgeNode/internal/configs" + "github.com/TeaOSLab/EdgeNode/internal/utils" "github.com/iwind/TeaGo/logs" ) var sharedNodeConfig *configs.NodeConfig = nil +var stop = make(chan bool) type Node struct { } @@ -15,6 +17,7 @@ func NewNode() *Node { } func (this *Node) Start() { + // 读取配置 nodeConfig, err := configs.SharedNodeConfig() if err != nil { logs.Println("[NODE]start failed: read node config failed: " + err.Error()) @@ -22,5 +25,15 @@ func (this *Node) Start() { } sharedNodeConfig = nodeConfig - logs.PrintAsJSON(nodeConfig) + // 设置rlimit + _ = utils.SetRLimit(1024 * 1024) + + // 启动端口 + err = sharedListenerManager.Start(nodeConfig) + if err != nil { + logs.Println("[NODE]start failed: " + err.Error()) + } + + // hold住进程 + <-stop } diff --git a/internal/nodes/rpc_client.go b/internal/nodes/rpc_client.go new file mode 100644 index 0000000..7b62be2 --- /dev/null +++ b/internal/nodes/rpc_client.go @@ -0,0 +1,50 @@ +package nodes + +import ( + "context" + "errors" + "github.com/TeaOSLab/EdgeNode/internal/configs" + "github.com/TeaOSLab/EdgeNode/internal/rpc/node" + "github.com/iwind/TeaGo/rands" + "google.golang.org/grpc" +) + +type RPCClient struct { + nodeClients []node.ServiceClient +} + +func NewRPCClient(apiConfig *configs.APIConfig) (*RPCClient, error) { + nodeClients := []node.ServiceClient{} + + conns := []*grpc.ClientConn{} + for _, endpoint := range apiConfig.RPC.Endpoints { + conn, err := grpc.Dial(endpoint, grpc.WithInsecure()) + if err != nil { + return nil, err + } + conns = append(conns, conn) + } + if len(conns) == 0 { + return nil, errors.New("[RPC]no available endpoints") + } + + // node clients + for _, conn := range conns { + nodeClients = append(nodeClients, node.NewServiceClient(conn)) + } + + return &RPCClient{ + nodeClients: nodeClients, + }, nil +} + +func (this *RPCClient) NodeRPC() node.ServiceClient { + if len(this.nodeClients) > 0 { + return this.nodeClients[rands.Int(0, len(this.nodeClients)-1)] + } + return nil +} + +func (this *RPCClient) Context() context.Context { + return context.Background() +} diff --git a/internal/nodes/rpc_client_test.go b/internal/nodes/rpc_client_test.go new file mode 100644 index 0000000..7a51e01 --- /dev/null +++ b/internal/nodes/rpc_client_test.go @@ -0,0 +1,30 @@ +package nodes + +import ( + "github.com/TeaOSLab/EdgeNode/internal/configs" + "github.com/TeaOSLab/EdgeNode/internal/rpc/node" + "testing" + "time" +) + +func TestRPCClient_NodeRPC(t *testing.T) { + before := time.Now() + defer func() { + t.Log(time.Since(before).Seconds()*1000, "ms") + }() + config, err := configs.LoadAPIConfig() + if err != nil { + t.Fatal(err) + } + rpc, err := NewRPCClient(config) + if err != nil { + t.Fatal(err) + } + resp, err := rpc.NodeRPC().Config(rpc.Context(), &node.ConfigRequest{ + NodeId: "123456", + }) + if err != nil { + t.Fatal(err) + } + t.Log(resp) +} diff --git a/internal/rpc/node/service.pb.go b/internal/rpc/node/service.pb.go new file mode 100644 index 0000000..629cd5b --- /dev/null +++ b/internal/rpc/node/service.pb.go @@ -0,0 +1,299 @@ +// Code generated by protoc-gen-go. DO NOT EDIT. +// versions: +// protoc-gen-go v1.25.0 +// protoc v3.12.3 +// source: node/service.proto + +package node + +import ( + context "context" + proto "github.com/golang/protobuf/proto" + grpc "google.golang.org/grpc" + codes "google.golang.org/grpc/codes" + status "google.golang.org/grpc/status" + protoreflect "google.golang.org/protobuf/reflect/protoreflect" + protoimpl "google.golang.org/protobuf/runtime/protoimpl" + reflect "reflect" + sync "sync" +) + +const ( + // Verify that this generated code is sufficiently up-to-date. + _ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) + // Verify that runtime/protoimpl is sufficiently up-to-date. + _ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) +) + +// This is a compile-time assertion that a sufficiently up-to-date version +// of the legacy proto package is being used. +const _ = proto.ProtoPackageIsVersion4 + +type ConfigRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + NodeId string `protobuf:"bytes,1,opt,name=nodeId,proto3" json:"nodeId,omitempty"` +} + +func (x *ConfigRequest) Reset() { + *x = ConfigRequest{} + if protoimpl.UnsafeEnabled { + mi := &file_node_service_proto_msgTypes[0] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *ConfigRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*ConfigRequest) ProtoMessage() {} + +func (x *ConfigRequest) ProtoReflect() protoreflect.Message { + mi := &file_node_service_proto_msgTypes[0] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use ConfigRequest.ProtoReflect.Descriptor instead. +func (*ConfigRequest) Descriptor() ([]byte, []int) { + return file_node_service_proto_rawDescGZIP(), []int{0} +} + +func (x *ConfigRequest) GetNodeId() string { + if x != nil { + return x.NodeId + } + return "" +} + +type ConfigResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + Id string `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"` +} + +func (x *ConfigResponse) Reset() { + *x = ConfigResponse{} + if protoimpl.UnsafeEnabled { + mi := &file_node_service_proto_msgTypes[1] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *ConfigResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*ConfigResponse) ProtoMessage() {} + +func (x *ConfigResponse) ProtoReflect() protoreflect.Message { + mi := &file_node_service_proto_msgTypes[1] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use ConfigResponse.ProtoReflect.Descriptor instead. +func (*ConfigResponse) Descriptor() ([]byte, []int) { + return file_node_service_proto_rawDescGZIP(), []int{1} +} + +func (x *ConfigResponse) GetId() string { + if x != nil { + return x.Id + } + return "" +} + +var File_node_service_proto protoreflect.FileDescriptor + +var file_node_service_proto_rawDesc = []byte{ + 0x0a, 0x12, 0x6e, 0x6f, 0x64, 0x65, 0x2f, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x2e, 0x70, + 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x04, 0x6e, 0x6f, 0x64, 0x65, 0x22, 0x27, 0x0a, 0x0d, 0x43, 0x6f, + 0x6e, 0x66, 0x69, 0x67, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x16, 0x0a, 0x06, 0x6e, + 0x6f, 0x64, 0x65, 0x49, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x06, 0x6e, 0x6f, 0x64, + 0x65, 0x49, 0x64, 0x22, 0x20, 0x0a, 0x0e, 0x43, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x52, 0x65, 0x73, + 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, + 0x09, 0x52, 0x02, 0x69, 0x64, 0x32, 0x40, 0x0a, 0x07, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, + 0x12, 0x35, 0x0a, 0x06, 0x63, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x12, 0x13, 0x2e, 0x6e, 0x6f, 0x64, + 0x65, 0x2e, 0x43, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, + 0x14, 0x2e, 0x6e, 0x6f, 0x64, 0x65, 0x2e, 0x43, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x52, 0x65, 0x73, + 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x42, 0x08, 0x5a, 0x06, 0x2e, 0x2f, 0x6e, 0x6f, 0x64, + 0x65, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, +} + +var ( + file_node_service_proto_rawDescOnce sync.Once + file_node_service_proto_rawDescData = file_node_service_proto_rawDesc +) + +func file_node_service_proto_rawDescGZIP() []byte { + file_node_service_proto_rawDescOnce.Do(func() { + file_node_service_proto_rawDescData = protoimpl.X.CompressGZIP(file_node_service_proto_rawDescData) + }) + return file_node_service_proto_rawDescData +} + +var file_node_service_proto_msgTypes = make([]protoimpl.MessageInfo, 2) +var file_node_service_proto_goTypes = []interface{}{ + (*ConfigRequest)(nil), // 0: node.ConfigRequest + (*ConfigResponse)(nil), // 1: node.ConfigResponse +} +var file_node_service_proto_depIdxs = []int32{ + 0, // 0: node.Service.config:input_type -> node.ConfigRequest + 1, // 1: node.Service.config:output_type -> node.ConfigResponse + 1, // [1:2] is the sub-list for method output_type + 0, // [0:1] is the sub-list for method input_type + 0, // [0:0] is the sub-list for extension type_name + 0, // [0:0] is the sub-list for extension extendee + 0, // [0:0] is the sub-list for field type_name +} + +func init() { file_node_service_proto_init() } +func file_node_service_proto_init() { + if File_node_service_proto != nil { + return + } + if !protoimpl.UnsafeEnabled { + file_node_service_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*ConfigRequest); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_node_service_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*ConfigResponse); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + } + type x struct{} + out := protoimpl.TypeBuilder{ + File: protoimpl.DescBuilder{ + GoPackagePath: reflect.TypeOf(x{}).PkgPath(), + RawDescriptor: file_node_service_proto_rawDesc, + NumEnums: 0, + NumMessages: 2, + NumExtensions: 0, + NumServices: 1, + }, + GoTypes: file_node_service_proto_goTypes, + DependencyIndexes: file_node_service_proto_depIdxs, + MessageInfos: file_node_service_proto_msgTypes, + }.Build() + File_node_service_proto = out.File + file_node_service_proto_rawDesc = nil + file_node_service_proto_goTypes = nil + file_node_service_proto_depIdxs = nil +} + +// Reference imports to suppress errors if they are not otherwise used. +var _ context.Context +var _ grpc.ClientConnInterface + +// This is a compile-time assertion to ensure that this generated file +// is compatible with the grpc package it is being compiled against. +const _ = grpc.SupportPackageIsVersion6 + +// ServiceClient is the client API for Service service. +// +// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://godoc.org/google.golang.org/grpc#ClientConn.NewStream. +type ServiceClient interface { + Config(ctx context.Context, in *ConfigRequest, opts ...grpc.CallOption) (*ConfigResponse, error) +} + +type serviceClient struct { + cc grpc.ClientConnInterface +} + +func NewServiceClient(cc grpc.ClientConnInterface) ServiceClient { + return &serviceClient{cc} +} + +func (c *serviceClient) Config(ctx context.Context, in *ConfigRequest, opts ...grpc.CallOption) (*ConfigResponse, error) { + out := new(ConfigResponse) + err := c.cc.Invoke(ctx, "/node.Service/config", in, out, opts...) + if err != nil { + return nil, err + } + return out, nil +} + +// ServiceServer is the server API for Service service. +type ServiceServer interface { + Config(context.Context, *ConfigRequest) (*ConfigResponse, error) +} + +// UnimplementedServiceServer can be embedded to have forward compatible implementations. +type UnimplementedServiceServer struct { +} + +func (*UnimplementedServiceServer) Config(context.Context, *ConfigRequest) (*ConfigResponse, error) { + return nil, status.Errorf(codes.Unimplemented, "method Config not implemented") +} + +func RegisterServiceServer(s *grpc.Server, srv ServiceServer) { + s.RegisterService(&_Service_serviceDesc, srv) +} + +func _Service_Config_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(ConfigRequest) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(ServiceServer).Config(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: "/node.Service/Config", + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(ServiceServer).Config(ctx, req.(*ConfigRequest)) + } + return interceptor(ctx, in, info, handler) +} + +var _Service_serviceDesc = grpc.ServiceDesc{ + ServiceName: "node.Service", + HandlerType: (*ServiceServer)(nil), + Methods: []grpc.MethodDesc{ + { + MethodName: "config", + Handler: _Service_Config_Handler, + }, + }, + Streams: []grpc.StreamDesc{}, + Metadata: "node/service.proto", +} diff --git a/internal/rpc/node/service.proto b/internal/rpc/node/service.proto new file mode 100644 index 0000000..e706eb2 --- /dev/null +++ b/internal/rpc/node/service.proto @@ -0,0 +1,19 @@ +syntax = "proto3"; + +package node; + +option go_package = "./node"; + +service Service { + rpc config (ConfigRequest) returns (ConfigResponse) { + + } +} + +message ConfigRequest { + string nodeId = 1; +} + +message ConfigResponse { + string id = 1; +} \ No newline at end of file diff --git a/internal/utils/rlimit_darwin.go b/internal/utils/rlimit_darwin.go new file mode 100644 index 0000000..af493c7 --- /dev/null +++ b/internal/utils/rlimit_darwin.go @@ -0,0 +1,29 @@ +// +build darwin + +package utils + +import ( + "syscall" +) + +// set resource limit +func SetRLimit(limit uint64) error { + rLimit := &syscall.Rlimit{} + err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, rLimit) + if err != nil { + return err + } + + if rLimit.Cur < limit { + rLimit.Cur = limit + } + if rLimit.Max < limit { + rLimit.Max = limit + } + return syscall.Setrlimit(syscall.RLIMIT_NOFILE, rLimit) +} + +// set best resource limit value +func SetSuitableRLimit() { + SetRLimit(4096 * 100) // 1M=100Files +} diff --git a/internal/utils/rlimit_linux.go b/internal/utils/rlimit_linux.go new file mode 100644 index 0000000..a33d8e5 --- /dev/null +++ b/internal/utils/rlimit_linux.go @@ -0,0 +1,29 @@ +// +build linux + +package utils + +import ( + "syscall" +) + +// set resource limit +func SetRLimit(limit uint64) error { + rLimit := &syscall.Rlimit{} + err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, rLimit) + if err != nil { + return err + } + + if rLimit.Cur < limit { + rLimit.Cur = limit + } + if rLimit.Max < limit { + rLimit.Max = limit + } + return syscall.Setrlimit(syscall.RLIMIT_NOFILE, rLimit) +} + +// set best resource limit value +func SetSuitableRLimit() { + SetRLimit(4096 * 100) // 1M=100Files +} diff --git a/internal/utils/rlimit_others.go b/internal/utils/rlimit_others.go new file mode 100644 index 0000000..6047f6e --- /dev/null +++ b/internal/utils/rlimit_others.go @@ -0,0 +1,13 @@ +// +build !linux,!darwin + +package utils + +// set resource limit +func SetRLimit(limit uint64) error { + return nil +} + +// set best resource limit value +func SetSuitableRLimit() { + +}