Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-02-04 12:40:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

✨ feat: 展示排序 close #29

Browse Source
This commit is contained in:
naiba
2021-01-08 21:04:50 +08:00
parent 35766e590d
commit 347bb7faa0
14 changed files with 63 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
service/rpc/auth.go
View File

@@ -2,6 +2,7 @@ package rpc
import (
"context"
"strconv"
"github.com/naiba/nezha/service/dao"
"google.golang.org/grpc/codes"
@@ -26,18 +27,15 @@ func (a *AuthHandler) RequireTransportSecurity() bool {
}
// Check ..
func (a *AuthHandler) Check(ctx context.Context) (clientID string, err error) {
func (a *AuthHandler) Check(ctx context.Context) (clientID uint64, err error) {
md, ok := metadata.FromIncomingContext(ctx)
if !ok {
err = status.Errorf(codes.Unauthenticated, "获取 metaData 失败")
return
}
var (
clientSecret string
)
var clientSecret string
if value, ok := md["client_id"]; ok {
clientID = value[0]
clientID, _ = strconv.ParseUint(value[0], 10, 64)
}
if value, ok := md["client_secret"]; ok {
clientSecret = value[0]

6
service/rpc/nezha.go
View File

@@ -17,7 +17,7 @@ type NezhaHandler struct {
// ReportState ..
func (s *NezhaHandler) ReportState(c context.Context, r *pb.State) (*pb.Receipt, error) {
var clientID string
var clientID uint64
var err error
if clientID, err = s.Auth.Check(c); err != nil {
return nil, err
@@ -32,7 +32,7 @@ func (s *NezhaHandler) ReportState(c context.Context, r *pb.State) (*pb.Receipt,
// Heartbeat ..
func (s *NezhaHandler) Heartbeat(r *pb.Beat, stream pb.NezhaService_HeartbeatServer) error {
var clientID string
var clientID uint64
var err error
defer log.Printf("Heartbeat exit server:%v err:%v", clientID, err)
if clientID, err = s.Auth.Check(stream.Context()); err != nil {
@@ -52,7 +52,7 @@ func (s *NezhaHandler) Heartbeat(r *pb.Beat, stream pb.NezhaService_HeartbeatSer
// Register ..
func (s *NezhaHandler) Register(c context.Context, r *pb.Host) (*pb.Receipt, error) {
var clientID string
var clientID uint64
var err error
if clientID, err = s.Auth.Check(c); err != nil {
return nil, err