blob: 32d5b0147a2f001e6855472447fbc32c175d6f35 [file] [log] [blame]
Naveen Sampath04696f72022-06-13 15:19:14 +05301/*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19
20package thrift
21
22import (
23 "context"
24)
25
26const (
27 UNKNOWN_APPLICATION_EXCEPTION = 0
28 UNKNOWN_METHOD = 1
29 INVALID_MESSAGE_TYPE_EXCEPTION = 2
30 WRONG_METHOD_NAME = 3
31 BAD_SEQUENCE_ID = 4
32 MISSING_RESULT = 5
33 INTERNAL_ERROR = 6
34 PROTOCOL_ERROR = 7
35 INVALID_TRANSFORM = 8
36 INVALID_PROTOCOL = 9
37 UNSUPPORTED_CLIENT_TYPE = 10
38)
39
40var defaultApplicationExceptionMessage = map[int32]string{
41 UNKNOWN_APPLICATION_EXCEPTION: "unknown application exception",
42 UNKNOWN_METHOD: "unknown method",
43 INVALID_MESSAGE_TYPE_EXCEPTION: "invalid message type",
44 WRONG_METHOD_NAME: "wrong method name",
45 BAD_SEQUENCE_ID: "bad sequence ID",
46 MISSING_RESULT: "missing result",
47 INTERNAL_ERROR: "unknown internal error",
48 PROTOCOL_ERROR: "unknown protocol error",
49 INVALID_TRANSFORM: "Invalid transform",
50 INVALID_PROTOCOL: "Invalid protocol",
51 UNSUPPORTED_CLIENT_TYPE: "Unsupported client type",
52}
53
54// Application level Thrift exception
55type TApplicationException interface {
56 TException
57 TypeId() int32
58 Read(ctx context.Context, iprot TProtocol) error
59 Write(ctx context.Context, oprot TProtocol) error
60}
61
62type tApplicationException struct {
63 message string
64 type_ int32
65}
66
67var _ TApplicationException = (*tApplicationException)(nil)
68
69func (tApplicationException) TExceptionType() TExceptionType {
70 return TExceptionTypeApplication
71}
72
73func (e tApplicationException) Error() string {
74 if e.message != "" {
75 return e.message
76 }
77 return defaultApplicationExceptionMessage[e.type_]
78}
79
80func NewTApplicationException(type_ int32, message string) TApplicationException {
81 return &tApplicationException{message, type_}
82}
83
84func (p *tApplicationException) TypeId() int32 {
85 return p.type_
86}
87
88func (p *tApplicationException) Read(ctx context.Context, iprot TProtocol) error {
89 // TODO: this should really be generated by the compiler
90 _, err := iprot.ReadStructBegin(ctx)
91 if err != nil {
92 return err
93 }
94
95 message := ""
96 type_ := int32(UNKNOWN_APPLICATION_EXCEPTION)
97
98 for {
99 _, ttype, id, err := iprot.ReadFieldBegin(ctx)
100 if err != nil {
101 return err
102 }
103 if ttype == STOP {
104 break
105 }
106 switch id {
107 case 1:
108 if ttype == STRING {
109 if message, err = iprot.ReadString(ctx); err != nil {
110 return err
111 }
112 } else {
113 if err = SkipDefaultDepth(ctx, iprot, ttype); err != nil {
114 return err
115 }
116 }
117 case 2:
118 if ttype == I32 {
119 if type_, err = iprot.ReadI32(ctx); err != nil {
120 return err
121 }
122 } else {
123 if err = SkipDefaultDepth(ctx, iprot, ttype); err != nil {
124 return err
125 }
126 }
127 default:
128 if err = SkipDefaultDepth(ctx, iprot, ttype); err != nil {
129 return err
130 }
131 }
132 if err = iprot.ReadFieldEnd(ctx); err != nil {
133 return err
134 }
135 }
136 if err := iprot.ReadStructEnd(ctx); err != nil {
137 return err
138 }
139
140 p.message = message
141 p.type_ = type_
142
143 return nil
144}
145
146func (p *tApplicationException) Write(ctx context.Context, oprot TProtocol) (err error) {
147 err = oprot.WriteStructBegin(ctx, "TApplicationException")
148 if len(p.Error()) > 0 {
149 err = oprot.WriteFieldBegin(ctx, "message", STRING, 1)
150 if err != nil {
151 return
152 }
153 err = oprot.WriteString(ctx, p.Error())
154 if err != nil {
155 return
156 }
157 err = oprot.WriteFieldEnd(ctx)
158 if err != nil {
159 return
160 }
161 }
162 err = oprot.WriteFieldBegin(ctx, "type", I32, 2)
163 if err != nil {
164 return
165 }
166 err = oprot.WriteI32(ctx, p.type_)
167 if err != nil {
168 return
169 }
170 err = oprot.WriteFieldEnd(ctx)
171 if err != nil {
172 return
173 }
174 err = oprot.WriteFieldStop(ctx)
175 if err != nil {
176 return
177 }
178 err = oprot.WriteStructEnd(ctx)
179 return
180}