This commit is contained in:
wyh 2023-11-03 15:53:43 +08:00
parent 48683558dc
commit efa2be9b7a
18 changed files with 795 additions and 120 deletions

2
.gitignore vendored
View File

@ -1,2 +1,4 @@
*.sum
logs
*.exe
.vscode

View File

@ -1,16 +1,45 @@
### grpc-server
- 获取组件
##### 获取组件
- go get -u google.golang.org/grpc
- 生成 pb 文件
- protoc --proto_path=./protos ./protos/\*.proto --go_out=plugins=grpc:./pb
- go install google.golang.org/protobuf/cmd/protoc-gen-go@v1.28
- go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.2
- 生成 ssl 证书
- cmd: openssl req -x509 -newkey rsa:4096 -keyout key.pem -out cert.pem -days 365 -nodes -subj /CN=localhost
- go1.15 之后需使用临时配置 添加扩展 SAN
- 临时配置 openssl.cnf 并使用
- -config 指定配置文件
- -extensions v3_req 扩展
- cmd: openssl req -x509 -newkey rsa:4096 -keyout key.pem -out cert.pem -days 365 -nodes -subj /CN=localhost -config ./openssl.cnf -extensions v3_req
##### pb文件 生成两个文件一个基础模型文件一个grpc文件
- protoc --go_out=../pb ./product.proto
- protoc --go-grpc_out=../pb ./product.proto
### TSL证书生成
1. openssl安装https://slproweb.com/products/Win32OpenSSL.html light版即可
2. 所以操作都在cert目录下执行
> 生成私钥文件
- openssl genrsa -des3 -out ca.key 2048
> 创建证书请求
- openssl req -new -key ca.key -out ca.csr
>生成crt文件 //windows需要在cmd命令窗口运行
- openssl x509 -req -days 365 -in ca.csr-signkey ca.key -out ca.crt
>生成证书私钥server.key
- openssl genpkey -algorithm RSA -out server.key
>通过server.key生成证书请求文件server.csr
- openssl req -new -nodes -key server.key -out server.csr -days 3650 -config ./openssl.cnf -extensions v3_req
>生成SAN证书
- openssl x509 -req -days 365 -in server.csr -out server.pem -CA ca.crt -CAkey ca.key -CAcreateserial -extfile ./openssl.cnf -extensions v3_req
##### 下面为之前版本应该也能使用
```
临时配置 openssl.cnf 并使用
-config 指定配置文件
-extensions v3_req 扩展
cmd: openssl req -x509 -newkey rsa:4096 -keyout key.pem -out cert.pem -days 365 -nodes -subj /CN=localhost -config ./openssl.cnf -extensions v3_req
```

View File

@ -1,29 +0,0 @@
-----BEGIN CERTIFICATE-----
MIIE4TCCAsmgAwIBAgIUJC8WZ7nPROEEwjUJwJpwsUkHQu0wDQYJKoZIhvcNAQEL
BQAwFDESMBAGA1UEAwwJbG9jYWxob3N0MB4XDTIyMDEyMDA4MTE0MFoXDTIzMDEy
MDA4MTE0MFowFDESMBAGA1UEAwwJbG9jYWxob3N0MIICIjANBgkqhkiG9w0BAQEF
AAOCAg8AMIICCgKCAgEAuNwDPBdmTlGrZgVEnYkj0/ajuN7l/9JuoquRpjjSzTUR
2bIDL/eFQRBUYXPFVuZGep02obpEgi4bjPwLf2eCeHZOz2CP7UU8OynKh4V5fIge
rUKIpVtZ+IKvhfztvn5ObNK5b4nZCcb20wmNnQjn5LpjoMTT4ENTyYFUQ08Y31XF
NgnK6UQml86BNMWN7N6g19LjOna2xK/PDroLilmhK162liHaw0S+WEE6XvlLtDRG
ojguB+3wEBBIFQJ4kvLB8y9ft13eh/UYACC7/4SrHH9UwTUKhTwM2+d9l4JAnPc7
rU103ORHu3kh4JOxC0/u6gJKFkpU1eFIQ/bhGAzS/cHL4M6+KF1+6IQoWrayNzPG
ghwf9udZhwkeVtAu+31IQmOECqnUYs/gzAsOj3C8kMdZLPA9J8oHpHzx5GeXlH1P
K+/zevL///YYlQhTWnNnDl0DEK1eFsAABrTwY1AWrm8IfTzmMwd/247vJ1ZsuA59
KsGumk/K6Rum0jkTqrlU7VT1Cv521AAoc1lGoLWDxfw+iqXRLk/LISX74eBMAUCg
BrxzaSPD5KYs/F2jJF/B/ZU0ldggSuqPunfLc2OkQdhM6mUiKOUjnp0aKPgLKBYA
ixXwxEdKMingSlb9DH69g0rjWd6tvVSMwSgNzmWjrQ7Ah4jJi/93KCIq0zcXHvUC
AwEAAaMrMCkwCQYDVR0TBAIwADALBgNVHQ8EBAMCBeAwDwYDVR0RBAgwBoIBKoIB
KjANBgkqhkiG9w0BAQsFAAOCAgEArr0HszS531ayi9ffPlmrSd8CLmCxSxQkA+lf
Bw3nC34U+v7hOzWqGiA8mhfLQUQmL+DwM2NQs4QVKs4GmgzNQFeDnQFuWPMJuMeF
POreC9wu6uhaj2W9MGm6/L2e1WVnH1dIPC2jb3OtgmLQQ4kWQQg2vR6gXZIoxMJG
rMOkyrmBZmbtnfC0vCg+anMP/D09fzeKTQ6RQuK2OWWPYpqpxZAt3Ha8jPPGkmCO
bnSiVsAA0qygd3tyuRrkKPQMnbixz2AG7ICxePdxXpRY3B02zgW9zOOhr09bbypz
SP/pQFq2TfV/1w6HGS1MKs+VN0DUQAzEmrGf7IR1MlpEv4ujFIXZTaxUPXfbdzu1
3fsc66FKDsyLd5I0/+DG1+kwFvYNwD/mjDs75S1nLCnOevYTZcdLN40p9OUPXtka
BtEHgBCtmSAqFM9HH6+szziSlRG6iPMX0gWIqB0YzfXadZi7QOiBIvrJhRsP9JgB
VywTEb81V9qHQKHb9rt2kqYgS5S8khW7wJpI6u2S9Dz6JZKvlfdKj6Syuo0c2Uxo
S961yvqiDXGQDRu1rzF8+xFHpVENlDeN2lKHh+vDL36kACWgsozRWYnCEuxgcTwn
bWCuLo8VeA6VQuVQ53Gw5Pd+aOp/fQX0pobv1bdN1HsHwmozjY04mL/1BmMFBbXN
XgDF/7k=
-----END CERTIFICATE-----

30
cert/cert.pem Normal file
View File

@ -0,0 +1,30 @@
-----BEGIN CERTIFICATE-----
MIIFDjCCAvagAwIBAgIUcqWf//49z/zT8kVDQvKLNjsuaAYwDQYJKoZIhvcNAQEL
BQAwFDESMBAGA1UEAwwJbG9jYWxob3N0MB4XDTIzMTEwMzAxMjQxOFoXDTI0MTEw
MjAxMjQxOFowFDESMBAGA1UEAwwJbG9jYWxob3N0MIICIjANBgkqhkiG9w0BAQEF
AAOCAg8AMIICCgKCAgEA2XvjMScP+cpYvMyHHdCYtDyOKh/wW5/vZq912BiyZK0o
MVF7HEF605z73tnpZ4lvW9QiNzlM+HTKGM/w6hzf9fJaaKh0lfpPTWztE2l6UaMU
ft4xOlBrDhZSkcuZByo2tWhsoMfi/4iAa6ndJbZsNNLKKGOwqJFJHO8zqg99Y+Z4
SUJa4N5DSjHPi1eyPUcpXibtF+ud/BcJptLFLjAvunCL/NxWUqtweOI+Y3kkrBSd
k1ScmTAtpI1YAx3MYvU6JWU+EZaEjq/BABkYd1IDD7+Bmtpsl2yrL0t/63W0XaWY
RLnZXpCVjLNne8545zzxNwcTa/okGG5Q9FfSlfZ5qx8DBz/0pIp8azzJV0z2/2PK
hZggwhsi4CWX6375iFBMLdF5OptnX2HAkbtbCst5om4BdBrOgg+FpbMOm1rcdo3R
Krywx04iVljlg96fBAwao+1GWziLTuf0bKkGRD/fiQmExGx/c09W1GBMw5PVzVZg
ncSAvwmK2MrhgJBL9jsqmK3Ev/2u2WBpy5o2BkFsrZC43z2cAIhYlbYrKF0YjI3y
uXSabZHkKtTHqFTHe2y3d7fa1VU27ubhZkp1rZ4kyK4fZMfuoIPZtw8Y8stmvGuO
Dqhr4bCsD1GzQpODQjpXHa2fqkp5yaJQED/tVckHOLFvKD6vd8FOcOMeNv14rHkC
AwEAAaNYMFYwCQYDVR0TBAIwADALBgNVHQ8EBAMCBeAwHQYDVR0RBBYwFIIJbG9j
YWxob3N0ggcwLjAuMC4wMB0GA1UdDgQWBBQwBG+mPjpEJdaLHkqHxYwqwz0IDDAN
BgkqhkiG9w0BAQsFAAOCAgEAstQ9MzrzYjhoY3n/I4ECRRZ8WGx6ddzxNFcf3U+W
i83GMfRcAPZWtCw0d/m+7IShM/nlZcgdS0iz6rrA3Jsi3VopL1Mt1gtwDi2aJaaW
kwd2u8mlxeJcTYNapcMEve2X7u+yVnjONI3tDu0py7uaCYa0/ci902awJeTRV06C
Nj8GHmkFFP1GIAlPM+MmqkGFAZ2fItPc3A97K7j12aLypd8zGaP2nh4gxrdAxBhg
9VtF5IXADiHDULJcqXC4pysv1DDk5mfV+P/b5cMOOt/+AD6idNmcvKxKiCr5Df53
PwoNeABk6nUxm2vLkTG0gVeMZOZus2qH9l6Jj4fyGVdAtPVRyf+6N/czya3EWOm2
HM7Po3pFTOVf9XddJY/49XWtMJaLHNjFSKjaYGH6A+6P6PHv6zAX59OT65ePwZkF
3FUHzfy6JSn/S6sOswwTg3Ne0adY8tE1qSTQ6/q2IkVFpwfguhRVZ/AewhHMhjaJ
bRDDd0ugJ71KLc7TtjNwPZHeqXx0QXnWmJUrpPEOMyBBEZsZtm8/+EZakez7bLkG
IVeRccBImuVCmmDQ/Djs5xew/gJMYJgwvLPLch/wEGUlVqvmMZ/oKK5taO/grV4h
gzARNVU2SB0I6oacrzXB8mqYkEu++yRJHqKQFqbOZLkhgQFRTz4qVhPf2/5Gb4Dz
X54=
-----END CERTIFICATE-----

52
cert/key.pem Normal file
View File

@ -0,0 +1,52 @@
-----BEGIN PRIVATE KEY-----
MIIJRAIBADANBgkqhkiG9w0BAQEFAASCCS4wggkqAgEAAoICAQDZe+MxJw/5yli8
zIcd0Ji0PI4qH/Bbn+9mr3XYGLJkrSgxUXscQXrTnPve2elniW9b1CI3OUz4dMoY
z/DqHN/18lpoqHSV+k9NbO0TaXpRoxR+3jE6UGsOFlKRy5kHKja1aGygx+L/iIBr
qd0ltmw00sooY7CokUkc7zOqD31j5nhJQlrg3kNKMc+LV7I9RyleJu0X6538Fwmm
0sUuMC+6cIv83FZSq3B44j5jeSSsFJ2TVJyZMC2kjVgDHcxi9TolZT4RloSOr8EA
GRh3UgMPv4Ga2myXbKsvS3/rdbRdpZhEudlekJWMs2d7znjnPPE3BxNr+iQYblD0
V9KV9nmrHwMHP/SkinxrPMlXTPb/Y8qFmCDCGyLgJZfrfvmIUEwt0Xk6m2dfYcCR
u1sKy3mibgF0Gs6CD4Wlsw6bWtx2jdEqvLDHTiJWWOWD3p8EDBqj7UZbOItO5/Rs
qQZEP9+JCYTEbH9zT1bUYEzDk9XNVmCdxIC/CYrYyuGAkEv2OyqYrcS//a7ZYGnL
mjYGQWytkLjfPZwAiFiVtisoXRiMjfK5dJptkeQq1MeoVMd7bLd3t9rVVTbu5uFm
SnWtniTIrh9kx+6gg9m3Dxjyy2a8a44OqGvhsKwPUbNCk4NCOlcdrZ+qSnnJolAQ
P+1VyQc4sW8oPq93wU5w4x42/XiseQIDAQABAoICAAx0gyAhQdG3JCpZ10IenGQ0
FJs2fYwtxmbd1RSlY3APF55tcWy+lkNwud77IVoX3ZwHc9cwTadr/tf5bq1lrnDW
KpXUUer17R08x/4E3Ce5PHcAkBasHNlvb1agr5igswXkH1Z85BPo1Bv2XC8o+Zzz
1naGMQuSWHrS+yU54l86h3iAl5ZJvFZPMHjlMpS+UtDsz0a/rGpPwnJ+nyLESTN2
9xlp3/q3jUcrA9IsEazWOGkD3z0JrI+ANcJl7jrbmvMhz3Jv6ZMHXzj7dvJ8bKb1
+3zsFex9IXehrgeE0+nKDkSfVSLMingGnT5Spottqs3RNqxIJNGOXc4Eio+ikDig
xB5o5zyTWqS6fRvCibiEgUeEDKlsPK4cNu7Ehdh5pXRWIf0eYglLPIeTQqkXy3wB
/6SFRksrq3oyJJ2DKNNBsqMEChYyFB+JWpq46S3ILW7Ez6eI6Gy5T/9tWRq8Zm1p
cjnnHT8k5h/Y71F6vAjjp51qVvYOrgnbPQ1gkO4bGFCQZU1+9Vw1Bn3aLtG+bBYS
HtuMwJ+AuWd3Avm4NkHhQ99dJKz0IgWALY84e97SxDjaHWWYbsT8ITNEUV/0BE7H
QKNtbaU/WZ2pYzki5nmRzJ1+uE1UDsJK2LYLowddE1+9z8VkhJ2rDM4C+YA4o/9V
Z/8yEumB/+v5Tdi9pB/BAoIBAQD9BLrgQNOlI0XZMnQghALpP3fBkU+Qje6idvfC
6oP3J2eKEsK6PiWtmB/JwB/bcitbGuKuWEefeQr4gxceXoUFMaIonixY3iL7G10W
K5YgMu+mC3M8OMS6mkrUI8ETf5+pwg5Nd8gEqtiqHoFe7RD5ZfLusFiz6tFLdEk3
JazVlIvihRPmKlwAiZ50jtZD7FzGrRmBdCiK27tp5VL3q99UR7iy8qt0oFLDDLBB
omSeV6D7K6yOt7CvezwgXGiI8ZP2SOFrddB2fZUcsca5I03Hz+Cm4QpTS1m6P8Xl
5eqsXM4AjzyfadKuS9MkVWziQTzxSb8yYaHfqjsm+nkxmJMhAoIBAQDcC/ZCotET
yFpTfIrVuX30kDPKThKHb+8mMWiLBtlpgE9r2iGSVi3eQQv4gxsyj3NKGrp9gmPX
mGNNRbZrFrvwIqo+blzPSYNTANTqdYfZpRprYqXJNCCTYB46IwOS4STGQATcVG+m
u9MfxZDdCi7jR5EJ7tyHTsqnn8jCOCmTh4LLs379JeifJEXcRGBzyOOOy6mo+j8h
Y2X+JSV542t0YKGA6Bai9Bmr27cwfCbx6EwH84pBbu4UU+jAjg6KjOVSBtwptBcS
18ZpgFJyXDpJ5qQM6anc4sY6IDripBPVnRV/13prnJiPRp4UcqNWT9mEVvTIkQxm
/fSvAT4LYsZZAoIBAQCQEdJrAN7CiEwseYsgPKr/r3D8WkaI1BiOj9lSNoEACYLu
XvJcZCjGKPnowiohpWfEj5GFnqThzGWPze0rEL0DkRh5sl8kFxCqFRGzHwxzty04
SgrKvrReYJ69HsLA3InIesP0HoFZDwMiEw6tQ6XHrOJgQS+w+sNECKnr6dr5HoVs
bURAadxLPhFo4zBkKsFFK6l7Se4P60BS8w6EEmyvK8uisYUpemTFxYLTTKmoc9Mp
UdwGw9eqhCxfbN7lI1GSCju5GbOi1K1EUoIKR2zxmo+b1kYv2Zb08303A+zVSl7O
eMEXMIWTU3+dj7rZNjaCWfCxumAW4HKgJj0+HVehAoIBAQCMYLE0p4v8M9ez5QLo
/9HjwE7Z496cOoN1eRW5g6AVVv7Tzi+WnJXccODDJFioc61X/B82zIUHWDcZZ7zN
Ko6wzlyPQXbB4qL9w80Q2fPVnCdzlFYxOqkBe9MswYOZIzqDovm22cL0umQgo7gb
21A4PELB9HkJdPGB6ER56rxTWqLNcvAzbQiu5cQPlit2EhVh1nmgzCyUD0ZU0LvM
u5Gkh7TgeK9gO5tY8O036CwsLUa719k+jxchLrqQhM+wAdyV7Lxtem8x3Se9zFhM
jpu843+KoVJ1ENLm6ALI7WWd4/2ANiXqGteUp7dXUh7zhDM+4v58KVShK6f6MS+0
scnZAoIBAQDLPq62TVSO6GyZNPD60kSSxtyleNyzgFwyWGbN7rPx4y3ezMSkKc06
QOg/BxjeGKrs1N4LGQptZo9UWqFcpaXDthc9OJQmZVvoNTORrsYwGlm68zDX8pno
dgRe/fQwgiLWB5p5yj4zqZh28Tu6tXcteVUud1h3xVhOk9J05bp6q7QcEdKKCQkJ
MHtJ0evMLakzgW17zxw0jfM6Y8qWHveb6A9gOBWnfxqNiHO2KN6Tf6/h5bZNRX6H
hiTaZ2EwFcL4Y9b47x8rAZdAcq9MzH76I1miijbS+nDWiN1hFsRPwMx8/on0LCKZ
nJPGSQUK2+IZTi4sUmCQcy3H9tTurpA3
-----END PRIVATE KEY-----

View File

@ -227,8 +227,8 @@ subjectAltName = @alt_names
[ alt_names ]
DNS.1 = *
DNS.2 = *
DNS.1 = localhost
DNS.2 = 0.0.0.0
[ v3_ca ]

32
go.mod
View File

@ -3,33 +3,21 @@ module myschools.me/wyh/grpcservice
go 1.17
require (
github.com/gin-gonic/gin v1.7.7
github.com/lestrrat-go/file-rotatelogs v2.4.0+incompatible
github.com/rifflock/lfshook v0.0.0-20180920164130-b9218ef580f5
github.com/sirupsen/logrus v1.8.1
golang.org/x/net v0.0.0-20220114011407-0dd24b26b47d
google.golang.org/grpc v1.43.0
google.golang.org/protobuf v1.27.1
github.com/sirupsen/logrus v1.9.3
google.golang.org/grpc v1.59.0
google.golang.org/protobuf v1.31.0
)
require (
github.com/gin-contrib/sse v0.1.0 // indirect
github.com/go-playground/locales v0.13.0 // indirect
github.com/go-playground/universal-translator v0.17.0 // indirect
github.com/go-playground/validator/v10 v10.4.1 // indirect
github.com/golang/protobuf v1.5.2 // indirect
github.com/golang/protobuf v1.5.3 // indirect
github.com/jonboulle/clockwork v0.2.2 // indirect
github.com/json-iterator/go v1.1.9 // indirect
github.com/leodido/go-urn v1.2.0 // indirect
github.com/lestrrat-go/strftime v1.0.5 // indirect
github.com/mattn/go-isatty v0.0.12 // indirect
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 // indirect
github.com/lestrrat-go/strftime v1.0.6 // indirect
github.com/pkg/errors v0.9.1 // indirect
github.com/ugorji/go/codec v1.1.7 // indirect
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9 // indirect
golang.org/x/sys v0.0.0-20220114195835-da31bd327af9 // indirect
golang.org/x/text v0.3.7 // indirect
google.golang.org/genproto v0.0.0-20220118154757-00ab72f36ad5 // indirect
gopkg.in/yaml.v2 v2.2.8 // indirect
golang.org/x/net v0.17.0 // indirect
golang.org/x/sys v0.13.0 // indirect
golang.org/x/text v0.13.0 // indirect
google.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b // indirect
google.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405 // indirect
)

52
key.pem
View File

@ -1,52 +0,0 @@
-----BEGIN PRIVATE KEY-----
MIIJQgIBADANBgkqhkiG9w0BAQEFAASCCSwwggkoAgEAAoICAQC43AM8F2ZOUatm
BUSdiSPT9qO43uX/0m6iq5GmONLNNRHZsgMv94VBEFRhc8VW5kZ6nTahukSCLhuM
/At/Z4J4dk7PYI/tRTw7KcqHhXl8iB6tQoilW1n4gq+F/O2+fk5s0rlvidkJxvbT
CY2dCOfkumOgxNPgQ1PJgVRDTxjfVcU2CcrpRCaXzoE0xY3s3qDX0uM6drbEr88O
uguKWaErXraWIdrDRL5YQTpe+Uu0NEaiOC4H7fAQEEgVAniS8sHzL1+3Xd6H9RgA
ILv/hKscf1TBNQqFPAzb532XgkCc9zutTXTc5Ee7eSHgk7ELT+7qAkoWSlTV4UhD
9uEYDNL9wcvgzr4oXX7ohChatrI3M8aCHB/251mHCR5W0C77fUhCY4QKqdRiz+DM
Cw6PcLyQx1ks8D0nygekfPHkZ5eUfU8r7/N68v//9hiVCFNac2cOXQMQrV4WwAAG
tPBjUBaubwh9POYzB3/bju8nVmy4Dn0qwa6aT8rpG6bSOROquVTtVPUK/nbUAChz
WUagtYPF/D6KpdEuT8shJfvh4EwBQKAGvHNpI8Pkpiz8XaMkX8H9lTSV2CBK6o+6
d8tzY6RB2EzqZSIo5SOenRoo+AsoFgCLFfDER0oyKeBKVv0Mfr2DSuNZ3q29VIzB
KA3OZaOtDsCHiMmL/3coIirTNxce9QIDAQABAoICACNoePIkpULxH7xt2+O05fju
sBN0Fmf23Zwo8r/8+Fu4A5BIKZYhehOpVhBIhY8Mkolf3ZwCuqaYwEwz5tvjbrk2
KAhOvaK79Tp72cA1j7yLshyLtvK9WbFhDs/CT6OCO/FaN4ymwby+sH5hwQ9IKmrE
TNr5eH/nRYA+Vm6CEwiMPDcKjyQ5Go9PTa5W/MVXK0XLrhx/qqhn6GgwuI8qddWB
0cVV/LoUS8VJ0S1p4NdGgBUIX8DDEY1yIOxtSb0T5JR5CYNs5nfyOUOlAX2C4ueV
eB191GMKyofAgmzn1Qp7zKF4SZ8Z1qeRHy0N/QZmrCOEhr5YRIGnIhlqi1j489y1
zvU/v7YJcl7HJe7mVxbVaCLVHyevY3p/7nJJMxJ4mgp7TdlQrTy/xFiO8NkW3ZJ3
udWN7GFHdKt6OENdOrhBKAmd2eYKwcNslIJxlx22IkP7S0Q1Jy42HPNHEYdk08eK
fs5XgPD5ioUwKllnTdOuhrC8c+wBnF01BVsnWdXVd953uDyKFpgQATLsAowJ+4uI
cv2imL+NnwZjOQdzPWas9TCgwORT43aGeqkGrPSEOkkIxcXYIXEItd5sum+4I1Wp
OA/iUBhl0guBjk6GgdXEF/hjcXzlQrfPuYCjKeGhn9SVhXHyb1bGmgXa8cLA/1l/
kzew3TvAWnkgGX58VuwNAoIBAQDccPn/2nsoUnii3g1Axuj0VjR/d8Hgy20BzSDz
p71mVo2YwOlrhlTdyAuZqDZfODplUXaCO0xJFonIjan3uzEMp2Kbt945aLu3ow2c
xOXdGARxFUCNVtlCJZ4zUHEWPsHGY9sHav5PfvptWMrEVddGDUFj+s4nxqIP1yf5
35z344sAFQkNNMpvC3U6u0gFWn4XrQwAcC4CxteSsLT9sWrL600P8mp+Lx9ZBG61
mGRrHO7tjJchhnAN2igUYtAIPH5e2F1QhQJV7newCDbetsYfh69arxJPK39Jjst8
6tnYnBN/n+HTyBOuQeXODHYseBkV7RtHHKswvbPVA4FcX/KvAoIBAQDWrbK6o6+h
njp1dIQmhwG4nh9f4nR6rCn9BQQG4NvCOA/sd0tIvDD2F4V/89Zu7RzTiMxGsJsA
lgH9bREw8fR2A1E2oZ9xv5nuO3fWYwLTsZksvArkBcjqjCFAP0BDxYWgAnJ7Usa0
yqNdmDSwH9TDk26IadIe6eRanuaJQKH9cl1DDp818M6OYAxPF31ivEmLeozeZDMs
2CfW48QzHtykoiz1qk9zULq9HXgnn9KXc0qsAQVC9r593GX6u0iyMvgEbCWYmen4
0RMHtTbziHZRJanMt0w7JBTGzLSylCEUbgfSEwuNKkhabr15yUTAnzg5F3h9sexW
mhmEobrs24GbAoIBAAqgK3zItD0KVKRNHchsjhgQ+7DVPfQu+g8nKxpzkG6ZZ3Pg
gJ/dNsQF+4EpmY1XcYEmvxah8xhmzLdtFK9LR577hoZJA54yzTGZ23MeDpJssPIe
eyEU+Z31oD2K+BiBZ+e/MnSWFLISRdH1gLicBqdqxicilMNYEMHAtj2WKMPOJQoY
+Xm4fLWDkOAG3P7dEMzlW3m97onVudhpSiAcaDByJCHLppR0gwdQbmG0ecqr91xd
9X75MveH93ycErkC6Xh652yVLLqlpwFGNBbiKoSkjrBJVhGY4CbUijx4OY32X/47
36/Mqh4xMPmQojWciGEFL9ZqrFnKGxv+ckLYoSMCggEAJ+DQSJdLcsw/s73XJokB
qtfEoD0iwf7lNdkOrHBsoiFj49stIu/CcJRh+ijDdouZRTouKBt6T5uN9xOhk+P+
M0XeejV/ttOOtpBZbHur9MMS3xp3M3BXqCorqxGU0IIKlnSd64au3rqdqi8x6l2f
Wk0yxShTHeb4wlSuImTdMoViCajXrl9zH0N8RdzpL8QmLUCwlkcqsnPcwrBAq/ZZ
qkNt8pyWcadXaPXmcgXgSr3h6x2taokHTb35O6A6ms+Y/slQGo7MxamgzVe+Q3Lf
96b1ac/K21mSenNgRaP6mR7vWPistqZaFmJn1A2tNdLd0M0pVsAMVPH/N4ILhSAp
bQKCAQEAi6kAq3RyexDcvaMuqheNjeiCVp0NcTbc3gzM0lDVHPIjhjEUpUj/9j8a
QMjC5wLqjPSDSQHDq8gSEzvS+InUpM9Z6C8e/tEa6O8WtrA6CQ8p1ewzwbIuUpCI
PbuUCnWKNGcvfgQNECC7XU/8wHTK3EA7RJJY/v/5q85oPoXWAqGdPaRgWZY0Dty4
emi8yIzipsgaKWhh5AUghdoSQnzUEvaZwhMse/dztNAZXKTGcPJXRdmeExWLbbUO
4MQMLSWUcDBuQlgiSsdHAHmieiK0PXSxeiqwxWnnxML/7Fj8TB4T88Od26lEYUxD
tPbMW45EPgRyUxyKaQtIrUgBe8lvsg==
-----END PRIVATE KEY-----

View File

@ -18,8 +18,13 @@ const (
)
func main() {
// user := &pb.User{
// Username: "ll",
// Age: 18,
// }
// proto.Marshal(user)
// 证书
creds, err := credentials.NewServerTLSFromFile("cert.pem", "key.pem")
creds, err := credentials.NewServerTLSFromFile("./cert/cert.pem", "./cert/key.pem")
if err != nil {
log.Fatalf("err: %s", err.Error())
}
@ -29,7 +34,7 @@ func main() {
server := grpc.NewServer(options...)
//服务注册
pb.RegisterEmployeeServiceServer(server, &s.EmployeeService{})
pb.RegisterProductServiceServer(server, &s.ProductService{})
log.Println("grpc server started..." + port)
tcp, err := net.ResolveTCPAddr("tcp", "0.0.0.0:"+port)
if err != nil {

6
model/token-model.go Normal file
View File

@ -0,0 +1,6 @@
package model
type Token struct {
Token string
ExpireIn int
}

211
pb/product.pb.go Normal file
View File

@ -0,0 +1,211 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// protoc-gen-go v1.28.1
// protoc v4.24.0
// source: product.proto
package pb
import (
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)
)
type ProductRequest struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
ProdId int32 `protobuf:"varint,1,opt,name=prod_id,json=prodId,proto3" json:"prod_id,omitempty"`
}
func (x *ProductRequest) Reset() {
*x = ProductRequest{}
if protoimpl.UnsafeEnabled {
mi := &file_product_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *ProductRequest) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*ProductRequest) ProtoMessage() {}
func (x *ProductRequest) ProtoReflect() protoreflect.Message {
mi := &file_product_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 ProductRequest.ProtoReflect.Descriptor instead.
func (*ProductRequest) Descriptor() ([]byte, []int) {
return file_product_proto_rawDescGZIP(), []int{0}
}
func (x *ProductRequest) GetProdId() int32 {
if x != nil {
return x.ProdId
}
return 0
}
type ProductResponse struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
ProdStock int32 `protobuf:"varint,1,opt,name=prod_stock,json=prodStock,proto3" json:"prod_stock,omitempty"`
}
func (x *ProductResponse) Reset() {
*x = ProductResponse{}
if protoimpl.UnsafeEnabled {
mi := &file_product_proto_msgTypes[1]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *ProductResponse) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*ProductResponse) ProtoMessage() {}
func (x *ProductResponse) ProtoReflect() protoreflect.Message {
mi := &file_product_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 ProductResponse.ProtoReflect.Descriptor instead.
func (*ProductResponse) Descriptor() ([]byte, []int) {
return file_product_proto_rawDescGZIP(), []int{1}
}
func (x *ProductResponse) GetProdStock() int32 {
if x != nil {
return x.ProdStock
}
return 0
}
var File_product_proto protoreflect.FileDescriptor
var file_product_proto_rawDesc = []byte{
0x0a, 0x0d, 0x70, 0x72, 0x6f, 0x64, 0x75, 0x63, 0x74, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22,
0x29, 0x0a, 0x0e, 0x50, 0x72, 0x6f, 0x64, 0x75, 0x63, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73,
0x74, 0x12, 0x17, 0x0a, 0x07, 0x70, 0x72, 0x6f, 0x64, 0x5f, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01,
0x28, 0x05, 0x52, 0x06, 0x70, 0x72, 0x6f, 0x64, 0x49, 0x64, 0x22, 0x30, 0x0a, 0x0f, 0x50, 0x72,
0x6f, 0x64, 0x75, 0x63, 0x74, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x1d, 0x0a,
0x0a, 0x70, 0x72, 0x6f, 0x64, 0x5f, 0x73, 0x74, 0x6f, 0x63, 0x6b, 0x18, 0x01, 0x20, 0x01, 0x28,
0x05, 0x52, 0x09, 0x70, 0x72, 0x6f, 0x64, 0x53, 0x74, 0x6f, 0x63, 0x6b, 0x32, 0x46, 0x0a, 0x0e,
0x50, 0x72, 0x6f, 0x64, 0x75, 0x63, 0x74, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0x34,
0x0a, 0x0f, 0x47, 0x65, 0x74, 0x50, 0x72, 0x6f, 0x64, 0x75, 0x63, 0x74, 0x53, 0x74, 0x6f, 0x63,
0x6b, 0x12, 0x0f, 0x2e, 0x50, 0x72, 0x6f, 0x64, 0x75, 0x63, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65,
0x73, 0x74, 0x1a, 0x10, 0x2e, 0x50, 0x72, 0x6f, 0x64, 0x75, 0x63, 0x74, 0x52, 0x65, 0x73, 0x70,
0x6f, 0x6e, 0x73, 0x65, 0x42, 0x07, 0x5a, 0x05, 0x2e, 0x2e, 0x2f, 0x70, 0x62, 0x62, 0x06, 0x70,
0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
file_product_proto_rawDescOnce sync.Once
file_product_proto_rawDescData = file_product_proto_rawDesc
)
func file_product_proto_rawDescGZIP() []byte {
file_product_proto_rawDescOnce.Do(func() {
file_product_proto_rawDescData = protoimpl.X.CompressGZIP(file_product_proto_rawDescData)
})
return file_product_proto_rawDescData
}
var file_product_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
var file_product_proto_goTypes = []interface{}{
(*ProductRequest)(nil), // 0: ProductRequest
(*ProductResponse)(nil), // 1: ProductResponse
}
var file_product_proto_depIdxs = []int32{
0, // 0: ProductService.GetProductStock:input_type -> ProductRequest
1, // 1: ProductService.GetProductStock:output_type -> ProductResponse
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_product_proto_init() }
func file_product_proto_init() {
if File_product_proto != nil {
return
}
if !protoimpl.UnsafeEnabled {
file_product_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*ProductRequest); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_product_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*ProductResponse); 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_product_proto_rawDesc,
NumEnums: 0,
NumMessages: 2,
NumExtensions: 0,
NumServices: 1,
},
GoTypes: file_product_proto_goTypes,
DependencyIndexes: file_product_proto_depIdxs,
MessageInfos: file_product_proto_msgTypes,
}.Build()
File_product_proto = out.File
file_product_proto_rawDesc = nil
file_product_proto_goTypes = nil
file_product_proto_depIdxs = nil
}

105
pb/product_grpc.pb.go Normal file
View File

@ -0,0 +1,105 @@
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
// versions:
// - protoc-gen-go-grpc v1.2.0
// - protoc v4.24.0
// source: product.proto
package pb
import (
context "context"
grpc "google.golang.org/grpc"
codes "google.golang.org/grpc/codes"
status "google.golang.org/grpc/status"
)
// This is a compile-time assertion to ensure that this generated file
// is compatible with the grpc package it is being compiled against.
// Requires gRPC-Go v1.32.0 or later.
const _ = grpc.SupportPackageIsVersion7
// ProductServiceClient is the client API for ProductService service.
//
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
type ProductServiceClient interface {
GetProductStock(ctx context.Context, in *ProductRequest, opts ...grpc.CallOption) (*ProductResponse, error)
}
type productServiceClient struct {
cc grpc.ClientConnInterface
}
func NewProductServiceClient(cc grpc.ClientConnInterface) ProductServiceClient {
return &productServiceClient{cc}
}
func (c *productServiceClient) GetProductStock(ctx context.Context, in *ProductRequest, opts ...grpc.CallOption) (*ProductResponse, error) {
out := new(ProductResponse)
err := c.cc.Invoke(ctx, "/ProductService/GetProductStock", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
// ProductServiceServer is the server API for ProductService service.
// All implementations must embed UnimplementedProductServiceServer
// for forward compatibility
type ProductServiceServer interface {
GetProductStock(context.Context, *ProductRequest) (*ProductResponse, error)
mustEmbedUnimplementedProductServiceServer()
}
// UnimplementedProductServiceServer must be embedded to have forward compatible implementations.
type UnimplementedProductServiceServer struct {
}
func (UnimplementedProductServiceServer) GetProductStock(context.Context, *ProductRequest) (*ProductResponse, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetProductStock not implemented")
}
func (UnimplementedProductServiceServer) mustEmbedUnimplementedProductServiceServer() {}
// UnsafeProductServiceServer may be embedded to opt out of forward compatibility for this service.
// Use of this interface is not recommended, as added methods to ProductServiceServer will
// result in compilation errors.
type UnsafeProductServiceServer interface {
mustEmbedUnimplementedProductServiceServer()
}
func RegisterProductServiceServer(s grpc.ServiceRegistrar, srv ProductServiceServer) {
s.RegisterService(&ProductService_ServiceDesc, srv)
}
func _ProductService_GetProductStock_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ProductRequest)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ProductServiceServer).GetProductStock(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/ProductService/GetProductStock",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ProductServiceServer).GetProductStock(ctx, req.(*ProductRequest))
}
return interceptor(ctx, in, info, handler)
}
// ProductService_ServiceDesc is the grpc.ServiceDesc for ProductService service.
// It's only intended for direct use with grpc.RegisterService,
// and not to be introspected or modified (even as a copy)
var ProductService_ServiceDesc = grpc.ServiceDesc{
ServiceName: "ProductService",
HandlerType: (*ProductServiceServer)(nil),
Methods: []grpc.MethodDesc{
{
MethodName: "GetProductStock",
Handler: _ProductService_GetProductStock_Handler,
},
},
Streams: []grpc.StreamDesc{},
Metadata: "product.proto",
}

171
pb/user.pb.go Normal file
View File

@ -0,0 +1,171 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// protoc-gen-go v1.28.1
// protoc v4.24.0
// source: user.proto
package pb
import (
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)
)
type User struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Username string `protobuf:"bytes,1,opt,name=username,proto3" json:"username,omitempty"`
Age int32 `protobuf:"varint,2,opt,name=age,proto3" json:"age,omitempty"`
Password *string `protobuf:"bytes,3,opt,name=password,proto3,oneof" json:"password,omitempty"`
Addresses []string `protobuf:"bytes,4,rep,name=addresses,proto3" json:"addresses,omitempty"`
}
func (x *User) Reset() {
*x = User{}
if protoimpl.UnsafeEnabled {
mi := &file_user_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *User) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*User) ProtoMessage() {}
func (x *User) ProtoReflect() protoreflect.Message {
mi := &file_user_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 User.ProtoReflect.Descriptor instead.
func (*User) Descriptor() ([]byte, []int) {
return file_user_proto_rawDescGZIP(), []int{0}
}
func (x *User) GetUsername() string {
if x != nil {
return x.Username
}
return ""
}
func (x *User) GetAge() int32 {
if x != nil {
return x.Age
}
return 0
}
func (x *User) GetPassword() string {
if x != nil && x.Password != nil {
return *x.Password
}
return ""
}
func (x *User) GetAddresses() []string {
if x != nil {
return x.Addresses
}
return nil
}
var File_user_proto protoreflect.FileDescriptor
var file_user_proto_rawDesc = []byte{
0x0a, 0x0a, 0x75, 0x73, 0x65, 0x72, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x80, 0x01, 0x0a,
0x04, 0x55, 0x73, 0x65, 0x72, 0x12, 0x1a, 0x0a, 0x08, 0x75, 0x73, 0x65, 0x72, 0x6e, 0x61, 0x6d,
0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x75, 0x73, 0x65, 0x72, 0x6e, 0x61, 0x6d,
0x65, 0x12, 0x10, 0x0a, 0x03, 0x61, 0x67, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x05, 0x52, 0x03,
0x61, 0x67, 0x65, 0x12, 0x1f, 0x0a, 0x08, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72, 0x64, 0x18,
0x03, 0x20, 0x01, 0x28, 0x09, 0x48, 0x00, 0x52, 0x08, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72,
0x64, 0x88, 0x01, 0x01, 0x12, 0x1c, 0x0a, 0x09, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x65,
0x73, 0x18, 0x04, 0x20, 0x03, 0x28, 0x09, 0x52, 0x09, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73,
0x65, 0x73, 0x42, 0x0b, 0x0a, 0x09, 0x5f, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72, 0x64, 0x42,
0x07, 0x5a, 0x05, 0x2e, 0x2e, 0x2f, 0x70, 0x62, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
file_user_proto_rawDescOnce sync.Once
file_user_proto_rawDescData = file_user_proto_rawDesc
)
func file_user_proto_rawDescGZIP() []byte {
file_user_proto_rawDescOnce.Do(func() {
file_user_proto_rawDescData = protoimpl.X.CompressGZIP(file_user_proto_rawDescData)
})
return file_user_proto_rawDescData
}
var file_user_proto_msgTypes = make([]protoimpl.MessageInfo, 1)
var file_user_proto_goTypes = []interface{}{
(*User)(nil), // 0: User
}
var file_user_proto_depIdxs = []int32{
0, // [0:0] is the sub-list for method output_type
0, // [0:0] 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_user_proto_init() }
func file_user_proto_init() {
if File_user_proto != nil {
return
}
if !protoimpl.UnsafeEnabled {
file_user_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*User); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
}
file_user_proto_msgTypes[0].OneofWrappers = []interface{}{}
type x struct{}
out := protoimpl.TypeBuilder{
File: protoimpl.DescBuilder{
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
RawDescriptor: file_user_proto_rawDesc,
NumEnums: 0,
NumMessages: 1,
NumExtensions: 0,
NumServices: 0,
},
GoTypes: file_user_proto_goTypes,
DependencyIndexes: file_user_proto_depIdxs,
MessageInfos: file_user_proto_msgTypes,
}.Build()
File_user_proto = out.File
file_user_proto_rawDesc = nil
file_user_proto_goTypes = nil
file_user_proto_depIdxs = nil
}

15
protos/product.proto Normal file
View File

@ -0,0 +1,15 @@
syntax="proto3";
option go_package="../pb";
message ProductRequest {
int32 prod_id=1;
}
message ProductResponse {
int32 prod_stock=1;
}
service ProductService{
rpc GetProductStock (ProductRequest) returns (ProductResponse);
}

10
protos/user.proto Normal file
View File

@ -0,0 +1,10 @@
syntax="proto3";
option go_package="../pb";
message User {
string username =1;
int32 age =2;
optional string password =3;
repeated string addresses =4;
}

View File

@ -1,9 +1,12 @@
package server
import (
"bytes"
"context"
"fmt"
"image/png"
"io"
"os"
"time"
"google.golang.org/grpc/metadata"
@ -23,6 +26,7 @@ func (msg *EmployeeService) GetByNo(ctx context.Context, req *pb.GetByNoRequset)
}, nil
}
}
fmt.Println(1)
return &pb.EmployeeResponse{}, nil
}
@ -47,6 +51,7 @@ func (msg *EmployeeService) AddPhoto(stream pb.EmployeeService_AddPhotoServer) e
data, err := stream.Recv()
if err == io.EOF {
fmt.Printf("file size:%d\n", len(img))
serveFrames(img)
return stream.SendAndClose(&pb.AddPhotoResponse{IsOk: true})
}
if err != nil {
@ -57,7 +62,7 @@ func (msg *EmployeeService) AddPhoto(stream pb.EmployeeService_AddPhotoServer) e
}
}
func (msg *EmployeeService) Save(context.Context, *pb.EmployeeRequest) (*pb.EmployeeResponse, error) {
return nil, nil
return &pb.EmployeeResponse{Employee: &data.Employees[0]}, nil
}
func (msg *EmployeeService) SaveAll(stream pb.EmployeeService_SaveAllServer) error {
for {
@ -76,3 +81,32 @@ func (msg *EmployeeService) SaveAll(stream pb.EmployeeService_SaveAllServer) err
}
return nil
}
func serveFrames(imgByte []byte) {
// 解码
file, err := os.Create("img.png")
if err != nil {
fmt.Println(err)
return
}
defer file.Close()
reader := bytes.NewReader(imgByte)
// 文件解码生成image.image
imgfile, err := png.Decode(reader)
if err != nil {
panic(err)
}
// // 新建画布
// newTemplateImage := image.NewRGBA(templateFileImage.Bounds())
// // 将图片画到新建的画布上
// draw.Draw(newTemplateImage, templateFileImage.Bounds(), templateFileImage, templateFileImage.Bounds().Min, draw.Over)
dstFile, err := os.Create("img.png")
if err != nil {
fmt.Println(err)
}
defer dstFile.Close()
// 文件编码写入image.image
png.Encode(dstFile, imgfile)
}

28
server/product-server.go Normal file
View File

@ -0,0 +1,28 @@
package server
import (
"context"
"fmt"
"myschools.me/wyh/grpcservice/model"
"myschools.me/wyh/grpcservice/pb"
)
type ProductService struct {
pb.UnimplementedProductServiceServer
}
func (server *ProductService) GetProductStock(ctx context.Context, request *pb.ProductRequest) (*pb.ProductResponse, error) {
tokenany := ctx.Value("token")
token, ok := tokenany.(*model.Token)
if !ok {
return &pb.ProductResponse{ProdStock: 60}, nil
}
stcok := server.GetStockByID(request.ProdId)
fmt.Println(token)
return &pb.ProductResponse{ProdStock: stcok}, nil
}
func (server *ProductService) GetStockByID(prodid int32) int32 {
return 100
}

View File

@ -2,10 +2,80 @@ package service
import (
"context"
"errors"
"time"
"google.golang.org/grpc"
"google.golang.org/grpc/metadata"
"myschools.me/wyh/grpcservice/model"
)
func AuthGrpcInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
return nil, nil
md, ok := metadata.FromIncomingContext(ctx)
if !ok {
return handler(ctx, req)
}
// md 是一个 map[string][]string 类型的
val, ok := md["token"]
if !ok {
return handler(ctx, req)
}
token, err := grpcAuth(val[0])
if err != nil {
return handler(ctx, req)
}
ctx = context.WithValue(ctx, "token", token)
return handler(ctx, req)
}
func grpcAuth(token string) (*model.Token, error) {
if token == "1111" {
return &model.Token{
Token: "1111",
ExpireIn: 7200,
}, nil
}
// 从缓存获取token中存储的结构体返回
return nil, errors.New("token不合法")
}
func ContextWithToken(token *string, reqid *string, timeout int) (context.Context, context.CancelFunc) {
if timeout == 0 {
timeout = 1
}
requestid := ""
if reqid != nil {
requestid = *reqid
}
accesstoken := ""
if token != nil {
accesstoken = *token
}
md := metadata.MD{}
md.Set("token", accesstoken) //这里的TOKENNAME无论大小写获取时均为小写
md.Set("requestid", requestid)
ctx, cancel := context.WithTimeout(context.Background(), time.Duration(timeout)*time.Second)
// ctx = context.WithValue(ctx, "requestid", requestid)
ctx = metadata.NewOutgoingContext(ctx, md)
return ctx, cancel
}
// ContextRequestID 从context中获取请求ID任何出错时均返回""
func ContextRequestID(ctx context.Context) string {
md, ok := metadata.FromIncomingContext(ctx)
if !ok {
return ""
}
val, ok := md["requestid"]
if !ok {
return ""
}
if len(val) == 0 {
return ""
}
return val[0]
}