init

cmd/server.go

@@ -0,0 +1,35 @@
+package cmd
+
+import (
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"git.hpds.cc/pavement/hpds_node"
+	discover "hpds_access_point/internal/discover/consul"
+)
+
+var (
+	consulConfigs chan *discover.ConsulConfig
+)
+
+func Run() {
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	// 退出channel
+	exitChannel := make(chan os.Signal)
+	defer close(exitChannel)
+
+	// 退出信号监听
+	go func(c chan os.Signal) {
+		signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
+	}(exitChannel)
+
+	ap := hpds_node.NewAccessPoint(
+		"hpds-ap",
+		hpds_node.WithMqAddr("localhost:27188"),
+		hpds_node.WithCredential("token:z1"),
+	)
+}

config/config.go

@@ -0,0 +1,7 @@
+package config
+
+type AccessPointConfig struct {
+	NodeName string `yaml:"name"`
+	Host     string `yaml:"host"`
+	Port     string `yaml:"port"`
+}

go.mod

@@ -0,0 +1,21 @@
+module hpds_access_point
+
+go 1.19
+
+require github.com/hashicorp/consul/api v1.15.2
+
+require (
+	github.com/armon/go-metrics v0.3.10 // indirect
+	github.com/fatih/color v1.9.0 // indirect
+	github.com/hashicorp/go-cleanhttp v0.5.1 // indirect
+	github.com/hashicorp/go-hclog v0.14.1 // indirect
+	github.com/hashicorp/go-immutable-radix v1.3.0 // indirect
+	github.com/hashicorp/go-rootcerts v1.0.2 // indirect
+	github.com/hashicorp/golang-lru v0.5.4 // indirect
+	github.com/hashicorp/serf v0.9.7 // indirect
+	github.com/mattn/go-colorable v0.1.6 // indirect
+	github.com/mattn/go-isatty v0.0.12 // indirect
+	github.com/mitchellh/go-homedir v1.1.0 // indirect
+	github.com/mitchellh/mapstructure v1.4.1 // indirect
+	golang.org/x/sys v0.0.0-20220412211240-33da011f77ad // indirect
+)

internal/discover/consul/consul.go

@@ -0,0 +1,79 @@
+package discover
+
+import (
+	"fmt"
+
+	"github.com/hashicorp/consul/api"
+)
+
+type ConsulConfig struct {
+	Client                         *api.Client `json:"client"`                         // consul client
+	ConsulAddress                  string      `json:"consulAddress"`                  // consul 服务地址:IP+port
+	ServiceId                      string      `json:"serviceId"`                      // 服务ID
+	ServiceName                    string      `json:"serviceName"`                    // 服务名称
+	ServiceIP                      string      `json:"serviceIP"`                      // 服务IP
+	ServicePort                    int         `json:"servicePort"`                    // 服务端口
+	Tags                           []string    `json:"tags"`                           // 服务标签列表
+	DeregisterCriticalServiceAfter int         `json:"deregisterCriticalServiceAfter"` // 指定与服务关联的检查应在此时间之后注销
+	Interval                       int         `json:"interval"`                       // 指定运行此检查的频率
+	Timeout                        int         `json:"timeout"`                        // 在脚本、HTTP、TCP 或 gRPC 检查的情况下指定传出连接的超时时间
+}
+
+func NewConsulConfig(consulAddress string, serviceId string, serviceName string, serviceIP string, servicePort int, tags []string, deregisterCriticalServiceAfter int, interval int, timeout int) (*ConsulConfig, error) {
+	// 1.consul配置
+	config := api.DefaultConfig()
+	config.Address = consulAddress
+	// 2.client
+	client, err := api.NewClient(config)
+	if err != nil {
+		return nil, err
+	}
+	return &ConsulConfig{
+		Client:                         client,
+		ConsulAddress:                  consulAddress,
+		ServiceId:                      serviceId,
+		ServiceName:                    serviceName,
+		ServiceIP:                      serviceIP,
+		ServicePort:                    servicePort,
+		Tags:                           tags,
+		DeregisterCriticalServiceAfter: deregisterCriticalServiceAfter,
+		Interval:                       interval,
+		Timeout:                        timeout,
+	}, nil
+
+}
+
+// ServiceRegister 服务注册
+func (cf *ConsulConfig) ServiceRegister() (err error) {
+	// 注册器
+	reg := &api.AgentServiceRegistration{
+		ID:      cf.ServiceId,
+		Name:    cf.ServiceName,
+		Address: cf.ServiceIP,
+		Port:    cf.ServicePort,
+		Tags:    cf.Tags,
+		Check: &api.AgentServiceCheck{
+			Interval:                       fmt.Sprintf("%vs", cf.Interval),                                  // 健康检查间隔
+			HTTP:                           fmt.Sprintf("http://%v:%v/health", cf.ServiceIP, cf.ServicePort), // HTTP 支持，执行健康检查的地址，service 会传到 Health.Check 函数中
+			Timeout:                        fmt.Sprintf("%vs", cf.Timeout),                                   // 健康检查超时时间
+			DeregisterCriticalServiceAfter: fmt.Sprintf("%vs", cf.DeregisterCriticalServiceAfter),            // 注销时间，相当于过期时间
+			Notes:                          "Consul check service health status.",
+		},
+	}
+	// 注册服务
+	err = cf.Client.Agent().ServiceRegister(reg)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+// ServiceDeregister 服务注销
+func (cf *ConsulConfig) ServiceDeregister() error {
+	return cf.Client.Agent().ServiceDeregister(cf.ServiceId)
+}
+
+// ServiceDiscover 服务发现
+func (cf *ConsulConfig) ServiceDiscover(service string, tags []string, q *api.QueryOptions) ([]*api.CatalogService, *api.QueryMeta, error) {
+	return cf.Client.Catalog().ServiceMultipleTags(service, tags, q)
+}

main.go

@@ -0,0 +1 @@
+package hpds_access_point