import (
    "github.com/hirpc/hrpc"
    "github.com/hirpc/hrpc/database/mysql"
    "github.com/hirpc/hrpc/database/redis"
    // "github.com/hirpc/hrpc/life"
    "github.com/hirpc/hrpc/option"
    
    pb "...."
)
func main() {
    // 需要携带配置
    s, err := hrpc.NewServer(
    	// 在TKE环境下的容器，可以忽略这个Consul部分的配置
        option.WithConsul(option.Consul{
            // Token 如果环境变量未配置，需要在此指定，否则连不上consul
	    Token: "xxxxxx",
            // Address 如果环境变量未配置，需要在此指定，否则连不上consul
            Address: "xxx",
            // DataCenter 默认数据中心是dc1，如果不一样，可以配置环境变量或者在此指定
            DataCenter: "",
	}),
        // 开启mysql和redis的支持
	option.WithDatabases(mysql.New(), redis.New()),
        // 定义该服务的名称
	option.WithServerName("服务名称"),
        // 定义环境，当前分为option.Development开发环境与option.Production生产环境
	option.WithEnvironment(option.Development),
        // 如果开启健康检查，则会启动http服务监听本地6688端口用于consul的http请求探测
	option.WithHealthCheck(),
    )
    if err != nil {
        panic(err)
    }
    pb.RegisterXXXXXServiceServer(s.Server(), &xxxxxServerImpl{})

    // 如果有任何需要的析构函数，可以在`s.Serve()`执行前，进行注册
    // life.WhenExit(xxxx.Destory)

    if err := s.Serve(); err != nil {
        panic(err)
    }
}

服务间调用

import (
    "github.com/hirpc/hrpc/service"
)

// Example 会调用consul获取对应健康状态下的服务
// 如果有多个实例，则会基于weight排序，返回weight最大的那个服务
// 注意：
//   返回的服务包含IP、名称、权重等等信息，不推荐在TKE环境中，使用Endpoint属性(POD IP)作为连接方式，因为POD IP重启变化
//   可以使用Target()属性，返回`服务名:端口`的字符串，利用TKE的DNS解析出服务IP，由TKE转发至POD中
func Example() error {
    // 默认是基于当前环境的设置（在NewServer阶段传入的option.Development或者option.Production），获取当前环境下的服务
    // 但支持传入tag标识，获取其他环境下的服务
    // 假设当前在开发环境，如果你的consul的Token允许的话，那么可以`service.Get("服务名称", service.Tag(option.Production))`获取生产环境下的服务
    p, err := service.Get("服务名称")
	if err != nil {
		return err
	}
	c, err := grpc.Dial(p.Target(), grpc.WithInsecure())
	if err != nil {
		return err
	}
    ...
    return nil
}

数据库访问

前提需要在main.go的hrpc.NewServer中，传入option.WithDatabases(mysql.New(), redis.New()),选项。之后在其他子包中，可直接采用如下方式访问

import (
    "github.com/hirpc/hrpc/database/mysql"
    "github.com/hirpc/hrpc/database/redis"
)

func UserInfo(uid string) {
    // MySQL的访问
    var v string
	if err := mysql.Client().QueryRow(ctx, `
	SELECT
		uid
	FROM
		users_info
	WHERE
		uid = ? AND is_admin = 1 AND status = 0`, []interface{}{&v}, uid,
	); err != nil {
		if err != sql.ErrNoRows 
		}
		return err
	

    // Redis的访问
    redis.Get().Set(xxx)
}

因为，在hrpc.NewServer阶段，会基于配置，维护MySQL或者Redis的连接，可以用对应包的Get()方法获取链接。