diff --git a/internal/db/models/ssl_cert_dao_test.go b/internal/db/models/ssl_cert_dao_test.go index f41504b9..5e0f0756 100644 --- a/internal/db/models/ssl_cert_dao_test.go +++ b/internal/db/models/ssl_cert_dao_test.go @@ -238,17 +238,3 @@ func TestSSLCertDAO_Update_JSON(t *testing.T) { t.Log(cert) } } - -func TestSSLCertDAO_FindAllCertsMatchDomains(t *testing.T) { - dbs.NotifyReady() - - var tx *dbs.Tx - var dao = models.NewSSLCertDAO() - certs, err := dao.FindAllCertsMatchDomains(tx, 0, []string{"goedge.cn", "teaos.cn", "www.goedge.cn", "'hello\"'", "中文.com", "xn---1.com", "global.dl.goedge.cn"}) - if err != nil { - t.Fatal(err) - } - for _, cert := range certs { - t.Log("id:", cert.Id, "userId:", cert.UserId, "name:", cert.Name, "dnsNames:", cert.DecodeDNSNames(), "end:", timeutil.FormatTime("Y-m-d H:i:s", int64(cert.TimeEndAt))) - } -} diff --git a/internal/rpc/services/service_server.go b/internal/rpc/services/service_server.go index 9d3ca50f..97835907 100644 --- a/internal/rpc/services/service_server.go +++ b/internal/rpc/services/service_server.go @@ -899,6 +899,7 @@ func (this *ServerService) ListEnabledServersMatch(ctx context.Context, req *pb. Name: clusterName, }, ServerGroups: pbGroups, + UserId: int64(server.UserId), User: pbUser, BandwidthTime: server.BandwidthTime, BandwidthBytes: int64(server.BandwidthBytes), @@ -1046,6 +1047,7 @@ func (this *ServerService) FindEnabledServer(ctx context.Context, req *pb.FindEn Name: clusterName, }, ServerGroups: pbGroups, + UserId: int64(server.UserId), User: pbUser, }}, nil } @@ -1571,6 +1573,7 @@ func (this *ServerService) FindEnabledUserServerBasic(ctx context.Context, req * Description: server.Description, IsOn: server.IsOn, Type: server.Type, + UserId: int64(server.UserId), NodeCluster: &pb.NodeCluster{ Id: int64(server.ClusterId), Name: clusterName, diff --git a/internal/rpc/services/service_ssl_cert.go b/internal/rpc/services/service_ssl_cert.go index a8d6d7ab..ed97e6e6 100644 --- a/internal/rpc/services/service_ssl_cert.go +++ b/internal/rpc/services/service_ssl_cert.go @@ -180,10 +180,10 @@ func (this *SSLCertService) CountSSLCerts(ctx context.Context, req *pb.CountSSLC var tx = this.NullTx() - if userId > 0 { - userId = req.UserId - } else if adminId > 0 { + if adminId > 0 { userId = req.UserId + } else if userId <= 0 { + return nil, errors.New("invalid user") } count, err := models.SharedSSLCertDAO.CountCerts(tx, req.IsCA, req.IsAvailable, req.IsExpired, int64(req.ExpiringDays), req.Keyword, userId, req.Domains) @@ -202,10 +202,10 @@ func (this *SSLCertService) ListSSLCerts(ctx context.Context, req *pb.ListSSLCer return nil, err } - if userId > 0 { - userId = req.UserId - } else if adminId > 0 { + if adminId > 0 { userId = req.UserId + } else if userId <= 0 { + return nil, errors.New("invalid user") } var tx = this.NullTx()