Compare commits

181 Commits

Author SHA1 Message Date
60c95c2e0e IAP 라이브러리 추가 from kd-branch 2024-04-25 11:37:07 +09:00
0a65c6009e 세션 consumer에 Revoke 추가 2024-04-23 15:57:25 +09:00
fba613f83c 세션 provider, consumer 인터페이스 변경 2024-04-23 11:19:08 +09:00
4aa071441b 세션 중복 로그인 금지 2024-04-22 18:05:43 +09:00
564827dd9c ServiceCode제거 2024-03-29 09:06:15 +09:00
c7ad1301c1 httpapibroker CallByHeader 함수 추가 2024-03-28 12:28:38 +09:00
dff2bfca76 에러나도 Fatal로 죽이지는 않도록 수정 2024-03-05 10:49:52 +09:00
08238b22d2 ttl 업데이트 2024-02-21 12:49:38 +09:00
0ca764d6be 세션 무효화 처리 강화 2024-02-21 12:08:33 +09:00
0c5ddac9f5 프로세스 종료 안되는 문제 수정 2024-02-15 12:06:31 +09:00
c08a3abf83 prometheus metric 오류 처리 2024-02-15 12:06:20 +09:00
eb3926e575 conn 정리시 널 체크 2024-02-14 11:08:22 +09:00
16b43ec31d 로그 제거 2024-02-14 10:36:43 +09:00
bef684b3fe 잘못된 호출 수정 2024-02-13 15:06:46 +09:00
d8156a96ee config 파일 경로를 절대 경로로 만드는 로직 다시 제거 2024-02-13 15:06:40 +09:00
e0ab399a9c metric 등록 로직 수정 2024-02-13 15:05:59 +09:00
ce86e0bf72 errwithstatus 2024-02-06 13:39:11 +09:00
163f52a5c1 로그 제거 2024-02-04 21:43:55 +09:00
2fef6d4064 현재 Exe 경로와 join 2024-02-04 21:03:48 +09:00
c71a74762d 웹소켓 메시지 전송 채널 단일화 2024-02-01 18:28:59 +09:00
d2f94b502b WriteMessage를 WritePreparedMessage로 변경 2024-02-01 13:15:09 +09:00
b100148f54 WriteMessage를 WritePreparedMessage로 변경 2024-01-22 17:14:45 +09:00
7e8afcd2a9 쿠폰 이름 해시 로직 변경 2024-01-22 16:51:10 +09:00
9ebfdfd049 음성채팅 중단 수정 2024-01-09 20:28:24 +09:00
4bd537bd96 음성채팅에 mid 대신 alias로 로그인 2024-01-09 19:42:37 +09:00
2439c0ef68 다음 틱 토큰 받아올때 크래시 수정 2024-01-09 09:43:05 +09:00
e03bf6e199 http request대신 직접 호출 2024-01-08 21:41:53 +09:00
e919473fa2 패키지 이름 변경 2024-01-08 20:26:04 +09:00
4098812556 voice chat을 gocommon으로 옮김 2024-01-08 20:22:30 +09:00
fdb9830fed 오타 수정 2023-12-28 16:55:06 +09:00
45883436c5 세션 무효화시 접속 종료 처리 2023-12-28 10:57:42 +09:00
d1e892d449 생성 삭제 채널을 밖으로 뺌 2023-12-26 00:18:15 +09:00
e42c1969a2 upgrade noauth 처리기 수정 2023-12-25 22:41:48 +09:00
29dcca9725 deleteChannel 오류 수정 2023-12-25 22:23:46 +09:00
46f7d358ed 세션 해제 콜백 추가 2023-12-25 22:06:57 +09:00
08802176cb 로그 변경
peer handler 코드 정리
2023-12-23 17:31:41 +09:00
00f4cab992 ws peer 를 제네릭으로 변경 2023-12-20 10:02:49 +09:00
b2ee40249a 테스트 코드 제거 2023-12-19 21:21:46 +09:00
a7d0384413 response 사이즈 오류 수정 2023-12-19 21:02:15 +09:00
99346fe7b2 size 리딩 오류 수정 2023-12-19 20:58:47 +09:00
51b2a48894 size 헤더 두번 읽는 오류 수정 2023-12-19 20:42:08 +09:00
c9ac5980a5 nonce 응답 추가 2023-12-19 17:04:14 +09:00
17593f74c6 alias대신 nonce를 받는 peer handler 2023-12-19 15:53:04 +09:00
43d3e8cbff wahandler peer 파일 분리 2023-12-18 23:22:49 +09:00
ef75de4f55 wshandler peer 추가 2023-12-18 22:21:40 +09:00
3e1febab56 ShrinkSliceAt 추가 2023-12-02 20:30:43 +09:00
1fd3e86ed3 빈 슬라이스 처리 2023-12-02 20:09:52 +09:00
137dbac273 FindONeInSlice 수정 2023-12-02 19:56:37 +09:00
0182ea7b01 FindOneInSlice 추가 2023-12-02 01:25:12 +09:00
51a1e352bc slice shrink 추가 2023-12-01 19:34:49 +09:00
188d310b42 metric.Export 인터페이스 변경 2023-11-30 14:47:31 +09:00
2260de6adb MakeRPCFail 추가 2023-11-29 22:22:05 +09:00
9b8757acda metric 해시 계산 로직 수정 2023-11-29 18:17:21 +09:00
4089ac0bc1 로그 제거 2023-11-29 17:35:05 +09:00
5787a384f9 같은 메트릭은 등록 패스 2023-11-28 22:40:48 +09:00
e5a5240f96 metric 등록 방법 변경 2023-11-28 22:29:52 +09:00
d33fada0a1 metric 키 계산 수정 2023-11-28 01:48:10 +09:00
9fba5d239d ccu 메트릭은 각 서비스로 분리 2023-11-28 01:18:30 +09:00
2c02116781 endian 변경 2023-11-27 23:50:09 +09:00
92711de4a1 location 레이블 변경 2023-11-26 18:42:05 +09:00
7e7d023252 인덱스 옵션 인자로 받음 2023-11-26 15:46:34 +09:00
f7ff7ae13b concurrent 메트릭에 location 레이블 추가 2023-11-26 15:46:22 +09:00
59b0a975f7 공통 메트릭 추가 2023-11-24 16:28:25 +09:00
d672b5dd90 metric exception 2023-11-23 20:08:57 +09:00
8adef2adb8 metric 사이즈 수정 2023-11-23 10:39:25 +09:00
9688731b3e metric head/tail 변경 2023-11-23 09:45:36 +09:00
64dece4f7e metricwriter nil 2023-11-22 23:41:49 +09:00
343754ba7f mongo ping go rountine 막음 2023-11-22 18:23:41 +09:00
6ad727c728 InternalServerAPI에 추가 header를 인자로 받도록 추가 2023-11-21 16:52:06 +09:00
49dabb35f0 proc/stat 분해 2023-11-20 17:06:53 +09:00
aa7a9661b9 로그 추가 2023-11-20 16:42:46 +09:00
c34cb485bd 메트릭 시작 로그 추가 2023-11-20 16:39:39 +09:00
a5d9809db7 metric 기본 타입을 int64로 2023-11-20 14:12:44 +09:00
6c5a82cc9b 메트릭 생성 파라미터 변경 2023-11-20 13:49:29 +09:00
72a62b678f metric 패키지 추가 2023-11-16 19:56:47 +09:00
770356d830 errWriter 제거하고 stdout으로 통일 2023-11-14 11:21:47 +09:00
81ce0dd9b6 웹 소켓 접속 종료 시그니쳐 변경 2023-11-10 16:40:42 +09:00
2e48ff0ca4 redis.Nil -> nil 2023-11-10 16:40:26 +09:00
60d0c19256 redis 접속 실패시 로그가 과도하게 남는 문제 수정 2023-10-26 16:42:17 +09:00
ebd1a89a58 http api broker call 변경 2023-10-26 11:02:45 +09:00
4fd7d303bf 로그 추가 2023-10-18 15:32:55 +09:00
0f30243725 JSONArrAppend 수정 2023-10-16 10:01:53 +09:00
0fb806bfda JSONArrAppend 호출 오류 수정 2023-10-12 12:03:11 +09:00
9c3d761d97 CallInternalServiceAPI 를 gocommon 으로 이동 2023-10-05 12:08:50 +09:00
06ca411e03 ErrorWithStatus 추가 2023-10-04 18:03:19 +09:00
dc54082a42 redis.Nil 처리 추가 2023-10-04 17:30:33 +09:00
3f29e0be86 로그 제거 2023-09-27 15:36:22 +09:00
2cec794483 로그 변경 2023-09-21 13:44:39 +09:00
49a3722a7e http, websocket api handler 추가 2023-09-19 18:50:17 +09:00
1af5d72819 json, gob 둘다 지원 2023-09-11 12:45:15 +09:00
23231dc6d7 room과 conn의 상호 의존성 제거 2023-09-08 18:19:16 +09:00
46ce5f0989 안쓰는 코드 정리 및 bson을 json으로 통일. 추후 gob으로 변경 예정 2023-09-08 15:26:30 +09:00
3603c0386b gob.Register 누락 수정 2023-09-08 11:50:07 +09:00
6cbf32c386 코드 정리 및 websocket도 http와 비슷하게 api handler로 통일 2023-09-08 11:35:57 +09:00
eb54fa2e3a Revert "NewWebsocketHandler에 redisClient를 받음"
This reverts commit db3b0c8d08.
2023-09-06 23:20:24 +09:00
a7ff64fdbd redisClient에서 redisUri로 변경 2023-09-06 19:16:24 +09:00
db3b0c8d08 NewWebsocketHandler에 redisClient를 받음 2023-09-06 19:13:34 +09:00
25fef67c07 만료된 세션도 확인 필요 2023-09-06 12:41:49 +09:00
7951814f12 함수 이름을 인자로 받음 2023-09-05 16:43:34 +09:00
3a3839a461 ReadBsonDocumentFromBody로 교체 2023-09-05 15:02:06 +09:00
7489fa657a ReadJsonDocumentFromBody 추가 2023-09-05 14:42:12 +09:00
0762d9311b receiverName을 인자로 받음 2023-09-05 13:52:03 +09:00
57464c6cf0 타입 체크 수정 2023-09-05 12:45:40 +09:00
8877bf88c0 MakeHttpApiReceiverByValue 제거 2023-09-05 12:28:25 +09:00
def9494dd3 nil exception 수정 2023-09-05 12:06:36 +09:00
5117847740 MakeHttpApiReceiverByValue 추가 2023-09-05 11:56:53 +09:00
84f56dfc50 HttpApiHandlerContainer 추가 2023-09-05 11:43:42 +09:00
821dc5c639 session 인터페이스 수정 2023-09-04 14:35:33 +09:00
0eefb438a6 session provider에도 touch 추가 2023-09-04 11:56:56 +09:00
d396a35713 로그 제거 2023-09-04 09:54:40 +09:00
26e968f7c9 wshandler에서 직접 state를 쓰는 코드 제거 2023-09-01 11:29:51 +09:00
68e40d580f mongo dbname도 url에서 받음 2023-09-01 10:42:15 +09:00
9e9d91b5a3 오류 처리 추가 2023-08-31 21:16:19 +09:00
90502c3029 이름 다시 변경 2023-08-31 21:12:08 +09:00
a011588509 helper 함수 추가 2023-08-31 21:09:31 +09:00
319dbf5a54 이름 변경 2023-08-31 21:07:06 +09:00
888abef7c0 SessionConfig 추가 2023-08-31 21:05:08 +09:00
58de9a3f0c wshandler config 수정 2023-08-31 20:57:17 +09:00
57b518562e consumer, provider 생성 방법 통일 2023-08-31 20:39:00 +09:00
cde46e6a5f 세션키를 자체 발급하도록 수정 2023-08-31 14:33:08 +09:00
e88df26ed7 wshandler도 session.Consumer로 교체 2023-08-30 18:23:19 +09:00
021f183157 provider에도 Query func 추가 2023-08-30 16:35:22 +09:00
66a191f494 mongodb용 세션 추가 2023-08-30 15:43:26 +09:00
d8458662fd consumer/provider에 Authorization타입으로 고정 2023-08-30 13:38:13 +09:00
a5d66a0249 session 패키지 추가 2023-08-30 13:15:44 +09:00
e4527aa5b3 쿠폰 헬퍼 함수 추가 2023-08-25 10:55:01 +09:00
6bae78a282 noauth에 인증 패스 2023-08-23 23:02:14 +09:00
400c7f6443 JSONObjKeys 에서 result가 없을 때 크래시 방지 2023-08-23 22:44:14 +09:00
c34045e215 DropIndex 추가 2023-08-23 17:40:14 +09:00
2addec5adf server stop 추가 2023-08-22 11:45:31 +09:00
81d069cddf 필드 이름 변경 2023-08-14 21:41:52 +09:00
a5b7e11964 종료시 콜백 호출 2023-08-12 19:30:33 +09:00
4e69b5b9fb sender callback 등록 함수 추가 2023-08-12 19:26:16 +09:00
05478f54ae sender에 DisconnectedCallbacks추가 2023-08-12 19:18:22 +09:00
b88bcff389 게스트 타입 추가 2023-08-11 19:00:12 +09:00
a57de9715c 리턴타입 되돌림. 키는 하나여도 리턴은 여러개 가능 2023-08-10 15:35:16 +09:00
e4ac505928 JSONGetString, JSONGetInt64 리턴 타입 변경 2023-08-10 15:25:40 +09:00
b32858cb88 JSONNumIncrBy, JSONNumMultBy 리턴타입 변경 2023-08-10 15:04:46 +09:00
11aabe0216 JSONNumIncrBy 버그 수정 2023-08-10 13:57:12 +09:00
0f1aeccde0 종료시 wsconn.Close 호출 2023-08-10 11:19:02 +09:00
bdfc8082a2 entireAlias 다시 제거 2023-08-08 17:20:45 +09:00
1a7df89c47 다시 count 제거 2023-08-06 12:41:47 +09:00
6f9f791f02 room 인원을 count에 저장 2023-08-06 12:28:11 +09:00
d6738b2b70 alias만으로도 메시지를 보낼 수 있다. 2023-08-06 11:49:17 +09:00
165d75a21f Update wshandler.go 2023-08-03 10:13:51 +09:00
f6441a9b4c html escape 비활성화 2023-08-02 16:21:34 +09:00
b501160efc nosession을 noauth로 교체 2023-08-01 14:17:47 +09:00
1dff16a86e sh mainloop도 waitgroup에 추가 2023-07-28 09:40:53 +09:00
8ea2a698ce 종료시 컨넥션 정리 2023-07-27 21:55:28 +09:00
854084d819 alias utf-8 변환 2023-07-27 21:31:18 +09:00
112e037f4d commandtype 이름 변경 2023-07-19 16:38:44 +09:00
e0e911f9e7 Squashed commit of the following:
commit 29b2f25850
Author: mountain <mountain@action2quare.com>
Date:   Wed Jul 19 09:33:37 2023 +0900

    타입 이름 변경

commit 256bfd030c
Author: mountain <mountain@action2quare.com>
Date:   Wed Jul 19 09:31:01 2023 +0900

    redison 추가

commit 72a683fed2
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 19:51:24 2023 +0900

    gob에 []any 추가

commit 89fa9e4ac5
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 17:45:12 2023 +0900

    write control 수정

commit d724cc84fa
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 17:38:04 2023 +0900

    redis pubsub 채널 이름에 디비 인덱스 추가

commit 8df248fa54
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 17:20:47 2023 +0900

    close를 writecontrol로 변경

commit 40a603522d
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 12:21:06 2023 +0900

    conn에 msg를 쓰는 쓰레드 단일화

commit c21017d2cd
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 11:08:38 2023 +0900

    redis call이 문제가 아니었음

commit 82abcddb49
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 11:04:15 2023 +0900

    잦은 redis call 회피

commit 289af24a8f
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 09:55:18 2023 +0900

    room create 메시지 전송

commit 4b35e0e638
Author: mountain <mountain@action2quare.com>
Date:   Tue Jul 18 09:45:27 2023 +0900

    EventReceiver 인터페이스 추가

commit 29843802ff
Author: mountain <mountain@action2quare.com>
Date:   Mon Jul 17 17:45:40 2023 +0900

    gob 등록

commit 66aea48fb7
Author: mountain <mountain@action2quare.com>
Date:   Sun Jul 16 18:39:11 2023 +0900

    채널간 publish marshalling을 gob으로 변경

commit 8f6c87a8ae
Author: mountain <mountain@action2quare.com>
Date:   Sun Jul 16 16:37:02 2023 +0900

    redis option을 copy로 변경

commit f0f459332d
Author: mountain <mountain@action2quare.com>
Date:   Sat Jul 15 17:08:33 2023 +0900

    wshandler에서 authcache제거하고 config 포맷 변경
2023-07-19 09:35:25 +09:00
269fa0f870 connection 도 room을 알고 있게 됨 2023-07-13 17:06:45 +09:00
522bd4a597 go version 1.20 올림 2023-07-13 15:40:12 +09:00
d48d4c0f21 Merge branch 'wshandler_mod' 2023-07-11 17:41:12 +09:00
74829b93ac tag 추가 2023-07-11 15:26:13 +09:00
4acb81a20d message body를 any로 변경 2023-07-11 14:30:10 +09:00
9fd0dd00cb mesasge에 태그 추가 2023-07-11 12:57:57 +09:00
d7b26608df 타입 미스매치 수정 2023-07-11 12:31:35 +09:00
a42eb2888e commandMessage를 private으로 만듬 2023-07-11 09:56:04 +09:00
3bb985d0b6 MessageReceiver signature 숮어 2023-07-11 09:36:21 +09:00
a842845685 parameter를 좀더 명확하게 2023-07-10 14:30:24 +09:00
8598d59ab4 receiver 함수를 region 별로 분리 2023-07-10 13:04:45 +09:00
1f668586f2 wshandler reciever 함수 추가 수정 2023-07-10 12:50:27 +09:00
bfebc67eb7 테스트 함수 수정 2023-07-10 12:28:32 +09:00
b5a72aad05 채널이름을 모든 public 함수의 signature를 고려 2023-07-10 12:13:51 +09:00
c859aeb75f 채널이름 고정으로 계산 2023-07-10 11:43:52 +09:00
27a3f2f08c funcname 오류 수정 2023-07-10 11:12:45 +09:00
98efbc2875 callctx 수정 2023-07-10 10:53:49 +09:00
e912c88993 rpc.call helper 함수 추가 2023-07-10 10:29:16 +09:00
550374ef6f rpc 패키지 추가 2023-07-10 09:42:34 +09:00
b85d271efe SendCloseMessage 추가 2023-07-06 20:37:32 +09:00
ba746d03fa local channel로 먼저 message를 보내봄 2023-07-06 17:30:57 +09:00
20803c67ed SendUpstreamMessage 추가 2023-07-06 16:47:36 +09:00
5dc9d4dca4 소켓 메세지 수신 함수 등록 타이밍 수정 2023-07-06 14:29:53 +09:00
4bdd72152e receiver 이동 2023-07-06 14:18:11 +09:00
8c94fc6e29 소켓 종료 처리 2023-07-06 13:07:57 +09:00
2b54d69b9e receiver 타입 변경 2023-07-06 12:35:39 +09:00
7ef3e68d16 GetState 함수 추가 2023-07-06 00:01:45 +09:00
822681bf74 close func를 밖으로 옮김 2023-07-05 22:26:57 +09:00
edd2f7aab5 방에 메세지를 브로드캐스팅하는 기능 추가 2023-07-05 11:27:44 +09:00
31 changed files with 4414 additions and 579 deletions

View File

@ -20,7 +20,6 @@ import (
type Authinfo struct { type Authinfo struct {
Accid primitive.ObjectID `bson:"_id,omitempty" json:"_id,omitempty"` Accid primitive.ObjectID `bson:"_id,omitempty" json:"_id,omitempty"`
ServiceCode string `bson:"code" json:"code"`
Platform string Platform string
Uid string Uid string
Email string Email string
@ -30,7 +29,7 @@ type Authinfo struct {
} }
const ( const (
sessionSyncChannelName = "session-sync-channel2" sessionSyncChannelNamePrefix = "session-sync-channel2"
) )
type AuthinfoCell interface { type AuthinfoCell interface {
@ -99,6 +98,8 @@ func (ac *redisAuthCell) ToBytes() []byte {
func newAuthCollectionWithRedis(redisClient *redis.Client, subctx context.Context, maingateURL string, apiToken string) *AuthCollection { func newAuthCollectionWithRedis(redisClient *redis.Client, subctx context.Context, maingateURL string, apiToken string) *AuthCollection {
sessionTTL := int64(3600) sessionTTL := int64(3600)
ac := MakeAuthCollection(time.Duration(sessionTTL * int64(time.Second))) ac := MakeAuthCollection(time.Duration(sessionTTL * int64(time.Second)))
sessionSyncChannelName := fmt.Sprintf("%s-%d", sessionSyncChannelNamePrefix, redisClient.Options().DB)
pubsub := redisClient.Subscribe(subctx, sessionSyncChannelName) pubsub := redisClient.Subscribe(subctx, sessionSyncChannelName)
ctx, cancel := context.WithCancel(context.TODO()) ctx, cancel := context.WithCancel(context.TODO())
go func(ctx context.Context, sub *redis.PubSub, authCache *AuthCollection) { go func(ctx context.Context, sub *redis.PubSub, authCache *AuthCollection) {
@ -205,7 +206,7 @@ func (acg *AuthCollectionGlobal) Reload(context context.Context) error {
for r, url := range config.RegionStorage { for r, url := range config.RegionStorage {
if _, ok := oldval[r]; !ok { if _, ok := oldval[r]; !ok {
// 새로 생겼네 // 새로 생겼네
redisClient, err := NewRedisClient(url.Redis.URL, url.Redis.Offset["session"]) redisClient, err := NewRedisClient(url.Redis["session"])
if err != nil { if err != nil {
return err return err
} }
@ -228,7 +229,7 @@ func NewAuthCollectionGlobal(context context.Context, apiToken string) (AuthColl
output := make(map[string]*AuthCollection) output := make(map[string]*AuthCollection)
for region, url := range config.RegionStorage { for region, url := range config.RegionStorage {
redisClient, err := NewRedisClient(url.Redis.URL, url.Redis.Offset["session"]) redisClient, err := NewRedisClient(url.Redis["session"])
if err != nil { if err != nil {
return AuthCollectionGlobal{}, err return AuthCollectionGlobal{}, err
} }

View File

@ -18,8 +18,6 @@ import (
"repositories.action2quare.com/ayo/gocommon/flagx" "repositories.action2quare.com/ayo/gocommon/flagx"
"repositories.action2quare.com/ayo/gocommon/logger" "repositories.action2quare.com/ayo/gocommon/logger"
"go.mongodb.org/mongo-driver/bson"
) )
var linkupdate = flagx.String("updatelink", "", "") var linkupdate = flagx.String("updatelink", "", "")
@ -239,17 +237,3 @@ func ReplyUpdateComplete() {
// return (uint32(b[0]) << 0) | (uint32(b[1]) << 8) | (uint32(b[2]) << 16) | (uint32(b[3]) << 24) // return (uint32(b[0]) << 0) | (uint32(b[1]) << 8) | (uint32(b[2]) << 16) | (uint32(b[3]) << 24)
// } // }
type BsonMarshaler[T any] struct {
val T
}
func NewBsonMarshaler[T any](val T) *BsonMarshaler[T] {
return &BsonMarshaler[T]{
val: val,
}
}
func (m *BsonMarshaler[T]) MarshalBinary() (data []byte, err error) {
return bson.Marshal(m.val)
}

59
coupon/helper.go Normal file
View File

@ -0,0 +1,59 @@
package coupon
import (
"crypto/md5"
"encoding/binary"
"encoding/hex"
"math/rand"
"strings"
)
func DisolveCouponCode(code string) (round string, key string) {
var final []byte
for _, n := range strings.Split(code, "-") {
nb, err := hex.DecodeString(n)
if err != nil {
// 형식 오류
return "", ""
}
final = append(final, nb...)
}
if len(final) != 8 {
// 형식 오류
return "", ""
}
uid := final[4:]
left := binary.BigEndian.Uint16(uid[0:2])
right := binary.BigEndian.Uint16(uid[2:4])
final = final[0:4]
xor := binary.LittleEndian.Uint32(final)
roundhashnum := xor ^ (uint32(left) * uint32(right))
roundhash := make([]byte, 4)
binary.BigEndian.PutUint32(roundhash, roundhashnum)
round = hex.EncodeToString(roundhash)
key = hex.EncodeToString(uid)
return
}
func MakeCouponRoundHash(name string) (hash string, roundNumber uint32) {
m5 := md5.New()
m5.Write([]byte(name))
hashbt := m5.Sum(nil)
roundbt := make([]byte, 8)
copy(roundbt, hashbt[:8])
roundseed := int64(binary.BigEndian.Uint64(roundbt))
roundhash := make([]byte, 4)
rand.New(rand.NewSource(roundseed)).Read(roundhash)
roundNumber = binary.BigEndian.Uint32(roundhash)
hash = hex.EncodeToString(roundhash)
return
}

19
go.mod
View File

@ -1,29 +1,36 @@
module repositories.action2quare.com/ayo/gocommon module repositories.action2quare.com/ayo/gocommon
go 1.19 go 1.20
replace repositories.action2quare.com/ayo/gocommon => ./
require ( require (
github.com/go-redis/redis/v8 v8.11.5 github.com/go-redis/redis/v8 v8.11.5
github.com/golang-jwt/jwt v3.2.2+incompatible github.com/golang-jwt/jwt v3.2.2+incompatible
github.com/gorilla/websocket v1.5.0 github.com/gorilla/websocket v1.5.0
github.com/pires/go-proxyproto v0.7.0 github.com/pires/go-proxyproto v0.7.0
github.com/prometheus/client_golang v1.17.0
go.mongodb.org/mongo-driver v1.11.6 go.mongodb.org/mongo-driver v1.11.6
golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d
golang.org/x/text v0.3.7 golang.org/x/text v0.9.0
) )
require ( require (
github.com/cespare/xxhash/v2 v2.1.2 // indirect github.com/beorn7/perks v1.0.1 // indirect
github.com/cespare/xxhash/v2 v2.2.0 // indirect
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
github.com/golang/protobuf v1.5.3 // indirect
github.com/golang/snappy v0.0.1 // indirect github.com/golang/snappy v0.0.1 // indirect
github.com/klauspost/compress v1.13.6 // indirect github.com/klauspost/compress v1.13.6 // indirect
github.com/matttproud/golang_protobuf_extensions v1.0.4 // indirect
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe // indirect github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe // indirect
github.com/pkg/errors v0.9.1 // indirect github.com/pkg/errors v0.9.1 // indirect
github.com/prometheus/client_model v0.4.1-0.20230718164431-9a2bf3000d16 // indirect
github.com/prometheus/common v0.44.0 // indirect
github.com/prometheus/procfs v0.11.1 // indirect
github.com/xdg-go/pbkdf2 v1.0.0 // indirect github.com/xdg-go/pbkdf2 v1.0.0 // indirect
github.com/xdg-go/scram v1.1.1 // indirect github.com/xdg-go/scram v1.1.1 // indirect
github.com/xdg-go/stringprep v1.0.3 // indirect github.com/xdg-go/stringprep v1.0.3 // indirect
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c // indirect golang.org/x/sync v0.3.0 // indirect
golang.org/x/sys v0.11.0 // indirect
google.golang.org/protobuf v1.31.0 // indirect
) )

43
go.sum
View File

@ -1,5 +1,7 @@
github.com/cespare/xxhash/v2 v2.1.2 h1:YRXhKfTDauu4ajMg1TPgFO5jnlC2HCbmLXMcTG5cbYE= github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
@ -10,19 +12,26 @@ github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC
github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo= github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo=
github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY= github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I= github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4= github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4=
github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
github.com/google/go-cmp v0.5.2 h1:X2ev0eStA3AbceY54o37/0PQ/UWqKEiiO2dKL5OPaFM=
github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc= github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
github.com/klauspost/compress v1.13.6 h1:P76CopJELS0TiO2mebmnzgWaajssP/EszplttgQxcgc= github.com/klauspost/compress v1.13.6 h1:P76CopJELS0TiO2mebmnzgWaajssP/EszplttgQxcgc=
github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk= github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE= github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
github.com/matttproud/golang_protobuf_extensions v1.0.4 h1:mmDVorXM7PCGKw94cs5zkfA9PSy5pEvNWRP0ET0TIVo=
github.com/matttproud/golang_protobuf_extensions v1.0.4/go.mod h1:BSXmuO+STAnVfrANrmjBb36TMTDstsz7MSK+HVaYKv4=
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe h1:iruDEfMl2E6fbMZ9s0scYfZQ84/6SPL6zC8ACM2oIL0= github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe h1:iruDEfMl2E6fbMZ9s0scYfZQ84/6SPL6zC8ACM2oIL0=
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc= github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE= github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE=
@ -34,6 +43,15 @@ github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/prometheus/client_golang v1.17.0 h1:rl2sfwZMtSthVU752MqfjQozy7blglC+1SOtjMAMh+Q=
github.com/prometheus/client_golang v1.17.0/go.mod h1:VeL+gMmOAxkS2IqfCq0ZmHSL+LjWfWDUmp1mBz9JgUY=
github.com/prometheus/client_model v0.4.1-0.20230718164431-9a2bf3000d16 h1:v7DLqVdK4VrYkVD5diGdl4sxJurKJEMnODWRJlxV9oM=
github.com/prometheus/client_model v0.4.1-0.20230718164431-9a2bf3000d16/go.mod h1:oMQmHW1/JoDwqLtg57MGgP/Fb1CJEYF2imWWhWtMkYU=
github.com/prometheus/common v0.44.0 h1:+5BrQJwiBB9xsMygAB3TNvpQKOwlkc25LbISbrdOOfY=
github.com/prometheus/common v0.44.0/go.mod h1:ofAIvZbQ1e/nugmZGz4/qCb9Ap1VoSTIO7x0VV9VvuY=
github.com/prometheus/procfs v0.11.1 h1:xRC8Iq1yyca5ypa9n1EZnWZkt7dwcoRPQwX/5gwaUuI=
github.com/prometheus/procfs v0.11.1/go.mod h1:eesXgaPo1q7lBpVMoMy0ZOFTth9hBn4W/y0/p/ScXhY=
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.6.1 h1:hDPOHmpOpP40lSULcqw7IrRb/u7w6RpDC9399XyoNd0= github.com/stretchr/testify v1.6.1 h1:hDPOHmpOpP40lSULcqw7IrRb/u7w6RpDC9399XyoNd0=
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
@ -51,21 +69,28 @@ go.mongodb.org/mongo-driver v1.11.6 h1:XM7G6PjiGAO5betLF13BIa5TlLUUE3uJ/2Ox3Lz1K
go.mongodb.org/mongo-driver v1.11.6/go.mod h1:G9TgswdsWjX4tmDA5zfs2+6AEPpYJwqblyjsfuh8oXY= go.mongodb.org/mongo-driver v1.11.6/go.mod h1:G9TgswdsWjX4tmDA5zfs2+6AEPpYJwqblyjsfuh8oXY=
golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d h1:sK3txAijHtOK88l68nt020reeT1ZdKLIYetKl95FzVY= golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d h1:sK3txAijHtOK88l68nt020reeT1ZdKLIYetKl95FzVY=
golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4= golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2 h1:CIJ76btIcR3eFI5EgSo6k1qKw9KJexJuRLI9G7Hp5wE=
golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c h1:5KslGYwFpkhGh+Q16bwMP3cOontH8FOep7tGV86Y7SQ= golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M=
golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.3.0 h1:ftCYgMx6zT/asHUrPw8BLLscYtGznsLAnjq5RH9P66E=
golang.org/x/sync v0.3.0/go.mod h1:FU7BRWz2tNW+3quACPkgCx/L+uEAv1htQ0V83Z9Rj+Y=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e h1:fLOSk5Q00efkSvAm+4xcoXD+RRmLmmulPn5I3Y9F2EM= golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM=
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk=
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE=
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
google.golang.org/protobuf v1.31.0 h1:g0LDEJHgrBl9N9r17Ru3sqWhkIx2NB67okBHPwC7hs8=
google.golang.org/protobuf v1.31.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ= gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=

110
iap/galaxystore.go Normal file
View File

@ -0,0 +1,110 @@
package iap
import (
"encoding/json"
"errors"
"io/ioutil"
"net/http"
"strings"
)
type GalaxyStoreIAPVerifier struct {
ItemId string `json:"itemId"`
PaymentId string `json:"paymentId"`
OrderId string `json:"orderId"`
PackageName string `json:"packageName"`
ItemName string `json:"itemName"`
ItemDesc string `json:"itemDesc"`
PurchaseDate string `json:"purchaseDate"`
PaymentAmount string `json:"paymentAmount"`
Status string `json:"status"`
PaymentMethod string `json:"paymentMethod"`
Mode string `json:"mode"`
ConsumeYN string `json:"consumeYN"`
ConsumeDate string `json:"consumeDate"`
ConsumeDeviceModel string `json:"consumeDeviceModel"`
PassThroughParam string `json:"passThroughParam"`
CurrencyCode string `json:"currencyCode"`
CurrencyUnit string `json:"currencyUnit"`
VerificationLog string
}
func (verifer *GalaxyStoreIAPVerifier) Verify(purchaseId string, paymentId string, developerPayload string, itemId string, packageName string) (bool, error) {
client := &http.Client{}
req, err := http.NewRequest("GET", "https://iap.samsungapps.com/iap/v6/receipt?purchaseID="+purchaseId, nil)
if err != nil {
return false, err
}
req.Header.Add("Content-Type", "application/json")
resp, err := client.Do(req)
if err != nil {
return false, err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return false, err
}
json.Unmarshal(body, verifer)
verifer.VerificationLog = string(body)
if verifer.Status != "success" {
return false, errors.New("Status is wrong:" + verifer.Status)
}
if verifer.ItemId != itemId {
return false, errors.New("itemId is different")
}
if verifer.PackageName != packageName {
return false, errors.New("packageName is different")
}
if !strings.EqualFold(verifer.PassThroughParam, developerPayload) {
return false, errors.New("developerPayload is different : " + verifer.PassThroughParam + "/" + developerPayload)
}
if verifer.PaymentId != paymentId {
return false, errors.New("paymentId is different")
}
return true, nil
}
// Owned Item - mItemId: kd_currency_gold_75
// Owned Item - mItemName: Gold x75
// Owned Item - mItemPrice: 1300.000000
// Owned Item - mItemPriceString: ₩1,300
// Owned Item - mCurrencyUnit: ₩
// Owned Item - mCurrencyCode: KRW
// Owned Item - mItemDesc: 75 Gold
// Owned Item - mPaymentId: TPMTID20240208KR32371566
// Owned Item - mPurchaseId: 34c50c9677e8e837e7b7a6eaac37a385fb7e4b36437820825bc7d824647c6aa7
// Owned Item - mPurchaseDate: -2147483648
// Owned Item - mPassThroughParam: testpayload
// {
// "itemId": "kd_currency_gold_75",
// "paymentId": "TPMTID20240208KR32367386",
// "orderId": "P20240208KR32367386",
// "packageName": "com.yjmgames.kingdom.gal",
// "itemName": "Gold x75",
// "itemDesc": "75 Gold",
// "purchaseDate": "2024-02-08 02:33:24",
// "paymentAmount": "1300.0",
// "status": "success",
// "paymentMethod": "Credit Card",
// "mode": "TEST",
// "consumeYN": "Y",
// "consumeDate": "2024-02-08 02:52:10",
// "consumeDeviceModel": "SM-S918N",
// "passThroughParam": "testpayload",
// "currencyCode": "KRW",
// "currencyUnit": "₩"
// }

173
iap/iap.go Normal file
View File

@ -0,0 +1,173 @@
package iap
import (
"context"
b64 "encoding/base64"
"encoding/json"
"io/ioutil"
"net/http"
"os"
"go-ayo/logger"
"github.com/awa/go-iap/appstore/api"
"github.com/awa/go-iap/playstore"
)
var PlayStorePublicKey string
var PlayStorePackageName string
var PlayStoreClient *playstore.Client
type PlayStoreReceiptInfo struct {
OrderId string `json:"orderId"`
PackageName string `json:"packageName"`
ProductId string `json:"productId"`
PurchaseTime int `json:"purchaseTime"`
PurchaseState int `json:"purchaseState"`
PurchaseToken string `json:"purchaseToken"`
Quantity int `json:"quantity"`
Acknowledged bool `json:"acknowledged"`
}
func PlayStoreInit(packageName string, publickey string) {
jsonKey, err := ioutil.ReadFile("playstore-iap-kingdom.json")
if err != nil {
logger.Fatal(err)
return
}
PlayStoreClient, err = playstore.New(jsonKey)
if err != nil {
logger.Fatal(err)
return
}
PlayStorePublicKey = publickey
PlayStorePackageName = packageName
}
func PlayStoreVerifyReceipt(receptData string, signature string) (isVerified bool, receiptInfo PlayStoreReceiptInfo) {
var info PlayStoreReceiptInfo
uDec, _ := b64.URLEncoding.DecodeString(receptData)
bOk, err := playstore.VerifySignature(PlayStorePublicKey, []byte(uDec), signature)
if bOk == false || err != nil {
logger.Println("playstore - VerifySignature fail :" + err.Error())
return bOk, info
}
json.Unmarshal(uDec, &info)
return bOk, info
}
func PlayStoreVerifyProduct(info PlayStoreReceiptInfo) (bool, int, int) {
productPurchases, err := PlayStoreClient.VerifyProduct(context.Background(), PlayStorePackageName, info.ProductId, info.PurchaseToken)
if err != nil {
logger.Println("playstore - PlayStoreVerifyProduct fail :" + err.Error())
return false, -1, -1
}
// fmt.Println(productPurchases)
// fmt.Println("================")
// fmt.Println(productPurchases.PurchaseState)
// fmt.Println(productPurchases.ConsumptionState)
// fmt.Println("================")
return true, int(productPurchases.PurchaseState), int(productPurchases.ConsumptionState)
}
func PlayStoreConsumeProduct(info PlayStoreReceiptInfo) bool {
err := PlayStoreClient.ConsumeProduct(context.Background(), PlayStorePackageName, info.ProductId, info.PurchaseToken)
if err != nil {
logger.Error("playstore - PlayStoreConsumeProduct fail :" + err.Error())
return false
}
return true
}
var AppStoreClient *api.StoreClient
func AppStoreInit(keyPath string, configFilePath string) {
keyContentBytes, err := os.ReadFile(keyPath)
if err != nil {
logger.Fatal(err)
return
}
configBytes, err := os.ReadFile(configFilePath)
if err != nil {
logger.Fatal(err)
return
}
var configMap map[string]any
err = json.Unmarshal(configBytes, &configMap)
config := &api.StoreConfig{
KeyContent: keyContentBytes,
KeyID: configMap["KeyID"].(string),
BundleID: configMap["BundleID"].(string),
Issuer: configMap["Issuer"].(string),
Sandbox: configMap["Sandbox"].(bool),
}
AppStoreClient = api.NewStoreClient(config)
if config.Sandbox {
logger.Println("IPA - Appstore : Use Sandbox")
} else {
logger.Println("IPA - Appstore : Normal")
}
}
func AppStoreVerifyReceipt(transactionId string) (bool, *api.JWSTransaction) {
res, err := AppStoreClient.GetTransactionInfo(context.Background(), transactionId)
if err != nil {
logger.Error(err)
return false, nil
}
transaction, err := AppStoreClient.ParseSignedTransaction(res.SignedTransactionInfo)
if err != nil {
logger.Error(err)
return false, nil
}
if transaction.TransactionID != transactionId {
return false, nil
}
if len(transaction.OriginalTransactionId) == 0 {
return false, nil
}
return true, transaction
}
func AppStoreConsumeProduct(transaction *api.JWSTransaction) bool {
//https://developer.apple.com/documentation/appstoreserverapi/consumptionrequest
consumeReqBody := api.ConsumptionRequestBody{
AccountTenure: 0,
AppAccountToken: transaction.AppAccountToken,
ConsumptionStatus: 3,
CustomerConsented: true,
DeliveryStatus: 0,
LifetimeDollarsPurchased: 0,
LifetimeDollarsRefunded: 0,
Platform: 1,
PlayTime: 0,
SampleContentProvided: false,
UserStatus: 1,
}
statusCode, err := AppStoreClient.SendConsumptionInfo(context.Background(), transaction.OriginalTransactionId, consumeReqBody)
if statusCode != http.StatusAccepted {
return false
}
if err != nil {
logger.Error(err)
return false
}
return true
}

249
iap/onestore.go Normal file
View File

@ -0,0 +1,249 @@
package iap
import (
"bytes"
"encoding/json"
"errors"
"go-ayo/logger"
"io/ioutil"
"net/http"
"strings"
"sync"
"time"
)
// Ref https://onestore-dev.gitbook.io/dev/tools/tools/v21/06.-api-api-v7#accesstoken
const shouldGenerateAccessTokenSec = 600
type OneStoreIAPVerificationInfo struct {
ConsumptionState int `json:"consumptionState"`
DeveloperPayload string `json:"developerPayload"`
PurchaseState int `json:"purchaseState"`
PurchaseTime int64 `json:"purchaseTime"`
PurchaseId string `json:"purchaseId"`
AcknowledgeState int `json:"acknowledgeState"`
Quantity int `json:"quantity"`
}
type OneStoreIAPClientPurcharinfo struct {
OrderId string `json:"orderId"`
PackageName string `json:"packageName"`
ProductId string `json:"productId"`
PurchaseTime int64 `json:"purchaseTime"`
PurchaseId string `json:"purchaseId"`
PurchaseToken string `json:"purchaseToken"`
DeveloperPayload string `json:"developerPayload"`
Quantity int `json:"quantity"`
}
type OneStoreConfig struct {
OneClientID string
OneClientSecret string
}
type OneStoreTokenData struct {
AccessToken string `json:"access_token"`
TokenType string `json:"token_type,omitempty"`
ClientId string `json:"client_id,omitempty"`
ExpiresIn int `json:"expires_in,omitempty"`
Scope string `json:"scope,omitempty"`
Expiry time.Time
}
func (s *OnestoreAccessToken) HostURL() string {
if s.isSandBox {
//-- Sandbox
return "https://sbpp.onestore.co.kr"
} else {
//-- RealURL
return "https://iap-apis.onestore.net"
}
}
func (s *OnestoreAccessToken) GenerateNewToken() error {
var newToken OneStoreTokenData
//==
HostURL := s.HostURL()
data := []byte("grant_type=client_credentials&client_id=" + s.config.OneClientID + "&client_secret=" + s.config.OneClientSecret)
client := &http.Client{}
req, err := http.NewRequest("POST", HostURL+"/v7/oauth/token", bytes.NewBuffer(data))
if err != nil {
return err
}
req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
req.Header.Add("x-market-code", s.marketCode)
resp, err := client.Do(req)
if err != nil {
return err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return err
}
json.Unmarshal(body, &newToken)
newExpiredSec := (newToken.ExpiresIn - shouldGenerateAccessTokenSec)
if newExpiredSec < 0 {
newExpiredSec = 0
}
newToken.Expiry = time.Now().Round(0).Add(time.Duration(newExpiredSec) * time.Second)
s.data = &newToken
logger.Println("IAP - OneStore : Generate Access Token :", s.data)
return nil
}
type OnestoreAccessToken struct {
mu sync.Mutex // guards t
data *OneStoreTokenData
marketCode string
isSandBox bool
config *OneStoreConfig
}
func (t *OnestoreAccessToken) Valid() bool {
return t != nil && t.data != nil && t.data.AccessToken != "" && !t.expired()
}
func (t *OnestoreAccessToken) expired() bool {
return t.data.Expiry.Before(time.Now())
}
func (s *OnestoreAccessToken) Token() (string, error) {
s.mu.Lock()
defer s.mu.Unlock()
if s.Valid() {
return s.data.AccessToken, nil
}
//== Expire 되면 새로 얻어 오자..
err := s.GenerateNewToken()
if err != nil {
return "", err
}
return s.data.AccessToken, nil
}
type OnestoreIAPVerifier struct {
config OneStoreConfig
SandBox_OnestoreAccessToken_MKT_ONE OnestoreAccessToken
SandBox_OnestoreAccessToken_MKT_GLB OnestoreAccessToken
Live_OnestoreAccessToken_MKT_ONE OnestoreAccessToken
Live_OnestoreAccessToken_MKT_GLB OnestoreAccessToken
}
func (s *OnestoreIAPVerifier) Init(clientId string, clientSecret string) {
s.config.OneClientID = clientId
s.config.OneClientSecret = clientSecret
s.SandBox_OnestoreAccessToken_MKT_ONE.marketCode = "MKT_ONE"
s.SandBox_OnestoreAccessToken_MKT_GLB.marketCode = "MKT_GLB"
s.Live_OnestoreAccessToken_MKT_ONE.marketCode = "MKT_ONE"
s.Live_OnestoreAccessToken_MKT_GLB.marketCode = "MKT_GLB"
s.SandBox_OnestoreAccessToken_MKT_ONE.isSandBox = true
s.SandBox_OnestoreAccessToken_MKT_GLB.isSandBox = true
s.Live_OnestoreAccessToken_MKT_ONE.isSandBox = false
s.Live_OnestoreAccessToken_MKT_GLB.isSandBox = false
s.SandBox_OnestoreAccessToken_MKT_ONE.config = &s.config
s.SandBox_OnestoreAccessToken_MKT_GLB.config = &s.config
s.Live_OnestoreAccessToken_MKT_ONE.config = &s.config
s.Live_OnestoreAccessToken_MKT_GLB.config = &s.config
}
func (s *OnestoreIAPVerifier) Verify(marketCode string, isSandBox bool, packageName string, clientInfo OneStoreIAPClientPurcharinfo) (bool, error) {
var accesstoken string
if marketCode != "MKT_ONE" && marketCode != "MKT_GLB" {
return false, errors.New("marketCode is wrong")
}
if clientInfo.ProductId == "" {
return false, errors.New("productId is empty")
}
if clientInfo.PurchaseToken == "" {
return false, errors.New("purchaseToken is empty")
}
if packageName != clientInfo.PackageName {
return false, errors.New("packageName is wrong")
}
var err error
var HostURL string
if marketCode == "MKT_ONE" {
if isSandBox {
accesstoken, err = s.SandBox_OnestoreAccessToken_MKT_ONE.Token()
HostURL = s.SandBox_OnestoreAccessToken_MKT_ONE.HostURL()
} else {
accesstoken, err = s.Live_OnestoreAccessToken_MKT_ONE.Token()
HostURL = s.Live_OnestoreAccessToken_MKT_ONE.HostURL()
}
} else {
if isSandBox {
accesstoken, err = s.SandBox_OnestoreAccessToken_MKT_GLB.Token()
HostURL = s.SandBox_OnestoreAccessToken_MKT_GLB.HostURL()
} else {
accesstoken, err = s.Live_OnestoreAccessToken_MKT_GLB.Token()
HostURL = s.Live_OnestoreAccessToken_MKT_GLB.HostURL()
}
}
if err != nil {
return false, err
}
client := &http.Client{}
req, err := http.NewRequest("GET", HostURL+"/v7/apps/"+s.config.OneClientID+"/purchases/inapp/products/"+clientInfo.ProductId+"/"+clientInfo.PurchaseToken, nil)
if err != nil {
return false, err
}
req.Header.Add("Authorization", "Bearer "+accesstoken)
req.Header.Add("Content-Type", "application/json")
req.Header.Add("x-market-code", marketCode)
resp, err := client.Do(req)
if err != nil {
return false, err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return false, err
}
var verificatioinfo OneStoreIAPVerificationInfo
json.Unmarshal(body, &verificatioinfo)
// fmt.Println("================")
// fmt.Println(string(body))
// fmt.Println("================")
if verificatioinfo.PurchaseState != 0 {
return false, errors.New("canceled payment")
}
if !strings.EqualFold(clientInfo.DeveloperPayload, verificatioinfo.DeveloperPayload) {
return false, errors.New("developerPayload is wrong :" + clientInfo.DeveloperPayload + "/" + verificatioinfo.DeveloperPayload)
}
if clientInfo.PurchaseId != verificatioinfo.PurchaseId {
return false, errors.New("purchaseId is wrong")
}
if clientInfo.PurchaseTime != verificatioinfo.PurchaseTime {
return false, errors.New("purchaseTime is wrong")
}
if clientInfo.Quantity != verificatioinfo.Quantity {
return false, errors.New("quantity is wrong")
}
return true, nil
}

View File

@ -16,7 +16,7 @@ var txSetArgs redis.SetArgs = redis.SetArgs{
} }
type LockerWithRedis struct { type LockerWithRedis struct {
key string locked_key string
} }
var ErrTransactionLocked = errors.New("transaction is already locked") var ErrTransactionLocked = errors.New("transaction is already locked")
@ -34,10 +34,10 @@ func (locker *LockerWithRedis) Lock(rc *redis.Client, key string) error {
return ErrTransactionLocked return ErrTransactionLocked
} }
locker.key = key locker.locked_key = key
return nil return nil
} }
func (locker *LockerWithRedis) Unlock(rc *redis.Client) { func (locker *LockerWithRedis) Unlock(rc *redis.Client) {
rc.Del(context.Background(), locker.key).Result() rc.Del(context.Background(), locker.locked_key).Result()
} }

View File

@ -7,12 +7,12 @@ import (
"log" "log"
"os" "os"
"path" "path"
"runtime"
"runtime/debug" "runtime/debug"
"strings" "strings"
) )
var stdlogger *log.Logger var stdlogger *log.Logger
var errlogger *log.Logger
var _ = flag.Bool("logfile", false, "") var _ = flag.Bool("logfile", false, "")
func init() { func init() {
@ -20,9 +20,7 @@ func init() {
binname := path.Base(strings.ReplaceAll(binpath, "\\", "/")) binname := path.Base(strings.ReplaceAll(binpath, "\\", "/"))
var outWriter io.Writer var outWriter io.Writer
var errWriter io.Writer
outWriter = os.Stdout outWriter = os.Stdout
errWriter = os.Stderr
args := os.Args args := os.Args
useLogFile := false useLogFile := false
@ -46,11 +44,9 @@ func init() {
} }
outWriter = io.MultiWriter(outWriter, logFile) outWriter = io.MultiWriter(outWriter, logFile)
errWriter = io.MultiWriter(errWriter, logFile)
} }
stdlogger = log.New(outWriter, "", log.LstdFlags) stdlogger = log.New(outWriter, "", log.LstdFlags)
errlogger = log.New(errWriter, "", log.LstdFlags)
} }
func Println(v ...interface{}) { func Println(v ...interface{}) {
@ -62,42 +58,83 @@ func Printf(format string, v ...interface{}) {
} }
func Error(v ...interface{}) { func Error(v ...interface{}) {
errlogger.Output(2, fmt.Sprintln(v...)) stdlogger.Output(2, fmt.Sprintln(v...))
errlogger.Output(2, string(debug.Stack())) stdlogger.Output(2, string(debug.Stack()))
} }
func Errorf(format string, v ...interface{}) { func Errorf(format string, v ...interface{}) {
errlogger.Output(2, fmt.Sprintf(format, v...)) stdlogger.Output(2, fmt.Sprintf(format, v...))
errlogger.Output(2, string(debug.Stack())) stdlogger.Output(2, string(debug.Stack()))
} }
func Fatal(v ...interface{}) { func Fatal(v ...interface{}) {
errlogger.Output(2, fmt.Sprint(v...)) stdlogger.Output(2, fmt.Sprint(v...))
errlogger.Output(2, string(debug.Stack())) stdlogger.Output(2, string(debug.Stack()))
os.Exit(1) os.Exit(1)
} }
func Fatalln(v ...interface{}) { func Fatalln(v ...interface{}) {
errlogger.Output(2, fmt.Sprintln(v...)) stdlogger.Output(2, fmt.Sprintln(v...))
errlogger.Output(2, string(debug.Stack())) stdlogger.Output(2, string(debug.Stack()))
os.Exit(1) os.Exit(1)
} }
func Panic(v ...interface{}) { func Panic(v ...interface{}) {
s := fmt.Sprint(v...) s := fmt.Sprint(v...)
errlogger.Output(2, s) stdlogger.Output(2, s)
errlogger.Output(2, string(debug.Stack())) stdlogger.Output(2, string(debug.Stack()))
panic(s) panic(s)
} }
func Panicf(format string, v ...interface{}) { func Panicf(format string, v ...interface{}) {
s := fmt.Sprintf(format, v...) s := fmt.Sprintf(format, v...)
errlogger.Output(2, s) stdlogger.Output(2, s)
errlogger.Output(2, string(debug.Stack())) stdlogger.Output(2, string(debug.Stack()))
panic(s) panic(s)
} }
func Panicln(v ...interface{}) { func Panicln(v ...interface{}) {
s := fmt.Sprintln(v...) s := fmt.Sprintln(v...)
errlogger.Output(2, s) stdlogger.Output(2, s)
errlogger.Output(2, string(debug.Stack())) stdlogger.Output(2, string(debug.Stack()))
panic(s) panic(s)
} }
type errWithCallstack struct {
inner error
frames []*runtime.Frame
}
func (ecs *errWithCallstack) Error() string {
if ecs.frames == nil {
return ecs.inner.Error()
}
out := make([]string, 0, len(ecs.frames)+1)
out = append(out, ecs.inner.Error())
for i := len(ecs.frames) - 1; i >= 0; i-- {
frame := ecs.frames[i]
out = append(out, fmt.Sprintf("%s\n\t%s:%d", frame.Function, frame.File, frame.Line))
}
return strings.Join(out, "\n")
}
func ErrorWithCallStack(err error) error {
var frames []*runtime.Frame
if recur, ok := err.(*errWithCallstack); ok {
err = recur.inner
frames = recur.frames
}
pc, _, _, ok := runtime.Caller(1)
if ok {
curframes := runtime.CallersFrames([]uintptr{pc})
f, _ := curframes.Next()
frames = append(frames, &f)
}
return &errWithCallstack{
inner: err,
frames: frames,
}
}

161
metric/common.go Normal file
View File

@ -0,0 +1,161 @@
package metric
import (
"crypto/md5"
"encoding/binary"
"encoding/hex"
"encoding/json"
"fmt"
"math"
"os"
"path"
"sort"
"strings"
"sync/atomic"
"repositories.action2quare.com/ayo/gocommon/logger"
)
const metric_value_line_size = 19
type MetricDescription struct {
Key string
Type MetricType
Name string `json:",omitempty"`
Help string `json:",omitempty"`
ConstLabels map[string]string `json:",omitempty"`
}
type Exporter interface {
RegisterMetric(*MetricDescription)
UpdateMetric(string, float64)
Shutdown()
}
type MetricWriter interface {
Add(int64)
Set(int64)
}
type metric_empty struct{}
func (mw *metric_empty) Set(int64) {}
func (mw *metric_empty) Add(int64) {}
var MetricWriterNil = MetricWriter(&metric_empty{})
type metric_int64 struct {
valptr *int64
buff [metric_value_line_size]byte
}
func (mw *metric_int64) printOut() {
binary.LittleEndian.PutUint64(mw.buff[9:], math.Float64bits(float64(atomic.LoadInt64(mw.valptr))))
os.Stdout.Write(mw.buff[:])
}
func (mw *metric_int64) Set(newval int64) {
atomic.StoreInt64(mw.valptr, newval)
mw.printOut()
}
func (mw *metric_int64) Add(inc int64) {
atomic.AddInt64(mw.valptr, inc)
mw.printOut()
}
func NewMetric(mt MetricType, name string, help string, constLabels map[string]string) (writer MetricWriter) {
if !metricEnabled {
return MetricWriterNil
}
var disorder []struct {
k string
v string
}
for k, v := range constLabels {
disorder = append(disorder, struct {
k string
v string
}{k: strings.ToLower(k), v: strings.ToLower(v)})
}
sort.Slice(disorder, func(i, j int) bool {
return disorder[i].k < disorder[j].k
})
hash := md5.New()
hash.Write([]byte(strings.ToLower(name)))
for _, d := range disorder {
hash.Write([]byte(d.k))
hash.Write([]byte(d.v))
}
key := hex.EncodeToString(hash.Sum(nil))[:metric_key_size]
temp, _ := json.Marshal(MetricDescription{
Key: key,
Type: mt,
Name: name,
Help: help,
ConstLabels: constLabels,
})
impl := &metric_int64{
valptr: new(int64),
}
impl.buff[0] = METRIC_HEAD_INLINE
impl.buff[17] = METRIC_TAIL_INLINE
impl.buff[18] = '\n'
copy(impl.buff[1:], []byte(key))
output := append([]byte{METRIC_HEAD_INLINE}, temp...)
output = append(output, METRIC_TAIL_INLINE, '\n')
os.Stdout.Write(output)
// writer
return impl
}
func ReadMetricValue(line []byte) (string, float64) {
if len(line) < 16 {
return "", 0
}
key := string(line[0:8])
valbits := binary.LittleEndian.Uint64(line[8:])
val := math.Float64frombits(valbits)
return key, val
}
var metricEnabled = false
func init() {
if path.Base(os.Args[0]) == "houston" {
logger.Println("metrics are going to be generated for myself(houston)")
metricEnabled = true
return
}
ppid := os.Getppid()
if parent, _ := os.FindProcess(ppid); parent != nil {
filename := fmt.Sprintf(`/proc/%d/stat`, os.Getppid())
if fn, err := os.ReadFile(filename); err == nil {
stats := strings.SplitN(string(fn), " ", 3)
parentname := strings.Trim(stats[1], "()")
if path.Base(parentname) == "houston" {
logger.Println("metrics are going to be generated for houston")
metricEnabled = true
} else {
logger.Println("metrics are NOT going to be generated. parent is not houston :", filename, string(fn))
}
} else {
logger.Println("metrics are NOT going to be generated. ppid proc is missing :", filename)
}
} else {
logger.Println("metrics are NOT going to be generated. parent process is missing. ppid :", ppid)
}
}

157
metric/prometheus.go Normal file
View File

@ -0,0 +1,157 @@
package metric
import (
"context"
"math"
"sync/atomic"
"github.com/prometheus/client_golang/prometheus"
"repositories.action2quare.com/ayo/gocommon/logger"
)
const (
METRIC_HEAD_INLINE = byte(14)
METRIC_TAIL_INLINE = byte(15)
)
type MetricType int
const (
MetricCounter = MetricType(1)
MetricGuage = MetricType(2)
metric_key_size = 8
)
func convertValueType(in MetricType) prometheus.ValueType {
switch in {
case MetricCounter:
return prometheus.CounterValue
case MetricGuage:
return prometheus.GaugeValue
}
return prometheus.UntypedValue
}
type writeRequest struct {
key string
val float64
}
type prometheusMetricDesc struct {
*prometheus.Desc
valueType prometheus.ValueType
valptr *uint64
key string
}
type prometheusExporter struct {
writerChan chan *writeRequest
registerChan chan *prometheusMetricDesc
namespace string
cancel context.CancelFunc
}
func (pe *prometheusExporter) RegisterMetric(nm *MetricDescription) {
pe.registerChan <- &prometheusMetricDesc{
Desc: prometheus.NewDesc(prometheus.BuildFQName(pe.namespace, "", nm.Name), nm.Help, nil, nm.ConstLabels),
valueType: convertValueType(nm.Type),
valptr: new(uint64),
key: nm.Key,
}
}
func (pe *prometheusExporter) UpdateMetric(key string, val float64) {
pe.writerChan <- &writeRequest{key: key, val: val}
}
func (pe *prometheusExporter) Shutdown() {
if pe.cancel != nil {
pe.cancel()
}
}
type prometheusCollector struct {
metrics map[string]*prometheusMetricDesc
}
func (pc *prometheusCollector) Describe(ch chan<- *prometheus.Desc) {
for _, v := range pc.metrics {
ch <- v.Desc
}
}
func (pc *prometheusCollector) Collect(ch chan<- prometheus.Metric) {
for _, v := range pc.metrics {
cm, err := prometheus.NewConstMetric(v.Desc, v.valueType, math.Float64frombits(atomic.LoadUint64(v.valptr)))
if err == nil {
ch <- cm
}
}
}
func (pe *prometheusExporter) loop(ctx context.Context) {
defer func() {
r := recover()
if r != nil {
logger.Error(r)
}
}()
var collector *prometheusCollector
defer func() {
if collector != nil {
prometheus.Unregister(collector)
}
}()
for {
select {
case <-ctx.Done():
return
case req := <-pe.writerChan:
if collector != nil {
if m := collector.metrics[req.key]; m != nil {
atomic.StoreUint64(m.valptr, math.Float64bits(req.val))
}
}
case nm := <-pe.registerChan:
var nextmetrics map[string]*prometheusMetricDesc
if collector != nil {
nextmetrics = collector.metrics
prometheus.Unregister(collector)
nextmetrics[nm.key] = nm
} else {
nextmetrics = map[string]*prometheusMetricDesc{
nm.key: nm,
}
}
nextcollector := &prometheusCollector{
metrics: nextmetrics,
}
if err := prometheus.Register(nextcollector); err != nil {
logger.Error("prometheus register err :", *nm, err)
} else {
collector = nextcollector
}
}
}
}
func NewPrometheusExport(namespace string) Exporter {
ctx, cancel := context.WithCancel(context.Background())
exp := &prometheusExporter{
registerChan: make(chan *prometheusMetricDesc, 10),
writerChan: make(chan *writeRequest, 100),
namespace: namespace,
cancel: cancel,
}
go exp.loop(ctx)
return exp
}

38
misc.go
View File

@ -103,3 +103,41 @@ func SerializeInterface(w io.Writer, val interface{}) (err error) {
return return
} }
func ShrinkSliceAt[T any](in []T, from int) []T {
if len(in) == 0 {
return in
}
cursor := from
for i := from + 1; i < len(in); i++ {
in[cursor] = in[i]
cursor++
}
return in[:len(in)-1]
}
func ShrinkSlice[T any](in []T, compare func(elem T) bool) []T {
if len(in) == 0 {
return in
}
cursor := 0
for i := 0; i < len(in); i++ {
if compare(in[i]) {
continue
}
in[cursor] = in[i]
cursor++
}
return in[:cursor]
}
func FindOneInSlice[T any](in []T, compare func(elem *T) bool) (int, bool) {
for i, e := range in {
if compare(&e) {
return i, true
}
}
return -1, false
}

View File

@ -15,6 +15,7 @@ import (
"go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options" "go.mongodb.org/mongo-driver/mongo/options"
"go.mongodb.org/mongo-driver/mongo/readpref" "go.mongodb.org/mongo-driver/mongo/readpref"
"go.mongodb.org/mongo-driver/x/mongo/driver/connstring"
) )
type MongoClient struct { type MongoClient struct {
@ -60,8 +61,19 @@ func (ci *ConnectionInfo) SetDatabase(dbname string) *ConnectionInfo {
return ci return ci
} }
func NewMongoClient(ctx context.Context, url string, dbname string) (MongoClient, error) { var errNoDatabaseNameInMongoUri = errors.New("mongo uri has no database name")
return newMongoClient(ctx, NewMongoConnectionInfo(url, dbname))
func NewMongoClient(ctx context.Context, url string) (MongoClient, error) {
connstr, err := connstring.ParseAndValidate(url)
if err != nil {
return MongoClient{}, err
}
if len(connstr.Database) == 0 {
return MongoClient{}, errNoDatabaseNameInMongoUri
}
return newMongoClient(ctx, NewMongoConnectionInfo(url, connstr.Database))
} }
func newMongoClient(ctx context.Context, ci *ConnectionInfo) (MongoClient, error) { func newMongoClient(ctx context.Context, ci *ConnectionInfo) (MongoClient, error) {
@ -94,21 +106,21 @@ func newMongoClient(ctx context.Context, ci *ConnectionInfo) (MongoClient, error
return MongoClient{}, err return MongoClient{}, err
} }
go func() { // go func() {
for { // for {
if err := client.Ping(ctx, nil); err != nil { // if err := client.Ping(ctx, nil); err != nil {
logger.Error("mongo client ping err :", err) // logger.Error("mongo client ping err :", err)
} // }
select { // select {
case <-time.After(10 * time.Second): // case <-time.After(10 * time.Second):
continue // continue
case <-ctx.Done(): // case <-ctx.Done():
return // return
} // }
} // }
}() // }()
mdb := client.Database(ci.Database, nil) mdb := client.Database(ci.Database, nil)
return MongoClient{c: client, db: mdb}, nil return MongoClient{c: client, db: mdb}, nil
@ -124,6 +136,18 @@ func (mc MongoClient) Close() {
} }
} }
func (mc MongoClient) DropIndex(coll CollectionName, name string) error {
matchcoll := mc.Collection(coll)
_, err := matchcoll.Indexes().DropOne(context.Background(), name)
if commanderr, ok := err.(mongo.CommandError); ok {
if commanderr.Code == 27 {
// 인덱스가 없는 것이므로 그냥 성공
return nil
}
}
return err
}
func (mc MongoClient) Watch(coll CollectionName, pipeline mongo.Pipeline, opts ...*options.ChangeStreamOptions) (*mongo.ChangeStream, error) { func (mc MongoClient) Watch(coll CollectionName, pipeline mongo.Pipeline, opts ...*options.ChangeStreamOptions) (*mongo.ChangeStream, error) {
if len(opts) == 0 { if len(opts) == 0 {
opts = []*options.ChangeStreamOptions{options.ChangeStream().SetFullDocument(options.UpdateLookup).SetMaxAwaitTime(0)} opts = []*options.ChangeStreamOptions{options.ChangeStream().SetFullDocument(options.UpdateLookup).SetMaxAwaitTime(0)}
@ -432,7 +456,7 @@ IndexSearchLabel:
return err return err
} }
func (mc MongoClient) makeIndicesWithOption(coll CollectionName, indices map[string]bson.D, option *options.IndexOptions) error { func (mc MongoClient) makeIndicesWithOption(coll CollectionName, indices map[string]bson.D, opts ...*options.IndexOptions) error {
collection := mc.Collection(coll) collection := mc.Collection(coll)
cursor, err := collection.Indexes().List(context.Background(), options.ListIndexes().SetMaxTime(time.Second)) cursor, err := collection.Indexes().List(context.Background(), options.ListIndexes().SetMaxTime(time.Second))
if err != nil { if err != nil {
@ -460,12 +484,12 @@ func (mc MongoClient) makeIndicesWithOption(coll CollectionName, indices map[str
if len(v) == 1 { if len(v) == 1 {
mod = mongo.IndexModel{ mod = mongo.IndexModel{
Keys: primitive.M{v[0].Key: v[0].Value}, Keys: primitive.M{v[0].Key: v[0].Value},
Options: options.MergeIndexOptions(options.Index().SetName(name), option), Options: options.MergeIndexOptions(append(opts, options.Index().SetName(name))...),
} }
} else { } else {
mod = mongo.IndexModel{ mod = mongo.IndexModel{
Keys: indices[name], Keys: indices[name],
Options: options.MergeIndexOptions(options.Index().SetName(name), option), Options: options.MergeIndexOptions(append(opts, options.Index().SetName(name))...),
} }
} }
@ -478,10 +502,10 @@ func (mc MongoClient) makeIndicesWithOption(coll CollectionName, indices map[str
return nil return nil
} }
func (mc MongoClient) MakeUniqueIndices(coll CollectionName, indices map[string]bson.D) error { func (mc MongoClient) MakeUniqueIndices(coll CollectionName, indices map[string]bson.D, opts ...*options.IndexOptions) error {
return mc.makeIndicesWithOption(coll, indices, options.Index().SetUnique(true)) return mc.makeIndicesWithOption(coll, indices, append(opts, options.Index().SetUnique(true))...)
} }
func (mc MongoClient) MakeIndices(coll CollectionName, indices map[string]bson.D) error { func (mc MongoClient) MakeIndices(coll CollectionName, indices map[string]bson.D, opts ...*options.IndexOptions) error {
return mc.makeIndicesWithOption(coll, indices, options.Index()) return mc.makeIndicesWithOption(coll, indices, opts...)
} }

439
redis.go
View File

@ -2,9 +2,11 @@ package gocommon
import ( import (
"context" "context"
"net/url" "encoding/json"
"fmt"
"os" "os"
"strconv" "strconv"
"strings"
"github.com/go-redis/redis/v8" "github.com/go-redis/redis/v8"
) )
@ -19,23 +21,30 @@ func newRedisClient(uri string, dbidxoffset int) *redis.Client {
return redis.NewClient(option) return redis.NewClient(option)
} }
func NewRedisClient(uri string, dbidx int) (*redis.Client, error) { func NewRedisClient(uri string) (*redis.Client, error) {
if !*devflag { if !*devflag {
return newRedisClient(uri, dbidx), nil return newRedisClient(uri, 0), nil
} }
rdb := newRedisClient(uri, 0) option, err := redis.ParseURL(uri)
devUrl, _ := url.Parse(uri) if err != nil {
return nil, err
}
zero := *option
zero.DB = 0
rdb := redis.NewClient(&zero)
defer rdb.Close()
hostname, _ := os.Hostname() hostname, _ := os.Hostname()
myidx, _ := rdb.HGet(context.Background(), "private_db", hostname).Result() myidx, _ := rdb.HGet(context.Background(), "private_db", hostname).Result()
if len(myidx) > 0 { if len(myidx) > 0 {
devUrl.Path = "/" + myidx offset, _ := strconv.Atoi(myidx)
return newRedisClient(devUrl.String(), dbidx), nil option.DB += offset
return redis.NewClient(option), nil
} }
alldbs, err := rdb.HGetAll(context.Background(), "private_db").Result() alldbs, err := rdb.HGetAll(context.Background(), "private_db").Result()
if err != nil { if err != nil {
rdb.Close()
return nil, err return nil, err
} }
@ -53,6 +62,416 @@ func NewRedisClient(uri string, dbidx int) (*redis.Client, error) {
return nil, err return nil, err
} }
devUrl.Path = "/" + strconv.Itoa(newidx) option.DB += newidx
return newRedisClient(devUrl.String(), dbidx), nil return redis.NewClient(option), nil
}
type RedisonSetOption = string
type RedisonGetOption = [2]any
const (
// JSONSET command Options
RedisonSetOptionNX RedisonSetOption = "NX"
RedisonSetOptionXX RedisonSetOption = "XX"
)
var (
RedisonGetOptionSPACE = RedisonGetOption{"SPACE", " "}
RedisonGetOptionINDENT = RedisonGetOption{"INDENT", "\t"}
RedisonGetOptionNEWLINE = RedisonGetOption{"NEWLINE", "\n"}
RedisonGetOptionNOESCAPE = RedisonGetOption{"NOESCAPE", ""}
)
// gocommon으로 옮길 거
type RedisonHandler struct {
*redis.Client
ctx context.Context
}
func NewRedisonHandler(ctx context.Context, redisClient *redis.Client) *RedisonHandler {
return &RedisonHandler{
Client: redisClient,
ctx: ctx,
}
}
func respToArray[T any](resp any, err error) ([]T, error) {
if err != nil {
if err == redis.Nil {
return nil, nil
}
return nil, err
}
resArr := resp.([]any)
v := make([]T, len(resArr))
for i, e := range resArr {
v[i] = e.(T)
}
return v, nil
}
func appendArgs[T any](args []any, ext ...T) []any {
for _, e := range ext {
args = append(args, e)
}
return args
}
func (rh *RedisonHandler) JSONMSetRel(key string, prefixPath string, kv map[string]any) error {
if len(prefixPath) > 0 && !strings.HasSuffix(prefixPath, ".") {
prefixPath += "."
}
pl := rh.Pipeline()
for path, obj := range kv {
b, err := json.Marshal(obj)
if err != nil {
return err
}
pl.Do(rh.ctx, "JSON.SET", key, prefixPath+path, b)
}
cmders, err := pl.Exec(rh.ctx)
if err != nil {
return err
}
for _, cmder := range cmders {
if cmder.Err() != nil {
return cmder.Err()
}
}
return nil
}
func (rh *RedisonHandler) JSONMSet(key string, kv map[string]any) error {
return rh.JSONMSetRel(key, "", kv)
}
func (rh *RedisonHandler) jsonSetMergeJSONSet(cmd, key, path string, obj any, opts ...RedisonSetOption) (bool, error) {
b, err := json.Marshal(obj)
if err != nil {
return false, err
}
args := []any{
"JSON.SET",
key,
path,
b,
}
if len(opts) > 0 {
args = append(args, opts[0])
}
res, err := rh.Do(rh.ctx, args...).Result()
if err != nil {
return false, err
}
return res.(string) == "OK", nil
}
func (rh *RedisonHandler) JSONSet(key, path string, obj any, opts ...RedisonSetOption) (bool, error) {
return rh.jsonSetMergeJSONSet("JSON.SET", key, path, obj, opts...)
}
func (rh *RedisonHandler) JSONMerge(key, path string, obj any, opts ...RedisonSetOption) (bool, error) {
return rh.jsonSetMergeJSONSet("JSON.MERGE", key, path, obj, opts...)
}
func (rh *RedisonHandler) JSONGet(key, path string, opts ...RedisonGetOption) (res any, err error) {
args := appendArgs[string]([]any{
"JSON.GET",
key,
}, strings.Split(path, " ")...)
for _, opt := range opts {
args = append(args, opt[:]...)
}
return rh.Do(rh.ctx, args...).Result()
}
func (rh *RedisonHandler) JSONGetString(key, path string) ([]string, error) {
return respToArray[string](rh.JSONResp(key, path))
}
func (rh *RedisonHandler) JSONGetDocuments(key, path string) ([]map[string]any, error) {
resp, err := rh.JSONGet(key, path)
if err != nil {
if err == redis.Nil {
return nil, nil
}
return nil, err
}
var objs []map[string]any
err = json.Unmarshal([]byte(resp.(string)), &objs)
return objs, err
}
func (rh *RedisonHandler) JSONGetInt64(key, path string) ([]int64, error) {
return respToArray[int64](rh.JSONResp(key, path))
}
func (rh *RedisonHandler) JSONMGet(path string, keys ...string) (res any, err error) {
args := appendArgs[string]([]any{
"JSON.MGET",
path,
}, keys...)
return rh.Do(rh.ctx, args...).Result()
}
func (rh *RedisonHandler) JSONMDel(key string, paths []string) error {
pl := rh.Pipeline()
for _, path := range paths {
args := []any{
"JSON.DEL",
key,
path,
}
pl.Do(rh.ctx, args...)
}
_, err := pl.Exec(rh.ctx)
return err
}
func (rh *RedisonHandler) JSONDel(key, path string) (int64, error) {
args := []any{
"JSON.DEL",
key,
path,
}
resp, err := rh.Do(rh.ctx, args...).Result()
if err != nil {
return 0, err
}
return resp.(int64), nil
}
func (rh *RedisonHandler) JSONType(key, path string) ([]string, error) {
args := []any{
"JSON.TYPE",
key,
path,
}
return respToArray[string](rh.Do(rh.ctx, args...).Result())
}
func (rh *RedisonHandler) JSONNumIncrBy(key, path string, number int) ([]int64, error) {
args := []any{
"JSON.NUMINCRBY",
key,
path,
number,
}
resp, err := rh.Do(rh.ctx, args...).Result()
if err != nil {
return nil, err
}
var cnts []int64
err = json.Unmarshal([]byte(resp.(string)), &cnts)
return cnts, err
}
func (rh *RedisonHandler) JSONNumMultBy(key, path string, number int) (res any, err error) {
args := []any{
"JSON.NUMMULTBY",
key,
path,
number,
}
resp, err := rh.Do(rh.ctx, args...).Result()
if err != nil {
return nil, err
}
return resp.([]any), nil
}
func (rh *RedisonHandler) JSONStrAppend(key, path string, jsonstring string) ([]int64, error) {
args := []any{
"JSON.STRAPPEND",
key,
path,
fmt.Sprintf(`'"%s"'`, jsonstring),
}
return respToArray[int64](rh.Do(rh.ctx, args...).Result())
}
func (rh *RedisonHandler) JSONStrLen(key, path string) (res []int64, err error) {
args := []any{
"JSON.STRLEN",
key,
path,
}
return respToArray[int64](rh.Do(rh.ctx, args...).Result())
}
func (rh *RedisonHandler) JSONArrAppend(key, path string, values ...any) (int64, error) {
args := appendValues([]any{
"JSON.ARRAPPEND",
key,
path,
}, values...)
resp, err := rh.Do(rh.ctx, args...).Result()
if err != nil {
return 0, err
}
return resp.(int64), nil
}
func (rh *RedisonHandler) JSONArrLen(key, path string) ([]int64, error) {
args := []any{
"JSON.ARRLEN",
key,
path,
}
return respToArray[int64](rh.Do(rh.ctx, args...).Result())
}
func (rh *RedisonHandler) JSONArrPop(key, path string, index int) (res any, err error) {
args := []any{
"JSON.ARRPOP",
key,
path,
index,
}
resp, err := rh.Do(rh.ctx, args...).Result()
if err != nil {
return nil, err
}
return resp.([]any)[0], nil
}
func appendValues(args []any, values ...any) []any {
for _, jsonValue := range values {
switch jsonValue := jsonValue.(type) {
case string:
args = append(args, fmt.Sprintf(`'"%s"'`, jsonValue))
case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64:
args = append(args, jsonValue)
default:
bt, _ := json.Marshal(jsonValue)
args = append(args, bt)
}
}
return args
}
func (rh *RedisonHandler) JSONArrIndex(key, path string, jsonValue any, optionalRange ...int) ([]int64, error) {
args := appendValues([]any{
"JSON.ARRINDEX",
key,
path,
}, jsonValue)
return respToArray[int64](rh.Do(rh.ctx, args...).Result())
}
func (rh *RedisonHandler) JSONArrTrim(key, path string, start, end int) (res any, err error) {
args := []any{
"JSON.ARRTRIM",
key,
path,
start,
end,
}
return respToArray[int64](rh.Do(rh.ctx, args...).Result())
}
func (rh *RedisonHandler) JSONArrInsert(key, path string, index int, values ...any) (res any, err error) {
args := appendValues([]any{
"JSON.ARRINSERT",
key,
path,
index,
}, values...)
return respToArray[int64](rh.Do(rh.ctx, args...).Result())
}
func (rh *RedisonHandler) JSONObjKeys(key, path string) ([]string, error) {
args := []any{
"JSON.OBJKEYS",
key,
path,
}
res, err := rh.Do(rh.ctx, args...).Result()
if err != nil {
return nil, err
}
resArr := res.([]any)
if len(resArr) == 0 {
return nil, nil
}
resArr = resArr[0].([]any)
slc := make([]string, len(resArr))
for i, r := range resArr {
slc[i] = r.(string)
}
return slc, nil
}
func (rh *RedisonHandler) JSONObjLen(key, path string) ([]int64, error) {
args := []any{
"JSON.OBJLEN",
key,
}
if path != "$" {
args = append(args, path)
}
resp, err := rh.Do(rh.ctx, args...).Result()
if err != nil {
return nil, err
}
switch resp := resp.(type) {
case []any:
return respToArray[int64](resp, nil)
case int64:
return []int64{resp}, nil
}
return []int64{0}, nil
}
func (rh *RedisonHandler) JSONDebug(key, path string) (res any, err error) {
args := []any{
"JSON.DEBUG",
"MEMORY",
key,
path,
}
return rh.Do(rh.ctx, args...).Result()
}
func (rh *RedisonHandler) JSONForget(key, path string) (int64, error) {
return rh.JSONDel(key, path)
}
func (rh *RedisonHandler) JSONResp(key, path string) (res any, err error) {
args := []any{
"JSON.RESP",
key,
path,
}
return rh.Do(rh.ctx, args...).Result()
} }

View File

@ -16,6 +16,11 @@ func configFilePath() string {
configfilepath = *configfileflag configfilepath = *configfileflag
} }
// if !strings.HasPrefix(configfilepath, "/") {
// exe, _ := os.Executable()
// configfilepath = path.Join(path.Dir(exe), configfilepath)
// }
return configfilepath return configfilepath
} }
@ -65,10 +70,7 @@ func LoadConfig[T any](outptr *T) error {
type StorageAddr struct { type StorageAddr struct {
Mongo string Mongo string
Redis struct { Redis map[string]string
URL string
Offset map[string]int
}
} }
type RegionStorageConfig struct { type RegionStorageConfig struct {

336
server.go
View File

@ -1,7 +1,9 @@
package gocommon package gocommon
import ( import (
"bytes"
"context" "context"
"encoding/gob"
"encoding/json" "encoding/json"
"errors" "errors"
"fmt" "fmt"
@ -20,6 +22,7 @@ import (
"sync/atomic" "sync/atomic"
"syscall" "syscall"
"time" "time"
"unsafe"
"repositories.action2quare.com/ayo/gocommon/flagx" "repositories.action2quare.com/ayo/gocommon/flagx"
"repositories.action2quare.com/ayo/gocommon/logger" "repositories.action2quare.com/ayo/gocommon/logger"
@ -28,6 +31,15 @@ import (
"go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/bson/primitive"
) )
func init() {
gob.Register(map[string]any{})
gob.Register(primitive.A{})
gob.Register(primitive.M{})
gob.Register(primitive.D{})
gob.Register(primitive.ObjectID{})
gob.Register([]any{})
}
const ( const (
// HTTPStatusReloginRequired : http status를 이걸 받으면 클라이언트는 로그아웃하고 로그인 화면으로 돌아가야 한다. // HTTPStatusReloginRequired : http status를 이걸 받으면 클라이언트는 로그아웃하고 로그인 화면으로 돌아가야 한다.
HTTPStatusReloginRequired = 599 HTTPStatusReloginRequired = 599
@ -46,6 +58,14 @@ const (
ShutdownFlagIdle = ShutdownFlag(3) ShutdownFlagIdle = ShutdownFlag(3)
) )
type ErrorWithStatus struct {
StatusCode int
}
func (e ErrorWithStatus) Error() string {
return fmt.Sprintf("%d", e.StatusCode)
}
type RpcReturnTypeInterface interface { type RpcReturnTypeInterface interface {
Value() any Value() any
Error() error Error() error
@ -97,24 +117,18 @@ func NewHTTPServerWithPort(serveMux *http.ServeMux, port int) *Server {
addr := fmt.Sprintf(":%d", port) addr := fmt.Sprintf(":%d", port)
serveMux.HandleFunc(MakeHttpHandlerPattern("welcome"), welcomeHandler) serveMux.HandleFunc(MakeHttpHandlerPattern("welcome"), welcomeHandler)
serveMux.HandleFunc(MakeHttpHandlerPattern("lb_health_chceck"), healthCheckHandler) serveMux.HandleFunc(MakeHttpHandlerPattern("lb_health_chceck"), healthCheckHandler)
serveMux.HandleFunc(MakeHttpHandlerPattern("lb_health_check"), healthCheckHandler)
server := &Server{ server := &Server{
httpserver: &http.Server{Addr: addr, Handler: serveMux}, httpserver: &http.Server{Addr: addr, Handler: serveMux},
interrupt: make(chan os.Signal, 1),
} }
server.httpserver.SetKeepAlivesEnabled(true) server.httpserver.SetKeepAlivesEnabled(true)
signal.Notify(server.interrupt, os.Interrupt, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
go func() {
c := <-server.interrupt
logger.Println("interrupt!!!!!!!! :", c.String())
server.shutdown()
}()
return server return server
} }
func NewHTTPServer(serveMux *http.ServeMux) *Server { func NewHTTPServer(serveMux *http.ServeMux) *Server {
// 시작시 자동으로 enable됨 // 시작시 자동으로 enable됨
if len(*tls) > 0 && *portptr == 80 { if len(*tls) > 0 && *portptr == 80 {
*portptr = 443 *portptr = 443
@ -153,6 +167,14 @@ func (server *Server) shutdown() {
} }
} }
func (server *Server) Stop() {
if server.interrupt != nil {
server.interrupt <- os.Interrupt
} else {
server.shutdown()
}
}
// Start : // Start :
func (server *Server) Start() error { func (server *Server) Start() error {
if server.httpserver != nil { if server.httpserver != nil {
@ -160,6 +182,15 @@ func (server *Server) Start() error {
if r != nil { if r != nil {
return r return r
} }
server.interrupt = make(chan os.Signal, 1)
signal.Notify(server.interrupt, os.Interrupt, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
go func() {
c := <-server.interrupt
logger.Println("interrupt!!!!!!!! :", c.String())
server.shutdown()
}()
proxyListener := &proxyproto.Listener{ proxyListener := &proxyproto.Listener{
Listener: ln, Listener: ln,
ReadHeaderTimeout: 10 * time.Second, ReadHeaderTimeout: 10 * time.Second,
@ -419,6 +450,46 @@ func ReadStringFormValue(r url.Values, key string) (string, bool) {
return strval, len(strval) > 0 return strval, len(strval) > 0
} }
type encoder interface {
Encode(any) error
}
type nilEncoder struct{}
func (ne *nilEncoder) Encode(any) error { return nil }
type decoder interface {
Decode(any) error
}
type nilDecoder struct{}
func (nd *nilDecoder) Decode(any) error { return nil }
func MakeDecoder(r *http.Request) decoder {
ct := r.Header.Get("Content-Type")
if ct == "application/gob" {
return gob.NewDecoder(r.Body)
} else if ct == "application/json" {
return json.NewDecoder(r.Body)
}
logger.Error("Content-Type is not supported :", ct)
return &nilDecoder{}
}
func MakeEncoder(w http.ResponseWriter, r *http.Request) encoder {
ct := r.Header.Get("Content-Type")
if ct == "application/gob" {
return gob.NewEncoder(w)
} else if ct == "application/json" {
return json.NewEncoder(w)
}
logger.Error("Content-Type is not supported :", ct)
return &nilEncoder{}
}
func DotStringToTimestamp(tv string) primitive.Timestamp { func DotStringToTimestamp(tv string) primitive.Timestamp {
if len(tv) == 0 { if len(tv) == 0 {
return primitive.Timestamp{T: 0, I: 0} return primitive.Timestamp{T: 0, I: 0}
@ -505,7 +576,11 @@ func (rt *RpcReturnError) WithCode(code int) *RpcReturnError {
} }
func (rt *RpcReturnError) WithError(err error) *RpcReturnError { func (rt *RpcReturnError) WithError(err error) *RpcReturnError {
rt.err = err if err2, ok := err.(*ErrorWithStatus); ok {
rt.WithCode(err2.StatusCode)
} else {
rt.err = err
}
return rt return rt
} }
@ -535,6 +610,14 @@ func MakeRPCReturn(value interface{}) *RpcReturnSimple {
} }
} }
func MakeRPCFail() *RpcReturnError {
return &RpcReturnError{
err: nil,
code: http.StatusInternalServerError,
h: nil,
}
}
func MakeRPCError() *RpcReturnError { func MakeRPCError() *RpcReturnError {
pc, _, _, ok := runtime.Caller(1) pc, _, _, ok := runtime.Caller(1)
if ok { if ok {
@ -597,3 +680,238 @@ func MakeHttpRequestForLogging(r *http.Request) *http.Request {
r.Body = ib r.Body = ib
return r return r
} }
type apiFuncType func(http.ResponseWriter, *http.Request)
type HttpApiHandler struct {
methods map[string]apiFuncType
originalReceiverName string
}
func MakeHttpApiHandler[T any](receiver *T, receiverName string) HttpApiHandler {
methods := make(map[string]apiFuncType)
tp := reflect.TypeOf(receiver)
if len(receiverName) == 0 {
receiverName = tp.Elem().Name()
}
writerType := reflect.TypeOf((*http.ResponseWriter)(nil)).Elem()
for i := 0; i < tp.NumMethod(); i++ {
method := tp.Method(i)
if method.Type.NumIn() != 3 {
continue
}
if method.Type.In(0) != tp {
continue
}
if !method.Type.In(1).Implements(writerType) {
continue
}
var r http.Request
if method.Type.In(2) != reflect.TypeOf(&r) {
continue
}
if method.Name == "ServeHTTP" {
continue
}
funcptr := method.Func.Pointer()
p1 := unsafe.Pointer(&funcptr)
p2 := unsafe.Pointer(&p1)
testfunc := (*func(*T, http.ResponseWriter, *http.Request))(p2)
methods[receiverName+"."+method.Name] = func(w http.ResponseWriter, r *http.Request) {
(*testfunc)(receiver, w, r)
}
}
return HttpApiHandler{
methods: methods,
originalReceiverName: tp.Elem().Name(),
}
}
type HttpApiBroker struct {
methods map[string]apiFuncType
methods_dup map[string][]apiFuncType
}
type bufferReadCloser struct {
*bytes.Reader
}
func (buff *bufferReadCloser) Close() error { return nil }
type readOnlyResponseWriter struct {
inner http.ResponseWriter
statusCode int
}
func (w *readOnlyResponseWriter) Header() http.Header {
return w.inner.Header()
}
func (w *readOnlyResponseWriter) Write(in []byte) (int, error) {
logger.Println("readOnlyResponseWriter cannot write")
return len(in), nil
}
func (w *readOnlyResponseWriter) WriteHeader(statusCode int) {
w.statusCode = statusCode
}
func (hc *HttpApiBroker) AddHandler(receiver HttpApiHandler) {
if hc.methods == nil {
hc.methods = make(map[string]apiFuncType)
hc.methods_dup = make(map[string][]apiFuncType)
}
for k, v := range receiver.methods {
ab := strings.Split(k, ".")
logger.Printf("http api registered : %s.%s -> %s\n", receiver.originalReceiverName, ab[1], k)
hc.methods_dup[k] = append(hc.methods_dup[k], v)
if len(hc.methods_dup[k]) > 1 {
chain := hc.methods_dup[k]
hc.methods[k] = func(w http.ResponseWriter, r *http.Request) {
body, _ := io.ReadAll(r.Body)
defer r.Body.Close()
wrap := &readOnlyResponseWriter{inner: w, statusCode: 200}
for _, f := range chain {
r.Body = &bufferReadCloser{bytes.NewReader(body)}
f(wrap, r)
}
if wrap.statusCode != 200 {
w.WriteHeader(wrap.statusCode)
}
}
} else {
hc.methods[k] = v
}
}
}
func (hc *HttpApiBroker) CallByHeader(w http.ResponseWriter, r *http.Request) {
funcname := r.Header.Get("AS-X-CALL")
if len(funcname) == 0 {
logger.Println("as-x-call header is missing")
w.WriteHeader(http.StatusBadRequest)
return
}
hc.call(funcname, w, r)
}
func (hc *HttpApiBroker) Call(w http.ResponseWriter, r *http.Request) {
funcname := r.URL.Query().Get("call")
if len(funcname) == 0 {
logger.Println("query param 'call' is missing")
w.WriteHeader(http.StatusBadRequest)
return
}
hc.call(funcname, w, r)
}
func (hc *HttpApiBroker) call(funcname string, w http.ResponseWriter, r *http.Request) {
if found := hc.methods[funcname]; found != nil {
found(w, r)
} else {
logger.Println("api is not found :", funcname)
}
}
func CallInternalServiceAPI[T any](url string, apitoken string, method string, data T, headers ...string) error {
tempHeader := make(http.Header)
tempHeader.Set("MG-X-API-TOKEN", apitoken)
tempHeader.Set("Content-Type", "application/gob")
for i := 1; i < len(headers); i += 2 {
tempHeader.Set(headers[i-1], headers[i])
}
buff := new(bytes.Buffer)
ct := tempHeader.Get("Content-Type")
if ct == "application/gob" {
enc := gob.NewEncoder(buff)
enc.Encode(data)
} else if ct == "application/json" {
enc := json.NewEncoder(buff)
enc.Encode(data)
}
reqURL := fmt.Sprintf("%s/api?call=%s", url, method)
req, err := http.NewRequest("POST", reqURL, buff)
if err != nil {
return err
}
req.Header = tempHeader
resp, err := http.DefaultClient.Do(req)
defer func() {
if resp != nil && resp.Body != nil {
resp.Body.Close()
}
}()
if err == nil {
if resp.StatusCode != http.StatusOK {
return &ErrorWithStatus{StatusCode: resp.StatusCode}
}
}
return err
}
func CallInternalServiceAPIAs[Tin any, Tout any](url string, apitoken string, method string, data Tin, out *Tout, headers ...string) error {
tempHeader := make(http.Header)
tempHeader.Set("MG-X-API-TOKEN", apitoken)
tempHeader.Set("Content-Type", "application/gob")
for i := 1; i < len(headers); i += 2 {
tempHeader.Set(headers[i-1], headers[i])
}
buff := new(bytes.Buffer)
ct := tempHeader.Get("Content-Type")
if ct == "application/gob" {
enc := gob.NewEncoder(buff)
enc.Encode(data)
} else if ct == "application/json" {
enc := json.NewEncoder(buff)
enc.Encode(data)
}
reqURL := fmt.Sprintf("%s/api?call=%s", url, method)
req, err := http.NewRequest("POST", reqURL, buff)
if err != nil {
return err
}
req.Header = tempHeader
resp, err := http.DefaultClient.Do(req)
if err != nil {
return err
}
if resp.StatusCode != http.StatusOK {
return &ErrorWithStatus{StatusCode: resp.StatusCode}
}
defer func() {
if resp != nil && resp.Body != nil {
resp.Body.Close()
}
}()
if out != nil && resp.Body != nil {
dec := gob.NewDecoder(resp.Body)
return dec.Decode(out)
}
return nil
}

161
session/common.go Normal file
View File

@ -0,0 +1,161 @@
package session
import (
"context"
"encoding/binary"
"encoding/hex"
"errors"
"math/rand"
"strings"
"time"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type Authorization struct {
Account primitive.ObjectID `bson:"a" json:"a"`
invalidated string
// by authorization provider
Platform string `bson:"p" json:"p"`
Uid string `bson:"u" json:"u"`
Email string `bson:"em" json:"em"`
}
func (auth *Authorization) ToStrings() []string {
return []string{
"a", auth.Account.Hex(),
"p", auth.Platform,
"u", auth.Uid,
"em", auth.Email,
"inv", auth.invalidated,
}
}
func (auth *Authorization) Invalidated() bool {
return len(auth.invalidated) > 0
}
func MakeAuthrizationFromStringMap(src map[string]string) Authorization {
accid, _ := primitive.ObjectIDFromHex(src["a"])
return Authorization{
Account: accid,
Platform: src["p"],
Uid: src["u"],
Email: src["em"],
invalidated: src["inv"],
}
}
type Provider interface {
New(*Authorization) (string, error)
RevokeAll(primitive.ObjectID) error
Query(string) (Authorization, error)
Touch(string) (bool, error)
}
type Consumer interface {
Query(string) (Authorization, error)
Touch(string) (Authorization, error)
IsRevoked(primitive.ObjectID) bool
Revoke(string)
RegisterOnSessionInvalidated(func(primitive.ObjectID))
}
type storagekey string
type publickey string
var r = rand.New(rand.NewSource(time.Now().UnixNano()))
func make_storagekey(acc primitive.ObjectID) storagekey {
bs := [4]byte{}
binary.LittleEndian.PutUint32(bs[:], r.Uint32())
return storagekey(acc.Hex() + hex.EncodeToString(bs[2:]))
}
func storagekey_to_publickey(sk storagekey) publickey {
bs, _ := hex.DecodeString(string(sk))
acc := bs[:12]
cs := bs[12:]
encoded := [14]byte{}
for i, v := range acc[:] {
encoded[i] = (v ^ cs[0]) ^ cs[1]
}
encoded[12] = cs[0]
encoded[13] = cs[1]
return publickey(hex.EncodeToString(encoded[:]))
}
func publickey_to_storagekey(pk publickey) storagekey {
bs, _ := hex.DecodeString(string(pk))
acc := bs[:12]
cs := bs[12:]
decoded := [14]byte{}
for i, v := range acc[:] {
decoded[i] = (v ^ cs[1]) ^ cs[0]
}
decoded[12] = cs[0]
decoded[13] = cs[1]
return storagekey(hex.EncodeToString(decoded[:]))
}
type SessionConfig struct {
SessionTTL int64 `json:"session_ttl"`
SessionStorage string `json:"session_storage"`
}
var errInvalidScheme = errors.New("storageAddr is not valid scheme")
var errSessionStorageMissing = errors.New("session_storageis missing")
func NewConsumer(ctx context.Context, storageAddr string, ttl time.Duration) (Consumer, error) {
if strings.HasPrefix(storageAddr, "mongodb") {
return newConsumerWithMongo(ctx, storageAddr, ttl)
}
if strings.HasPrefix(storageAddr, "redis") {
return newConsumerWithRedis(ctx, storageAddr, ttl)
}
return nil, errInvalidScheme
}
func NewConsumerWithConfig(ctx context.Context, cfg SessionConfig) (Consumer, error) {
if len(cfg.SessionStorage) == 0 {
return nil, errSessionStorageMissing
}
if cfg.SessionTTL == 0 {
cfg.SessionTTL = 3600
}
return NewConsumer(ctx, cfg.SessionStorage, time.Duration(cfg.SessionTTL)*time.Second)
}
func NewProvider(ctx context.Context, storageAddr string, ttl time.Duration) (Provider, error) {
if strings.HasPrefix(storageAddr, "mongodb") {
return newProviderWithMongo(ctx, storageAddr, ttl)
}
if strings.HasPrefix(storageAddr, "redis") {
return newProviderWithRedis(ctx, storageAddr, ttl)
}
return nil, errInvalidScheme
}
func NewProviderWithConfig(ctx context.Context, cfg SessionConfig) (Provider, error) {
if len(cfg.SessionStorage) == 0 {
return nil, errSessionStorageMissing
}
if cfg.SessionTTL == 0 {
cfg.SessionTTL = 3600
}
return NewProvider(ctx, cfg.SessionStorage, time.Duration(cfg.SessionTTL)*time.Second)
}

View File

@ -0,0 +1,70 @@
package session
import (
"context"
"sync"
"time"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type cache_stage[T any] struct {
cache map[storagekey]T
deleted map[storagekey]T
}
func make_cache_stage[T any]() *cache_stage[T] {
return &cache_stage[T]{
cache: make(map[storagekey]T),
deleted: make(map[storagekey]T),
}
}
type consumer_common[T any] struct {
lock sync.Mutex
ttl time.Duration
ctx context.Context
stages [2]*cache_stage[T]
startTime time.Time
onSessionInvalidated []func(primitive.ObjectID)
}
func (c *consumer_common[T]) add_internal(sk storagekey, si T) {
c.stages[0].cache[sk] = si
delete(c.stages[0].deleted, sk)
c.stages[1].cache[sk] = si
delete(c.stages[1].deleted, sk)
}
func (c *consumer_common[T]) delete_internal(sk storagekey) (old T) {
if v, ok := c.stages[0].cache[sk]; ok {
old = v
c.stages[0].deleted[sk] = old
c.stages[1].deleted[sk] = old
delete(c.stages[0].cache, sk)
delete(c.stages[1].cache, sk)
} else if v, ok = c.stages[1].cache[sk]; ok {
old = v
c.stages[1].deleted[sk] = old
delete(c.stages[1].cache, sk)
}
return
}
func (c *consumer_common[T]) delete(sk storagekey) T {
c.lock.Lock()
defer c.lock.Unlock()
return c.delete_internal(sk)
}
func (c *consumer_common[T]) changeStage() {
c.lock.Lock()
defer c.lock.Unlock()
c.stages[1] = c.stages[0]
c.stages[0] = make_cache_stage[T]()
}

383
session/impl_mongo.go Normal file
View File

@ -0,0 +1,383 @@
package session
import (
"context"
"time"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"repositories.action2quare.com/ayo/gocommon"
"repositories.action2quare.com/ayo/gocommon/logger"
)
const (
session_collection_name = gocommon.CollectionName("session")
)
type provider_mongo struct {
mongoClient gocommon.MongoClient
}
type sessionMongo struct {
Id primitive.ObjectID `bson:"_id,omitempty"`
Auth *Authorization `bson:"auth"`
Key storagekey `bson:"key"`
Ts primitive.DateTime `bson:"_ts"`
}
func newProviderWithMongo(ctx context.Context, mongoUrl string, ttl time.Duration) (Provider, error) {
mc, err := gocommon.NewMongoClient(ctx, mongoUrl)
if err != nil {
return nil, err
}
if err = mc.MakeUniqueIndices(session_collection_name, map[string]bson.D{
"key": {{Key: "key", Value: 1}},
}); err != nil {
return nil, err
}
if err := mc.MakeExpireIndex(session_collection_name, int32(ttl.Seconds())); err != nil {
return nil, err
}
return &provider_mongo{
mongoClient: mc,
}, nil
}
func (p *provider_mongo) New(input *Authorization) (string, error) {
sk := make_storagekey(input.Account)
_, _, err := p.mongoClient.Update(session_collection_name, bson.M{
"_id": input.Account,
}, bson.M{
"$set": sessionMongo{
Auth: input,
Key: sk,
Ts: primitive.NewDateTimeFromTime(time.Now().UTC()),
},
}, options.Update().SetUpsert(true))
return string(storagekey_to_publickey(sk)), err
}
func (p *provider_mongo) RevokeAll(acc primitive.ObjectID) error {
_, err := p.mongoClient.Delete(session_collection_name, bson.M{
"_id": acc,
})
return err
}
func (p *provider_mongo) Query(pk string) (Authorization, error) {
sk := publickey_to_storagekey(publickey(pk))
var auth Authorization
err := p.mongoClient.FindOneAs(session_collection_name, bson.M{
"key": sk,
}, &auth)
return auth, err
}
func (p *provider_mongo) Touch(pk string) (bool, error) {
sk := publickey_to_storagekey(publickey(pk))
worked, _, err := p.mongoClient.Update(session_collection_name, bson.M{
"key": sk,
}, bson.M{
"$currentDate": bson.M{
"_ts": bson.M{"$type": "date"},
},
}, options.Update().SetUpsert(false))
if err != nil {
logger.Println("provider Touch :", err)
return false, err
}
return worked, nil
}
type consumer_mongo struct {
consumer_common[*sessionMongo]
ids map[primitive.ObjectID]storagekey
mongoClient gocommon.MongoClient
ttl time.Duration
}
type sessionPipelineDocument struct {
OperationType string `bson:"operationType"`
DocumentKey struct {
Id primitive.ObjectID `bson:"_id"`
} `bson:"documentKey"`
Session *sessionMongo `bson:"fullDocument"`
}
func newConsumerWithMongo(ctx context.Context, mongoUrl string, ttl time.Duration) (Consumer, error) {
mc, err := gocommon.NewMongoClient(ctx, mongoUrl)
if err != nil {
return nil, err
}
consumer := &consumer_mongo{
consumer_common: consumer_common[*sessionMongo]{
ttl: ttl,
ctx: ctx,
stages: [2]*cache_stage[*sessionMongo]{make_cache_stage[*sessionMongo](), make_cache_stage[*sessionMongo]()},
startTime: time.Now(),
},
ids: make(map[primitive.ObjectID]storagekey),
ttl: ttl,
mongoClient: mc,
}
go func() {
matchStage := bson.D{
{
Key: "$match", Value: bson.D{
{Key: "operationType", Value: bson.D{
{Key: "$in", Value: bson.A{
"delete",
"insert",
"update",
}},
}},
},
}}
projectStage := bson.D{
{
Key: "$project", Value: bson.D{
{Key: "documentKey", Value: 1},
{Key: "operationType", Value: 1},
{Key: "fullDocument", Value: 1},
},
},
}
var stream *mongo.ChangeStream
nextswitch := time.Now().Add(ttl)
for {
if stream == nil {
stream, err = mc.Watch(session_collection_name, mongo.Pipeline{matchStage, projectStage})
if err != nil {
logger.Error("watchAuthCollection watch failed :", err)
time.Sleep(time.Minute)
continue
}
}
changed := stream.TryNext(ctx)
if ctx.Err() != nil {
logger.Error("watchAuthCollection stream.TryNext failed. process should be restarted! :", ctx.Err().Error())
break
}
if changed {
var data sessionPipelineDocument
if err := stream.Decode(&data); err == nil {
ot := data.OperationType
switch ot {
case "insert":
consumer.add(data.Session.Key, data.DocumentKey.Id, data.Session)
case "update":
if data.Session == nil {
if old := consumer.deleteById(data.DocumentKey.Id); old != nil {
for _, f := range consumer.onSessionInvalidated {
f(old.Auth.Account)
}
}
} else {
consumer.add(data.Session.Key, data.DocumentKey.Id, data.Session)
}
case "delete":
if old := consumer.deleteById(data.DocumentKey.Id); old != nil {
for _, f := range consumer.onSessionInvalidated {
f(old.Auth.Account)
}
}
}
} else {
logger.Error("watchAuthCollection stream.Decode failed :", err)
}
} else if stream.Err() != nil || stream.ID() == 0 {
select {
case <-ctx.Done():
logger.Println("watchAuthCollection is done")
stream.Close(ctx)
return
case <-time.After(time.Second):
logger.Error("watchAuthCollection stream error :", stream.Err())
stream.Close(ctx)
stream = nil
}
} else {
time.Sleep(time.Second)
}
now := time.Now()
for now.After(nextswitch) {
consumer.changeStage()
nextswitch = nextswitch.Add(ttl)
}
}
}()
return consumer, nil
}
func (c *consumer_mongo) query_internal(sk storagekey) (*sessionMongo, bool, error) {
if _, deleted := c.stages[0].deleted[sk]; deleted {
return nil, false, nil
}
if _, deleted := c.stages[1].deleted[sk]; deleted {
return nil, false, nil
}
found, ok := c.stages[0].cache[sk]
if !ok {
found, ok = c.stages[1].cache[sk]
}
if ok {
return found, false, nil
}
var si sessionMongo
err := c.mongoClient.FindOneAs(session_collection_name, bson.M{
"key": sk,
}, &si)
if err != nil {
logger.Println("consumer Query :", err)
return nil, false, err
}
if len(si.Key) > 0 {
siptr := &si
c.add_internal(sk, siptr)
return siptr, true, nil
}
return nil, false, nil
}
func (c *consumer_mongo) Query(pk string) (Authorization, error) {
c.lock.Lock()
defer c.lock.Unlock()
sk := publickey_to_storagekey(publickey(pk))
si, _, err := c.query_internal(sk)
if err != nil {
return Authorization{}, err
}
if si == nil {
return Authorization{}, nil
}
if time.Now().After(si.Ts.Time().Add(c.ttl)) {
return Authorization{}, nil
}
return *si.Auth, nil
}
func (c *consumer_mongo) Touch(pk string) (Authorization, error) {
c.lock.Lock()
defer c.lock.Unlock()
sk := publickey_to_storagekey(publickey(pk))
worked, _, err := c.mongoClient.Update(session_collection_name, bson.M{
"key": sk,
}, bson.M{
"$currentDate": bson.M{
"_ts": bson.M{"$type": "date"},
},
}, options.Update().SetUpsert(false))
if err != nil {
logger.Println("consumer Touch :", err)
return Authorization{}, err
}
if !worked {
// 이미 만료되서 사라짐
return Authorization{}, nil
}
si, added, err := c.query_internal(sk)
if err != nil {
return Authorization{}, err
}
if si == nil {
return Authorization{}, nil
}
if !added {
var doc sessionMongo
err := c.mongoClient.FindOneAs(session_collection_name, bson.M{
"key": sk,
}, &doc)
if err != nil {
logger.Println("consumer Query :", err)
return Authorization{}, err
}
if len(si.Key) > 0 {
c.add_internal(sk, &doc)
c.ids[doc.Id] = sk
return *doc.Auth, nil
}
}
return *si.Auth, nil
}
func (c *consumer_mongo) Revoke(pk string) {
sk := publickey_to_storagekey(publickey(pk))
_, err := c.mongoClient.Delete(session_collection_name, bson.M{
"key": sk,
})
if err == nil {
for id, v := range c.ids {
if v == sk {
delete(c.ids, id)
break
}
}
}
}
func (c *consumer_mongo) IsRevoked(id primitive.ObjectID) bool {
_, ok := c.ids[id]
return !ok
}
func (c *consumer_mongo) add(sk storagekey, id primitive.ObjectID, si *sessionMongo) {
c.lock.Lock()
defer c.lock.Unlock()
c.consumer_common.add_internal(sk, si)
c.ids[id] = sk
}
func (c *consumer_mongo) deleteById(id primitive.ObjectID) (old *sessionMongo) {
c.lock.Lock()
defer c.lock.Unlock()
if sk, ok := c.ids[id]; ok {
old = c.consumer_common.delete_internal(sk)
delete(c.ids, id)
}
return
}
func (c *consumer_mongo) RegisterOnSessionInvalidated(cb func(primitive.ObjectID)) {
c.onSessionInvalidated = append(c.onSessionInvalidated, cb)
}

358
session/impl_redis.go Normal file
View File

@ -0,0 +1,358 @@
package session
import (
"context"
"fmt"
"time"
"github.com/go-redis/redis/v8"
"go.mongodb.org/mongo-driver/bson/primitive"
"repositories.action2quare.com/ayo/gocommon"
"repositories.action2quare.com/ayo/gocommon/logger"
)
const (
communication_channel_name_prefix = "_sess_comm_chan_name"
)
type sessionRedis struct {
*Authorization
expireAt time.Time
}
type provider_redis struct {
redisClient *redis.Client
deleteChannel string
ttl time.Duration
ctx context.Context
}
func newProviderWithRedis(ctx context.Context, redisUrl string, ttl time.Duration) (Provider, error) {
redisClient, err := gocommon.NewRedisClient(redisUrl)
if err != nil {
return nil, err
}
return &provider_redis{
redisClient: redisClient,
deleteChannel: fmt.Sprintf("%s_%d_d", communication_channel_name_prefix, redisClient.Options().DB),
ttl: ttl,
ctx: ctx,
}, nil
}
func (p *provider_redis) New(input *Authorization) (string, error) {
newsk := make_storagekey(input.Account)
prefix := input.Account.Hex()
sks, err := p.redisClient.Keys(p.ctx, prefix+"*").Result()
if err != nil {
logger.Println("session provider delete :", sks, err)
return "", err
}
p.redisClient.Del(p.ctx, sks...)
for _, sk := range sks {
p.redisClient.Publish(p.ctx, p.deleteChannel, sk).Result()
}
for {
duplicated := false
for _, sk := range sks {
if sk == string(newsk) {
duplicated = true
break
}
}
if !duplicated {
break
}
newsk = make_storagekey(input.Account)
}
_, err = p.redisClient.HSet(p.ctx, string(newsk), input.ToStrings()).Result()
if err != nil {
return "", err
}
_, err = p.redisClient.Expire(p.ctx, string(newsk), p.ttl).Result()
if err != nil {
return "", err
}
pk := storagekey_to_publickey(newsk)
return string(pk), err
}
func (p *provider_redis) RevokeAll(account primitive.ObjectID) error {
prefix := account.Hex()
sks, err := p.redisClient.Keys(p.ctx, prefix+"*").Result()
if err != nil {
logger.Println("session provider delete :", sks, err)
return err
}
for _, sk := range sks {
p.redisClient.HSet(p.ctx, sk, "inv", "true")
p.redisClient.Publish(p.ctx, p.deleteChannel, sk).Result()
}
return nil
}
func (p *provider_redis) Query(pk string) (Authorization, error) {
sk := publickey_to_storagekey(publickey(pk))
src, err := p.redisClient.HGetAll(p.ctx, string(sk)).Result()
if err == redis.Nil {
logger.Println("session provider query :", pk, err)
return Authorization{}, nil
} else if err != nil {
logger.Println("session provider query :", pk, err)
return Authorization{}, err
}
auth := MakeAuthrizationFromStringMap(src)
return auth, nil
}
func (p *provider_redis) Touch(pk string) (bool, error) {
sk := publickey_to_storagekey(publickey(pk))
ok, err := p.redisClient.Expire(p.ctx, string(sk), p.ttl).Result()
if err == redis.Nil {
// 이미 만료됨
logger.Println("session provider touch :", pk, err)
return false, nil
} else if err != nil {
logger.Println("session provider touch :", pk, err)
return false, err
}
return ok, nil
}
type consumer_redis struct {
consumer_common[*sessionRedis]
redisClient *redis.Client
deleteChannel string
}
func newConsumerWithRedis(ctx context.Context, redisUrl string, ttl time.Duration) (Consumer, error) {
redisClient, err := gocommon.NewRedisClient(redisUrl)
if err != nil {
return nil, err
}
deleteChannel := fmt.Sprintf("%s_%d_d", communication_channel_name_prefix, redisClient.Options().DB)
sub := redisClient.Subscribe(ctx, deleteChannel)
consumer := &consumer_redis{
consumer_common: consumer_common[*sessionRedis]{
ttl: ttl,
ctx: ctx,
stages: [2]*cache_stage[*sessionRedis]{make_cache_stage[*sessionRedis](), make_cache_stage[*sessionRedis]()},
startTime: time.Now(),
},
redisClient: redisClient,
deleteChannel: deleteChannel,
}
go func() {
stageswitch := time.Now().Add(ttl)
tickTimer := time.After(ttl)
for {
select {
case <-ctx.Done():
return
case <-tickTimer:
consumer.changeStage()
stageswitch = stageswitch.Add(ttl)
tempttl := time.Until(stageswitch)
tickTimer = time.After(tempttl)
case msg := <-sub.Channel():
if msg == nil {
return
}
if len(msg.Payload) == 0 {
continue
}
switch msg.Channel {
case deleteChannel:
sk := storagekey(msg.Payload)
old := consumer.delete(sk)
if old != nil {
for _, f := range consumer.onSessionInvalidated {
f(old.Account)
}
}
}
}
}
}()
return consumer, nil
}
func (c *consumer_redis) query_internal(sk storagekey) (*sessionRedis, error) {
if old, deleted := c.stages[0].deleted[sk]; deleted {
return old, nil
}
if old, deleted := c.stages[1].deleted[sk]; deleted {
return old, nil
}
found, ok := c.stages[0].cache[sk]
if !ok {
found, ok = c.stages[1].cache[sk]
}
if ok {
if time.Now().Before(found.expireAt) {
// 만료전 세션
return found, nil
}
// 다른 Consumer가 Touch했을 수도 있으므로 redis에서 읽어본다.
}
payload, err := c.redisClient.HGetAll(c.ctx, string(sk)).Result()
if err != nil && err != redis.Nil {
logger.Println("consumer Query :", err)
return nil, err
}
if len(payload) == 0 {
return nil, nil
}
ttl, err := c.redisClient.TTL(c.ctx, string(sk)).Result()
if err != nil {
logger.Println("consumer Query :", err)
return nil, err
}
if ttl < 0 {
ttl = time.Duration(time.Hour * 24)
}
auth := MakeAuthrizationFromStringMap(payload)
si := &sessionRedis{
Authorization: &auth,
expireAt: time.Now().Add(ttl),
}
if auth.Invalidated() {
c.stages[0].deleted[sk] = si
} else {
c.add_internal(sk, si)
}
return si, nil
}
func (c *consumer_redis) Query(pk string) (Authorization, error) {
c.lock.Lock()
defer c.lock.Unlock()
sk := publickey_to_storagekey(publickey(pk))
si, err := c.query_internal(sk)
if err != nil {
logger.Println("session consumer query :", pk, err)
return Authorization{}, err
}
if si == nil {
logger.Println("session consumer query(si nil) :", pk, nil)
return Authorization{}, nil
}
if time.Now().After(si.expireAt) {
logger.Println("session consumer query(expired):", pk, nil)
return Authorization{}, nil
}
return *si.Authorization, nil
}
func (c *consumer_redis) Touch(pk string) (Authorization, error) {
c.lock.Lock()
defer c.lock.Unlock()
sk := publickey_to_storagekey(publickey(pk))
if _, deleted := c.stages[0].deleted[sk]; deleted {
return Authorization{}, nil
}
if _, deleted := c.stages[1].deleted[sk]; deleted {
return Authorization{}, nil
}
ok, err := c.redisClient.Expire(c.ctx, string(sk), c.ttl).Result()
if err == redis.Nil {
logger.Println("session consumer touch :", pk, err)
return Authorization{}, nil
} else if err != nil {
logger.Println("session consumer touch :", pk, err)
return Authorization{}, err
}
if ok {
// redis에 살아있다.
si, err := c.query_internal(sk)
if err != nil {
logger.Println("session consumer touch(ok) :", pk, err)
return Authorization{}, err
}
if si == nil {
logger.Println("session consumer touch(ok, si nil) :", pk)
return Authorization{}, nil
}
return *si.Authorization, nil
}
return Authorization{}, nil
}
func (c *consumer_redis) Revoke(pk string) {
sk := publickey_to_storagekey(publickey(pk))
c.redisClient.Del(c.ctx, string(sk))
c.lock.Lock()
defer c.lock.Unlock()
if sr, ok := c.stages[0].cache[sk]; ok {
c.stages[0].deleted[sk] = sr
}
if sr, ok := c.stages[1].cache[sk]; ok {
c.stages[1].deleted[sk] = sr
}
}
func (c *consumer_redis) IsRevoked(accid primitive.ObjectID) bool {
sk := make_storagekey(accid)
c.lock.Lock()
defer c.lock.Unlock()
if _, deleted := c.stages[0].deleted[sk]; deleted {
return true
}
if _, deleted := c.stages[1].deleted[sk]; deleted {
return true
}
return false
}
func (c *consumer_redis) RegisterOnSessionInvalidated(cb func(primitive.ObjectID)) {
c.onSessionInvalidated = append(c.onSessionInvalidated, cb)
}

93
session/session_test.go Normal file
View File

@ -0,0 +1,93 @@
// package main ...
package session
import (
"context"
"testing"
"time"
"go.mongodb.org/mongo-driver/bson/primitive"
"repositories.action2quare.com/ayo/gocommon/logger"
)
func TestExpTable(t *testing.T) {
// pv, err := NewProvider(context.Background(), "redis://192.168.8.94:6380/1", 10*time.Second)
// if err != nil {
// t.Error(err)
// }
// cs, err := NewConsumer(context.Background(), "redis://192.168.8.94:6380/1", 10*time.Second)
// if err != nil {
// t.Error(err)
// }
pv, err := NewProvider(context.Background(), "mongodb://192.168.8.94:27017/maingate?replicaSet=repl01&retrywrites=false", 10*time.Second)
if err != nil {
t.Error(err)
}
cs, err := NewConsumer(context.Background(), "mongodb://192.168.8.94:27017/maingate?replicaSet=repl01&retrywrites=false", 10*time.Second)
if err != nil {
t.Error(err)
}
test := primitive.NewObjectID()
sk := make_storagekey(test)
pk := storagekey_to_publickey(sk)
if publickey_to_storagekey(pk) != sk {
t.Errorf("pk != sk : %s, %s", pk, sk)
}
au1 := &Authorization{
Account: primitive.NewObjectID(),
Platform: "editor",
Uid: "uid-1",
}
sk1, err := pv.New(au1)
if err != nil {
t.Error(err)
}
au2 := &Authorization{
Account: primitive.NewObjectID(),
Platform: "editor",
Uid: "uid-2",
}
sk2, err := pv.New(au2)
if err != nil {
t.Error(err)
}
go func() {
for {
q1, err := cs.Query(sk1)
logger.Println("query :", q1, err)
q2, err := cs.Query(sk2)
logger.Println("query :", q2, err)
time.Sleep(time.Second)
}
}()
cs.Touch(sk1)
time.Sleep(2 * time.Second)
cs.Touch(sk2)
time.Sleep(2 * time.Second)
time.Sleep(2 * time.Second)
pv.RevokeAll(au1.Account)
cs.Touch(sk1)
time.Sleep(2 * time.Second)
cs.Touch(sk2)
time.Sleep(2 * time.Second)
cs2, err := NewConsumer(context.Background(), "redis://192.168.8.94:6380/1", 10*time.Second)
if err != nil {
t.Error(err)
}
q2, err := cs2.Query(sk2)
logger.Println("queryf :", q2, err)
time.Sleep(20 * time.Second)
}

146
voicechat/eos_impl.go Normal file
View File

@ -0,0 +1,146 @@
package voicechat
import (
"bytes"
"context"
"encoding/base64"
"encoding/json"
"fmt"
"net/http"
"sync/atomic"
"time"
"unsafe"
"repositories.action2quare.com/ayo/gocommon/logger"
)
type eosauth struct {
AccessToken string `json:"access_token"`
ExpiresAt time.Time `json:"expires_at"`
ExpiresIn int64 `json:"expires_in"`
DeploymentId string `json:"deployment_id"`
ProductId string `json:"product_id"`
SandboxId string `json:"sandbox_id"`
TokenType string `json:"token_type"`
}
func eosTokenRefresh(target *unsafe.Pointer, ctx context.Context) {
defer func() {
r := recover()
if r != nil {
logger.Error(r)
}
}()
endpoint := "https://api.epicgames.dev/auth/v1/oauth/token"
auth := base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("%s:%s", config.EosClientId, config.EosClientSecret)))
for {
req, _ := http.NewRequest("POST", endpoint, bytes.NewBufferString("grant_type=client_credentials&deployment_id="+config.EosDeploymentId))
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
req.Header.Set("Accept", "application/json")
req.Header.Set("Authorization", fmt.Sprintf("Basic %s", auth))
resp, err := http.DefaultClient.Do(req)
if err != nil {
logger.Println("eosTokenRefresh failed. eos token reqeust err :", err)
time.Sleep(time.Minute)
continue
}
var neweos eosauth
err = json.NewDecoder(resp.Body).Decode(&neweos)
resp.Body.Close()
if err != nil {
logger.Println("eosTokenRefresh failed. decode err :", err)
return
}
logger.Printf("eos access_token retreived : %s...", neweos.AccessToken[:20])
atomic.StorePointer(target, unsafe.Pointer(&neweos))
select {
case <-ctx.Done():
return
case <-time.After(time.Duration(neweos.ExpiresIn-60) * time.Second):
}
}
}
type eosRoomParticipantRequests struct {
Puid string `json:"puid"`
ClientIP string `json:"clientIP"`
HardMuted bool `json:"hardMuted"`
}
type eosRoomParticipants struct {
Participants []eosRoomParticipantRequests `json:"participants"`
}
func (gv *eosauth) joinVoiceChat(data JoinVoiceChatRequst) map[string]any {
// https://dev.epicgames.com/docs/web-api-ref/voice-web-api
accessToken := gv.AccessToken
if len(accessToken) == 0 {
logger.Println("eos voice chat is not ready. access_token is empty")
return nil
}
voiceendpoint := fmt.Sprintf("https://api.epicgames.dev/rtc/v1/%s/room/%s", config.EosDeploymentId, data.Gid)
participants := eosRoomParticipants{
Participants: []eosRoomParticipantRequests{
{Puid: data.Alias},
},
}
body, _ := json.Marshal(participants)
req, _ := http.NewRequest("POST", voiceendpoint, bytes.NewBuffer(body))
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Accept", "application/json")
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", accessToken))
resp, err := http.DefaultClient.Do(req)
if err != nil {
logger.Println("join voice room failed. api.epicgames.dev return err :", err)
return nil
}
var result map[string]any
json.NewDecoder(resp.Body).Decode(&result)
resp.Body.Close()
result["client_id"] = config.EosClientId
result["client_secret"] = config.EosClientSecret
par := result["participants"].([]any)[0]
participant := par.(map[string]any)
channelCredentials := map[string]any{
"override_userid": data.Alias,
"client_base_url": result["clientBaseUrl"],
"participant_token": participant["token"],
}
marshaled, _ := json.Marshal(channelCredentials)
result["channel_credentials"] = base64.StdEncoding.EncodeToString(marshaled)
return result
}
func (gv *eosauth) leaveVoiceChat(data LeaveVoiceChatRequst) {
voiceendpoint := fmt.Sprintf("https://api.epicgames.dev/rtc/v1/%s/room/%s/participants/%s", config.EosDeploymentId, data.Gid, data.Alias)
accessToken := gv.AccessToken
req, _ := http.NewRequest("DELETE", voiceendpoint, nil)
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", accessToken))
resp, err := http.DefaultClient.Do(req)
if err != nil {
logger.Println("LeaveVoiceChat failed. err :", err)
return
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusNoContent {
// 204가 정상 응답
logger.Println("LeaveVoiceChat failed. status code :", resp.StatusCode)
}
}

83
voicechat/voice_chat.go Normal file
View File

@ -0,0 +1,83 @@
package voicechat
import (
"context"
"sync/atomic"
"unsafe"
"repositories.action2quare.com/ayo/gocommon"
"repositories.action2quare.com/ayo/gocommon/logger"
)
type JoinVoiceChatRequst struct {
Gid string
Alias string
Service string
}
type LeaveVoiceChatRequst struct {
Gid string
Alias string
Service string
}
type voiceChatConfig struct {
EosClientId string `json:"eos_client_id"`
EosClientSecret string `json:"eos_client_secret"`
EosDeploymentId string `json:"eos_deployment_id"`
}
type voiceServiceImpl interface {
joinVoiceChat(JoinVoiceChatRequst) map[string]any
leaveVoiceChat(LeaveVoiceChatRequst)
}
var config voiceChatConfig
type VoiceChatService struct {
eosptr unsafe.Pointer
}
func (gv *VoiceChatService) Initialize(ctx context.Context) error {
if err := gocommon.LoadConfig(&config); err != nil {
return logger.ErrorWithCallStack(err)
}
if len(config.EosClientId) == 0 {
logger.Println("eos voice chat is disabled. 'eos_client_id' is empty")
}
if len(config.EosClientSecret) == 0 {
logger.Println("eos voice chat is disabled. 'eos_client_secret' is empty")
}
if len(config.EosDeploymentId) == 0 {
logger.Println("eos voice chat is disabled. 'eos_deployment_id' is empty")
}
gv.eosptr = unsafe.Pointer(&eosauth{})
if len(config.EosClientId) > 0 && len(config.EosClientSecret) > 0 && len(config.EosDeploymentId) > 0 {
go eosTokenRefresh(&gv.eosptr, ctx)
}
return nil
}
func (gv *VoiceChatService) eos() voiceServiceImpl {
ptr := atomic.LoadPointer(&gv.eosptr)
return (*eosauth)(ptr)
}
func (gv *VoiceChatService) JoinVoiceChat(req JoinVoiceChatRequst) map[string]any {
switch req.Service {
case "eos":
return gv.eos().joinVoiceChat(req)
}
return nil
}
func (gv *VoiceChatService) LeaveVoiceChat(req LeaveVoiceChatRequst) {
switch req.Service {
case "eos":
gv.eos().leaveVoiceChat(req)
}
}

185
wshandler/api_handler.go Normal file
View File

@ -0,0 +1,185 @@
package wshandler
import (
"encoding/json"
"io"
"reflect"
"strings"
"unsafe"
"github.com/gorilla/websocket"
"repositories.action2quare.com/ayo/gocommon/logger"
)
const (
ClientConnected = "ClientConnected"
ClientDisconnected = "ClientDisconnected"
)
type apiFuncType func(ApiCallContext)
type connFuncType func(*websocket.Conn, *Sender)
type disconnFuncType func(string, *Sender)
type WebsocketApiHandler struct {
methods map[string]apiFuncType
connfunc connFuncType
disconnfunc disconnFuncType
originalReceiverName string
}
type ApiCallContext struct {
CallBy *Sender
Arguments []any
}
func MakeWebsocketApiHandler[T any](receiver *T, receiverName string) WebsocketApiHandler {
methods := make(map[string]apiFuncType)
tp := reflect.TypeOf(receiver)
if len(receiverName) == 0 {
receiverName = tp.Elem().Name()
}
var connfunc connFuncType
var disconnfunc disconnFuncType
for i := 0; i < tp.NumMethod(); i++ {
method := tp.Method(i)
if method.Type.In(0) != tp {
continue
}
if method.Name == ClientConnected {
if method.Type.NumIn() != 3 {
continue
}
if method.Type.In(1) != reflect.TypeOf((*websocket.Conn)(nil)) {
continue
}
if method.Type.In(2) != reflect.TypeOf((*Sender)(nil)) {
continue
}
funcptr := method.Func.Pointer()
p1 := unsafe.Pointer(&funcptr)
p2 := unsafe.Pointer(&p1)
connfuncptr := (*func(*T, *websocket.Conn, *Sender))(p2)
connfunc = func(c *websocket.Conn, s *Sender) {
(*connfuncptr)(receiver, c, s)
}
} else if method.Name == ClientDisconnected {
if method.Type.NumIn() != 3 {
continue
}
if method.Type.In(1) != reflect.TypeOf("") {
continue
}
if method.Type.In(2) != reflect.TypeOf((*Sender)(nil)) {
continue
}
funcptr := method.Func.Pointer()
p1 := unsafe.Pointer(&funcptr)
p2 := unsafe.Pointer(&p1)
disconnfuncptr := (*func(*T, string, *Sender))(p2)
disconnfunc = func(msg string, s *Sender) {
(*disconnfuncptr)(receiver, msg, s)
}
} else {
if method.Type.NumIn() != 2 {
continue
}
if method.Type.In(1) != reflect.TypeOf((*ApiCallContext)(nil)).Elem() {
continue
}
funcptr := method.Func.Pointer()
p1 := unsafe.Pointer(&funcptr)
p2 := unsafe.Pointer(&p1)
apifuncptr := (*func(*T, ApiCallContext))(p2)
methods[receiverName+"."+method.Name] = func(ctx ApiCallContext) {
(*apifuncptr)(receiver, ctx)
}
}
}
return WebsocketApiHandler{
methods: methods,
connfunc: connfunc,
disconnfunc: disconnfunc,
originalReceiverName: tp.Elem().Name(),
}
}
type WebsocketApiBroker struct {
methods map[string]apiFuncType
methods_dup map[string][]apiFuncType
connFuncs []connFuncType
disconnFuncs []disconnFuncType
}
func (hc *WebsocketApiBroker) AddHandler(receiver WebsocketApiHandler) {
if hc.methods == nil {
hc.methods = make(map[string]apiFuncType)
hc.methods_dup = make(map[string][]apiFuncType)
}
for k, v := range receiver.methods {
ab := strings.Split(k, ".")
logger.Printf("ws api registered : %s.%s -> %s\n", receiver.originalReceiverName, ab[1], k)
hc.methods_dup[k] = append(hc.methods_dup[k], v)
if len(hc.methods_dup[k]) > 1 {
chain := hc.methods_dup[k]
hc.methods[k] = func(ctx ApiCallContext) {
for _, f := range chain {
f(ctx)
}
}
} else {
hc.methods[k] = v
}
}
if receiver.connfunc != nil {
logger.Printf("ws api registered : %s.ClientConnected\n", receiver.originalReceiverName)
hc.connFuncs = append(hc.connFuncs, receiver.connfunc)
}
if receiver.disconnfunc != nil {
// disconnfunc은 역순
logger.Printf("ws api registered : %s.ClientDisconnected\n", receiver.originalReceiverName)
hc.disconnFuncs = append([]disconnFuncType{receiver.disconnfunc}, hc.disconnFuncs...)
}
}
func (hc *WebsocketApiBroker) ClientConnected(c *wsconn) {
for _, v := range hc.connFuncs {
v(c.Conn, c.sender)
}
}
func (hc *WebsocketApiBroker) ClientDisconnected(c *wsconn) {
for _, v := range hc.disconnFuncs {
v(c.closeMessage, c.sender)
}
}
func (hc *WebsocketApiBroker) Call(callby *Sender, funcname string, r io.Reader) {
if found := hc.methods[funcname]; found != nil {
var args []any
if r != nil {
dec := json.NewDecoder(r)
if err := dec.Decode(&args); err != nil {
logger.Println("WebsocketApiBroker.Call failed. decode returns err :", err)
}
}
found(ApiCallContext{
CallBy: callby,
Arguments: args,
})
} else {
logger.Println("api is not found :", funcname)
}
}

View File

@ -0,0 +1,66 @@
// package main ...
package wshandler
import (
"bytes"
"encoding/json"
"fmt"
"reflect"
"testing"
)
type TestReceiver struct {
}
func (tr *TestReceiver) Func1([]any) {
}
func (tr *TestReceiver) Func2(args []any) {
fmt.Println(args...)
}
func TestUnmarshalToken(t *testing.T) {
// 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7
// 012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890
insrc := []byte(`["string value 1",200,["inner string value 1","inner string value 2"],{"inner map key 1":"inner map string value 1","inner map key 2":"inner map string value 2"},500.1]`)
dec := json.NewDecoder(bytes.NewBuffer(insrc))
dec.Token()
for {
token_start := dec.InputOffset()
tok, _ := dec.Token()
token_end := dec.InputOffset()
var string_val_1 string
stringtype := reflect.TypeOf(string_val_1)
stringvalue := reflect.New(stringtype)
castptr := stringvalue.Interface()
err := json.Unmarshal(insrc[token_start:token_end], castptr)
fmt.Println(err)
if tok == nil {
break
}
fmt.Println(tok, dec.InputOffset())
}
// src := []any{"a", 1, false}
// payload, _ := json.Marshal(src)
// tr := new(TestReceiver)
// receiver := MakeWebsocketApiHandler(tr, "test")
// var con WebsocketApiBroker
// con.AddHandler(receiver)
}
func TestUnmarshal(t *testing.T) {
src := []byte(`{"123" :"str1", "456": "str2"}`)
var test map[int]string
err := json.Unmarshal(src, &test)
if err != nil {
t.Error(err)
}
}

1
wshandler/config.json Normal file
View File

@ -0,0 +1 @@
{}

108
wshandler/room.go Normal file
View File

@ -0,0 +1,108 @@
package wshandler
import (
"bytes"
"context"
"encoding/json"
"github.com/gorilla/websocket"
"go.mongodb.org/mongo-driver/bson/primitive"
"repositories.action2quare.com/ayo/gocommon/logger"
)
type room struct {
inChan chan *wsconn
outChan chan primitive.ObjectID
messageChan chan *UpstreamMessage
name string
destroyChan chan<- string
sendMsgChan chan<- send_msg_queue_elem
}
// 만약 destroyChan가 nil이면 room이 비어도 파괴되지 않는다. 영구 유지되는 room
func makeRoom(name string, destroyChan chan<- string, sendMsgChan chan<- send_msg_queue_elem) *room {
return &room{
inChan: make(chan *wsconn, 10),
outChan: make(chan primitive.ObjectID, 10),
messageChan: make(chan *UpstreamMessage, 100),
name: name,
destroyChan: destroyChan,
sendMsgChan: sendMsgChan,
}
}
func (r *room) broadcast(msg *UpstreamMessage) {
r.messageChan <- msg
}
func (r *room) in(conn *wsconn) *room {
r.inChan <- conn
return r
}
func (r *room) out(accid primitive.ObjectID) *room {
r.outChan <- accid
return r
}
func (r *room) start(ctx context.Context) {
go func(ctx context.Context) {
conns := make(map[string]*wsconn)
normal := false
for !normal {
normal = r.loop(ctx, &conns)
}
}(ctx)
}
func (r *room) loop(ctx context.Context, conns *map[string]*wsconn) (normalEnd bool) {
defer func() {
s := recover()
if s != nil {
logger.Error(s)
normalEnd = false
}
}()
tag := "#" + r.name
for {
select {
case <-ctx.Done():
return true
case conn := <-r.inChan:
(*conns)[conn.sender.Accid.Hex()] = conn
case accid := <-r.outChan:
delete((*conns), accid.Hex())
if len(*conns) == 0 && r.destroyChan != nil {
r.destroyChan <- r.name
return true
}
case msg := <-r.messageChan:
ds := DownstreamMessage{
Alias: msg.Alias,
Body: msg.Body,
Tag: append(msg.Tag, tag),
}
buff := new(bytes.Buffer)
enc := json.NewEncoder(buff)
enc.SetEscapeHTML(false)
enc.Encode(ds)
for _, conn := range *conns {
pmsg, err := websocket.NewPreparedMessage(websocket.TextMessage, buff.Bytes())
if err != nil {
logger.Println("websocket.NewPreparedMessage failed :", err)
} else {
r.sendMsgChan <- send_msg_queue_elem{
to: conn.Conn,
pmsg: pmsg,
}
}
}
}
}
}

File diff suppressed because it is too large Load Diff

391
wshandler/wshandler_peer.go Normal file
View File

@ -0,0 +1,391 @@
package wshandler
import (
"encoding/hex"
"encoding/json"
"fmt"
"io"
"net/http"
"reflect"
"strings"
"time"
"go.mongodb.org/mongo-driver/bson/primitive"
"repositories.action2quare.com/ayo/gocommon/logger"
"repositories.action2quare.com/ayo/gocommon/session"
"github.com/gorilla/websocket"
)
type WebsocketPeerHandler interface {
RegisterHandlers(serveMux *http.ServeMux, prefix string) error
}
type peerCtorChannelValue struct {
accid primitive.ObjectID
conn *websocket.Conn
}
type peerDtorChannelValue struct {
accid primitive.ObjectID
sk string
}
type websocketPeerHandler[T PeerInterface] struct {
methods map[string]peerApiFuncType[T]
createPeer func(primitive.ObjectID) T
sessionConsumer session.Consumer
peerCtorChannel chan peerCtorChannelValue
peerDtorChannel chan peerDtorChannelValue
}
type PeerInterface interface {
ClientDisconnected(string)
ClientConnected(*websocket.Conn)
}
type peerApiFuncType[T PeerInterface] func(T, io.Reader) (any, error)
type websocketPeerApiHandler[T PeerInterface] struct {
methods map[string]peerApiFuncType[T]
originalReceiverName string
}
func (hc *websocketPeerHandler[T]) call(recv T, funcname string, r io.Reader) (v any, e error) {
defer func() {
r := recover()
if r != nil {
logger.Error(r)
e = fmt.Errorf("%v", r)
}
}()
if found := hc.methods[funcname]; found != nil {
return found(recv, r)
}
return nil, fmt.Errorf("api is not found : %s", funcname)
}
func makeWebsocketPeerApiHandler[T PeerInterface]() websocketPeerApiHandler[T] {
methods := make(map[string]peerApiFuncType[T])
var archetype T
tp := reflect.TypeOf(archetype)
for i := 0; i < tp.NumMethod(); i++ {
method := tp.Method(i)
if method.Type.In(0) != tp {
continue
}
if method.Name == ClientDisconnected {
continue
}
var intypes []reflect.Type
for i := 1; i < method.Type.NumIn(); i++ {
intypes = append(intypes, method.Type.In(i))
}
var outconv func([]reflect.Value) (any, error)
if method.Type.NumOut() == 0 {
outconv = func([]reflect.Value) (any, error) { return nil, nil }
} else if method.Type.NumOut() == 1 {
if method.Type.Out(0).Implements(reflect.TypeOf((*error)(nil)).Elem()) {
outconv = func(out []reflect.Value) (any, error) {
if out[0].Interface() == nil {
return nil, nil
}
return nil, out[0].Interface().(error)
}
} else {
outconv = func(out []reflect.Value) (any, error) {
return out[0].Interface(), nil
}
}
} else if method.Type.NumOut() == 2 && method.Type.Out(1).Implements(reflect.TypeOf((*error)(nil)).Elem()) {
outconv = func(out []reflect.Value) (any, error) {
if out[1].Interface() == nil {
return out[0].Interface(), nil
}
return out[0].Interface(), out[1].Interface().(error)
}
}
methods[method.Name] = func(recv T, r io.Reader) (any, error) {
decoder := json.NewDecoder(r)
inargs := make([]any, len(intypes))
for i, intype := range intypes {
zerovalueptr := reflect.New(intype)
inargs[i] = zerovalueptr.Interface()
}
err := decoder.Decode(&inargs)
if err != nil {
return nil, err
}
reflectargs := make([]reflect.Value, 0, len(inargs)+1)
reflectargs = append(reflectargs, reflect.ValueOf(recv))
for _, p := range inargs {
reflectargs = append(reflectargs, reflect.ValueOf(p).Elem())
}
return outconv(method.Func.Call(reflectargs))
}
}
return websocketPeerApiHandler[T]{
methods: methods,
originalReceiverName: tp.Elem().Name(),
}
}
func NewWebsocketPeerHandler[T PeerInterface](consumer session.Consumer, creator func(primitive.ObjectID) T) WebsocketPeerHandler {
methods := make(map[string]peerApiFuncType[T])
receiver := makeWebsocketPeerApiHandler[T]()
for k, v := range receiver.methods {
logger.Printf("ws api registered : %s.%s\n", receiver.originalReceiverName, k)
methods[k] = v
}
wsh := &websocketPeerHandler[T]{
sessionConsumer: consumer,
methods: methods,
createPeer: creator,
peerCtorChannel: make(chan peerCtorChannelValue),
peerDtorChannel: make(chan peerDtorChannelValue),
}
consumer.RegisterOnSessionInvalidated(wsh.onSessionInvalidated)
return wsh
}
func (ws *websocketPeerHandler[T]) RegisterHandlers(serveMux *http.ServeMux, prefix string) error {
if *noAuthFlag {
serveMux.HandleFunc(prefix, ws.upgrade_noauth)
} else {
serveMux.HandleFunc(prefix, ws.upgrade)
}
go ws.sessionMonitoring()
return nil
}
func (ws *websocketPeerHandler[T]) onSessionInvalidated(accid primitive.ObjectID) {
ws.peerDtorChannel <- peerDtorChannelValue{
accid: accid,
}
}
func (ws *websocketPeerHandler[T]) sessionMonitoring() {
all := make(map[primitive.ObjectID]*websocket.Conn)
unauthdata := []byte{0x03, 0xec}
unauthdata = append(unauthdata, []byte("unauthorized")...)
for {
select {
case estVal := <-ws.peerCtorChannel:
all[estVal.accid] = estVal.conn
case disVal := <-ws.peerDtorChannel:
if c := all[disVal.accid]; c != nil {
c.WriteControl(websocket.CloseMessage, unauthdata, time.Time{})
delete(all, disVal.accid)
}
if len(disVal.sk) > 0 {
ws.sessionConsumer.Revoke(disVal.sk)
delete(all, disVal.accid)
}
}
}
}
func (ws *websocketPeerHandler[T]) upgrade_core(conn *websocket.Conn, accid primitive.ObjectID, sk string) {
go func(c *websocket.Conn, accid primitive.ObjectID, sk string) {
peer := ws.createPeer(accid)
var closeReason string
peer.ClientConnected(conn)
ws.peerCtorChannel <- peerCtorChannelValue{accid: accid, conn: conn}
defer func() {
ws.peerDtorChannel <- peerDtorChannelValue{accid: accid, sk: sk}
peer.ClientDisconnected(closeReason)
}()
response := make([]byte, 255)
for {
response = response[:5]
messageType, r, err := c.NextReader()
if err != nil {
if ce, ok := err.(*websocket.CloseError); ok {
closeReason = ce.Text
}
c.Close()
break
}
if messageType == websocket.CloseMessage {
closeMsg, _ := io.ReadAll(r)
closeReason = string(closeMsg)
break
}
if messageType == websocket.BinaryMessage {
var flag [1]byte
r.Read(flag[:])
if flag[0] == 0xff {
// nonce
r.Read(response[1:5])
var size [1]byte
r.Read(size[:])
cmd := make([]byte, size[0])
r.Read(cmd)
result, err := ws.call(peer, string(cmd), r)
if err != nil {
response[0] = 21 // 21 : Negative Ack
response = append(response, []byte(err.Error())...)
} else {
response[0] = 6 // 6 : Acknowledgement
switch result := result.(type) {
case string:
response = append(response, []byte(result)...)
case int8, int16, int32, int64, uint8, uint16, uint32, uint64:
response = append(response, []byte(fmt.Sprintf("%d", result))...)
case float32, float64:
response = append(response, []byte(fmt.Sprintf("%f", result))...)
case []byte:
response = append(response, result...)
default:
j, _ := json.Marshal(result)
response = append(response, j...)
}
}
pmsg, err := websocket.NewPreparedMessage(websocket.BinaryMessage, response)
if err != nil {
logger.Println("websocket.NewPreparedMessage failed :", err)
} else {
c.WritePreparedMessage(pmsg)
}
} else {
cmd := make([]byte, flag[0])
r.Read(cmd)
ws.call(peer, string(cmd), r)
}
}
}
}(conn, accid, sk)
}
func (ws *websocketPeerHandler[T]) upgrade_noauth(w http.ResponseWriter, r *http.Request) {
// 클라이언트 접속
defer func() {
s := recover()
if s != nil {
logger.Error(s)
}
io.Copy(io.Discard, r.Body)
r.Body.Close()
}()
sk := r.Header.Get("AS-X-SESSION")
var accid primitive.ObjectID
if len(sk) > 0 {
authinfo, err := ws.sessionConsumer.Query(sk)
if err == nil {
accid = authinfo.Account
}
}
if accid.IsZero() {
auth := strings.Split(r.Header.Get("Authorization"), " ")
if len(auth) != 2 {
w.WriteHeader(http.StatusBadRequest)
return
}
temp, err := hex.DecodeString(auth[1])
if err != nil {
w.WriteHeader(http.StatusBadRequest)
return
}
if len(temp) != len(primitive.NilObjectID) {
w.WriteHeader(http.StatusBadRequest)
return
}
raw := (*[12]byte)(temp)
accid = primitive.ObjectID(*raw)
}
var upgrader = websocket.Upgrader{} // use default options
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
// var alias string
// if v := r.Header.Get("AS-X-ALIAS"); len(v) > 0 {
// vt, _ := base64.StdEncoding.DecodeString(v)
// alias = string(vt)
// } else {
// alias = accid.Hex()
// }
ws.upgrade_core(conn, accid, sk)
}
func (ws *websocketPeerHandler[T]) upgrade(w http.ResponseWriter, r *http.Request) {
// 클라이언트 접속
defer func() {
s := recover()
if s != nil {
logger.Error(s)
}
io.Copy(io.Discard, r.Body)
r.Body.Close()
}()
sk := r.Header.Get("AS-X-SESSION")
authinfo, err := ws.sessionConsumer.Query(sk)
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
logger.Error("authorize query failed :", err)
return
}
if authinfo.Account.IsZero() || authinfo.Invalidated() {
w.WriteHeader(http.StatusUnauthorized)
return
}
var upgrader = websocket.Upgrader{} // use default options
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
// var alias string
// if v := r.Header.Get("AS-X-ALIAS"); len(v) > 0 {
// vt, _ := base64.StdEncoding.DecodeString(v)
// alias = string(vt)
// } else {
// alias = authinfo.Account.Hex()
// }
ws.upgrade_core(conn, authinfo.Account, sk)
}

View File

@ -13,7 +13,6 @@ import (
"crypto/x509" "crypto/x509"
"errors" "errors"
"io" "io"
"log"
"net/http" "net/http"
"os" "os"
"strings" "strings"
@ -235,7 +234,7 @@ func deflate(inflated []byte) string {
zr := flate.NewReader(byteReader) zr := flate.NewReader(byteReader)
if _, err := io.Copy(wBuf, zr); err != nil { if _, err := io.Copy(wBuf, zr); err != nil {
log.Fatal(err) panic(err)
} }
return wBuf.String() return wBuf.String()