-
Notifications
You must be signed in to change notification settings - Fork 568
/
provider.ts
86 lines (80 loc) 路 1.82 KB
/
provider.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import {
saveClassMetadata,
saveModule,
MS_PROVIDER_KEY,
attachClassMetadata,
savePropertyMetadata,
MS_GRPC_METHOD_KEY,
MS_DUBBO_METHOD_KEY,
Provide,
Scope,
} from '../';
import { GRPCMetadata, MSProviderType, ScopeEnum } from '../../interface';
export function Provider(
type: MSProviderType.GRPC,
metadata?: GRPCMetadata.ProviderOptions
): ClassDecorator;
export function Provider(
type: MSProviderType.DUBBO,
metadata?: any
): ClassDecorator;
export function Provider(
type: MSProviderType,
metadata: any = {}
): ClassDecorator {
return (target: any) => {
saveModule(MS_PROVIDER_KEY, target);
saveClassMetadata(
MS_PROVIDER_KEY,
{
type,
metadata,
},
target
);
Scope(ScopeEnum.Request)(target);
Provide()(target);
};
}
export enum GrpcStreamTypeEnum {
BASE = 'base',
DUPLEX = 'ServerDuplexStream',
READABLE = 'ServerReadableStream',
WRITEABLE = 'ServerWritableStream',
}
export function GrpcMethod(
methodOptions: {
methodName?: string;
type?: GrpcStreamTypeEnum;
onEnd?: string;
} = {}
): MethodDecorator {
return (target, propertyName, descriptor: PropertyDescriptor) => {
if (!methodOptions.type) {
methodOptions.type = GrpcStreamTypeEnum.BASE;
}
savePropertyMetadata(
MS_GRPC_METHOD_KEY,
{
methodName: methodOptions.methodName || propertyName,
type: methodOptions.type,
onEnd: methodOptions.onEnd,
},
target,
propertyName
);
return descriptor;
};
}
export function DubboMethod(methodName?: string): MethodDecorator {
return (target, propertyName, descriptor: PropertyDescriptor) => {
attachClassMetadata(
MS_DUBBO_METHOD_KEY,
{
methodName: methodName || propertyName,
},
target
);
return descriptor;
};
}