1// Autogenerated by Thrift Compiler (0.9.3)
2// DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
3
4package jaeger
5
6import (
7	"bytes"
8	"fmt"
9	"github.com/uber/jaeger-client-go/thrift"
10)
11
12// (needed to ensure safety because of naive import list construction.)
13var _ = thrift.ZERO
14var _ = fmt.Printf
15var _ = bytes.Equal
16
17type Agent interface {
18	// Parameters:
19	//  - Batch
20	EmitBatch(batch *Batch) (err error)
21}
22
23type AgentClient struct {
24	Transport       thrift.TTransport
25	ProtocolFactory thrift.TProtocolFactory
26	InputProtocol   thrift.TProtocol
27	OutputProtocol  thrift.TProtocol
28	SeqId           int32
29}
30
31func NewAgentClientFactory(t thrift.TTransport, f thrift.TProtocolFactory) *AgentClient {
32	return &AgentClient{Transport: t,
33		ProtocolFactory: f,
34		InputProtocol:   f.GetProtocol(t),
35		OutputProtocol:  f.GetProtocol(t),
36		SeqId:           0,
37	}
38}
39
40func NewAgentClientProtocol(t thrift.TTransport, iprot thrift.TProtocol, oprot thrift.TProtocol) *AgentClient {
41	return &AgentClient{Transport: t,
42		ProtocolFactory: nil,
43		InputProtocol:   iprot,
44		OutputProtocol:  oprot,
45		SeqId:           0,
46	}
47}
48
49// Parameters:
50//  - Batch
51func (p *AgentClient) EmitBatch(batch *Batch) (err error) {
52	if err = p.sendEmitBatch(batch); err != nil {
53		return
54	}
55	return
56}
57
58func (p *AgentClient) sendEmitBatch(batch *Batch) (err error) {
59	oprot := p.OutputProtocol
60	if oprot == nil {
61		oprot = p.ProtocolFactory.GetProtocol(p.Transport)
62		p.OutputProtocol = oprot
63	}
64	p.SeqId++
65	if err = oprot.WriteMessageBegin("emitBatch", thrift.ONEWAY, p.SeqId); err != nil {
66		return
67	}
68	args := AgentEmitBatchArgs{
69		Batch: batch,
70	}
71	if err = args.Write(oprot); err != nil {
72		return
73	}
74	if err = oprot.WriteMessageEnd(); err != nil {
75		return
76	}
77	return oprot.Flush()
78}
79
80type AgentProcessor struct {
81	processorMap map[string]thrift.TProcessorFunction
82	handler      Agent
83}
84
85func (p *AgentProcessor) AddToProcessorMap(key string, processor thrift.TProcessorFunction) {
86	p.processorMap[key] = processor
87}
88
89func (p *AgentProcessor) GetProcessorFunction(key string) (processor thrift.TProcessorFunction, ok bool) {
90	processor, ok = p.processorMap[key]
91	return processor, ok
92}
93
94func (p *AgentProcessor) ProcessorMap() map[string]thrift.TProcessorFunction {
95	return p.processorMap
96}
97
98func NewAgentProcessor(handler Agent) *AgentProcessor {
99
100	self6 := &AgentProcessor{handler: handler, processorMap: make(map[string]thrift.TProcessorFunction)}
101	self6.processorMap["emitBatch"] = &agentProcessorEmitBatch{handler: handler}
102	return self6
103}
104
105func (p *AgentProcessor) Process(iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) {
106	name, _, seqId, err := iprot.ReadMessageBegin()
107	if err != nil {
108		return false, err
109	}
110	if processor, ok := p.GetProcessorFunction(name); ok {
111		return processor.Process(seqId, iprot, oprot)
112	}
113	iprot.Skip(thrift.STRUCT)
114	iprot.ReadMessageEnd()
115	x7 := thrift.NewTApplicationException(thrift.UNKNOWN_METHOD, "Unknown function "+name)
116	oprot.WriteMessageBegin(name, thrift.EXCEPTION, seqId)
117	x7.Write(oprot)
118	oprot.WriteMessageEnd()
119	oprot.Flush()
120	return false, x7
121
122}
123
124type agentProcessorEmitBatch struct {
125	handler Agent
126}
127
128func (p *agentProcessorEmitBatch) Process(seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) {
129	args := AgentEmitBatchArgs{}
130	if err = args.Read(iprot); err != nil {
131		iprot.ReadMessageEnd()
132		return false, err
133	}
134
135	iprot.ReadMessageEnd()
136	var err2 error
137	if err2 = p.handler.EmitBatch(args.Batch); err2 != nil {
138		return true, err2
139	}
140	return true, nil
141}
142
143// HELPER FUNCTIONS AND STRUCTURES
144
145// Attributes:
146//  - Batch
147type AgentEmitBatchArgs struct {
148	Batch *Batch `thrift:"batch,1" json:"batch"`
149}
150
151func NewAgentEmitBatchArgs() *AgentEmitBatchArgs {
152	return &AgentEmitBatchArgs{}
153}
154
155var AgentEmitBatchArgs_Batch_DEFAULT *Batch
156
157func (p *AgentEmitBatchArgs) GetBatch() *Batch {
158	if !p.IsSetBatch() {
159		return AgentEmitBatchArgs_Batch_DEFAULT
160	}
161	return p.Batch
162}
163func (p *AgentEmitBatchArgs) IsSetBatch() bool {
164	return p.Batch != nil
165}
166
167func (p *AgentEmitBatchArgs) Read(iprot thrift.TProtocol) error {
168	if _, err := iprot.ReadStructBegin(); err != nil {
169		return thrift.PrependError(fmt.Sprintf("%T read error: ", p), err)
170	}
171
172	for {
173		_, fieldTypeId, fieldId, err := iprot.ReadFieldBegin()
174		if err != nil {
175			return thrift.PrependError(fmt.Sprintf("%T field %d read error: ", p, fieldId), err)
176		}
177		if fieldTypeId == thrift.STOP {
178			break
179		}
180		switch fieldId {
181		case 1:
182			if err := p.readField1(iprot); err != nil {
183				return err
184			}
185		default:
186			if err := iprot.Skip(fieldTypeId); err != nil {
187				return err
188			}
189		}
190		if err := iprot.ReadFieldEnd(); err != nil {
191			return err
192		}
193	}
194	if err := iprot.ReadStructEnd(); err != nil {
195		return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err)
196	}
197	return nil
198}
199
200func (p *AgentEmitBatchArgs) readField1(iprot thrift.TProtocol) error {
201	p.Batch = &Batch{}
202	if err := p.Batch.Read(iprot); err != nil {
203		return thrift.PrependError(fmt.Sprintf("%T error reading struct: ", p.Batch), err)
204	}
205	return nil
206}
207
208func (p *AgentEmitBatchArgs) Write(oprot thrift.TProtocol) error {
209	if err := oprot.WriteStructBegin("emitBatch_args"); err != nil {
210		return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err)
211	}
212	if err := p.writeField1(oprot); err != nil {
213		return err
214	}
215	if err := oprot.WriteFieldStop(); err != nil {
216		return thrift.PrependError("write field stop error: ", err)
217	}
218	if err := oprot.WriteStructEnd(); err != nil {
219		return thrift.PrependError("write struct stop error: ", err)
220	}
221	return nil
222}
223
224func (p *AgentEmitBatchArgs) writeField1(oprot thrift.TProtocol) (err error) {
225	if err := oprot.WriteFieldBegin("batch", thrift.STRUCT, 1); err != nil {
226		return thrift.PrependError(fmt.Sprintf("%T write field begin error 1:batch: ", p), err)
227	}
228	if err := p.Batch.Write(oprot); err != nil {
229		return thrift.PrependError(fmt.Sprintf("%T error writing struct: ", p.Batch), err)
230	}
231	if err := oprot.WriteFieldEnd(); err != nil {
232		return thrift.PrependError(fmt.Sprintf("%T write field end error 1:batch: ", p), err)
233	}
234	return err
235}
236
237func (p *AgentEmitBatchArgs) String() string {
238	if p == nil {
239		return "<nil>"
240	}
241	return fmt.Sprintf("AgentEmitBatchArgs(%+v)", *p)
242}
243