diff --git a/backend/api/handler/coze/loop/apis/observability_trace_service.go b/backend/api/handler/coze/loop/apis/observability_trace_service.go index a8cdd4f18..c6ee025d5 100644 --- a/backend/api/handler/coze/loop/apis/observability_trace_service.go +++ b/backend/api/handler/coze/loop/apis/observability_trace_service.go @@ -178,3 +178,15 @@ func GetTrajectoryConfig(ctx context.Context, c *app.RequestContext) { func ListTrajectory(ctx context.Context, c *app.RequestContext) { invokeAndRender(ctx, c, observabilityClient.ListTrajectory) } + +// ListWorkspaceAnnotations . +// @router /api/observability/v1/annotations/list_by_workspace [POST] +func ListWorkspaceAnnotations(ctx context.Context, c *app.RequestContext) { + invokeAndRender(ctx, c, observabilityClient.ListWorkspaceAnnotations) +} + +// ListMetadata . +// @router /api/observability/v1/traces/metadata/list [POST] +func ListMetadata(ctx context.Context, c *app.RequestContext) { + invokeAndRender(ctx, c, observabilityClient.ListMetadata) +} diff --git a/backend/api/handler/coze/loop/apis/wire_gen.go b/backend/api/handler/coze/loop/apis/wire_gen.go index e0ef0528a..2b20f061c 100644 --- a/backend/api/handler/coze/loop/apis/wire_gen.go +++ b/backend/api/handler/coze/loop/apis/wire_gen.go @@ -45,6 +45,7 @@ import ( application3 "github.com/coze-dev/coze-loop/backend/modules/llm/application" application6 "github.com/coze-dev/coze-loop/backend/modules/observability/application" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/storage" + taskhook "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/task" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/service/taskexe/processor" application2 "github.com/coze-dev/coze-loop/backend/modules/prompt/application" "github.com/coze-dev/coze-loop/backend/pkg/conf" @@ -181,7 +182,7 @@ func InitObservabilityHandler(ctx context.Context, db2 db.Provider, ckDb ck.Prov if err != nil { return nil, err } - iTaskApplication, err := application6.InitTaskApplication(db2, idgen2, configFactory, benefit2, ckDb, meter, redis2, mqFactory, userClient, authCli, evalClient, evalSetClient, experimentClient, datasetClient, fileClient, taskProcessor, aid, persistentCmdable) + iTaskApplication, err := application6.InitTaskApplication(db2, idgen2, configFactory, benefit2, ckDb, meter, redis2, mqFactory, userClient, authCli, evalClient, evalSetClient, experimentClient, datasetClient, fileClient, taskProcessor, aid, persistentCmdable, taskhook.NewNoopTaskHookProvider()) if err != nil { return nil, err } diff --git a/backend/api/router/coze/loop/apis/coze.loop.apis.go b/backend/api/router/coze/loop/apis/coze.loop.apis.go index f8a89cb2e..fad4e2cf5 100644 --- a/backend/api/router/coze/loop/apis/coze.loop.apis.go +++ b/backend/api/router/coze/loop/apis/coze.loop.apis.go @@ -329,6 +329,7 @@ func Register(r *server.Hertz, handler *apis.APIHandler) { _annotations.DELETE("/:annotation_id", append(_deletemanualannotationMw(handler), apis.DeleteManualAnnotation)...) _annotations.PUT("/:annotation_id", append(_updatemanualannotationMw(handler), apis.UpdateManualAnnotation)...) _annotations.POST("/list", append(_listannotationsMw(handler), apis.ListAnnotations)...) + _annotations.POST("/list_by_workspace", append(_listworkspaceannotationsMw(handler), apis.ListWorkspaceAnnotations)...) _v14.POST("/tasks", append(_tasksMw(handler), apis.CreateTask)...) _tasks := _v14.Group("/tasks", _tasksMw(handler)...) _tasks.POST("/list", append(_listtasksMw(handler), apis.ListTasks)...) @@ -373,6 +374,10 @@ func Register(r *server.Hertz, handler *apis.APIHandler) { _traces.POST("/trajectory", append(_listtrajectoryMw(handler), apis.ListTrajectory)...) _traces.GET("/trajectory_config", append(_gettrajectoryconfigMw(handler), apis.GetTrajectoryConfig)...) _traces.POST("/trajectory_config", append(_upserttrajectoryconfigMw(handler), apis.UpsertTrajectoryConfig)...) + { + _metadata := _traces.Group("/metadata", _metadataMw(handler)...) + _metadata.POST("/list", append(_listmetadataMw(handler), apis.ListMetadata)...) + } } } } diff --git a/backend/api/router/coze/loop/apis/middleware.go b/backend/api/router/coze/loop/apis/middleware.go index 7271f85ae..4a65c3194 100644 --- a/backend/api/router/coze/loop/apis/middleware.go +++ b/backend/api/router/coze/loop/apis/middleware.go @@ -1850,6 +1850,21 @@ func _checkexperimenttemplatenameMw(handler *apis.APIHandler) []app.HandlerFunc return nil } +func _listworkspaceannotationsMw(handler *apis.APIHandler) []app.HandlerFunc { + // your code... + return nil +} + +func _metadataMw(handler *apis.APIHandler) []app.HandlerFunc { + // your code... + return nil +} + +func _listmetadataMw(handler *apis.APIHandler) []app.HandlerFunc { + // your code... + return nil +} + func _deleteevaluatoroapiMw(handler *apis.APIHandler) []app.HandlerFunc { // your code... return nil diff --git a/backend/kitex_gen/coze/loop/apis/observabilitytraceservice/client.go b/backend/kitex_gen/coze/loop/apis/observabilitytraceservice/client.go index 146ab0d78..3a1673bca 100644 --- a/backend/kitex_gen/coze/loop/apis/observabilitytraceservice/client.go +++ b/backend/kitex_gen/coze/loop/apis/observabilitytraceservice/client.go @@ -26,6 +26,7 @@ type Client interface { UpdateManualAnnotation(ctx context.Context, req *trace.UpdateManualAnnotationRequest, callOptions ...callopt.Option) (r *trace.UpdateManualAnnotationResponse, err error) DeleteManualAnnotation(ctx context.Context, req *trace.DeleteManualAnnotationRequest, callOptions ...callopt.Option) (r *trace.DeleteManualAnnotationResponse, err error) ListAnnotations(ctx context.Context, req *trace.ListAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListAnnotationsResponse, err error) + ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListWorkspaceAnnotationsResponse, err error) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.ExportTracesToDatasetResponse, err error) PreviewExportTracesToDataset(ctx context.Context, req *trace.PreviewExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.PreviewExportTracesToDatasetResponse, err error) ChangeEvaluatorScore(ctx context.Context, req *trace.ChangeEvaluatorScoreRequest, callOptions ...callopt.Option) (r *trace.ChangeEvaluatorScoreResponse, err error) @@ -34,6 +35,7 @@ type Client interface { UpsertTrajectoryConfig(ctx context.Context, req *trace.UpsertTrajectoryConfigRequest, callOptions ...callopt.Option) (r *trace.UpsertTrajectoryConfigResponse, err error) GetTrajectoryConfig(ctx context.Context, req *trace.GetTrajectoryConfigRequest, callOptions ...callopt.Option) (r *trace.GetTrajectoryConfigResponse, err error) ListTrajectory(ctx context.Context, req *trace.ListTrajectoryRequest, callOptions ...callopt.Option) (r *trace.ListTrajectoryResponse, err error) + ListMetadata(ctx context.Context, req *trace.ListMetadataRequest, callOptions ...callopt.Option) (r *trace.ListMetadataResponse, err error) } // NewClient creates a client for the service defined in IDL. @@ -140,6 +142,11 @@ func (p *kObservabilityTraceServiceClient) ListAnnotations(ctx context.Context, return p.kClient.ListAnnotations(ctx, req) } +func (p *kObservabilityTraceServiceClient) ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListWorkspaceAnnotationsResponse, err error) { + ctx = client.NewCtxWithCallOptions(ctx, callOptions) + return p.kClient.ListWorkspaceAnnotations(ctx, req) +} + func (p *kObservabilityTraceServiceClient) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.ExportTracesToDatasetResponse, err error) { ctx = client.NewCtxWithCallOptions(ctx, callOptions) return p.kClient.ExportTracesToDataset(ctx, req) @@ -179,3 +186,8 @@ func (p *kObservabilityTraceServiceClient) ListTrajectory(ctx context.Context, r ctx = client.NewCtxWithCallOptions(ctx, callOptions) return p.kClient.ListTrajectory(ctx, req) } + +func (p *kObservabilityTraceServiceClient) ListMetadata(ctx context.Context, req *trace.ListMetadataRequest, callOptions ...callopt.Option) (r *trace.ListMetadataResponse, err error) { + ctx = client.NewCtxWithCallOptions(ctx, callOptions) + return p.kClient.ListMetadata(ctx, req) +} diff --git a/backend/kitex_gen/coze/loop/apis/observabilitytraceservice/observabilitytraceservice.go b/backend/kitex_gen/coze/loop/apis/observabilitytraceservice/observabilitytraceservice.go index 694adf6b3..6ec9c5fa0 100644 --- a/backend/kitex_gen/coze/loop/apis/observabilitytraceservice/observabilitytraceservice.go +++ b/backend/kitex_gen/coze/loop/apis/observabilitytraceservice/observabilitytraceservice.go @@ -119,6 +119,13 @@ var serviceMethods = map[string]kitex.MethodInfo{ false, kitex.WithStreamingMode(kitex.StreamingNone), ), + "ListWorkspaceAnnotations": kitex.NewMethodInfo( + listWorkspaceAnnotationsHandler, + newTraceServiceListWorkspaceAnnotationsArgs, + newTraceServiceListWorkspaceAnnotationsResult, + false, + kitex.WithStreamingMode(kitex.StreamingNone), + ), "ExportTracesToDataset": kitex.NewMethodInfo( exportTracesToDatasetHandler, newTraceServiceExportTracesToDatasetArgs, @@ -175,6 +182,13 @@ var serviceMethods = map[string]kitex.MethodInfo{ false, kitex.WithStreamingMode(kitex.StreamingNone), ), + "ListMetadata": kitex.NewMethodInfo( + listMetadataHandler, + newTraceServiceListMetadataArgs, + newTraceServiceListMetadataResult, + false, + kitex.WithStreamingMode(kitex.StreamingNone), + ), } var ( @@ -493,6 +507,25 @@ func newTraceServiceListAnnotationsResult() interface{} { return trace.NewTraceServiceListAnnotationsResult() } +func listWorkspaceAnnotationsHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { + realArg := arg.(*trace.TraceServiceListWorkspaceAnnotationsArgs) + realResult := result.(*trace.TraceServiceListWorkspaceAnnotationsResult) + success, err := handler.(trace.TraceService).ListWorkspaceAnnotations(ctx, realArg.Req) + if err != nil { + return err + } + realResult.Success = success + return nil +} + +func newTraceServiceListWorkspaceAnnotationsArgs() interface{} { + return trace.NewTraceServiceListWorkspaceAnnotationsArgs() +} + +func newTraceServiceListWorkspaceAnnotationsResult() interface{} { + return trace.NewTraceServiceListWorkspaceAnnotationsResult() +} + func exportTracesToDatasetHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { realArg := arg.(*trace.TraceServiceExportTracesToDatasetArgs) realResult := result.(*trace.TraceServiceExportTracesToDatasetResult) @@ -645,6 +678,25 @@ func newTraceServiceListTrajectoryResult() interface{} { return trace.NewTraceServiceListTrajectoryResult() } +func listMetadataHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { + realArg := arg.(*trace.TraceServiceListMetadataArgs) + realResult := result.(*trace.TraceServiceListMetadataResult) + success, err := handler.(trace.TraceService).ListMetadata(ctx, realArg.Req) + if err != nil { + return err + } + realResult.Success = success + return nil +} + +func newTraceServiceListMetadataArgs() interface{} { + return trace.NewTraceServiceListMetadataArgs() +} + +func newTraceServiceListMetadataResult() interface{} { + return trace.NewTraceServiceListMetadataResult() +} + type kClient struct { c client.Client sc client.Streaming @@ -807,6 +859,16 @@ func (p *kClient) ListAnnotations(ctx context.Context, req *trace.ListAnnotation return _result.GetSuccess(), nil } +func (p *kClient) ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest) (r *trace.ListWorkspaceAnnotationsResponse, err error) { + var _args trace.TraceServiceListWorkspaceAnnotationsArgs + _args.Req = req + var _result trace.TraceServiceListWorkspaceAnnotationsResult + if err = p.c.Call(ctx, "ListWorkspaceAnnotations", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} + func (p *kClient) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest) (r *trace.ExportTracesToDatasetResponse, err error) { var _args trace.TraceServiceExportTracesToDatasetArgs _args.Req = req @@ -886,3 +948,13 @@ func (p *kClient) ListTrajectory(ctx context.Context, req *trace.ListTrajectoryR } return _result.GetSuccess(), nil } + +func (p *kClient) ListMetadata(ctx context.Context, req *trace.ListMetadataRequest) (r *trace.ListMetadataResponse, err error) { + var _args trace.TraceServiceListMetadataArgs + _args.Req = req + var _result trace.TraceServiceListMetadataResult + if err = p.c.Call(ctx, "ListMetadata", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} diff --git a/backend/kitex_gen/coze/loop/observability/domain/annotation/annotation.go b/backend/kitex_gen/coze/loop/observability/domain/annotation/annotation.go index 3f494f41b..a1b49f0e3 100644 --- a/backend/kitex_gen/coze/loop/observability/domain/annotation/annotation.go +++ b/backend/kitex_gen/coze/loop/observability/domain/annotation/annotation.go @@ -698,6 +698,8 @@ type AutoEvaluate struct { EvaluatorResult_ *EvaluatorResult_ `thrift:"evaluator_result,4,optional" frugal:"4,optional,EvaluatorResult_" form:"evaluator_result" json:"evaluator_result,omitempty" query:"evaluator_result"` RecordID int64 `thrift:"record_id,5,required" frugal:"5,required,i64" json:"record_id" form:"record_id,required" query:"record_id,required"` TaskID string `thrift:"task_id,6,required" frugal:"6,required,string" form:"task_id,required" json:"task_id,required" query:"task_id,required"` + ExptID *int64 `thrift:"expt_id,7,optional" frugal:"7,optional,i64" json:"expt_id" form:"expt_id" query:"expt_id"` + ExptTemplateID *int64 `thrift:"expt_template_id,8,optional" frugal:"8,optional,i64" json:"expt_template_id" form:"expt_template_id" query:"expt_template_id"` } func NewAutoEvaluate() *AutoEvaluate { @@ -753,6 +755,30 @@ func (p *AutoEvaluate) GetTaskID() (v string) { } return } + +var AutoEvaluate_ExptID_DEFAULT int64 + +func (p *AutoEvaluate) GetExptID() (v int64) { + if p == nil { + return + } + if !p.IsSetExptID() { + return AutoEvaluate_ExptID_DEFAULT + } + return *p.ExptID +} + +var AutoEvaluate_ExptTemplateID_DEFAULT int64 + +func (p *AutoEvaluate) GetExptTemplateID() (v int64) { + if p == nil { + return + } + if !p.IsSetExptTemplateID() { + return AutoEvaluate_ExptTemplateID_DEFAULT + } + return *p.ExptTemplateID +} func (p *AutoEvaluate) SetEvaluatorVersionID(val int64) { p.EvaluatorVersionID = val } @@ -771,6 +797,12 @@ func (p *AutoEvaluate) SetRecordID(val int64) { func (p *AutoEvaluate) SetTaskID(val string) { p.TaskID = val } +func (p *AutoEvaluate) SetExptID(val *int64) { + p.ExptID = val +} +func (p *AutoEvaluate) SetExptTemplateID(val *int64) { + p.ExptTemplateID = val +} var fieldIDToName_AutoEvaluate = map[int16]string{ 1: "evaluator_version_id", @@ -779,12 +811,22 @@ var fieldIDToName_AutoEvaluate = map[int16]string{ 4: "evaluator_result", 5: "record_id", 6: "task_id", + 7: "expt_id", + 8: "expt_template_id", } func (p *AutoEvaluate) IsSetEvaluatorResult_() bool { return p.EvaluatorResult_ != nil } +func (p *AutoEvaluate) IsSetExptID() bool { + return p.ExptID != nil +} + +func (p *AutoEvaluate) IsSetExptTemplateID() bool { + return p.ExptTemplateID != nil +} + func (p *AutoEvaluate) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -861,6 +903,22 @@ func (p *AutoEvaluate) Read(iprot thrift.TProtocol) (err error) { } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } + case 7: + if fieldTypeId == thrift.I64 { + if err = p.ReadField7(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 8: + if fieldTypeId == thrift.I64 { + if err = p.ReadField8(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } default: if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError @@ -979,6 +1037,28 @@ func (p *AutoEvaluate) ReadField6(iprot thrift.TProtocol) error { p.TaskID = _field return nil } +func (p *AutoEvaluate) ReadField7(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v + } + p.ExptID = _field + return nil +} +func (p *AutoEvaluate) ReadField8(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v + } + p.ExptTemplateID = _field + return nil +} func (p *AutoEvaluate) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 @@ -1010,6 +1090,14 @@ func (p *AutoEvaluate) Write(oprot thrift.TProtocol) (err error) { fieldId = 6 goto WriteFieldError } + if err = p.writeField7(oprot); err != nil { + fieldId = 7 + goto WriteFieldError + } + if err = p.writeField8(oprot); err != nil { + fieldId = 8 + goto WriteFieldError + } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -1126,6 +1214,42 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) } +func (p *AutoEvaluate) writeField7(oprot thrift.TProtocol) (err error) { + if p.IsSetExptID() { + if err = oprot.WriteFieldBegin("expt_id", thrift.I64, 7); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.ExptID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 7 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 7 end error: ", p), err) +} +func (p *AutoEvaluate) writeField8(oprot thrift.TProtocol) (err error) { + if p.IsSetExptTemplateID() { + if err = oprot.WriteFieldBegin("expt_template_id", thrift.I64, 8); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.ExptTemplateID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 8 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 8 end error: ", p), err) +} func (p *AutoEvaluate) String() string { if p == nil { @@ -1159,6 +1283,12 @@ func (p *AutoEvaluate) DeepEqual(ano *AutoEvaluate) bool { if !p.Field6DeepEqual(ano.TaskID) { return false } + if !p.Field7DeepEqual(ano.ExptID) { + return false + } + if !p.Field8DeepEqual(ano.ExptTemplateID) { + return false + } return true } @@ -1204,6 +1334,30 @@ func (p *AutoEvaluate) Field6DeepEqual(src string) bool { } return true } +func (p *AutoEvaluate) Field7DeepEqual(src *int64) bool { + + if p.ExptID == src { + return true + } else if p.ExptID == nil || src == nil { + return false + } + if *p.ExptID != *src { + return false + } + return true +} +func (p *AutoEvaluate) Field8DeepEqual(src *int64) bool { + + if p.ExptTemplateID == src { + return true + } else if p.ExptTemplateID == nil || src == nil { + return false + } + if *p.ExptTemplateID != *src { + return false + } + return true +} type ManualFeedback struct { TagKeyID int64 `thrift:"tag_key_id,1,required" frugal:"1,required,i64" json:"tag_key_id" form:"tag_key_id,required" query:"tag_key_id,required"` @@ -3066,3 +3220,330 @@ func (p *AnnotationEvaluator) Field3DeepEqual(src string) bool { } return true } + +type SimpleAnnotationInfo struct { + Key string `thrift:"key,1,required" frugal:"1,required,string" form:"key,required" json:"key,required" query:"key,required"` + AnnotationType *AnnotationType `thrift:"annotation_type,2,optional" frugal:"2,optional,string" form:"annotation_type" json:"annotation_type,omitempty" query:"annotation_type"` + OriginalKey *string `thrift:"original_key,3,optional" frugal:"3,optional,string" form:"original_key" json:"original_key,omitempty" query:"original_key"` +} + +func NewSimpleAnnotationInfo() *SimpleAnnotationInfo { + return &SimpleAnnotationInfo{} +} + +func (p *SimpleAnnotationInfo) InitDefault() { +} + +func (p *SimpleAnnotationInfo) GetKey() (v string) { + if p != nil { + return p.Key + } + return +} + +var SimpleAnnotationInfo_AnnotationType_DEFAULT AnnotationType + +func (p *SimpleAnnotationInfo) GetAnnotationType() (v AnnotationType) { + if p == nil { + return + } + if !p.IsSetAnnotationType() { + return SimpleAnnotationInfo_AnnotationType_DEFAULT + } + return *p.AnnotationType +} + +var SimpleAnnotationInfo_OriginalKey_DEFAULT string + +func (p *SimpleAnnotationInfo) GetOriginalKey() (v string) { + if p == nil { + return + } + if !p.IsSetOriginalKey() { + return SimpleAnnotationInfo_OriginalKey_DEFAULT + } + return *p.OriginalKey +} +func (p *SimpleAnnotationInfo) SetKey(val string) { + p.Key = val +} +func (p *SimpleAnnotationInfo) SetAnnotationType(val *AnnotationType) { + p.AnnotationType = val +} +func (p *SimpleAnnotationInfo) SetOriginalKey(val *string) { + p.OriginalKey = val +} + +var fieldIDToName_SimpleAnnotationInfo = map[int16]string{ + 1: "key", + 2: "annotation_type", + 3: "original_key", +} + +func (p *SimpleAnnotationInfo) IsSetAnnotationType() bool { + return p.AnnotationType != nil +} + +func (p *SimpleAnnotationInfo) IsSetOriginalKey() bool { + return p.OriginalKey != nil +} + +func (p *SimpleAnnotationInfo) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + var issetKey bool = false + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.STRING { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + issetKey = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 2: + if fieldTypeId == thrift.STRING { + if err = p.ReadField2(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 3: + if fieldTypeId == thrift.STRING { + if err = p.ReadField3(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + if !issetKey { + fieldId = 1 + goto RequiredFieldNotSetError + } + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SimpleAnnotationInfo[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +RequiredFieldNotSetError: + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_SimpleAnnotationInfo[fieldId])) +} + +func (p *SimpleAnnotationInfo) ReadField1(iprot thrift.TProtocol) error { + + var _field string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v + } + p.Key = _field + return nil +} +func (p *SimpleAnnotationInfo) ReadField2(iprot thrift.TProtocol) error { + + var _field *AnnotationType + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.AnnotationType = _field + return nil +} +func (p *SimpleAnnotationInfo) ReadField3(iprot thrift.TProtocol) error { + + var _field *string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.OriginalKey = _field + return nil +} + +func (p *SimpleAnnotationInfo) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("SimpleAnnotationInfo"); err != nil { + goto WriteStructBeginError + } + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } + if err = p.writeField3(oprot); err != nil { + fieldId = 3 + goto WriteFieldError + } + } + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError + } + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) +} + +func (p *SimpleAnnotationInfo) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("key", thrift.STRING, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(p.Key); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *SimpleAnnotationInfo) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetAnnotationType() { + if err = oprot.WriteFieldBegin("annotation_type", thrift.STRING, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.AnnotationType); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} +func (p *SimpleAnnotationInfo) writeField3(oprot thrift.TProtocol) (err error) { + if p.IsSetOriginalKey() { + if err = oprot.WriteFieldBegin("original_key", thrift.STRING, 3); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.OriginalKey); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) +} + +func (p *SimpleAnnotationInfo) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("SimpleAnnotationInfo(%+v)", *p) + +} + +func (p *SimpleAnnotationInfo) DeepEqual(ano *SimpleAnnotationInfo) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.Key) { + return false + } + if !p.Field2DeepEqual(ano.AnnotationType) { + return false + } + if !p.Field3DeepEqual(ano.OriginalKey) { + return false + } + return true +} + +func (p *SimpleAnnotationInfo) Field1DeepEqual(src string) bool { + + if strings.Compare(p.Key, src) != 0 { + return false + } + return true +} +func (p *SimpleAnnotationInfo) Field2DeepEqual(src *AnnotationType) bool { + + if p.AnnotationType == src { + return true + } else if p.AnnotationType == nil || src == nil { + return false + } + if strings.Compare(*p.AnnotationType, *src) != 0 { + return false + } + return true +} +func (p *SimpleAnnotationInfo) Field3DeepEqual(src *string) bool { + + if p.OriginalKey == src { + return true + } else if p.OriginalKey == nil || src == nil { + return false + } + if strings.Compare(*p.OriginalKey, *src) != 0 { + return false + } + return true +} diff --git a/backend/kitex_gen/coze/loop/observability/domain/annotation/annotation_validator.go b/backend/kitex_gen/coze/loop/observability/domain/annotation/annotation_validator.go index 76b7d2d3b..a5b22b7ef 100644 --- a/backend/kitex_gen/coze/loop/observability/domain/annotation/annotation_validator.go +++ b/backend/kitex_gen/coze/loop/observability/domain/annotation/annotation_validator.go @@ -99,3 +99,6 @@ func (p *Annotation) IsValid() error { func (p *AnnotationEvaluator) IsValid() error { return nil } +func (p *SimpleAnnotationInfo) IsValid() error { + return nil +} diff --git a/backend/kitex_gen/coze/loop/observability/domain/annotation/k-annotation.go b/backend/kitex_gen/coze/loop/observability/domain/annotation/k-annotation.go index ba5cf5fb2..8c9366a77 100644 --- a/backend/kitex_gen/coze/loop/observability/domain/annotation/k-annotation.go +++ b/backend/kitex_gen/coze/loop/observability/domain/annotation/k-annotation.go @@ -594,6 +594,34 @@ func (p *AutoEvaluate) FastRead(buf []byte) (int, error) { goto SkipFieldError } } + case 7: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField7(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 8: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField8(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } default: l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -720,6 +748,34 @@ func (p *AutoEvaluate) FastReadField6(buf []byte) (int, error) { return offset, nil } +func (p *AutoEvaluate) FastReadField7(buf []byte) (int, error) { + offset := 0 + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.ExptID = _field + return offset, nil +} + +func (p *AutoEvaluate) FastReadField8(buf []byte) (int, error) { + offset := 0 + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.ExptTemplateID = _field + return offset, nil +} + func (p *AutoEvaluate) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } @@ -729,6 +785,8 @@ func (p *AutoEvaluate) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { if p != nil { offset += p.fastWriteField1(buf[offset:], w) offset += p.fastWriteField5(buf[offset:], w) + offset += p.fastWriteField7(buf[offset:], w) + offset += p.fastWriteField8(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) offset += p.fastWriteField3(buf[offset:], w) offset += p.fastWriteField4(buf[offset:], w) @@ -747,6 +805,8 @@ func (p *AutoEvaluate) BLength() int { l += p.field4Length() l += p.field5Length() l += p.field6Length() + l += p.field7Length() + l += p.field8Length() } l += thrift.Binary.FieldStopLength() return l @@ -796,6 +856,24 @@ func (p *AutoEvaluate) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { return offset } +func (p *AutoEvaluate) fastWriteField7(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetExptID() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 7) + offset += thrift.Binary.WriteI64(buf[offset:], *p.ExptID) + } + return offset +} + +func (p *AutoEvaluate) fastWriteField8(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetExptTemplateID() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 8) + offset += thrift.Binary.WriteI64(buf[offset:], *p.ExptTemplateID) + } + return offset +} + func (p *AutoEvaluate) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() @@ -840,6 +918,24 @@ func (p *AutoEvaluate) field6Length() int { return l } +func (p *AutoEvaluate) field7Length() int { + l := 0 + if p.IsSetExptID() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + } + return l +} + +func (p *AutoEvaluate) field8Length() int { + l := 0 + if p.IsSetExptTemplateID() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + } + return l +} + func (p *AutoEvaluate) DeepCopy(s interface{}) error { src, ok := s.(*AutoEvaluate) if !ok { @@ -871,6 +967,16 @@ func (p *AutoEvaluate) DeepCopy(s interface{}) error { p.TaskID = kutils.StringDeepCopy(src.TaskID) } + if src.ExptID != nil { + tmp := *src.ExptID + p.ExptID = &tmp + } + + if src.ExptTemplateID != nil { + tmp := *src.ExptTemplateID + p.ExptTemplateID = &tmp + } + return nil } @@ -2219,3 +2325,232 @@ func (p *AnnotationEvaluator) DeepCopy(s interface{}) error { return nil } + +func (p *SimpleAnnotationInfo) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + var issetKey bool = false + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetKey = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 3: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField3(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + if !issetKey { + fieldId = 1 + goto RequiredFieldNotSetError + } + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SimpleAnnotationInfo[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_SimpleAnnotationInfo[fieldId])) +} + +func (p *SimpleAnnotationInfo) FastReadField1(buf []byte) (int, error) { + offset := 0 + + var _field string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.Key = _field + return offset, nil +} + +func (p *SimpleAnnotationInfo) FastReadField2(buf []byte) (int, error) { + offset := 0 + + var _field *AnnotationType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.AnnotationType = _field + return offset, nil +} + +func (p *SimpleAnnotationInfo) FastReadField3(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.OriginalKey = _field + return offset, nil +} + +func (p *SimpleAnnotationInfo) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *SimpleAnnotationInfo) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *SimpleAnnotationInfo) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + l += p.field3Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *SimpleAnnotationInfo) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 1) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.Key) + return offset +} + +func (p *SimpleAnnotationInfo) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetAnnotationType() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.AnnotationType) + } + return offset +} + +func (p *SimpleAnnotationInfo) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetOriginalKey() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 3) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.OriginalKey) + } + return offset +} + +func (p *SimpleAnnotationInfo) field1Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(p.Key) + return l +} + +func (p *SimpleAnnotationInfo) field2Length() int { + l := 0 + if p.IsSetAnnotationType() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.AnnotationType) + } + return l +} + +func (p *SimpleAnnotationInfo) field3Length() int { + l := 0 + if p.IsSetOriginalKey() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.OriginalKey) + } + return l +} + +func (p *SimpleAnnotationInfo) DeepCopy(s interface{}) error { + src, ok := s.(*SimpleAnnotationInfo) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + if src.Key != "" { + p.Key = kutils.StringDeepCopy(src.Key) + } + + if src.AnnotationType != nil { + tmp := *src.AnnotationType + p.AnnotationType = &tmp + } + + if src.OriginalKey != nil { + var tmp string + if *src.OriginalKey != "" { + tmp = kutils.StringDeepCopy(*src.OriginalKey) + } + p.OriginalKey = &tmp + } + + return nil +} diff --git a/backend/kitex_gen/coze/loop/observability/domain/filter/filter.go b/backend/kitex_gen/coze/loop/observability/domain/filter/filter.go index 21c65f847..0134a3ae5 100644 --- a/backend/kitex_gen/coze/loop/observability/domain/filter/filter.go +++ b/backend/kitex_gen/coze/loop/observability/domain/filter/filter.go @@ -55,6 +55,12 @@ const ( TaskFieldNameSampleRate = "sample_rate" TaskFieldNameCreatedBy = "created_by" + + TaskFieldNameTaskID = "task_id" + + TaskFieldNameWorkflowID = "workflow_id" + + TaskFieldNameUpdatedBy = "updated_by" ) type QueryType = string diff --git a/backend/kitex_gen/coze/loop/observability/domain/span/span.go b/backend/kitex_gen/coze/loop/observability/domain/span/span.go index d00487e82..22d9a597a 100644 --- a/backend/kitex_gen/coze/loop/observability/domain/span/span.go +++ b/backend/kitex_gen/coze/loop/observability/domain/span/span.go @@ -47,12 +47,24 @@ const ( SpanTypeAgent = "agent" SpanTypeLLMCall = "LLMCall" + + ValueTypeString = "string" + + ValueTypeByte = "byte" + + ValueTypeLong = "long" + + ValueTypeDouble = "double" + + ValueTypeBool = "bool" ) type SpanStatus = string type SpanType = string +type MetadataValueType = string + type AttrTos struct { InputDataURL *string `thrift:"input_data_url,1,optional" frugal:"1,optional,string" form:"input_data_url" json:"input_data_url,omitempty" query:"input_data_url"` OutputDataURL *string `thrift:"output_data_url,2,optional" frugal:"2,optional,string" form:"output_data_url" json:"output_data_url,omitempty" query:"output_data_url"` diff --git a/backend/kitex_gen/coze/loop/observability/domain/task/k-task.go b/backend/kitex_gen/coze/loop/observability/domain/task/k-task.go index e7001c375..2e5b98908 100644 --- a/backend/kitex_gen/coze/loop/observability/domain/task/k-task.go +++ b/backend/kitex_gen/coze/loop/observability/domain/task/k-task.go @@ -11,12 +11,14 @@ import ( "github.com/cloudwego/gopkg/protocol/thrift" kutils "github.com/cloudwego/kitex/pkg/utils" + dataset0 "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/common" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/dataset" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/filter" ) var ( + _ = dataset0.KitexUnusedProtection _ = common.KitexUnusedProtection _ = dataset.KitexUnusedProtection _ = filter.KitexUnusedProtection @@ -206,6 +208,34 @@ func (p *Task) FastRead(buf []byte) (int, error) { goto SkipFieldError } } + case 12: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField12(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 13: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField13(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } case 100: if fieldTypeId == thrift.STRUCT { l, err = p.FastReadField100(buf[offset:]) @@ -395,6 +425,45 @@ func (p *Task) FastReadField11(buf []byte) (int, error) { return offset, nil } +func (p *Task) FastReadField12(buf []byte) (int, error) { + offset := 0 + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.WorkflowID = _field + return offset, nil +} + +func (p *Task) FastReadField13(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*TaskRun, 0, size) + values := make([]TaskRun, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.TaskRuns = _field + return offset, nil +} + func (p *Task) FastReadField100(buf []byte) (int, error) { offset := 0 _field := common.NewBaseInfo() @@ -416,6 +485,7 @@ func (p *Task) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { if p != nil { offset += p.fastWriteField1(buf[offset:], w) offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField12(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) offset += p.fastWriteField3(buf[offset:], w) offset += p.fastWriteField5(buf[offset:], w) @@ -425,6 +495,7 @@ func (p *Task) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset += p.fastWriteField9(buf[offset:], w) offset += p.fastWriteField10(buf[offset:], w) offset += p.fastWriteField11(buf[offset:], w) + offset += p.fastWriteField13(buf[offset:], w) offset += p.fastWriteField100(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) @@ -445,6 +516,8 @@ func (p *Task) BLength() int { l += p.field9Length() l += p.field10Length() l += p.field11Length() + l += p.field12Length() + l += p.field13Length() l += p.field100Length() } l += thrift.Binary.FieldStopLength() @@ -546,6 +619,31 @@ func (p *Task) fastWriteField11(buf []byte, w thrift.NocopyWriter) int { return offset } +func (p *Task) fastWriteField12(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetWorkflowID() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 12) + offset += thrift.Binary.WriteI64(buf[offset:], *p.WorkflowID) + } + return offset +} + +func (p *Task) fastWriteField13(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetTaskRuns() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 13) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.TaskRuns { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + func (p *Task) fastWriteField100(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetBaseInfo() { @@ -650,6 +748,28 @@ func (p *Task) field11Length() int { return l } +func (p *Task) field12Length() int { + l := 0 + if p.IsSetWorkflowID() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + } + return l +} + +func (p *Task) field13Length() int { + l := 0 + if p.IsSetTaskRuns() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.TaskRuns { + _ = v + l += v.BLength() + } + } + return l +} + func (p *Task) field100Length() int { l := 0 if p.IsSetBaseInfo() { @@ -735,6 +855,26 @@ func (p *Task) DeepCopy(s interface{}) error { p.TaskSource = &tmp } + if src.WorkflowID != nil { + tmp := *src.WorkflowID + p.WorkflowID = &tmp + } + + if src.TaskRuns != nil { + p.TaskRuns = make([]*TaskRun, 0, len(src.TaskRuns)) + for _, elem := range src.TaskRuns { + var _elem *TaskRun + if elem != nil { + _elem = &TaskRun{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.TaskRuns = append(p.TaskRuns, _elem) + } + } + var _baseInfo *common.BaseInfo if src.BaseInfo != nil { _baseInfo = &common.BaseInfo{} @@ -1583,7 +1723,7 @@ func (p *EffectiveTime) DeepCopy(s interface{}) error { return nil } -func (p *TaskConfig) FastRead(buf []byte) (int, error) { +func (p *EvaluationExperimentConfig) FastRead(buf []byte) (int, error) { var err error var offset int @@ -1601,7 +1741,7 @@ func (p *TaskConfig) FastRead(buf []byte) (int, error) { } switch fieldId { case 1: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.I32 { l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { @@ -1615,7 +1755,7 @@ func (p *TaskConfig) FastRead(buf []byte) (int, error) { } } case 2: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.I32 { l, err = p.FastReadField2(buf[offset:]) offset += l if err != nil { @@ -1628,6 +1768,62 @@ func (p *TaskConfig) FastRead(buf []byte) (int, error) { goto SkipFieldError } } + case 3: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField3(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 4: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField4(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 5: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField5(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 6: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField6(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } default: l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -1641,37 +1837,82 @@ func (p *TaskConfig) FastRead(buf []byte) (int, error) { ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TaskConfig[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_EvaluationExperimentConfig[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) } -func (p *TaskConfig) FastReadField1(buf []byte) (int, error) { +func (p *EvaluationExperimentConfig) FastReadField1(buf []byte) (int, error) { offset := 0 - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { + var _field *int32 + if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { return offset, err + } else { + offset += l + _field = &v } - _field := make([]*AutoEvaluateConfig, 0, size) - values := make([]AutoEvaluateConfig, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } + p.ItemConcurrencyCount = _field + return offset, nil +} - _field = append(_field, _elem) +func (p *EvaluationExperimentConfig) FastReadField2(buf []byte) (int, error) { + offset := 0 + + var _field *int32 + if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v } - p.AutoEvaluateConfigs = _field + p.ItemMaxRetryCount = _field return offset, nil } -func (p *TaskConfig) FastReadField2(buf []byte) (int, error) { +func (p *EvaluationExperimentConfig) FastReadField3(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.SourceTargetID = _field + return offset, nil +} + +func (p *EvaluationExperimentConfig) FastReadField4(buf []byte) (int, error) { + offset := 0 + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.ExptTemplateID = _field + return offset, nil +} + +func (p *EvaluationExperimentConfig) FastReadField5(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.SourceTargetVersion = _field + return offset, nil +} + +func (p *EvaluationExperimentConfig) FastReadField6(buf []byte) (int, error) { offset := 0 _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) @@ -1679,8 +1920,8 @@ func (p *TaskConfig) FastReadField2(buf []byte) (int, error) { if err != nil { return offset, err } - _field := make([]*DataReflowConfig, 0, size) - values := make([]DataReflowConfig, size) + _field := make([]*EvaluateFieldMapping, 0, size) + values := make([]EvaluateFieldMapping, size) for i := 0; i < size; i++ { _elem := &values[i] _elem.InitDefault() @@ -1692,54 +1933,648 @@ func (p *TaskConfig) FastReadField2(buf []byte) (int, error) { _field = append(_field, _elem) } - p.DataReflowConfig = _field + p.FullEvalSetFieldMappings = _field return offset, nil } -func (p *TaskConfig) FastWrite(buf []byte) int { +func (p *EvaluationExperimentConfig) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *TaskConfig) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *EvaluationExperimentConfig) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) + offset += p.fastWriteField5(buf[offset:], w) + offset += p.fastWriteField6(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *TaskConfig) BLength() int { +func (p *EvaluationExperimentConfig) BLength() int { l := 0 if p != nil { l += p.field1Length() l += p.field2Length() + l += p.field3Length() + l += p.field4Length() + l += p.field5Length() + l += p.field6Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *TaskConfig) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *EvaluationExperimentConfig) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetAutoEvaluateConfigs() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.AutoEvaluateConfigs { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + if p.IsSetItemConcurrencyCount() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 1) + offset += thrift.Binary.WriteI32(buf[offset:], *p.ItemConcurrencyCount) } return offset } -func (p *TaskConfig) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { +func (p *EvaluationExperimentConfig) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetDataReflowConfig() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) + if p.IsSetItemMaxRetryCount() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 2) + offset += thrift.Binary.WriteI32(buf[offset:], *p.ItemMaxRetryCount) + } + return offset +} + +func (p *EvaluationExperimentConfig) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetSourceTargetID() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 3) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.SourceTargetID) + } + return offset +} + +func (p *EvaluationExperimentConfig) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetExptTemplateID() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 4) + offset += thrift.Binary.WriteI64(buf[offset:], *p.ExptTemplateID) + } + return offset +} + +func (p *EvaluationExperimentConfig) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetSourceTargetVersion() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 5) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.SourceTargetVersion) + } + return offset +} + +func (p *EvaluationExperimentConfig) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetFullEvalSetFieldMappings() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 6) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.FullEvalSetFieldMappings { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + +func (p *EvaluationExperimentConfig) field1Length() int { + l := 0 + if p.IsSetItemConcurrencyCount() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I32Length() + } + return l +} + +func (p *EvaluationExperimentConfig) field2Length() int { + l := 0 + if p.IsSetItemMaxRetryCount() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I32Length() + } + return l +} + +func (p *EvaluationExperimentConfig) field3Length() int { + l := 0 + if p.IsSetSourceTargetID() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.SourceTargetID) + } + return l +} + +func (p *EvaluationExperimentConfig) field4Length() int { + l := 0 + if p.IsSetExptTemplateID() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + } + return l +} + +func (p *EvaluationExperimentConfig) field5Length() int { + l := 0 + if p.IsSetSourceTargetVersion() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.SourceTargetVersion) + } + return l +} + +func (p *EvaluationExperimentConfig) field6Length() int { + l := 0 + if p.IsSetFullEvalSetFieldMappings() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.FullEvalSetFieldMappings { + _ = v + l += v.BLength() + } + } + return l +} + +func (p *EvaluationExperimentConfig) DeepCopy(s interface{}) error { + src, ok := s.(*EvaluationExperimentConfig) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + if src.ItemConcurrencyCount != nil { + tmp := *src.ItemConcurrencyCount + p.ItemConcurrencyCount = &tmp + } + + if src.ItemMaxRetryCount != nil { + tmp := *src.ItemMaxRetryCount + p.ItemMaxRetryCount = &tmp + } + + if src.SourceTargetID != nil { + var tmp string + if *src.SourceTargetID != "" { + tmp = kutils.StringDeepCopy(*src.SourceTargetID) + } + p.SourceTargetID = &tmp + } + + if src.ExptTemplateID != nil { + tmp := *src.ExptTemplateID + p.ExptTemplateID = &tmp + } + + if src.SourceTargetVersion != nil { + var tmp string + if *src.SourceTargetVersion != "" { + tmp = kutils.StringDeepCopy(*src.SourceTargetVersion) + } + p.SourceTargetVersion = &tmp + } + + if src.FullEvalSetFieldMappings != nil { + p.FullEvalSetFieldMappings = make([]*EvaluateFieldMapping, 0, len(src.FullEvalSetFieldMappings)) + for _, elem := range src.FullEvalSetFieldMappings { + var _elem *EvaluateFieldMapping + if elem != nil { + _elem = &EvaluateFieldMapping{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.FullEvalSetFieldMappings = append(p.FullEvalSetFieldMappings, _elem) + } + } + + return nil +} + +func (p *SourceInfo) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SourceInfo[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +} + +func (p *SourceInfo) FastReadField1(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.Name = _field + return offset, nil +} + +func (p *SourceInfo) FastReadField2(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.Version = _field + return offset, nil +} + +func (p *SourceInfo) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *SourceInfo) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *SourceInfo) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *SourceInfo) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetName() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 1) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.Name) + } + return offset +} + +func (p *SourceInfo) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetVersion() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.Version) + } + return offset +} + +func (p *SourceInfo) field1Length() int { + l := 0 + if p.IsSetName() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.Name) + } + return l +} + +func (p *SourceInfo) field2Length() int { + l := 0 + if p.IsSetVersion() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.Version) + } + return l +} + +func (p *SourceInfo) DeepCopy(s interface{}) error { + src, ok := s.(*SourceInfo) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + if src.Name != nil { + var tmp string + if *src.Name != "" { + tmp = kutils.StringDeepCopy(*src.Name) + } + p.Name = &tmp + } + + if src.Version != nil { + var tmp string + if *src.Version != "" { + tmp = kutils.StringDeepCopy(*src.Version) + } + p.Version = &tmp + } + + return nil +} + +func (p *TaskConfig) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 3: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField3(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 4: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField4(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 5: + if fieldTypeId == thrift.BOOL { + l, err = p.FastReadField5(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TaskConfig[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +} + +func (p *TaskConfig) FastReadField1(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*AutoEvaluateConfig, 0, size) + values := make([]AutoEvaluateConfig, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.AutoEvaluateConfigs = _field + return offset, nil +} + +func (p *TaskConfig) FastReadField2(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*DataReflowConfig, 0, size) + values := make([]DataReflowConfig, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.DataReflowConfig = _field + return offset, nil +} + +func (p *TaskConfig) FastReadField3(buf []byte) (int, error) { + offset := 0 + _field := NewEvaluationExperimentConfig() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.EvaluationExperimentConfig = _field + return offset, nil +} + +func (p *TaskConfig) FastReadField4(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*SourceInfo, 0, size) + values := make([]SourceInfo, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.SourceInfo = _field + return offset, nil +} + +func (p *TaskConfig) FastReadField5(buf []byte) (int, error) { + offset := 0 + + var _field *bool + if v, l, err := thrift.Binary.ReadBool(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.IsWorkflowScheduled = _field + return offset, nil +} + +func (p *TaskConfig) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *TaskConfig) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField5(buf[offset:], w) + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) + offset += p.fastWriteField4(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *TaskConfig) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + l += p.field3Length() + l += p.field4Length() + l += p.field5Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *TaskConfig) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetAutoEvaluateConfigs() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.AutoEvaluateConfigs { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + +func (p *TaskConfig) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetDataReflowConfig() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) listBeginOffset := offset offset += thrift.Binary.ListBeginLength() var length int @@ -1752,6 +2587,40 @@ func (p *TaskConfig) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { return offset } +func (p *TaskConfig) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetEvaluationExperimentConfig() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 3) + offset += p.EvaluationExperimentConfig.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *TaskConfig) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetSourceInfo() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 4) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.SourceInfo { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + +func (p *TaskConfig) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetIsWorkflowScheduled() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.BOOL, 5) + offset += thrift.Binary.WriteBool(buf[offset:], *p.IsWorkflowScheduled) + } + return offset +} + func (p *TaskConfig) field1Length() int { l := 0 if p.IsSetAutoEvaluateConfigs() { @@ -1778,6 +2647,37 @@ func (p *TaskConfig) field2Length() int { return l } +func (p *TaskConfig) field3Length() int { + l := 0 + if p.IsSetEvaluationExperimentConfig() { + l += thrift.Binary.FieldBeginLength() + l += p.EvaluationExperimentConfig.BLength() + } + return l +} + +func (p *TaskConfig) field4Length() int { + l := 0 + if p.IsSetSourceInfo() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.SourceInfo { + _ = v + l += v.BLength() + } + } + return l +} + +func (p *TaskConfig) field5Length() int { + l := 0 + if p.IsSetIsWorkflowScheduled() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.BoolLength() + } + return l +} + func (p *TaskConfig) DeepCopy(s interface{}) error { src, ok := s.(*TaskConfig) if !ok { @@ -1814,6 +2714,35 @@ func (p *TaskConfig) DeepCopy(s interface{}) error { } } + var _evaluationExperimentConfig *EvaluationExperimentConfig + if src.EvaluationExperimentConfig != nil { + _evaluationExperimentConfig = &EvaluationExperimentConfig{} + if err := _evaluationExperimentConfig.DeepCopy(src.EvaluationExperimentConfig); err != nil { + return err + } + } + p.EvaluationExperimentConfig = _evaluationExperimentConfig + + if src.SourceInfo != nil { + p.SourceInfo = make([]*SourceInfo, 0, len(src.SourceInfo)) + for _, elem := range src.SourceInfo { + var _elem *SourceInfo + if elem != nil { + _elem = &SourceInfo{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.SourceInfo = append(p.SourceInfo, _elem) + } + } + + if src.IsWorkflowScheduled != nil { + tmp := *src.IsWorkflowScheduled + p.IsWorkflowScheduled = &tmp + } + return nil } @@ -1890,6 +2819,20 @@ func (p *DataReflowConfig) FastRead(buf []byte) (int, error) { goto SkipFieldError } } + case 5: + if fieldTypeId == thrift.I32 { + l, err = p.FastReadField5(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } default: l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -1973,6 +2916,22 @@ func (p *DataReflowConfig) FastReadField4(buf []byte) (int, error) { return offset, nil } +func (p *DataReflowConfig) FastReadField5(buf []byte) (int, error) { + offset := 0 + + var _field *dataset0.DatasetCategory + if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + + tmp := dataset0.DatasetCategory(v) + _field = &tmp + } + p.DatasetCategory = _field + return offset, nil +} + func (p *DataReflowConfig) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } @@ -1984,6 +2943,7 @@ func (p *DataReflowConfig) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) in offset += p.fastWriteField2(buf[offset:], w) offset += p.fastWriteField3(buf[offset:], w) offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField5(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset @@ -1996,6 +2956,7 @@ func (p *DataReflowConfig) BLength() int { l += p.field2Length() l += p.field3Length() l += p.field4Length() + l += p.field5Length() } l += thrift.Binary.FieldStopLength() return l @@ -2044,6 +3005,15 @@ func (p *DataReflowConfig) fastWriteField4(buf []byte, w thrift.NocopyWriter) in return offset } +func (p *DataReflowConfig) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetDatasetCategory() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 5) + offset += thrift.Binary.WriteI32(buf[offset:], int32(*p.DatasetCategory)) + } + return offset +} + func (p *DataReflowConfig) field1Length() int { l := 0 if p.IsSetDatasetID() { @@ -2084,6 +3054,15 @@ func (p *DataReflowConfig) field4Length() int { return l } +func (p *DataReflowConfig) field5Length() int { + l := 0 + if p.IsSetDatasetCategory() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I32Length() + } + return l +} + func (p *DataReflowConfig) DeepCopy(s interface{}) error { src, ok := s.(*DataReflowConfig) if !ok { @@ -2127,6 +3106,11 @@ func (p *DataReflowConfig) DeepCopy(s interface{}) error { } } + if src.DatasetCategory != nil { + tmp := *src.DatasetCategory + p.DatasetCategory = &tmp + } + return nil } @@ -2195,6 +3179,34 @@ func (p *AutoEvaluateConfig) FastRead(buf []byte) (int, error) { goto SkipFieldError } } + case 4: + if fieldTypeId == thrift.DOUBLE { + l, err = p.FastReadField4(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 5: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField5(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } default: l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -2282,6 +3294,34 @@ func (p *AutoEvaluateConfig) FastReadField3(buf []byte) (int, error) { return offset, nil } +func (p *AutoEvaluateConfig) FastReadField4(buf []byte) (int, error) { + offset := 0 + + var _field *float64 + if v, l, err := thrift.Binary.ReadDouble(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.ScoreWeight = _field + return offset, nil +} + +func (p *AutoEvaluateConfig) FastReadField5(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.EvaluatorVersion = _field + return offset, nil +} + func (p *AutoEvaluateConfig) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } @@ -2291,7 +3331,9 @@ func (p *AutoEvaluateConfig) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) if p != nil { offset += p.fastWriteField1(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField4(buf[offset:], w) offset += p.fastWriteField3(buf[offset:], w) + offset += p.fastWriteField5(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset @@ -2303,6 +3345,8 @@ func (p *AutoEvaluateConfig) BLength() int { l += p.field1Length() l += p.field2Length() l += p.field3Length() + l += p.field4Length() + l += p.field5Length() } l += thrift.Binary.FieldStopLength() return l @@ -2336,6 +3380,24 @@ func (p *AutoEvaluateConfig) fastWriteField3(buf []byte, w thrift.NocopyWriter) return offset } +func (p *AutoEvaluateConfig) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetScoreWeight() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.DOUBLE, 4) + offset += thrift.Binary.WriteDouble(buf[offset:], *p.ScoreWeight) + } + return offset +} + +func (p *AutoEvaluateConfig) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetEvaluatorVersion() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 5) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.EvaluatorVersion) + } + return offset +} + func (p *AutoEvaluateConfig) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() @@ -2361,6 +3423,24 @@ func (p *AutoEvaluateConfig) field3Length() int { return l } +func (p *AutoEvaluateConfig) field4Length() int { + l := 0 + if p.IsSetScoreWeight() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.DoubleLength() + } + return l +} + +func (p *AutoEvaluateConfig) field5Length() int { + l := 0 + if p.IsSetEvaluatorVersion() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.EvaluatorVersion) + } + return l +} + func (p *AutoEvaluateConfig) DeepCopy(s interface{}) error { src, ok := s.(*AutoEvaluateConfig) if !ok { @@ -2386,6 +3466,19 @@ func (p *AutoEvaluateConfig) DeepCopy(s interface{}) error { } } + if src.ScoreWeight != nil { + tmp := *src.ScoreWeight + p.ScoreWeight = &tmp + } + + if src.EvaluatorVersion != nil { + var tmp string + if *src.EvaluatorVersion != "" { + tmp = kutils.StringDeepCopy(*src.EvaluatorVersion) + } + p.EvaluatorVersion = &tmp + } + return nil } diff --git a/backend/kitex_gen/coze/loop/observability/domain/task/task.go b/backend/kitex_gen/coze/loop/observability/domain/task/task.go index 44dad94e2..efaf07e0e 100644 --- a/backend/kitex_gen/coze/loop/observability/domain/task/task.go +++ b/backend/kitex_gen/coze/loop/observability/domain/task/task.go @@ -5,6 +5,7 @@ package task import ( "fmt" "github.com/apache/thrift/lib/go/thrift" + dataset0 "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/common" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/dataset" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/filter" @@ -83,6 +84,10 @@ type Task struct { BackfillTaskDetail *RunDetail `thrift:"backfill_task_detail,10,optional" frugal:"10,optional,RunDetail" form:"backfill_task_detail" json:"backfill_task_detail,omitempty" query:"backfill_task_detail"` // 创建来源 TaskSource *TaskSource `thrift:"task_source,11,optional" frugal:"11,optional,string" form:"task_source" json:"task_source,omitempty" query:"task_source"` + // 对应工作流 ID + WorkflowID *int64 `thrift:"workflow_id,12,optional" frugal:"12,optional,i64" json:"workflow_id" form:"workflow_id" query:"workflow_id"` + // task runs + TaskRuns []*TaskRun `thrift:"task_runs,13,optional" frugal:"13,optional,list" form:"task_runs" json:"task_runs,omitempty" query:"task_runs"` // 基础信息 BaseInfo *common.BaseInfo `thrift:"base_info,100,optional" frugal:"100,optional,common.BaseInfo" form:"base_info" json:"base_info,omitempty" query:"base_info"` } @@ -216,6 +221,30 @@ func (p *Task) GetTaskSource() (v TaskSource) { return *p.TaskSource } +var Task_WorkflowID_DEFAULT int64 + +func (p *Task) GetWorkflowID() (v int64) { + if p == nil { + return + } + if !p.IsSetWorkflowID() { + return Task_WorkflowID_DEFAULT + } + return *p.WorkflowID +} + +var Task_TaskRuns_DEFAULT []*TaskRun + +func (p *Task) GetTaskRuns() (v []*TaskRun) { + if p == nil { + return + } + if !p.IsSetTaskRuns() { + return Task_TaskRuns_DEFAULT + } + return p.TaskRuns +} + var Task_BaseInfo_DEFAULT *common.BaseInfo func (p *Task) GetBaseInfo() (v *common.BaseInfo) { @@ -260,6 +289,12 @@ func (p *Task) SetBackfillTaskDetail(val *RunDetail) { func (p *Task) SetTaskSource(val *TaskSource) { p.TaskSource = val } +func (p *Task) SetWorkflowID(val *int64) { + p.WorkflowID = val +} +func (p *Task) SetTaskRuns(val []*TaskRun) { + p.TaskRuns = val +} func (p *Task) SetBaseInfo(val *common.BaseInfo) { p.BaseInfo = val } @@ -276,6 +311,8 @@ var fieldIDToName_Task = map[int16]string{ 9: "task_detail", 10: "backfill_task_detail", 11: "task_source", + 12: "workflow_id", + 13: "task_runs", 100: "base_info", } @@ -315,6 +352,14 @@ func (p *Task) IsSetTaskSource() bool { return p.TaskSource != nil } +func (p *Task) IsSetWorkflowID() bool { + return p.WorkflowID != nil +} + +func (p *Task) IsSetTaskRuns() bool { + return p.TaskRuns != nil +} + func (p *Task) IsSetBaseInfo() bool { return p.BaseInfo != nil } @@ -429,6 +474,22 @@ func (p *Task) Read(iprot thrift.TProtocol) (err error) { } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } + case 12: + if fieldTypeId == thrift.I64 { + if err = p.ReadField12(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 13: + if fieldTypeId == thrift.LIST { + if err = p.ReadField13(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } case 100: if fieldTypeId == thrift.STRUCT { if err = p.ReadField100(iprot); err != nil { @@ -586,6 +647,40 @@ func (p *Task) ReadField11(iprot thrift.TProtocol) error { p.TaskSource = _field return nil } +func (p *Task) ReadField12(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v + } + p.WorkflowID = _field + return nil +} +func (p *Task) ReadField13(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*TaskRun, 0, size) + values := make([]TaskRun, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.TaskRuns = _field + return nil +} func (p *Task) ReadField100(iprot thrift.TProtocol) error { _field := common.NewBaseInfo() if err := _field.Read(iprot); err != nil { @@ -645,6 +740,14 @@ func (p *Task) Write(oprot thrift.TProtocol) (err error) { fieldId = 11 goto WriteFieldError } + if err = p.writeField12(oprot); err != nil { + fieldId = 12 + goto WriteFieldError + } + if err = p.writeField13(oprot); err != nil { + fieldId = 13 + goto WriteFieldError + } if err = p.writeField100(oprot); err != nil { fieldId = 100 goto WriteFieldError @@ -861,6 +964,50 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 11 end error: ", p), err) } +func (p *Task) writeField12(oprot thrift.TProtocol) (err error) { + if p.IsSetWorkflowID() { + if err = oprot.WriteFieldBegin("workflow_id", thrift.I64, 12); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.WorkflowID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 12 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 12 end error: ", p), err) +} +func (p *Task) writeField13(oprot thrift.TProtocol) (err error) { + if p.IsSetTaskRuns() { + if err = oprot.WriteFieldBegin("task_runs", thrift.LIST, 13); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.TaskRuns)); err != nil { + return err + } + for _, v := range p.TaskRuns { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 13 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 13 end error: ", p), err) +} func (p *Task) writeField100(oprot thrift.TProtocol) (err error) { if p.IsSetBaseInfo() { if err = oprot.WriteFieldBegin("base_info", thrift.STRUCT, 100); err != nil { @@ -927,6 +1074,12 @@ func (p *Task) DeepEqual(ano *Task) bool { if !p.Field11DeepEqual(ano.TaskSource) { return false } + if !p.Field12DeepEqual(ano.WorkflowID) { + return false + } + if !p.Field13DeepEqual(ano.TaskRuns) { + return false + } if !p.Field100DeepEqual(ano.BaseInfo) { return false } @@ -1035,6 +1188,31 @@ func (p *Task) Field11DeepEqual(src *TaskSource) bool { } return true } +func (p *Task) Field12DeepEqual(src *int64) bool { + + if p.WorkflowID == src { + return true + } else if p.WorkflowID == nil || src == nil { + return false + } + if *p.WorkflowID != *src { + return false + } + return true +} +func (p *Task) Field13DeepEqual(src []*TaskRun) bool { + + if len(p.TaskRuns) != len(src) { + return false + } + for i, v := range p.TaskRuns { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true +} func (p *Task) Field100DeepEqual(src *common.BaseInfo) bool { if !p.BaseInfo.DeepEqual(src) { @@ -2260,65 +2438,146 @@ func (p *EffectiveTime) Field2DeepEqual(src *int64) bool { return true } -// TaskConfig -type TaskConfig struct { - // 配置的评测规则信息 - AutoEvaluateConfigs []*AutoEvaluateConfig `thrift:"auto_evaluate_configs,1,optional" frugal:"1,optional,list" form:"auto_evaluate_configs" json:"auto_evaluate_configs,omitempty" query:"auto_evaluate_configs"` - // 配置的数据回流的数据集信息 - DataReflowConfig []*DataReflowConfig `thrift:"data_reflow_config,2,optional" frugal:"2,optional,list" form:"data_reflow_config" json:"data_reflow_config,omitempty" query:"data_reflow_config"` +type EvaluationExperimentConfig struct { + ItemConcurrencyCount *int32 `thrift:"item_concurrency_count,1,optional" frugal:"1,optional,i32" json:"item_concurrency_count" form:"item_concurrency_count" query:"item_concurrency_count"` + ItemMaxRetryCount *int32 `thrift:"item_max_retry_count,2,optional" frugal:"2,optional,i32" json:"item_max_retry_count" form:"item_max_retry_count" query:"item_max_retry_count"` + SourceTargetID *string `thrift:"source_target_id,3,optional" frugal:"3,optional,string" json:"source_target_id" form:"source_target_id" query:"source_target_id"` + ExptTemplateID *int64 `thrift:"expt_template_id,4,optional" frugal:"4,optional,i64" json:"expt_template_id" form:"expt_template_id" query:"expt_template_id"` + SourceTargetVersion *string `thrift:"source_target_version,5,optional" frugal:"5,optional,string" json:"source_target_version" form:"source_target_version" query:"source_target_version"` + FullEvalSetFieldMappings []*EvaluateFieldMapping `thrift:"full_eval_set_field_mappings,6,optional" frugal:"6,optional,list" json:"full_eval_set_field_mappings" form:"full_eval_set_field_mappings" query:"full_eval_set_field_mappings"` } -func NewTaskConfig() *TaskConfig { - return &TaskConfig{} +func NewEvaluationExperimentConfig() *EvaluationExperimentConfig { + return &EvaluationExperimentConfig{} } -func (p *TaskConfig) InitDefault() { +func (p *EvaluationExperimentConfig) InitDefault() { } -var TaskConfig_AutoEvaluateConfigs_DEFAULT []*AutoEvaluateConfig +var EvaluationExperimentConfig_ItemConcurrencyCount_DEFAULT int32 -func (p *TaskConfig) GetAutoEvaluateConfigs() (v []*AutoEvaluateConfig) { +func (p *EvaluationExperimentConfig) GetItemConcurrencyCount() (v int32) { if p == nil { return } - if !p.IsSetAutoEvaluateConfigs() { - return TaskConfig_AutoEvaluateConfigs_DEFAULT + if !p.IsSetItemConcurrencyCount() { + return EvaluationExperimentConfig_ItemConcurrencyCount_DEFAULT } - return p.AutoEvaluateConfigs + return *p.ItemConcurrencyCount } -var TaskConfig_DataReflowConfig_DEFAULT []*DataReflowConfig +var EvaluationExperimentConfig_ItemMaxRetryCount_DEFAULT int32 -func (p *TaskConfig) GetDataReflowConfig() (v []*DataReflowConfig) { +func (p *EvaluationExperimentConfig) GetItemMaxRetryCount() (v int32) { if p == nil { return } - if !p.IsSetDataReflowConfig() { - return TaskConfig_DataReflowConfig_DEFAULT + if !p.IsSetItemMaxRetryCount() { + return EvaluationExperimentConfig_ItemMaxRetryCount_DEFAULT } - return p.DataReflowConfig + return *p.ItemMaxRetryCount } -func (p *TaskConfig) SetAutoEvaluateConfigs(val []*AutoEvaluateConfig) { - p.AutoEvaluateConfigs = val + +var EvaluationExperimentConfig_SourceTargetID_DEFAULT string + +func (p *EvaluationExperimentConfig) GetSourceTargetID() (v string) { + if p == nil { + return + } + if !p.IsSetSourceTargetID() { + return EvaluationExperimentConfig_SourceTargetID_DEFAULT + } + return *p.SourceTargetID } -func (p *TaskConfig) SetDataReflowConfig(val []*DataReflowConfig) { - p.DataReflowConfig = val + +var EvaluationExperimentConfig_ExptTemplateID_DEFAULT int64 + +func (p *EvaluationExperimentConfig) GetExptTemplateID() (v int64) { + if p == nil { + return + } + if !p.IsSetExptTemplateID() { + return EvaluationExperimentConfig_ExptTemplateID_DEFAULT + } + return *p.ExptTemplateID } -var fieldIDToName_TaskConfig = map[int16]string{ - 1: "auto_evaluate_configs", - 2: "data_reflow_config", +var EvaluationExperimentConfig_SourceTargetVersion_DEFAULT string + +func (p *EvaluationExperimentConfig) GetSourceTargetVersion() (v string) { + if p == nil { + return + } + if !p.IsSetSourceTargetVersion() { + return EvaluationExperimentConfig_SourceTargetVersion_DEFAULT + } + return *p.SourceTargetVersion } -func (p *TaskConfig) IsSetAutoEvaluateConfigs() bool { - return p.AutoEvaluateConfigs != nil +var EvaluationExperimentConfig_FullEvalSetFieldMappings_DEFAULT []*EvaluateFieldMapping + +func (p *EvaluationExperimentConfig) GetFullEvalSetFieldMappings() (v []*EvaluateFieldMapping) { + if p == nil { + return + } + if !p.IsSetFullEvalSetFieldMappings() { + return EvaluationExperimentConfig_FullEvalSetFieldMappings_DEFAULT + } + return p.FullEvalSetFieldMappings +} +func (p *EvaluationExperimentConfig) SetItemConcurrencyCount(val *int32) { + p.ItemConcurrencyCount = val +} +func (p *EvaluationExperimentConfig) SetItemMaxRetryCount(val *int32) { + p.ItemMaxRetryCount = val +} +func (p *EvaluationExperimentConfig) SetSourceTargetID(val *string) { + p.SourceTargetID = val +} +func (p *EvaluationExperimentConfig) SetExptTemplateID(val *int64) { + p.ExptTemplateID = val +} +func (p *EvaluationExperimentConfig) SetSourceTargetVersion(val *string) { + p.SourceTargetVersion = val +} +func (p *EvaluationExperimentConfig) SetFullEvalSetFieldMappings(val []*EvaluateFieldMapping) { + p.FullEvalSetFieldMappings = val } -func (p *TaskConfig) IsSetDataReflowConfig() bool { - return p.DataReflowConfig != nil +var fieldIDToName_EvaluationExperimentConfig = map[int16]string{ + 1: "item_concurrency_count", + 2: "item_max_retry_count", + 3: "source_target_id", + 4: "expt_template_id", + 5: "source_target_version", + 6: "full_eval_set_field_mappings", } -func (p *TaskConfig) Read(iprot thrift.TProtocol) (err error) { +func (p *EvaluationExperimentConfig) IsSetItemConcurrencyCount() bool { + return p.ItemConcurrencyCount != nil +} + +func (p *EvaluationExperimentConfig) IsSetItemMaxRetryCount() bool { + return p.ItemMaxRetryCount != nil +} + +func (p *EvaluationExperimentConfig) IsSetSourceTargetID() bool { + return p.SourceTargetID != nil +} + +func (p *EvaluationExperimentConfig) IsSetExptTemplateID() bool { + return p.ExptTemplateID != nil +} + +func (p *EvaluationExperimentConfig) IsSetSourceTargetVersion() bool { + return p.SourceTargetVersion != nil +} + +func (p *EvaluationExperimentConfig) IsSetFullEvalSetFieldMappings() bool { + return p.FullEvalSetFieldMappings != nil +} + +func (p *EvaluationExperimentConfig) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -2337,7 +2596,7 @@ func (p *TaskConfig) Read(iprot thrift.TProtocol) (err error) { switch fieldId { case 1: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.I32 { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } @@ -2345,13 +2604,45 @@ func (p *TaskConfig) Read(iprot thrift.TProtocol) (err error) { goto SkipFieldError } case 2: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.I32 { if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } + case 3: + if fieldTypeId == thrift.STRING { + if err = p.ReadField3(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 4: + if fieldTypeId == thrift.I64 { + if err = p.ReadField4(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 5: + if fieldTypeId == thrift.STRING { + if err = p.ReadField5(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 6: + if fieldTypeId == thrift.LIST { + if err = p.ReadField6(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } default: if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError @@ -2371,7 +2662,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TaskConfig[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_EvaluationExperimentConfig[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -2381,37 +2672,890 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TaskConfig) ReadField1(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { +func (p *EvaluationExperimentConfig) ReadField1(iprot thrift.TProtocol) error { + + var _field *int32 + if v, err := iprot.ReadI32(); err != nil { return err + } else { + _field = &v } - _field := make([]*AutoEvaluateConfig, 0, size) - values := make([]AutoEvaluateConfig, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } + p.ItemConcurrencyCount = _field + return nil +} +func (p *EvaluationExperimentConfig) ReadField2(iprot thrift.TProtocol) error { - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { + var _field *int32 + if v, err := iprot.ReadI32(); err != nil { return err + } else { + _field = &v } - p.AutoEvaluateConfigs = _field + p.ItemMaxRetryCount = _field return nil } -func (p *TaskConfig) ReadField2(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { +func (p *EvaluationExperimentConfig) ReadField3(iprot thrift.TProtocol) error { + + var _field *string + if v, err := iprot.ReadString(); err != nil { return err + } else { + _field = &v } - _field := make([]*DataReflowConfig, 0, size) - values := make([]DataReflowConfig, size) - for i := 0; i < size; i++ { + p.SourceTargetID = _field + return nil +} +func (p *EvaluationExperimentConfig) ReadField4(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v + } + p.ExptTemplateID = _field + return nil +} +func (p *EvaluationExperimentConfig) ReadField5(iprot thrift.TProtocol) error { + + var _field *string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.SourceTargetVersion = _field + return nil +} +func (p *EvaluationExperimentConfig) ReadField6(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*EvaluateFieldMapping, 0, size) + values := make([]EvaluateFieldMapping, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.FullEvalSetFieldMappings = _field + return nil +} + +func (p *EvaluationExperimentConfig) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("EvaluationExperimentConfig"); err != nil { + goto WriteStructBeginError + } + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } + if err = p.writeField3(oprot); err != nil { + fieldId = 3 + goto WriteFieldError + } + if err = p.writeField4(oprot); err != nil { + fieldId = 4 + goto WriteFieldError + } + if err = p.writeField5(oprot); err != nil { + fieldId = 5 + goto WriteFieldError + } + if err = p.writeField6(oprot); err != nil { + fieldId = 6 + goto WriteFieldError + } + } + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError + } + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) +} + +func (p *EvaluationExperimentConfig) writeField1(oprot thrift.TProtocol) (err error) { + if p.IsSetItemConcurrencyCount() { + if err = oprot.WriteFieldBegin("item_concurrency_count", thrift.I32, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI32(*p.ItemConcurrencyCount); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *EvaluationExperimentConfig) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetItemMaxRetryCount() { + if err = oprot.WriteFieldBegin("item_max_retry_count", thrift.I32, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI32(*p.ItemMaxRetryCount); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} +func (p *EvaluationExperimentConfig) writeField3(oprot thrift.TProtocol) (err error) { + if p.IsSetSourceTargetID() { + if err = oprot.WriteFieldBegin("source_target_id", thrift.STRING, 3); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.SourceTargetID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) +} +func (p *EvaluationExperimentConfig) writeField4(oprot thrift.TProtocol) (err error) { + if p.IsSetExptTemplateID() { + if err = oprot.WriteFieldBegin("expt_template_id", thrift.I64, 4); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.ExptTemplateID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) +} +func (p *EvaluationExperimentConfig) writeField5(oprot thrift.TProtocol) (err error) { + if p.IsSetSourceTargetVersion() { + if err = oprot.WriteFieldBegin("source_target_version", thrift.STRING, 5); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.SourceTargetVersion); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) +} +func (p *EvaluationExperimentConfig) writeField6(oprot thrift.TProtocol) (err error) { + if p.IsSetFullEvalSetFieldMappings() { + if err = oprot.WriteFieldBegin("full_eval_set_field_mappings", thrift.LIST, 6); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FullEvalSetFieldMappings)); err != nil { + return err + } + for _, v := range p.FullEvalSetFieldMappings { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) +} + +func (p *EvaluationExperimentConfig) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("EvaluationExperimentConfig(%+v)", *p) + +} + +func (p *EvaluationExperimentConfig) DeepEqual(ano *EvaluationExperimentConfig) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.ItemConcurrencyCount) { + return false + } + if !p.Field2DeepEqual(ano.ItemMaxRetryCount) { + return false + } + if !p.Field3DeepEqual(ano.SourceTargetID) { + return false + } + if !p.Field4DeepEqual(ano.ExptTemplateID) { + return false + } + if !p.Field5DeepEqual(ano.SourceTargetVersion) { + return false + } + if !p.Field6DeepEqual(ano.FullEvalSetFieldMappings) { + return false + } + return true +} + +func (p *EvaluationExperimentConfig) Field1DeepEqual(src *int32) bool { + + if p.ItemConcurrencyCount == src { + return true + } else if p.ItemConcurrencyCount == nil || src == nil { + return false + } + if *p.ItemConcurrencyCount != *src { + return false + } + return true +} +func (p *EvaluationExperimentConfig) Field2DeepEqual(src *int32) bool { + + if p.ItemMaxRetryCount == src { + return true + } else if p.ItemMaxRetryCount == nil || src == nil { + return false + } + if *p.ItemMaxRetryCount != *src { + return false + } + return true +} +func (p *EvaluationExperimentConfig) Field3DeepEqual(src *string) bool { + + if p.SourceTargetID == src { + return true + } else if p.SourceTargetID == nil || src == nil { + return false + } + if strings.Compare(*p.SourceTargetID, *src) != 0 { + return false + } + return true +} +func (p *EvaluationExperimentConfig) Field4DeepEqual(src *int64) bool { + + if p.ExptTemplateID == src { + return true + } else if p.ExptTemplateID == nil || src == nil { + return false + } + if *p.ExptTemplateID != *src { + return false + } + return true +} +func (p *EvaluationExperimentConfig) Field5DeepEqual(src *string) bool { + + if p.SourceTargetVersion == src { + return true + } else if p.SourceTargetVersion == nil || src == nil { + return false + } + if strings.Compare(*p.SourceTargetVersion, *src) != 0 { + return false + } + return true +} +func (p *EvaluationExperimentConfig) Field6DeepEqual(src []*EvaluateFieldMapping) bool { + + if len(p.FullEvalSetFieldMappings) != len(src) { + return false + } + for i, v := range p.FullEvalSetFieldMappings { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true +} + +type SourceInfo struct { + Name *string `thrift:"name,1,optional" frugal:"1,optional,string" form:"name" json:"name,omitempty" query:"name"` + Version *string `thrift:"version,2,optional" frugal:"2,optional,string" form:"version" json:"version,omitempty" query:"version"` +} + +func NewSourceInfo() *SourceInfo { + return &SourceInfo{} +} + +func (p *SourceInfo) InitDefault() { +} + +var SourceInfo_Name_DEFAULT string + +func (p *SourceInfo) GetName() (v string) { + if p == nil { + return + } + if !p.IsSetName() { + return SourceInfo_Name_DEFAULT + } + return *p.Name +} + +var SourceInfo_Version_DEFAULT string + +func (p *SourceInfo) GetVersion() (v string) { + if p == nil { + return + } + if !p.IsSetVersion() { + return SourceInfo_Version_DEFAULT + } + return *p.Version +} +func (p *SourceInfo) SetName(val *string) { + p.Name = val +} +func (p *SourceInfo) SetVersion(val *string) { + p.Version = val +} + +var fieldIDToName_SourceInfo = map[int16]string{ + 1: "name", + 2: "version", +} + +func (p *SourceInfo) IsSetName() bool { + return p.Name != nil +} + +func (p *SourceInfo) IsSetVersion() bool { + return p.Version != nil +} + +func (p *SourceInfo) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.STRING { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 2: + if fieldTypeId == thrift.STRING { + if err = p.ReadField2(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SourceInfo[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +} + +func (p *SourceInfo) ReadField1(iprot thrift.TProtocol) error { + + var _field *string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.Name = _field + return nil +} +func (p *SourceInfo) ReadField2(iprot thrift.TProtocol) error { + + var _field *string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.Version = _field + return nil +} + +func (p *SourceInfo) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("SourceInfo"); err != nil { + goto WriteStructBeginError + } + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } + } + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError + } + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) +} + +func (p *SourceInfo) writeField1(oprot thrift.TProtocol) (err error) { + if p.IsSetName() { + if err = oprot.WriteFieldBegin("name", thrift.STRING, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.Name); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *SourceInfo) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetVersion() { + if err = oprot.WriteFieldBegin("version", thrift.STRING, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.Version); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} + +func (p *SourceInfo) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("SourceInfo(%+v)", *p) + +} + +func (p *SourceInfo) DeepEqual(ano *SourceInfo) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.Name) { + return false + } + if !p.Field2DeepEqual(ano.Version) { + return false + } + return true +} + +func (p *SourceInfo) Field1DeepEqual(src *string) bool { + + if p.Name == src { + return true + } else if p.Name == nil || src == nil { + return false + } + if strings.Compare(*p.Name, *src) != 0 { + return false + } + return true +} +func (p *SourceInfo) Field2DeepEqual(src *string) bool { + + if p.Version == src { + return true + } else if p.Version == nil || src == nil { + return false + } + if strings.Compare(*p.Version, *src) != 0 { + return false + } + return true +} + +// TaskConfig +type TaskConfig struct { + // 配置的评测规则信息 evaluator 维度 + AutoEvaluateConfigs []*AutoEvaluateConfig `thrift:"auto_evaluate_configs,1,optional" frugal:"1,optional,list" form:"auto_evaluate_configs" json:"auto_evaluate_configs,omitempty" query:"auto_evaluate_configs"` + // 配置的数据回流的数据集信息 + DataReflowConfig []*DataReflowConfig `thrift:"data_reflow_config,2,optional" frugal:"2,optional,list" form:"data_reflow_config" json:"data_reflow_config,omitempty" query:"data_reflow_config"` + // 评测实验配置 task 维度 + EvaluationExperimentConfig *EvaluationExperimentConfig `thrift:"evaluation_experiment_config,3,optional" frugal:"3,optional,EvaluationExperimentConfig" form:"evaluation_experiment_config" json:"evaluation_experiment_config,omitempty" query:"evaluation_experiment_config"` + // 数据源信息 + SourceInfo []*SourceInfo `thrift:"source_info,4,optional" frugal:"4,optional,list" form:"source_info" json:"source_info,omitempty" query:"source_info"` + // 工作流周期性任务 + IsWorkflowScheduled *bool `thrift:"is_workflow_scheduled,5,optional" frugal:"5,optional,bool" form:"is_workflow_scheduled" json:"is_workflow_scheduled,omitempty" query:"is_workflow_scheduled"` +} + +func NewTaskConfig() *TaskConfig { + return &TaskConfig{} +} + +func (p *TaskConfig) InitDefault() { +} + +var TaskConfig_AutoEvaluateConfigs_DEFAULT []*AutoEvaluateConfig + +func (p *TaskConfig) GetAutoEvaluateConfigs() (v []*AutoEvaluateConfig) { + if p == nil { + return + } + if !p.IsSetAutoEvaluateConfigs() { + return TaskConfig_AutoEvaluateConfigs_DEFAULT + } + return p.AutoEvaluateConfigs +} + +var TaskConfig_DataReflowConfig_DEFAULT []*DataReflowConfig + +func (p *TaskConfig) GetDataReflowConfig() (v []*DataReflowConfig) { + if p == nil { + return + } + if !p.IsSetDataReflowConfig() { + return TaskConfig_DataReflowConfig_DEFAULT + } + return p.DataReflowConfig +} + +var TaskConfig_EvaluationExperimentConfig_DEFAULT *EvaluationExperimentConfig + +func (p *TaskConfig) GetEvaluationExperimentConfig() (v *EvaluationExperimentConfig) { + if p == nil { + return + } + if !p.IsSetEvaluationExperimentConfig() { + return TaskConfig_EvaluationExperimentConfig_DEFAULT + } + return p.EvaluationExperimentConfig +} + +var TaskConfig_SourceInfo_DEFAULT []*SourceInfo + +func (p *TaskConfig) GetSourceInfo() (v []*SourceInfo) { + if p == nil { + return + } + if !p.IsSetSourceInfo() { + return TaskConfig_SourceInfo_DEFAULT + } + return p.SourceInfo +} + +var TaskConfig_IsWorkflowScheduled_DEFAULT bool + +func (p *TaskConfig) GetIsWorkflowScheduled() (v bool) { + if p == nil { + return + } + if !p.IsSetIsWorkflowScheduled() { + return TaskConfig_IsWorkflowScheduled_DEFAULT + } + return *p.IsWorkflowScheduled +} +func (p *TaskConfig) SetAutoEvaluateConfigs(val []*AutoEvaluateConfig) { + p.AutoEvaluateConfigs = val +} +func (p *TaskConfig) SetDataReflowConfig(val []*DataReflowConfig) { + p.DataReflowConfig = val +} +func (p *TaskConfig) SetEvaluationExperimentConfig(val *EvaluationExperimentConfig) { + p.EvaluationExperimentConfig = val +} +func (p *TaskConfig) SetSourceInfo(val []*SourceInfo) { + p.SourceInfo = val +} +func (p *TaskConfig) SetIsWorkflowScheduled(val *bool) { + p.IsWorkflowScheduled = val +} + +var fieldIDToName_TaskConfig = map[int16]string{ + 1: "auto_evaluate_configs", + 2: "data_reflow_config", + 3: "evaluation_experiment_config", + 4: "source_info", + 5: "is_workflow_scheduled", +} + +func (p *TaskConfig) IsSetAutoEvaluateConfigs() bool { + return p.AutoEvaluateConfigs != nil +} + +func (p *TaskConfig) IsSetDataReflowConfig() bool { + return p.DataReflowConfig != nil +} + +func (p *TaskConfig) IsSetEvaluationExperimentConfig() bool { + return p.EvaluationExperimentConfig != nil +} + +func (p *TaskConfig) IsSetSourceInfo() bool { + return p.SourceInfo != nil +} + +func (p *TaskConfig) IsSetIsWorkflowScheduled() bool { + return p.IsWorkflowScheduled != nil +} + +func (p *TaskConfig) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.LIST { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 2: + if fieldTypeId == thrift.LIST { + if err = p.ReadField2(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 3: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField3(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 4: + if fieldTypeId == thrift.LIST { + if err = p.ReadField4(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 5: + if fieldTypeId == thrift.BOOL { + if err = p.ReadField5(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TaskConfig[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +} + +func (p *TaskConfig) ReadField1(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*AutoEvaluateConfig, 0, size) + values := make([]AutoEvaluateConfig, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.AutoEvaluateConfigs = _field + return nil +} +func (p *TaskConfig) ReadField2(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*DataReflowConfig, 0, size) + values := make([]DataReflowConfig, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.DataReflowConfig = _field + return nil +} +func (p *TaskConfig) ReadField3(iprot thrift.TProtocol) error { + _field := NewEvaluationExperimentConfig() + if err := _field.Read(iprot); err != nil { + return err + } + p.EvaluationExperimentConfig = _field + return nil +} +func (p *TaskConfig) ReadField4(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*SourceInfo, 0, size) + values := make([]SourceInfo, size) + for i := 0; i < size; i++ { _elem := &values[i] _elem.InitDefault() @@ -2424,7 +3568,18 @@ func (p *TaskConfig) ReadField2(iprot thrift.TProtocol) error { if err := iprot.ReadListEnd(); err != nil { return err } - p.DataReflowConfig = _field + p.SourceInfo = _field + return nil +} +func (p *TaskConfig) ReadField5(iprot thrift.TProtocol) error { + + var _field *bool + if v, err := iprot.ReadBool(); err != nil { + return err + } else { + _field = &v + } + p.IsWorkflowScheduled = _field return nil } @@ -2442,6 +3597,18 @@ func (p *TaskConfig) Write(oprot thrift.TProtocol) (err error) { fieldId = 2 goto WriteFieldError } + if err = p.writeField3(oprot); err != nil { + fieldId = 3 + goto WriteFieldError + } + if err = p.writeField4(oprot); err != nil { + fieldId = 4 + goto WriteFieldError + } + if err = p.writeField5(oprot); err != nil { + fieldId = 5 + goto WriteFieldError + } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -2512,6 +3679,68 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } +func (p *TaskConfig) writeField3(oprot thrift.TProtocol) (err error) { + if p.IsSetEvaluationExperimentConfig() { + if err = oprot.WriteFieldBegin("evaluation_experiment_config", thrift.STRUCT, 3); err != nil { + goto WriteFieldBeginError + } + if err := p.EvaluationExperimentConfig.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) +} +func (p *TaskConfig) writeField4(oprot thrift.TProtocol) (err error) { + if p.IsSetSourceInfo() { + if err = oprot.WriteFieldBegin("source_info", thrift.LIST, 4); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.SourceInfo)); err != nil { + return err + } + for _, v := range p.SourceInfo { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) +} +func (p *TaskConfig) writeField5(oprot thrift.TProtocol) (err error) { + if p.IsSetIsWorkflowScheduled() { + if err = oprot.WriteFieldBegin("is_workflow_scheduled", thrift.BOOL, 5); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteBool(*p.IsWorkflowScheduled); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) +} func (p *TaskConfig) String() string { if p == nil { @@ -2533,6 +3762,15 @@ func (p *TaskConfig) DeepEqual(ano *TaskConfig) bool { if !p.Field2DeepEqual(ano.DataReflowConfig) { return false } + if !p.Field3DeepEqual(ano.EvaluationExperimentConfig) { + return false + } + if !p.Field4DeepEqual(ano.SourceInfo) { + return false + } + if !p.Field5DeepEqual(ano.IsWorkflowScheduled) { + return false + } return true } @@ -2562,6 +3800,38 @@ func (p *TaskConfig) Field2DeepEqual(src []*DataReflowConfig) bool { } return true } +func (p *TaskConfig) Field3DeepEqual(src *EvaluationExperimentConfig) bool { + + if !p.EvaluationExperimentConfig.DeepEqual(src) { + return false + } + return true +} +func (p *TaskConfig) Field4DeepEqual(src []*SourceInfo) bool { + + if len(p.SourceInfo) != len(src) { + return false + } + for i, v := range p.SourceInfo { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true +} +func (p *TaskConfig) Field5DeepEqual(src *bool) bool { + + if p.IsWorkflowScheduled == src { + return true + } else if p.IsWorkflowScheduled == nil || src == nil { + return false + } + if *p.IsWorkflowScheduled != *src { + return false + } + return true +} type DataReflowConfig struct { // 数据集id，新增数据集时可为空 @@ -2571,6 +3841,8 @@ type DataReflowConfig struct { // 数据集列数据schema DatasetSchema *dataset.DatasetSchema `thrift:"dataset_schema,3,optional" frugal:"3,optional,dataset.DatasetSchema" form:"dataset_schema" json:"dataset_schema,omitempty" query:"dataset_schema"` FieldMappings []*dataset.FieldMapping `thrift:"field_mappings,4,optional" frugal:"4,optional,list" form:"field_mappings" json:"field_mappings,omitempty" query:"field_mappings"` + // 数据集类型 + DatasetCategory *dataset0.DatasetCategory `thrift:"dataset_category,5,optional" frugal:"5,optional,DatasetCategory" form:"dataset_category" json:"dataset_category,omitempty" query:"dataset_category"` } func NewDataReflowConfig() *DataReflowConfig { @@ -2627,6 +3899,18 @@ func (p *DataReflowConfig) GetFieldMappings() (v []*dataset.FieldMapping) { } return p.FieldMappings } + +var DataReflowConfig_DatasetCategory_DEFAULT dataset0.DatasetCategory + +func (p *DataReflowConfig) GetDatasetCategory() (v dataset0.DatasetCategory) { + if p == nil { + return + } + if !p.IsSetDatasetCategory() { + return DataReflowConfig_DatasetCategory_DEFAULT + } + return *p.DatasetCategory +} func (p *DataReflowConfig) SetDatasetID(val *int64) { p.DatasetID = val } @@ -2639,12 +3923,16 @@ func (p *DataReflowConfig) SetDatasetSchema(val *dataset.DatasetSchema) { func (p *DataReflowConfig) SetFieldMappings(val []*dataset.FieldMapping) { p.FieldMappings = val } +func (p *DataReflowConfig) SetDatasetCategory(val *dataset0.DatasetCategory) { + p.DatasetCategory = val +} var fieldIDToName_DataReflowConfig = map[int16]string{ 1: "dataset_id", 2: "dataset_name", 3: "dataset_schema", 4: "field_mappings", + 5: "dataset_category", } func (p *DataReflowConfig) IsSetDatasetID() bool { @@ -2663,6 +3951,10 @@ func (p *DataReflowConfig) IsSetFieldMappings() bool { return p.FieldMappings != nil } +func (p *DataReflowConfig) IsSetDatasetCategory() bool { + return p.DatasetCategory != nil +} + func (p *DataReflowConfig) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -2713,6 +4005,14 @@ func (p *DataReflowConfig) Read(iprot thrift.TProtocol) (err error) { } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } + case 5: + if fieldTypeId == thrift.I32 { + if err = p.ReadField5(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } default: if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError @@ -2795,6 +4095,18 @@ func (p *DataReflowConfig) ReadField4(iprot thrift.TProtocol) error { p.FieldMappings = _field return nil } +func (p *DataReflowConfig) ReadField5(iprot thrift.TProtocol) error { + + var _field *dataset0.DatasetCategory + if v, err := iprot.ReadI32(); err != nil { + return err + } else { + tmp := dataset0.DatasetCategory(v) + _field = &tmp + } + p.DatasetCategory = _field + return nil +} func (p *DataReflowConfig) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 @@ -2818,6 +4130,10 @@ func (p *DataReflowConfig) Write(oprot thrift.TProtocol) (err error) { fieldId = 4 goto WriteFieldError } + if err = p.writeField5(oprot); err != nil { + fieldId = 5 + goto WriteFieldError + } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -2916,6 +4232,24 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) } +func (p *DataReflowConfig) writeField5(oprot thrift.TProtocol) (err error) { + if p.IsSetDatasetCategory() { + if err = oprot.WriteFieldBegin("dataset_category", thrift.I32, 5); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI32(int32(*p.DatasetCategory)); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) +} func (p *DataReflowConfig) String() string { if p == nil { @@ -2943,6 +4277,9 @@ func (p *DataReflowConfig) DeepEqual(ano *DataReflowConfig) bool { if !p.Field4DeepEqual(ano.FieldMappings) { return false } + if !p.Field5DeepEqual(ano.DatasetCategory) { + return false + } return true } @@ -2990,11 +4327,25 @@ func (p *DataReflowConfig) Field4DeepEqual(src []*dataset.FieldMapping) bool { } return true } +func (p *DataReflowConfig) Field5DeepEqual(src *dataset0.DatasetCategory) bool { + + if p.DatasetCategory == src { + return true + } else if p.DatasetCategory == nil || src == nil { + return false + } + if *p.DatasetCategory != *src { + return false + } + return true +} type AutoEvaluateConfig struct { EvaluatorVersionID int64 `thrift:"evaluator_version_id,1,required" frugal:"1,required,i64" json:"evaluator_version_id" form:"evaluator_version_id,required" query:"evaluator_version_id,required"` EvaluatorID int64 `thrift:"evaluator_id,2,required" frugal:"2,required,i64" json:"evaluator_id" form:"evaluator_id,required" query:"evaluator_id,required"` FieldMappings []*EvaluateFieldMapping `thrift:"field_mappings,3,required" frugal:"3,required,list" form:"field_mappings,required" json:"field_mappings,required" query:"field_mappings,required"` + ScoreWeight *float64 `thrift:"score_weight,4,optional" frugal:"4,optional,double" json:"score_weight" form:"score_weight" query:"score_weight"` + EvaluatorVersion *string `thrift:"evaluator_version,5,optional" frugal:"5,optional,string" form:"evaluator_version" json:"evaluator_version,omitempty" query:"evaluator_version"` } func NewAutoEvaluateConfig() *AutoEvaluateConfig { @@ -3024,6 +4375,30 @@ func (p *AutoEvaluateConfig) GetFieldMappings() (v []*EvaluateFieldMapping) { } return } + +var AutoEvaluateConfig_ScoreWeight_DEFAULT float64 + +func (p *AutoEvaluateConfig) GetScoreWeight() (v float64) { + if p == nil { + return + } + if !p.IsSetScoreWeight() { + return AutoEvaluateConfig_ScoreWeight_DEFAULT + } + return *p.ScoreWeight +} + +var AutoEvaluateConfig_EvaluatorVersion_DEFAULT string + +func (p *AutoEvaluateConfig) GetEvaluatorVersion() (v string) { + if p == nil { + return + } + if !p.IsSetEvaluatorVersion() { + return AutoEvaluateConfig_EvaluatorVersion_DEFAULT + } + return *p.EvaluatorVersion +} func (p *AutoEvaluateConfig) SetEvaluatorVersionID(val int64) { p.EvaluatorVersionID = val } @@ -3033,11 +4408,27 @@ func (p *AutoEvaluateConfig) SetEvaluatorID(val int64) { func (p *AutoEvaluateConfig) SetFieldMappings(val []*EvaluateFieldMapping) { p.FieldMappings = val } +func (p *AutoEvaluateConfig) SetScoreWeight(val *float64) { + p.ScoreWeight = val +} +func (p *AutoEvaluateConfig) SetEvaluatorVersion(val *string) { + p.EvaluatorVersion = val +} var fieldIDToName_AutoEvaluateConfig = map[int16]string{ 1: "evaluator_version_id", 2: "evaluator_id", 3: "field_mappings", + 4: "score_weight", + 5: "evaluator_version", +} + +func (p *AutoEvaluateConfig) IsSetScoreWeight() bool { + return p.ScoreWeight != nil +} + +func (p *AutoEvaluateConfig) IsSetEvaluatorVersion() bool { + return p.EvaluatorVersion != nil } func (p *AutoEvaluateConfig) Read(iprot thrift.TProtocol) (err error) { @@ -3088,6 +4479,22 @@ func (p *AutoEvaluateConfig) Read(iprot thrift.TProtocol) (err error) { } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } + case 4: + if fieldTypeId == thrift.DOUBLE { + if err = p.ReadField4(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 5: + if fieldTypeId == thrift.STRING { + if err = p.ReadField5(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } default: if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError @@ -3178,6 +4585,28 @@ func (p *AutoEvaluateConfig) ReadField3(iprot thrift.TProtocol) error { p.FieldMappings = _field return nil } +func (p *AutoEvaluateConfig) ReadField4(iprot thrift.TProtocol) error { + + var _field *float64 + if v, err := iprot.ReadDouble(); err != nil { + return err + } else { + _field = &v + } + p.ScoreWeight = _field + return nil +} +func (p *AutoEvaluateConfig) ReadField5(iprot thrift.TProtocol) error { + + var _field *string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.EvaluatorVersion = _field + return nil +} func (p *AutoEvaluateConfig) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 @@ -3197,6 +4626,14 @@ func (p *AutoEvaluateConfig) Write(oprot thrift.TProtocol) (err error) { fieldId = 3 goto WriteFieldError } + if err = p.writeField4(oprot); err != nil { + fieldId = 4 + goto WriteFieldError + } + if err = p.writeField5(oprot); err != nil { + fieldId = 5 + goto WriteFieldError + } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -3271,6 +4708,42 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) } +func (p *AutoEvaluateConfig) writeField4(oprot thrift.TProtocol) (err error) { + if p.IsSetScoreWeight() { + if err = oprot.WriteFieldBegin("score_weight", thrift.DOUBLE, 4); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteDouble(*p.ScoreWeight); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) +} +func (p *AutoEvaluateConfig) writeField5(oprot thrift.TProtocol) (err error) { + if p.IsSetEvaluatorVersion() { + if err = oprot.WriteFieldBegin("evaluator_version", thrift.STRING, 5); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.EvaluatorVersion); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) +} func (p *AutoEvaluateConfig) String() string { if p == nil { @@ -3295,6 +4768,12 @@ func (p *AutoEvaluateConfig) DeepEqual(ano *AutoEvaluateConfig) bool { if !p.Field3DeepEqual(ano.FieldMappings) { return false } + if !p.Field4DeepEqual(ano.ScoreWeight) { + return false + } + if !p.Field5DeepEqual(ano.EvaluatorVersion) { + return false + } return true } @@ -3325,6 +4804,30 @@ func (p *AutoEvaluateConfig) Field3DeepEqual(src []*EvaluateFieldMapping) bool { } return true } +func (p *AutoEvaluateConfig) Field4DeepEqual(src *float64) bool { + + if p.ScoreWeight == src { + return true + } else if p.ScoreWeight == nil || src == nil { + return false + } + if *p.ScoreWeight != *src { + return false + } + return true +} +func (p *AutoEvaluateConfig) Field5DeepEqual(src *string) bool { + + if p.EvaluatorVersion == src { + return true + } else if p.EvaluatorVersion == nil || src == nil { + return false + } + if strings.Compare(*p.EvaluatorVersion, *src) != 0 { + return false + } + return true +} // RunDetail type RunDetail struct { diff --git a/backend/kitex_gen/coze/loop/observability/domain/task/task_validator.go b/backend/kitex_gen/coze/loop/observability/domain/task/task_validator.go index c157b3ca8..1a19cbe80 100644 --- a/backend/kitex_gen/coze/loop/observability/domain/task/task_validator.go +++ b/backend/kitex_gen/coze/loop/observability/domain/task/task_validator.go @@ -78,7 +78,18 @@ func (p *Sampler) IsValid() error { func (p *EffectiveTime) IsValid() error { return nil } +func (p *EvaluationExperimentConfig) IsValid() error { + return nil +} +func (p *SourceInfo) IsValid() error { + return nil +} func (p *TaskConfig) IsValid() error { + if p.EvaluationExperimentConfig != nil { + if err := p.EvaluationExperimentConfig.IsValid(); err != nil { + return fmt.Errorf("field EvaluationExperimentConfig not valid, %w", err) + } + } return nil } func (p *DataReflowConfig) IsValid() error { diff --git a/backend/kitex_gen/coze/loop/observability/observabilitytraceservice/client.go b/backend/kitex_gen/coze/loop/observability/observabilitytraceservice/client.go index 146ab0d78..3a1673bca 100644 --- a/backend/kitex_gen/coze/loop/observability/observabilitytraceservice/client.go +++ b/backend/kitex_gen/coze/loop/observability/observabilitytraceservice/client.go @@ -26,6 +26,7 @@ type Client interface { UpdateManualAnnotation(ctx context.Context, req *trace.UpdateManualAnnotationRequest, callOptions ...callopt.Option) (r *trace.UpdateManualAnnotationResponse, err error) DeleteManualAnnotation(ctx context.Context, req *trace.DeleteManualAnnotationRequest, callOptions ...callopt.Option) (r *trace.DeleteManualAnnotationResponse, err error) ListAnnotations(ctx context.Context, req *trace.ListAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListAnnotationsResponse, err error) + ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListWorkspaceAnnotationsResponse, err error) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.ExportTracesToDatasetResponse, err error) PreviewExportTracesToDataset(ctx context.Context, req *trace.PreviewExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.PreviewExportTracesToDatasetResponse, err error) ChangeEvaluatorScore(ctx context.Context, req *trace.ChangeEvaluatorScoreRequest, callOptions ...callopt.Option) (r *trace.ChangeEvaluatorScoreResponse, err error) @@ -34,6 +35,7 @@ type Client interface { UpsertTrajectoryConfig(ctx context.Context, req *trace.UpsertTrajectoryConfigRequest, callOptions ...callopt.Option) (r *trace.UpsertTrajectoryConfigResponse, err error) GetTrajectoryConfig(ctx context.Context, req *trace.GetTrajectoryConfigRequest, callOptions ...callopt.Option) (r *trace.GetTrajectoryConfigResponse, err error) ListTrajectory(ctx context.Context, req *trace.ListTrajectoryRequest, callOptions ...callopt.Option) (r *trace.ListTrajectoryResponse, err error) + ListMetadata(ctx context.Context, req *trace.ListMetadataRequest, callOptions ...callopt.Option) (r *trace.ListMetadataResponse, err error) } // NewClient creates a client for the service defined in IDL. @@ -140,6 +142,11 @@ func (p *kObservabilityTraceServiceClient) ListAnnotations(ctx context.Context, return p.kClient.ListAnnotations(ctx, req) } +func (p *kObservabilityTraceServiceClient) ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListWorkspaceAnnotationsResponse, err error) { + ctx = client.NewCtxWithCallOptions(ctx, callOptions) + return p.kClient.ListWorkspaceAnnotations(ctx, req) +} + func (p *kObservabilityTraceServiceClient) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.ExportTracesToDatasetResponse, err error) { ctx = client.NewCtxWithCallOptions(ctx, callOptions) return p.kClient.ExportTracesToDataset(ctx, req) @@ -179,3 +186,8 @@ func (p *kObservabilityTraceServiceClient) ListTrajectory(ctx context.Context, r ctx = client.NewCtxWithCallOptions(ctx, callOptions) return p.kClient.ListTrajectory(ctx, req) } + +func (p *kObservabilityTraceServiceClient) ListMetadata(ctx context.Context, req *trace.ListMetadataRequest, callOptions ...callopt.Option) (r *trace.ListMetadataResponse, err error) { + ctx = client.NewCtxWithCallOptions(ctx, callOptions) + return p.kClient.ListMetadata(ctx, req) +} diff --git a/backend/kitex_gen/coze/loop/observability/observabilitytraceservice/observabilitytraceservice.go b/backend/kitex_gen/coze/loop/observability/observabilitytraceservice/observabilitytraceservice.go index dc730040e..0c87bd26f 100644 --- a/backend/kitex_gen/coze/loop/observability/observabilitytraceservice/observabilitytraceservice.go +++ b/backend/kitex_gen/coze/loop/observability/observabilitytraceservice/observabilitytraceservice.go @@ -119,6 +119,13 @@ var serviceMethods = map[string]kitex.MethodInfo{ false, kitex.WithStreamingMode(kitex.StreamingNone), ), + "ListWorkspaceAnnotations": kitex.NewMethodInfo( + listWorkspaceAnnotationsHandler, + newTraceServiceListWorkspaceAnnotationsArgs, + newTraceServiceListWorkspaceAnnotationsResult, + false, + kitex.WithStreamingMode(kitex.StreamingNone), + ), "ExportTracesToDataset": kitex.NewMethodInfo( exportTracesToDatasetHandler, newTraceServiceExportTracesToDatasetArgs, @@ -175,6 +182,13 @@ var serviceMethods = map[string]kitex.MethodInfo{ false, kitex.WithStreamingMode(kitex.StreamingNone), ), + "ListMetadata": kitex.NewMethodInfo( + listMetadataHandler, + newTraceServiceListMetadataArgs, + newTraceServiceListMetadataResult, + false, + kitex.WithStreamingMode(kitex.StreamingNone), + ), } var ( @@ -493,6 +507,25 @@ func newTraceServiceListAnnotationsResult() interface{} { return trace.NewTraceServiceListAnnotationsResult() } +func listWorkspaceAnnotationsHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { + realArg := arg.(*trace.TraceServiceListWorkspaceAnnotationsArgs) + realResult := result.(*trace.TraceServiceListWorkspaceAnnotationsResult) + success, err := handler.(trace.TraceService).ListWorkspaceAnnotations(ctx, realArg.Req) + if err != nil { + return err + } + realResult.Success = success + return nil +} + +func newTraceServiceListWorkspaceAnnotationsArgs() interface{} { + return trace.NewTraceServiceListWorkspaceAnnotationsArgs() +} + +func newTraceServiceListWorkspaceAnnotationsResult() interface{} { + return trace.NewTraceServiceListWorkspaceAnnotationsResult() +} + func exportTracesToDatasetHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { realArg := arg.(*trace.TraceServiceExportTracesToDatasetArgs) realResult := result.(*trace.TraceServiceExportTracesToDatasetResult) @@ -645,6 +678,25 @@ func newTraceServiceListTrajectoryResult() interface{} { return trace.NewTraceServiceListTrajectoryResult() } +func listMetadataHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { + realArg := arg.(*trace.TraceServiceListMetadataArgs) + realResult := result.(*trace.TraceServiceListMetadataResult) + success, err := handler.(trace.TraceService).ListMetadata(ctx, realArg.Req) + if err != nil { + return err + } + realResult.Success = success + return nil +} + +func newTraceServiceListMetadataArgs() interface{} { + return trace.NewTraceServiceListMetadataArgs() +} + +func newTraceServiceListMetadataResult() interface{} { + return trace.NewTraceServiceListMetadataResult() +} + type kClient struct { c client.Client sc client.Streaming @@ -807,6 +859,16 @@ func (p *kClient) ListAnnotations(ctx context.Context, req *trace.ListAnnotation return _result.GetSuccess(), nil } +func (p *kClient) ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest) (r *trace.ListWorkspaceAnnotationsResponse, err error) { + var _args trace.TraceServiceListWorkspaceAnnotationsArgs + _args.Req = req + var _result trace.TraceServiceListWorkspaceAnnotationsResult + if err = p.c.Call(ctx, "ListWorkspaceAnnotations", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} + func (p *kClient) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest) (r *trace.ExportTracesToDatasetResponse, err error) { var _args trace.TraceServiceExportTracesToDatasetArgs _args.Req = req @@ -886,3 +948,13 @@ func (p *kClient) ListTrajectory(ctx context.Context, req *trace.ListTrajectoryR } return _result.GetSuccess(), nil } + +func (p *kClient) ListMetadata(ctx context.Context, req *trace.ListMetadataRequest) (r *trace.ListMetadataResponse, err error) { + var _args trace.TraceServiceListMetadataArgs + _args.Req = req + var _result trace.TraceServiceListMetadataResult + if err = p.c.Call(ctx, "ListMetadata", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} diff --git a/backend/kitex_gen/coze/loop/observability/trace/coze.loop.observability.trace.go b/backend/kitex_gen/coze/loop/observability/trace/coze.loop.observability.trace.go index 2eba5a47c..26c5cba3c 100644 --- a/backend/kitex_gen/coze/loop/observability/trace/coze.loop.observability.trace.go +++ b/backend/kitex_gen/coze/loop/observability/trace/coze.loop.observability.trace.go @@ -12926,186 +12926,119 @@ func (p *ListAnnotationsResponse) Field255DeepEqual(src *base.BaseResp) bool { return true } -type ExportTracesToDatasetRequest struct { - WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` - SpanIds []*SpanID `thrift:"span_ids,2,required" frugal:"2,required,list" form:"span_ids,required" json:"span_ids,required"` - Category dataset.DatasetCategory `thrift:"category,3,required" frugal:"3,required,DatasetCategory" form:"category,required" json:"category,required"` - Config *DatasetConfig `thrift:"config,4,required" frugal:"4,required,DatasetConfig" form:"config,required" json:"config,required"` - StartTime int64 `thrift:"start_time,5,required" frugal:"5,required,i64" json:"start_time" form:"start_time,required" ` - EndTime int64 `thrift:"end_time,6,required" frugal:"6,required,i64" json:"end_time" form:"end_time,required" ` - PlatformType *common.PlatformType `thrift:"platform_type,7,optional" frugal:"7,optional,string" form:"platform_type" json:"platform_type,omitempty"` - // 导入方式，不填默认为追加 - ExportType dataset0.ExportType `thrift:"export_type,8,required" frugal:"8,required,string" form:"export_type,required" json:"export_type,required"` - FieldMappings []*dataset0.FieldMapping `thrift:"field_mappings,9,optional" frugal:"9,optional,list" form:"field_mappings" json:"field_mappings,omitempty"` - Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` +type ListWorkspaceAnnotationsRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` + PlatformType *common.PlatformType `thrift:"platform_type,2,optional" frugal:"2,optional,string" form:"platform_type" json:"platform_type,omitempty"` + AnnotationType *annotation.AnnotationType `thrift:"annotation_type,3,optional" frugal:"3,optional,string" form:"annotation_type" json:"annotation_type,omitempty"` + SpanListType *common.SpanListType `thrift:"span_list_type,4,optional" frugal:"4,optional,string" form:"span_list_type" json:"span_list_type,omitempty"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` } -func NewExportTracesToDatasetRequest() *ExportTracesToDatasetRequest { - return &ExportTracesToDatasetRequest{} +func NewListWorkspaceAnnotationsRequest() *ListWorkspaceAnnotationsRequest { + return &ListWorkspaceAnnotationsRequest{} } -func (p *ExportTracesToDatasetRequest) InitDefault() { +func (p *ListWorkspaceAnnotationsRequest) InitDefault() { } -func (p *ExportTracesToDatasetRequest) GetWorkspaceID() (v int64) { +func (p *ListWorkspaceAnnotationsRequest) GetWorkspaceID() (v int64) { if p != nil { return p.WorkspaceID } return } -func (p *ExportTracesToDatasetRequest) GetSpanIds() (v []*SpanID) { - if p != nil { - return p.SpanIds - } - return -} - -func (p *ExportTracesToDatasetRequest) GetCategory() (v dataset.DatasetCategory) { - if p != nil { - return p.Category - } - return -} - -var ExportTracesToDatasetRequest_Config_DEFAULT *DatasetConfig +var ListWorkspaceAnnotationsRequest_PlatformType_DEFAULT common.PlatformType -func (p *ExportTracesToDatasetRequest) GetConfig() (v *DatasetConfig) { +func (p *ListWorkspaceAnnotationsRequest) GetPlatformType() (v common.PlatformType) { if p == nil { return } - if !p.IsSetConfig() { - return ExportTracesToDatasetRequest_Config_DEFAULT - } - return p.Config -} - -func (p *ExportTracesToDatasetRequest) GetStartTime() (v int64) { - if p != nil { - return p.StartTime - } - return -} - -func (p *ExportTracesToDatasetRequest) GetEndTime() (v int64) { - if p != nil { - return p.EndTime + if !p.IsSetPlatformType() { + return ListWorkspaceAnnotationsRequest_PlatformType_DEFAULT } - return + return *p.PlatformType } -var ExportTracesToDatasetRequest_PlatformType_DEFAULT common.PlatformType +var ListWorkspaceAnnotationsRequest_AnnotationType_DEFAULT annotation.AnnotationType -func (p *ExportTracesToDatasetRequest) GetPlatformType() (v common.PlatformType) { +func (p *ListWorkspaceAnnotationsRequest) GetAnnotationType() (v annotation.AnnotationType) { if p == nil { return } - if !p.IsSetPlatformType() { - return ExportTracesToDatasetRequest_PlatformType_DEFAULT - } - return *p.PlatformType -} - -func (p *ExportTracesToDatasetRequest) GetExportType() (v dataset0.ExportType) { - if p != nil { - return p.ExportType + if !p.IsSetAnnotationType() { + return ListWorkspaceAnnotationsRequest_AnnotationType_DEFAULT } - return + return *p.AnnotationType } -var ExportTracesToDatasetRequest_FieldMappings_DEFAULT []*dataset0.FieldMapping +var ListWorkspaceAnnotationsRequest_SpanListType_DEFAULT common.SpanListType -func (p *ExportTracesToDatasetRequest) GetFieldMappings() (v []*dataset0.FieldMapping) { +func (p *ListWorkspaceAnnotationsRequest) GetSpanListType() (v common.SpanListType) { if p == nil { return } - if !p.IsSetFieldMappings() { - return ExportTracesToDatasetRequest_FieldMappings_DEFAULT + if !p.IsSetSpanListType() { + return ListWorkspaceAnnotationsRequest_SpanListType_DEFAULT } - return p.FieldMappings + return *p.SpanListType } -var ExportTracesToDatasetRequest_Base_DEFAULT *base.Base +var ListWorkspaceAnnotationsRequest_Base_DEFAULT *base.Base -func (p *ExportTracesToDatasetRequest) GetBase() (v *base.Base) { +func (p *ListWorkspaceAnnotationsRequest) GetBase() (v *base.Base) { if p == nil { return } if !p.IsSetBase() { - return ExportTracesToDatasetRequest_Base_DEFAULT + return ListWorkspaceAnnotationsRequest_Base_DEFAULT } return p.Base } -func (p *ExportTracesToDatasetRequest) SetWorkspaceID(val int64) { +func (p *ListWorkspaceAnnotationsRequest) SetWorkspaceID(val int64) { p.WorkspaceID = val } -func (p *ExportTracesToDatasetRequest) SetSpanIds(val []*SpanID) { - p.SpanIds = val -} -func (p *ExportTracesToDatasetRequest) SetCategory(val dataset.DatasetCategory) { - p.Category = val -} -func (p *ExportTracesToDatasetRequest) SetConfig(val *DatasetConfig) { - p.Config = val -} -func (p *ExportTracesToDatasetRequest) SetStartTime(val int64) { - p.StartTime = val -} -func (p *ExportTracesToDatasetRequest) SetEndTime(val int64) { - p.EndTime = val -} -func (p *ExportTracesToDatasetRequest) SetPlatformType(val *common.PlatformType) { +func (p *ListWorkspaceAnnotationsRequest) SetPlatformType(val *common.PlatformType) { p.PlatformType = val } -func (p *ExportTracesToDatasetRequest) SetExportType(val dataset0.ExportType) { - p.ExportType = val +func (p *ListWorkspaceAnnotationsRequest) SetAnnotationType(val *annotation.AnnotationType) { + p.AnnotationType = val } -func (p *ExportTracesToDatasetRequest) SetFieldMappings(val []*dataset0.FieldMapping) { - p.FieldMappings = val +func (p *ListWorkspaceAnnotationsRequest) SetSpanListType(val *common.SpanListType) { + p.SpanListType = val } -func (p *ExportTracesToDatasetRequest) SetBase(val *base.Base) { +func (p *ListWorkspaceAnnotationsRequest) SetBase(val *base.Base) { p.Base = val } -var fieldIDToName_ExportTracesToDatasetRequest = map[int16]string{ +var fieldIDToName_ListWorkspaceAnnotationsRequest = map[int16]string{ 1: "workspace_id", - 2: "span_ids", - 3: "category", - 4: "config", - 5: "start_time", - 6: "end_time", - 7: "platform_type", - 8: "export_type", - 9: "field_mappings", + 2: "platform_type", + 3: "annotation_type", + 4: "span_list_type", 255: "Base", } -func (p *ExportTracesToDatasetRequest) IsSetConfig() bool { - return p.Config != nil +func (p *ListWorkspaceAnnotationsRequest) IsSetPlatformType() bool { + return p.PlatformType != nil } -func (p *ExportTracesToDatasetRequest) IsSetPlatformType() bool { - return p.PlatformType != nil +func (p *ListWorkspaceAnnotationsRequest) IsSetAnnotationType() bool { + return p.AnnotationType != nil } -func (p *ExportTracesToDatasetRequest) IsSetFieldMappings() bool { - return p.FieldMappings != nil +func (p *ListWorkspaceAnnotationsRequest) IsSetSpanListType() bool { + return p.SpanListType != nil } -func (p *ExportTracesToDatasetRequest) IsSetBase() bool { +func (p *ListWorkspaceAnnotationsRequest) IsSetBase() bool { return p.Base != nil } -func (p *ExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err error) { +func (p *ListWorkspaceAnnotationsRequest) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 var issetWorkspaceID bool = false - var issetSpanIds bool = false - var issetCategory bool = false - var issetConfig bool = false - var issetStartTime bool = false - var issetEndTime bool = false - var issetExportType bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -13131,70 +13064,24 @@ func (p *ExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err error) goto SkipFieldError } case 2: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.STRING { if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } - issetSpanIds = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 3: - if fieldTypeId == thrift.I32 { + if fieldTypeId == thrift.STRING { if err = p.ReadField3(iprot); err != nil { goto ReadFieldError } - issetCategory = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 4: - if fieldTypeId == thrift.STRUCT { - if err = p.ReadField4(iprot); err != nil { - goto ReadFieldError - } - issetConfig = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 5: - if fieldTypeId == thrift.I64 { - if err = p.ReadField5(iprot); err != nil { - goto ReadFieldError - } - issetStartTime = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 6: - if fieldTypeId == thrift.I64 { - if err = p.ReadField6(iprot); err != nil { - goto ReadFieldError - } - issetEndTime = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 7: - if fieldTypeId == thrift.STRING { - if err = p.ReadField7(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 8: if fieldTypeId == thrift.STRING { - if err = p.ReadField8(iprot); err != nil { - goto ReadFieldError - } - issetExportType = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 9: - if fieldTypeId == thrift.LIST { - if err = p.ReadField9(iprot); err != nil { + if err = p.ReadField4(iprot); err != nil { goto ReadFieldError } } else if err = iprot.Skip(fieldTypeId); err != nil { @@ -13225,43 +13112,13 @@ func (p *ExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err error) fieldId = 1 goto RequiredFieldNotSetError } - - if !issetSpanIds { - fieldId = 2 - goto RequiredFieldNotSetError - } - - if !issetCategory { - fieldId = 3 - goto RequiredFieldNotSetError - } - - if !issetConfig { - fieldId = 4 - goto RequiredFieldNotSetError - } - - if !issetStartTime { - fieldId = 5 - goto RequiredFieldNotSetError - } - - if !issetEndTime { - fieldId = 6 - goto RequiredFieldNotSetError - } - - if !issetExportType { - fieldId = 8 - goto RequiredFieldNotSetError - } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExportTracesToDatasetRequest[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListWorkspaceAnnotationsRequest[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -13270,10 +13127,10 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ExportTracesToDatasetRequest[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListWorkspaceAnnotationsRequest[fieldId])) } -func (p *ExportTracesToDatasetRequest) ReadField1(iprot thrift.TProtocol) error { +func (p *ListWorkspaceAnnotationsRequest) ReadField1(iprot thrift.TProtocol) error { var _field int64 if v, err := iprot.ReadI64(); err != nil { @@ -13284,128 +13141,52 @@ func (p *ExportTracesToDatasetRequest) ReadField1(iprot thrift.TProtocol) error p.WorkspaceID = _field return nil } -func (p *ExportTracesToDatasetRequest) ReadField2(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { +func (p *ListWorkspaceAnnotationsRequest) ReadField2(iprot thrift.TProtocol) error { + + var _field *common.PlatformType + if v, err := iprot.ReadString(); err != nil { return err + } else { + _field = &v } - _field := make([]*SpanID, 0, size) - values := make([]SpanID, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } + p.PlatformType = _field + return nil +} +func (p *ListWorkspaceAnnotationsRequest) ReadField3(iprot thrift.TProtocol) error { - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { + var _field *annotation.AnnotationType + if v, err := iprot.ReadString(); err != nil { return err + } else { + _field = &v } - p.SpanIds = _field + p.AnnotationType = _field return nil } -func (p *ExportTracesToDatasetRequest) ReadField3(iprot thrift.TProtocol) error { +func (p *ListWorkspaceAnnotationsRequest) ReadField4(iprot thrift.TProtocol) error { - var _field dataset.DatasetCategory - if v, err := iprot.ReadI32(); err != nil { + var _field *common.SpanListType + if v, err := iprot.ReadString(); err != nil { return err } else { - _field = dataset.DatasetCategory(v) + _field = &v } - p.Category = _field + p.SpanListType = _field return nil } -func (p *ExportTracesToDatasetRequest) ReadField4(iprot thrift.TProtocol) error { - _field := NewDatasetConfig() +func (p *ListWorkspaceAnnotationsRequest) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBase() if err := _field.Read(iprot); err != nil { return err } - p.Config = _field + p.Base = _field return nil } -func (p *ExportTracesToDatasetRequest) ReadField5(iprot thrift.TProtocol) error { - var _field int64 - if v, err := iprot.ReadI64(); err != nil { - return err - } else { - _field = v - } - p.StartTime = _field - return nil -} -func (p *ExportTracesToDatasetRequest) ReadField6(iprot thrift.TProtocol) error { - - var _field int64 - if v, err := iprot.ReadI64(); err != nil { - return err - } else { - _field = v - } - p.EndTime = _field - return nil -} -func (p *ExportTracesToDatasetRequest) ReadField7(iprot thrift.TProtocol) error { - - var _field *common.PlatformType - if v, err := iprot.ReadString(); err != nil { - return err - } else { - _field = &v - } - p.PlatformType = _field - return nil -} -func (p *ExportTracesToDatasetRequest) ReadField8(iprot thrift.TProtocol) error { - - var _field dataset0.ExportType - if v, err := iprot.ReadString(); err != nil { - return err - } else { - _field = v - } - p.ExportType = _field - return nil -} -func (p *ExportTracesToDatasetRequest) ReadField9(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]*dataset0.FieldMapping, 0, size) - values := make([]dataset0.FieldMapping, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { - return err - } - p.FieldMappings = _field - return nil -} -func (p *ExportTracesToDatasetRequest) ReadField255(iprot thrift.TProtocol) error { - _field := base.NewBase() - if err := _field.Read(iprot); err != nil { - return err - } - p.Base = _field - return nil -} - -func (p *ExportTracesToDatasetRequest) Write(oprot thrift.TProtocol) (err error) { - var fieldId int16 - if err = oprot.WriteStructBegin("ExportTracesToDatasetRequest"); err != nil { - goto WriteStructBeginError +func (p *ListWorkspaceAnnotationsRequest) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListWorkspaceAnnotationsRequest"); err != nil { + goto WriteStructBeginError } if p != nil { if err = p.writeField1(oprot); err != nil { @@ -13424,26 +13205,6 @@ func (p *ExportTracesToDatasetRequest) Write(oprot thrift.TProtocol) (err error) fieldId = 4 goto WriteFieldError } - if err = p.writeField5(oprot); err != nil { - fieldId = 5 - goto WriteFieldError - } - if err = p.writeField6(oprot); err != nil { - fieldId = 6 - goto WriteFieldError - } - if err = p.writeField7(oprot); err != nil { - fieldId = 7 - goto WriteFieldError - } - if err = p.writeField8(oprot); err != nil { - fieldId = 8 - goto WriteFieldError - } - if err = p.writeField9(oprot); err != nil { - fieldId = 9 - goto WriteFieldError - } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -13466,7 +13227,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ExportTracesToDatasetRequest) writeField1(oprot thrift.TProtocol) (err error) { +func (p *ListWorkspaceAnnotationsRequest) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { goto WriteFieldBeginError } @@ -13482,23 +13243,17 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ExportTracesToDatasetRequest) writeField2(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("span_ids", thrift.LIST, 2); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.SpanIds)); err != nil { - return err - } - for _, v := range p.SpanIds { - if err := v.Write(oprot); err != nil { +func (p *ListWorkspaceAnnotationsRequest) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetPlatformType() { + if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.PlatformType); err != nil { return err } - } - if err := oprot.WriteListEnd(); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } return nil WriteFieldBeginError: @@ -13506,76 +13261,12 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *ExportTracesToDatasetRequest) writeField3(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("category", thrift.I32, 3); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI32(int32(p.Category)); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) -} -func (p *ExportTracesToDatasetRequest) writeField4(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("config", thrift.STRUCT, 4); err != nil { - goto WriteFieldBeginError - } - if err := p.Config.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) -} -func (p *ExportTracesToDatasetRequest) writeField5(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("start_time", thrift.I64, 5); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI64(p.StartTime); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) -} -func (p *ExportTracesToDatasetRequest) writeField6(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("end_time", thrift.I64, 6); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI64(p.EndTime); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) -} -func (p *ExportTracesToDatasetRequest) writeField7(oprot thrift.TProtocol) (err error) { - if p.IsSetPlatformType() { - if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 7); err != nil { +func (p *ListWorkspaceAnnotationsRequest) writeField3(oprot thrift.TProtocol) (err error) { + if p.IsSetAnnotationType() { + if err = oprot.WriteFieldBegin("annotation_type", thrift.STRING, 3); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteString(*p.PlatformType); err != nil { + if err := oprot.WriteString(*p.AnnotationType); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -13584,40 +13275,16 @@ func (p *ExportTracesToDatasetRequest) writeField7(oprot thrift.TProtocol) (err } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 7 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 7 end error: ", p), err) -} -func (p *ExportTracesToDatasetRequest) writeField8(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("export_type", thrift.STRING, 8); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(p.ExportType); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 8 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 8 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) } -func (p *ExportTracesToDatasetRequest) writeField9(oprot thrift.TProtocol) (err error) { - if p.IsSetFieldMappings() { - if err = oprot.WriteFieldBegin("field_mappings", thrift.LIST, 9); err != nil { +func (p *ListWorkspaceAnnotationsRequest) writeField4(oprot thrift.TProtocol) (err error) { + if p.IsSetSpanListType() { + if err = oprot.WriteFieldBegin("span_list_type", thrift.STRING, 4); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FieldMappings)); err != nil { - return err - } - for _, v := range p.FieldMappings { - if err := v.Write(oprot); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { + if err := oprot.WriteString(*p.SpanListType); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -13626,11 +13293,11 @@ func (p *ExportTracesToDatasetRequest) writeField9(oprot thrift.TProtocol) (err } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 9 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 9 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) } -func (p *ExportTracesToDatasetRequest) writeField255(oprot thrift.TProtocol) (err error) { +func (p *ListWorkspaceAnnotationsRequest) writeField255(oprot thrift.TProtocol) (err error) { if p.IsSetBase() { if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError @@ -13649,15 +13316,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *ExportTracesToDatasetRequest) String() string { +func (p *ListWorkspaceAnnotationsRequest) String() string { if p == nil { return "" } - return fmt.Sprintf("ExportTracesToDatasetRequest(%+v)", *p) + return fmt.Sprintf("ListWorkspaceAnnotationsRequest(%+v)", *p) } -func (p *ExportTracesToDatasetRequest) DeepEqual(ano *ExportTracesToDatasetRequest) bool { +func (p *ListWorkspaceAnnotationsRequest) DeepEqual(ano *ListWorkspaceAnnotationsRequest) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -13666,28 +13333,13 @@ func (p *ExportTracesToDatasetRequest) DeepEqual(ano *ExportTracesToDatasetReque if !p.Field1DeepEqual(ano.WorkspaceID) { return false } - if !p.Field2DeepEqual(ano.SpanIds) { - return false - } - if !p.Field3DeepEqual(ano.Category) { - return false - } - if !p.Field4DeepEqual(ano.Config) { - return false - } - if !p.Field5DeepEqual(ano.StartTime) { - return false - } - if !p.Field6DeepEqual(ano.EndTime) { - return false - } - if !p.Field7DeepEqual(ano.PlatformType) { + if !p.Field2DeepEqual(ano.PlatformType) { return false } - if !p.Field8DeepEqual(ano.ExportType) { + if !p.Field3DeepEqual(ano.AnnotationType) { return false } - if !p.Field9DeepEqual(ano.FieldMappings) { + if !p.Field4DeepEqual(ano.SpanListType) { return false } if !p.Field255DeepEqual(ano.Base) { @@ -13696,87 +13348,50 @@ func (p *ExportTracesToDatasetRequest) DeepEqual(ano *ExportTracesToDatasetReque return true } -func (p *ExportTracesToDatasetRequest) Field1DeepEqual(src int64) bool { +func (p *ListWorkspaceAnnotationsRequest) Field1DeepEqual(src int64) bool { if p.WorkspaceID != src { return false } return true } -func (p *ExportTracesToDatasetRequest) Field2DeepEqual(src []*SpanID) bool { +func (p *ListWorkspaceAnnotationsRequest) Field2DeepEqual(src *common.PlatformType) bool { - if len(p.SpanIds) != len(src) { + if p.PlatformType == src { + return true + } else if p.PlatformType == nil || src == nil { return false } - for i, v := range p.SpanIds { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } - return true -} -func (p *ExportTracesToDatasetRequest) Field3DeepEqual(src dataset.DatasetCategory) bool { - - if p.Category != src { - return false - } - return true -} -func (p *ExportTracesToDatasetRequest) Field4DeepEqual(src *DatasetConfig) bool { - - if !p.Config.DeepEqual(src) { - return false - } - return true -} -func (p *ExportTracesToDatasetRequest) Field5DeepEqual(src int64) bool { - - if p.StartTime != src { - return false - } - return true -} -func (p *ExportTracesToDatasetRequest) Field6DeepEqual(src int64) bool { - - if p.EndTime != src { + if strings.Compare(*p.PlatformType, *src) != 0 { return false } return true } -func (p *ExportTracesToDatasetRequest) Field7DeepEqual(src *common.PlatformType) bool { +func (p *ListWorkspaceAnnotationsRequest) Field3DeepEqual(src *annotation.AnnotationType) bool { - if p.PlatformType == src { + if p.AnnotationType == src { return true - } else if p.PlatformType == nil || src == nil { + } else if p.AnnotationType == nil || src == nil { return false } - if strings.Compare(*p.PlatformType, *src) != 0 { + if strings.Compare(*p.AnnotationType, *src) != 0 { return false } return true } -func (p *ExportTracesToDatasetRequest) Field8DeepEqual(src dataset0.ExportType) bool { +func (p *ListWorkspaceAnnotationsRequest) Field4DeepEqual(src *common.SpanListType) bool { - if strings.Compare(p.ExportType, src) != 0 { + if p.SpanListType == src { + return true + } else if p.SpanListType == nil || src == nil { return false } - return true -} -func (p *ExportTracesToDatasetRequest) Field9DeepEqual(src []*dataset0.FieldMapping) bool { - - if len(p.FieldMappings) != len(src) { + if strings.Compare(*p.SpanListType, *src) != 0 { return false } - for i, v := range p.FieldMappings { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } return true } -func (p *ExportTracesToDatasetRequest) Field255DeepEqual(src *base.Base) bool { +func (p *ListWorkspaceAnnotationsRequest) Field255DeepEqual(src *base.Base) bool { if !p.Base.DeepEqual(src) { return false @@ -13784,48 +13399,56 @@ func (p *ExportTracesToDatasetRequest) Field255DeepEqual(src *base.Base) bool { return true } -type SpanID struct { - TraceID string `thrift:"trace_id,1,required" frugal:"1,required,string" form:"trace_id,required" json:"trace_id,required" query:"trace_id,required"` - SpanID string `thrift:"span_id,2,required" frugal:"2,required,string" form:"span_id,required" json:"span_id,required" query:"span_id,required"` +type ListWorkspaceAnnotationsResponse struct { + SimpleAnnotationList []*annotation.SimpleAnnotationInfo `thrift:"simple_annotation_list,1,required" frugal:"1,required,list" form:"simple_annotation_list,required" json:"simple_annotation_list,required" query:"simple_annotation_list,required"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` } -func NewSpanID() *SpanID { - return &SpanID{} +func NewListWorkspaceAnnotationsResponse() *ListWorkspaceAnnotationsResponse { + return &ListWorkspaceAnnotationsResponse{} } -func (p *SpanID) InitDefault() { +func (p *ListWorkspaceAnnotationsResponse) InitDefault() { } -func (p *SpanID) GetTraceID() (v string) { +func (p *ListWorkspaceAnnotationsResponse) GetSimpleAnnotationList() (v []*annotation.SimpleAnnotationInfo) { if p != nil { - return p.TraceID + return p.SimpleAnnotationList } return } -func (p *SpanID) GetSpanID() (v string) { - if p != nil { - return p.SpanID +var ListWorkspaceAnnotationsResponse_BaseResp_DEFAULT *base.BaseResp + +func (p *ListWorkspaceAnnotationsResponse) GetBaseResp() (v *base.BaseResp) { + if p == nil { + return } - return + if !p.IsSetBaseResp() { + return ListWorkspaceAnnotationsResponse_BaseResp_DEFAULT + } + return p.BaseResp } -func (p *SpanID) SetTraceID(val string) { - p.TraceID = val +func (p *ListWorkspaceAnnotationsResponse) SetSimpleAnnotationList(val []*annotation.SimpleAnnotationInfo) { + p.SimpleAnnotationList = val } -func (p *SpanID) SetSpanID(val string) { - p.SpanID = val +func (p *ListWorkspaceAnnotationsResponse) SetBaseResp(val *base.BaseResp) { + p.BaseResp = val } -var fieldIDToName_SpanID = map[int16]string{ - 1: "trace_id", - 2: "span_id", +var fieldIDToName_ListWorkspaceAnnotationsResponse = map[int16]string{ + 1: "simple_annotation_list", + 255: "BaseResp", } -func (p *SpanID) Read(iprot thrift.TProtocol) (err error) { +func (p *ListWorkspaceAnnotationsResponse) IsSetBaseResp() bool { + return p.BaseResp != nil +} + +func (p *ListWorkspaceAnnotationsResponse) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 - var issetTraceID bool = false - var issetSpanID bool = false + var issetSimpleAnnotationList bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -13842,20 +13465,19 @@ func (p *SpanID) Read(iprot thrift.TProtocol) (err error) { switch fieldId { case 1: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.LIST { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } - issetTraceID = true + issetSimpleAnnotationList = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - case 2: - if fieldTypeId == thrift.STRING { - if err = p.ReadField2(iprot); err != nil { + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { goto ReadFieldError } - issetSpanID = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -13872,22 +13494,17 @@ func (p *SpanID) Read(iprot thrift.TProtocol) (err error) { goto ReadStructEndError } - if !issetTraceID { + if !issetSimpleAnnotationList { fieldId = 1 goto RequiredFieldNotSetError } - - if !issetSpanID { - fieldId = 2 - goto RequiredFieldNotSetError - } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SpanID[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListWorkspaceAnnotationsResponse[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -13896,35 +13513,44 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_SpanID[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListWorkspaceAnnotationsResponse[fieldId])) } -func (p *SpanID) ReadField1(iprot thrift.TProtocol) error { +func (p *ListWorkspaceAnnotationsResponse) ReadField1(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*annotation.SimpleAnnotationInfo, 0, size) + values := make([]annotation.SimpleAnnotationInfo, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() - var _field string - if v, err := iprot.ReadString(); err != nil { + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { return err - } else { - _field = v } - p.TraceID = _field + p.SimpleAnnotationList = _field return nil } -func (p *SpanID) ReadField2(iprot thrift.TProtocol) error { - - var _field string - if v, err := iprot.ReadString(); err != nil { +func (p *ListWorkspaceAnnotationsResponse) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBaseResp() + if err := _field.Read(iprot); err != nil { return err - } else { - _field = v } - p.SpanID = _field + p.BaseResp = _field return nil } -func (p *SpanID) Write(oprot thrift.TProtocol) (err error) { +func (p *ListWorkspaceAnnotationsResponse) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("SpanID"); err != nil { + if err = oprot.WriteStructBegin("ListWorkspaceAnnotationsResponse"); err != nil { goto WriteStructBeginError } if p != nil { @@ -13932,8 +13558,8 @@ func (p *SpanID) Write(oprot thrift.TProtocol) (err error) { fieldId = 1 goto WriteFieldError } - if err = p.writeField2(oprot); err != nil { - fieldId = 2 + if err = p.writeField255(oprot); err != nil { + fieldId = 255 goto WriteFieldError } } @@ -13954,11 +13580,19 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *SpanID) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("trace_id", thrift.STRING, 1); err != nil { +func (p *ListWorkspaceAnnotationsResponse) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("simple_annotation_list", thrift.LIST, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteString(p.TraceID); err != nil { + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.SimpleAnnotationList)); err != nil { + return err + } + for _, v := range p.SimpleAnnotationList { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -13970,157 +13604,249 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *SpanID) writeField2(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("span_id", thrift.STRING, 2); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(p.SpanID); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError +func (p *ListWorkspaceAnnotationsResponse) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBaseResp() { + if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.BaseResp.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *SpanID) String() string { +func (p *ListWorkspaceAnnotationsResponse) String() string { if p == nil { return "" } - return fmt.Sprintf("SpanID(%+v)", *p) + return fmt.Sprintf("ListWorkspaceAnnotationsResponse(%+v)", *p) } -func (p *SpanID) DeepEqual(ano *SpanID) bool { +func (p *ListWorkspaceAnnotationsResponse) DeepEqual(ano *ListWorkspaceAnnotationsResponse) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.TraceID) { + if !p.Field1DeepEqual(ano.SimpleAnnotationList) { return false } - if !p.Field2DeepEqual(ano.SpanID) { + if !p.Field255DeepEqual(ano.BaseResp) { return false } return true } -func (p *SpanID) Field1DeepEqual(src string) bool { +func (p *ListWorkspaceAnnotationsResponse) Field1DeepEqual(src []*annotation.SimpleAnnotationInfo) bool { - if strings.Compare(p.TraceID, src) != 0 { + if len(p.SimpleAnnotationList) != len(src) { return false } + for i, v := range p.SimpleAnnotationList { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } return true } -func (p *SpanID) Field2DeepEqual(src string) bool { +func (p *ListWorkspaceAnnotationsResponse) Field255DeepEqual(src *base.BaseResp) bool { - if strings.Compare(p.SpanID, src) != 0 { + if !p.BaseResp.DeepEqual(src) { return false } return true } -type DatasetConfig struct { - // 是否是新增数据集 - IsNewDataset bool `thrift:"is_new_dataset,1,required" frugal:"1,required,bool" form:"is_new_dataset,required" json:"is_new_dataset,required" query:"is_new_dataset,required"` - // 数据集id，新增数据集时可为空 - DatasetID *int64 `thrift:"dataset_id,2,optional" frugal:"2,optional,i64" json:"dataset_id" form:"dataset_id" query:"dataset_id"` - // 数据集名称，选择已有数据集时可为空 - DatasetName *string `thrift:"dataset_name,3,optional" frugal:"3,optional,string" form:"dataset_name" json:"dataset_name,omitempty" query:"dataset_name"` - // 数据集列数据schema - DatasetSchema *dataset0.DatasetSchema `thrift:"dataset_schema,4,optional" frugal:"4,optional,dataset.DatasetSchema" form:"dataset_schema" json:"dataset_schema,omitempty" query:"dataset_schema"` +type ExportTracesToDatasetRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` + SpanIds []*SpanID `thrift:"span_ids,2,required" frugal:"2,required,list" form:"span_ids,required" json:"span_ids,required"` + Category dataset.DatasetCategory `thrift:"category,3,required" frugal:"3,required,DatasetCategory" form:"category,required" json:"category,required"` + Config *DatasetConfig `thrift:"config,4,required" frugal:"4,required,DatasetConfig" form:"config,required" json:"config,required"` + StartTime int64 `thrift:"start_time,5,required" frugal:"5,required,i64" json:"start_time" form:"start_time,required" ` + EndTime int64 `thrift:"end_time,6,required" frugal:"6,required,i64" json:"end_time" form:"end_time,required" ` + PlatformType *common.PlatformType `thrift:"platform_type,7,optional" frugal:"7,optional,string" form:"platform_type" json:"platform_type,omitempty"` + // 导入方式，不填默认为追加 + ExportType dataset0.ExportType `thrift:"export_type,8,required" frugal:"8,required,string" form:"export_type,required" json:"export_type,required"` + FieldMappings []*dataset0.FieldMapping `thrift:"field_mappings,9,optional" frugal:"9,optional,list" form:"field_mappings" json:"field_mappings,omitempty"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` } -func NewDatasetConfig() *DatasetConfig { - return &DatasetConfig{} +func NewExportTracesToDatasetRequest() *ExportTracesToDatasetRequest { + return &ExportTracesToDatasetRequest{} } -func (p *DatasetConfig) InitDefault() { +func (p *ExportTracesToDatasetRequest) InitDefault() { } -func (p *DatasetConfig) GetIsNewDataset() (v bool) { +func (p *ExportTracesToDatasetRequest) GetWorkspaceID() (v int64) { if p != nil { - return p.IsNewDataset + return p.WorkspaceID } return } -var DatasetConfig_DatasetID_DEFAULT int64 - -func (p *DatasetConfig) GetDatasetID() (v int64) { - if p == nil { - return - } - if !p.IsSetDatasetID() { - return DatasetConfig_DatasetID_DEFAULT +func (p *ExportTracesToDatasetRequest) GetSpanIds() (v []*SpanID) { + if p != nil { + return p.SpanIds } - return *p.DatasetID + return } -var DatasetConfig_DatasetName_DEFAULT string - -func (p *DatasetConfig) GetDatasetName() (v string) { - if p == nil { - return - } - if !p.IsSetDatasetName() { - return DatasetConfig_DatasetName_DEFAULT +func (p *ExportTracesToDatasetRequest) GetCategory() (v dataset.DatasetCategory) { + if p != nil { + return p.Category } - return *p.DatasetName + return } -var DatasetConfig_DatasetSchema_DEFAULT *dataset0.DatasetSchema +var ExportTracesToDatasetRequest_Config_DEFAULT *DatasetConfig -func (p *DatasetConfig) GetDatasetSchema() (v *dataset0.DatasetSchema) { +func (p *ExportTracesToDatasetRequest) GetConfig() (v *DatasetConfig) { if p == nil { return } - if !p.IsSetDatasetSchema() { - return DatasetConfig_DatasetSchema_DEFAULT + if !p.IsSetConfig() { + return ExportTracesToDatasetRequest_Config_DEFAULT } - return p.DatasetSchema + return p.Config } -func (p *DatasetConfig) SetIsNewDataset(val bool) { - p.IsNewDataset = val + +func (p *ExportTracesToDatasetRequest) GetStartTime() (v int64) { + if p != nil { + return p.StartTime + } + return } -func (p *DatasetConfig) SetDatasetID(val *int64) { - p.DatasetID = val + +func (p *ExportTracesToDatasetRequest) GetEndTime() (v int64) { + if p != nil { + return p.EndTime + } + return } -func (p *DatasetConfig) SetDatasetName(val *string) { - p.DatasetName = val + +var ExportTracesToDatasetRequest_PlatformType_DEFAULT common.PlatformType + +func (p *ExportTracesToDatasetRequest) GetPlatformType() (v common.PlatformType) { + if p == nil { + return + } + if !p.IsSetPlatformType() { + return ExportTracesToDatasetRequest_PlatformType_DEFAULT + } + return *p.PlatformType } -func (p *DatasetConfig) SetDatasetSchema(val *dataset0.DatasetSchema) { - p.DatasetSchema = val + +func (p *ExportTracesToDatasetRequest) GetExportType() (v dataset0.ExportType) { + if p != nil { + return p.ExportType + } + return } -var fieldIDToName_DatasetConfig = map[int16]string{ - 1: "is_new_dataset", - 2: "dataset_id", - 3: "dataset_name", - 4: "dataset_schema", +var ExportTracesToDatasetRequest_FieldMappings_DEFAULT []*dataset0.FieldMapping + +func (p *ExportTracesToDatasetRequest) GetFieldMappings() (v []*dataset0.FieldMapping) { + if p == nil { + return + } + if !p.IsSetFieldMappings() { + return ExportTracesToDatasetRequest_FieldMappings_DEFAULT + } + return p.FieldMappings } -func (p *DatasetConfig) IsSetDatasetID() bool { - return p.DatasetID != nil +var ExportTracesToDatasetRequest_Base_DEFAULT *base.Base + +func (p *ExportTracesToDatasetRequest) GetBase() (v *base.Base) { + if p == nil { + return + } + if !p.IsSetBase() { + return ExportTracesToDatasetRequest_Base_DEFAULT + } + return p.Base +} +func (p *ExportTracesToDatasetRequest) SetWorkspaceID(val int64) { + p.WorkspaceID = val +} +func (p *ExportTracesToDatasetRequest) SetSpanIds(val []*SpanID) { + p.SpanIds = val +} +func (p *ExportTracesToDatasetRequest) SetCategory(val dataset.DatasetCategory) { + p.Category = val +} +func (p *ExportTracesToDatasetRequest) SetConfig(val *DatasetConfig) { + p.Config = val +} +func (p *ExportTracesToDatasetRequest) SetStartTime(val int64) { + p.StartTime = val +} +func (p *ExportTracesToDatasetRequest) SetEndTime(val int64) { + p.EndTime = val +} +func (p *ExportTracesToDatasetRequest) SetPlatformType(val *common.PlatformType) { + p.PlatformType = val +} +func (p *ExportTracesToDatasetRequest) SetExportType(val dataset0.ExportType) { + p.ExportType = val +} +func (p *ExportTracesToDatasetRequest) SetFieldMappings(val []*dataset0.FieldMapping) { + p.FieldMappings = val +} +func (p *ExportTracesToDatasetRequest) SetBase(val *base.Base) { + p.Base = val } -func (p *DatasetConfig) IsSetDatasetName() bool { - return p.DatasetName != nil +var fieldIDToName_ExportTracesToDatasetRequest = map[int16]string{ + 1: "workspace_id", + 2: "span_ids", + 3: "category", + 4: "config", + 5: "start_time", + 6: "end_time", + 7: "platform_type", + 8: "export_type", + 9: "field_mappings", + 255: "Base", } -func (p *DatasetConfig) IsSetDatasetSchema() bool { - return p.DatasetSchema != nil +func (p *ExportTracesToDatasetRequest) IsSetConfig() bool { + return p.Config != nil } -func (p *DatasetConfig) Read(iprot thrift.TProtocol) (err error) { +func (p *ExportTracesToDatasetRequest) IsSetPlatformType() bool { + return p.PlatformType != nil +} + +func (p *ExportTracesToDatasetRequest) IsSetFieldMappings() bool { + return p.FieldMappings != nil +} + +func (p *ExportTracesToDatasetRequest) IsSetBase() bool { + return p.Base != nil +} + +func (p *ExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 - var issetIsNewDataset bool = false + var issetWorkspaceID bool = false + var issetSpanIds bool = false + var issetCategory bool = false + var issetConfig bool = false + var issetStartTime bool = false + var issetEndTime bool = false + var issetExportType bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -14137,27 +13863,29 @@ func (p *DatasetConfig) Read(iprot thrift.TProtocol) (err error) { switch fieldId { case 1: - if fieldTypeId == thrift.BOOL { + if fieldTypeId == thrift.I64 { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } - issetIsNewDataset = true + issetWorkspaceID = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 2: - if fieldTypeId == thrift.I64 { + if fieldTypeId == thrift.LIST { if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } + issetSpanIds = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 3: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.I32 { if err = p.ReadField3(iprot); err != nil { goto ReadFieldError } + issetCategory = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -14166,6 +13894,58 @@ func (p *DatasetConfig) Read(iprot thrift.TProtocol) (err error) { if err = p.ReadField4(iprot); err != nil { goto ReadFieldError } + issetConfig = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 5: + if fieldTypeId == thrift.I64 { + if err = p.ReadField5(iprot); err != nil { + goto ReadFieldError + } + issetStartTime = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 6: + if fieldTypeId == thrift.I64 { + if err = p.ReadField6(iprot); err != nil { + goto ReadFieldError + } + issetEndTime = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 7: + if fieldTypeId == thrift.STRING { + if err = p.ReadField7(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 8: + if fieldTypeId == thrift.STRING { + if err = p.ReadField8(iprot); err != nil { + goto ReadFieldError + } + issetExportType = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 9: + if fieldTypeId == thrift.LIST { + if err = p.ReadField9(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { + goto ReadFieldError + } } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -14182,17 +13962,47 @@ func (p *DatasetConfig) Read(iprot thrift.TProtocol) (err error) { goto ReadStructEndError } - if !issetIsNewDataset { + if !issetWorkspaceID { fieldId = 1 goto RequiredFieldNotSetError } + + if !issetSpanIds { + fieldId = 2 + goto RequiredFieldNotSetError + } + + if !issetCategory { + fieldId = 3 + goto RequiredFieldNotSetError + } + + if !issetConfig { + fieldId = 4 + goto RequiredFieldNotSetError + } + + if !issetStartTime { + fieldId = 5 + goto RequiredFieldNotSetError + } + + if !issetEndTime { + fieldId = 6 + goto RequiredFieldNotSetError + } + + if !issetExportType { + fieldId = 8 + goto RequiredFieldNotSetError + } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_DatasetConfig[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExportTracesToDatasetRequest[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -14201,54 +14011,141 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_DatasetConfig[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ExportTracesToDatasetRequest[fieldId])) } -func (p *DatasetConfig) ReadField1(iprot thrift.TProtocol) error { +func (p *ExportTracesToDatasetRequest) ReadField1(iprot thrift.TProtocol) error { - var _field bool - if v, err := iprot.ReadBool(); err != nil { + var _field int64 + if v, err := iprot.ReadI64(); err != nil { return err } else { _field = v } - p.IsNewDataset = _field + p.WorkspaceID = _field return nil } -func (p *DatasetConfig) ReadField2(iprot thrift.TProtocol) error { - - var _field *int64 - if v, err := iprot.ReadI64(); err != nil { +func (p *ExportTracesToDatasetRequest) ReadField2(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { return err - } else { - _field = &v } - p.DatasetID = _field - return nil -} -func (p *DatasetConfig) ReadField3(iprot thrift.TProtocol) error { + _field := make([]*SpanID, 0, size) + values := make([]SpanID, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() - var _field *string + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.SpanIds = _field + return nil +} +func (p *ExportTracesToDatasetRequest) ReadField3(iprot thrift.TProtocol) error { + + var _field dataset.DatasetCategory + if v, err := iprot.ReadI32(); err != nil { + return err + } else { + _field = dataset.DatasetCategory(v) + } + p.Category = _field + return nil +} +func (p *ExportTracesToDatasetRequest) ReadField4(iprot thrift.TProtocol) error { + _field := NewDatasetConfig() + if err := _field.Read(iprot); err != nil { + return err + } + p.Config = _field + return nil +} +func (p *ExportTracesToDatasetRequest) ReadField5(iprot thrift.TProtocol) error { + + var _field int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = v + } + p.StartTime = _field + return nil +} +func (p *ExportTracesToDatasetRequest) ReadField6(iprot thrift.TProtocol) error { + + var _field int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = v + } + p.EndTime = _field + return nil +} +func (p *ExportTracesToDatasetRequest) ReadField7(iprot thrift.TProtocol) error { + + var _field *common.PlatformType if v, err := iprot.ReadString(); err != nil { return err } else { _field = &v } - p.DatasetName = _field + p.PlatformType = _field return nil } -func (p *DatasetConfig) ReadField4(iprot thrift.TProtocol) error { - _field := dataset0.NewDatasetSchema() +func (p *ExportTracesToDatasetRequest) ReadField8(iprot thrift.TProtocol) error { + + var _field dataset0.ExportType + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v + } + p.ExportType = _field + return nil +} +func (p *ExportTracesToDatasetRequest) ReadField9(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*dataset0.FieldMapping, 0, size) + values := make([]dataset0.FieldMapping, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.FieldMappings = _field + return nil +} +func (p *ExportTracesToDatasetRequest) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBase() if err := _field.Read(iprot); err != nil { return err } - p.DatasetSchema = _field + p.Base = _field return nil } -func (p *DatasetConfig) Write(oprot thrift.TProtocol) (err error) { +func (p *ExportTracesToDatasetRequest) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("DatasetConfig"); err != nil { + if err = oprot.WriteStructBegin("ExportTracesToDatasetRequest"); err != nil { goto WriteStructBeginError } if p != nil { @@ -14268,6 +14165,30 @@ func (p *DatasetConfig) Write(oprot thrift.TProtocol) (err error) { fieldId = 4 goto WriteFieldError } + if err = p.writeField5(oprot); err != nil { + fieldId = 5 + goto WriteFieldError + } + if err = p.writeField6(oprot); err != nil { + fieldId = 6 + goto WriteFieldError + } + if err = p.writeField7(oprot); err != nil { + fieldId = 7 + goto WriteFieldError + } + if err = p.writeField8(oprot); err != nil { + fieldId = 8 + goto WriteFieldError + } + if err = p.writeField9(oprot); err != nil { + fieldId = 9 + goto WriteFieldError + } + if err = p.writeField255(oprot); err != nil { + fieldId = 255 + goto WriteFieldError + } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -14286,11 +14207,11 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *DatasetConfig) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("is_new_dataset", thrift.BOOL, 1); err != nil { +func (p *ExportTracesToDatasetRequest) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteBool(p.IsNewDataset); err != nil { + if err := oprot.WriteI64(p.WorkspaceID); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -14302,17 +14223,23 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *DatasetConfig) writeField2(oprot thrift.TProtocol) (err error) { - if p.IsSetDatasetID() { - if err = oprot.WriteFieldBegin("dataset_id", thrift.I64, 2); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI64(*p.DatasetID); err != nil { +func (p *ExportTracesToDatasetRequest) writeField2(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("span_ids", thrift.LIST, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.SpanIds)); err != nil { + return err + } + for _, v := range p.SpanIds { + if err := v.Write(oprot); err != nil { return err } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } return nil WriteFieldBeginError: @@ -14320,17 +14247,15 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *DatasetConfig) writeField3(oprot thrift.TProtocol) (err error) { - if p.IsSetDatasetName() { - if err = oprot.WriteFieldBegin("dataset_name", thrift.STRING, 3); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(*p.DatasetName); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } +func (p *ExportTracesToDatasetRequest) writeField3(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("category", thrift.I32, 3); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI32(int32(p.Category)); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } return nil WriteFieldBeginError: @@ -14338,17 +14263,15 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) } -func (p *DatasetConfig) writeField4(oprot thrift.TProtocol) (err error) { - if p.IsSetDatasetSchema() { - if err = oprot.WriteFieldBegin("dataset_schema", thrift.STRUCT, 4); err != nil { - goto WriteFieldBeginError - } - if err := p.DatasetSchema.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } +func (p *ExportTracesToDatasetRequest) writeField4(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("config", thrift.STRUCT, 4); err != nil { + goto WriteFieldBeginError + } + if err := p.Config.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } return nil WriteFieldBeginError: @@ -14356,244 +14279,294 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) } - -func (p *DatasetConfig) String() string { - if p == nil { - return "" - } - return fmt.Sprintf("DatasetConfig(%+v)", *p) - -} - -func (p *DatasetConfig) DeepEqual(ano *DatasetConfig) bool { - if p == ano { - return true - } else if p == nil || ano == nil { - return false - } - if !p.Field1DeepEqual(ano.IsNewDataset) { - return false - } - if !p.Field2DeepEqual(ano.DatasetID) { - return false +func (p *ExportTracesToDatasetRequest) writeField5(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("start_time", thrift.I64, 5); err != nil { + goto WriteFieldBeginError } - if !p.Field3DeepEqual(ano.DatasetName) { - return false + if err := oprot.WriteI64(p.StartTime); err != nil { + return err } - if !p.Field4DeepEqual(ano.DatasetSchema) { - return false + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } - return true + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) } - -func (p *DatasetConfig) Field1DeepEqual(src bool) bool { - - if p.IsNewDataset != src { - return false +func (p *ExportTracesToDatasetRequest) writeField6(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("end_time", thrift.I64, 6); err != nil { + goto WriteFieldBeginError } - return true -} -func (p *DatasetConfig) Field2DeepEqual(src *int64) bool { - - if p.DatasetID == src { - return true - } else if p.DatasetID == nil || src == nil { - return false + if err := oprot.WriteI64(p.EndTime); err != nil { + return err } - if *p.DatasetID != *src { - return false + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } - return true + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) +} +func (p *ExportTracesToDatasetRequest) writeField7(oprot thrift.TProtocol) (err error) { + if p.IsSetPlatformType() { + if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 7); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.PlatformType); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 7 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 7 end error: ", p), err) +} +func (p *ExportTracesToDatasetRequest) writeField8(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("export_type", thrift.STRING, 8); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(p.ExportType); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 8 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 8 end error: ", p), err) +} +func (p *ExportTracesToDatasetRequest) writeField9(oprot thrift.TProtocol) (err error) { + if p.IsSetFieldMappings() { + if err = oprot.WriteFieldBegin("field_mappings", thrift.LIST, 9); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FieldMappings)); err != nil { + return err + } + for _, v := range p.FieldMappings { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 9 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 9 end error: ", p), err) +} +func (p *ExportTracesToDatasetRequest) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBase() { + if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.Base.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *DatasetConfig) Field3DeepEqual(src *string) bool { - if p.DatasetName == src { +func (p *ExportTracesToDatasetRequest) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("ExportTracesToDatasetRequest(%+v)", *p) + +} + +func (p *ExportTracesToDatasetRequest) DeepEqual(ano *ExportTracesToDatasetRequest) bool { + if p == ano { return true - } else if p.DatasetName == nil || src == nil { + } else if p == nil || ano == nil { return false } - if strings.Compare(*p.DatasetName, *src) != 0 { + if !p.Field1DeepEqual(ano.WorkspaceID) { + return false + } + if !p.Field2DeepEqual(ano.SpanIds) { + return false + } + if !p.Field3DeepEqual(ano.Category) { + return false + } + if !p.Field4DeepEqual(ano.Config) { + return false + } + if !p.Field5DeepEqual(ano.StartTime) { + return false + } + if !p.Field6DeepEqual(ano.EndTime) { + return false + } + if !p.Field7DeepEqual(ano.PlatformType) { + return false + } + if !p.Field8DeepEqual(ano.ExportType) { + return false + } + if !p.Field9DeepEqual(ano.FieldMappings) { + return false + } + if !p.Field255DeepEqual(ano.Base) { return false } return true } -func (p *DatasetConfig) Field4DeepEqual(src *dataset0.DatasetSchema) bool { - if !p.DatasetSchema.DeepEqual(src) { +func (p *ExportTracesToDatasetRequest) Field1DeepEqual(src int64) bool { + + if p.WorkspaceID != src { return false } return true } +func (p *ExportTracesToDatasetRequest) Field2DeepEqual(src []*SpanID) bool { -type ExportTracesToDatasetResponse struct { - // 成功导入的数量 - SuccessCount *int32 `thrift:"success_count,1,optional" frugal:"1,optional,i32" form:"success_count" json:"success_count,omitempty" query:"success_count"` - // 错误信息 - Errors []*dataset.ItemErrorGroup `thrift:"errors,2,optional" frugal:"2,optional,list" form:"errors" json:"errors,omitempty" query:"errors"` - // 数据集id - DatasetID *int64 `thrift:"dataset_id,3,optional" frugal:"3,optional,i64" json:"dataset_id" form:"dataset_id" query:"dataset_id"` - // 数据集名称 - DatasetName *string `thrift:"dataset_name,4,optional" frugal:"4,optional,string" form:"dataset_name" json:"dataset_name,omitempty" query:"dataset_name"` - BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"-" json:"-" query:"-"` - // 仅供http请求使用; 内部RPC不予使用，统一通过BaseResp获取Code和Msg - Code *int32 `thrift:"Code,256,optional" frugal:"256,optional,i32" form:"Code" json:"Code,omitempty" query:"Code"` - // 仅供http请求使用; 内部RPC不予使用，统一通过BaseResp获取Code和Msg - Msg *string `thrift:"Msg,257,optional" frugal:"257,optional,string" form:"Msg" json:"Msg,omitempty" query:"Msg"` + if len(p.SpanIds) != len(src) { + return false + } + for i, v := range p.SpanIds { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true } +func (p *ExportTracesToDatasetRequest) Field3DeepEqual(src dataset.DatasetCategory) bool { -func NewExportTracesToDatasetResponse() *ExportTracesToDatasetResponse { - return &ExportTracesToDatasetResponse{} + if p.Category != src { + return false + } + return true } +func (p *ExportTracesToDatasetRequest) Field4DeepEqual(src *DatasetConfig) bool { -func (p *ExportTracesToDatasetResponse) InitDefault() { + if !p.Config.DeepEqual(src) { + return false + } + return true } +func (p *ExportTracesToDatasetRequest) Field5DeepEqual(src int64) bool { -var ExportTracesToDatasetResponse_SuccessCount_DEFAULT int32 - -func (p *ExportTracesToDatasetResponse) GetSuccessCount() (v int32) { - if p == nil { - return - } - if !p.IsSetSuccessCount() { - return ExportTracesToDatasetResponse_SuccessCount_DEFAULT + if p.StartTime != src { + return false } - return *p.SuccessCount + return true } +func (p *ExportTracesToDatasetRequest) Field6DeepEqual(src int64) bool { -var ExportTracesToDatasetResponse_Errors_DEFAULT []*dataset.ItemErrorGroup - -func (p *ExportTracesToDatasetResponse) GetErrors() (v []*dataset.ItemErrorGroup) { - if p == nil { - return - } - if !p.IsSetErrors() { - return ExportTracesToDatasetResponse_Errors_DEFAULT + if p.EndTime != src { + return false } - return p.Errors + return true } +func (p *ExportTracesToDatasetRequest) Field7DeepEqual(src *common.PlatformType) bool { -var ExportTracesToDatasetResponse_DatasetID_DEFAULT int64 - -func (p *ExportTracesToDatasetResponse) GetDatasetID() (v int64) { - if p == nil { - return + if p.PlatformType == src { + return true + } else if p.PlatformType == nil || src == nil { + return false } - if !p.IsSetDatasetID() { - return ExportTracesToDatasetResponse_DatasetID_DEFAULT + if strings.Compare(*p.PlatformType, *src) != 0 { + return false } - return *p.DatasetID + return true } +func (p *ExportTracesToDatasetRequest) Field8DeepEqual(src dataset0.ExportType) bool { -var ExportTracesToDatasetResponse_DatasetName_DEFAULT string - -func (p *ExportTracesToDatasetResponse) GetDatasetName() (v string) { - if p == nil { - return - } - if !p.IsSetDatasetName() { - return ExportTracesToDatasetResponse_DatasetName_DEFAULT + if strings.Compare(p.ExportType, src) != 0 { + return false } - return *p.DatasetName + return true } +func (p *ExportTracesToDatasetRequest) Field9DeepEqual(src []*dataset0.FieldMapping) bool { -var ExportTracesToDatasetResponse_BaseResp_DEFAULT *base.BaseResp - -func (p *ExportTracesToDatasetResponse) GetBaseResp() (v *base.BaseResp) { - if p == nil { - return + if len(p.FieldMappings) != len(src) { + return false } - if !p.IsSetBaseResp() { - return ExportTracesToDatasetResponse_BaseResp_DEFAULT + for i, v := range p.FieldMappings { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } } - return p.BaseResp + return true } +func (p *ExportTracesToDatasetRequest) Field255DeepEqual(src *base.Base) bool { -var ExportTracesToDatasetResponse_Code_DEFAULT int32 - -func (p *ExportTracesToDatasetResponse) GetCode() (v int32) { - if p == nil { - return - } - if !p.IsSetCode() { - return ExportTracesToDatasetResponse_Code_DEFAULT + if !p.Base.DeepEqual(src) { + return false } - return *p.Code + return true } -var ExportTracesToDatasetResponse_Msg_DEFAULT string - -func (p *ExportTracesToDatasetResponse) GetMsg() (v string) { - if p == nil { - return - } - if !p.IsSetMsg() { - return ExportTracesToDatasetResponse_Msg_DEFAULT - } - return *p.Msg -} -func (p *ExportTracesToDatasetResponse) SetSuccessCount(val *int32) { - p.SuccessCount = val -} -func (p *ExportTracesToDatasetResponse) SetErrors(val []*dataset.ItemErrorGroup) { - p.Errors = val -} -func (p *ExportTracesToDatasetResponse) SetDatasetID(val *int64) { - p.DatasetID = val -} -func (p *ExportTracesToDatasetResponse) SetDatasetName(val *string) { - p.DatasetName = val -} -func (p *ExportTracesToDatasetResponse) SetBaseResp(val *base.BaseResp) { - p.BaseResp = val -} -func (p *ExportTracesToDatasetResponse) SetCode(val *int32) { - p.Code = val -} -func (p *ExportTracesToDatasetResponse) SetMsg(val *string) { - p.Msg = val -} - -var fieldIDToName_ExportTracesToDatasetResponse = map[int16]string{ - 1: "success_count", - 2: "errors", - 3: "dataset_id", - 4: "dataset_name", - 255: "BaseResp", - 256: "Code", - 257: "Msg", +type SpanID struct { + TraceID string `thrift:"trace_id,1,required" frugal:"1,required,string" form:"trace_id,required" json:"trace_id,required" query:"trace_id,required"` + SpanID string `thrift:"span_id,2,required" frugal:"2,required,string" form:"span_id,required" json:"span_id,required" query:"span_id,required"` } -func (p *ExportTracesToDatasetResponse) IsSetSuccessCount() bool { - return p.SuccessCount != nil +func NewSpanID() *SpanID { + return &SpanID{} } -func (p *ExportTracesToDatasetResponse) IsSetErrors() bool { - return p.Errors != nil +func (p *SpanID) InitDefault() { } -func (p *ExportTracesToDatasetResponse) IsSetDatasetID() bool { - return p.DatasetID != nil +func (p *SpanID) GetTraceID() (v string) { + if p != nil { + return p.TraceID + } + return } -func (p *ExportTracesToDatasetResponse) IsSetDatasetName() bool { - return p.DatasetName != nil +func (p *SpanID) GetSpanID() (v string) { + if p != nil { + return p.SpanID + } + return } - -func (p *ExportTracesToDatasetResponse) IsSetBaseResp() bool { - return p.BaseResp != nil +func (p *SpanID) SetTraceID(val string) { + p.TraceID = val } - -func (p *ExportTracesToDatasetResponse) IsSetCode() bool { - return p.Code != nil +func (p *SpanID) SetSpanID(val string) { + p.SpanID = val } -func (p *ExportTracesToDatasetResponse) IsSetMsg() bool { - return p.Msg != nil +var fieldIDToName_SpanID = map[int16]string{ + 1: "trace_id", + 2: "span_id", } -func (p *ExportTracesToDatasetResponse) Read(iprot thrift.TProtocol) (err error) { +func (p *SpanID) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 + var issetTraceID bool = false + var issetSpanID bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -14610,58 +14583,20 @@ func (p *ExportTracesToDatasetResponse) Read(iprot thrift.TProtocol) (err error) switch fieldId { case 1: - if fieldTypeId == thrift.I32 { + if fieldTypeId == thrift.STRING { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } + issetTraceID = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 2: - if fieldTypeId == thrift.LIST { - if err = p.ReadField2(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 3: - if fieldTypeId == thrift.I64 { - if err = p.ReadField3(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 4: - if fieldTypeId == thrift.STRING { - if err = p.ReadField4(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 255: - if fieldTypeId == thrift.STRUCT { - if err = p.ReadField255(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 256: - if fieldTypeId == thrift.I32 { - if err = p.ReadField256(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 257: if fieldTypeId == thrift.STRING { - if err = p.ReadField257(iprot); err != nil { + if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } + issetSpanID = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -14678,13 +14613,22 @@ func (p *ExportTracesToDatasetResponse) Read(iprot thrift.TProtocol) (err error) goto ReadStructEndError } + if !issetTraceID { + fieldId = 1 + goto RequiredFieldNotSetError + } + + if !issetSpanID { + fieldId = 2 + goto RequiredFieldNotSetError + } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExportTracesToDatasetResponse[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SpanID[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -14692,98 +14636,36 @@ ReadFieldEndError: return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +RequiredFieldNotSetError: + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_SpanID[fieldId])) } -func (p *ExportTracesToDatasetResponse) ReadField1(iprot thrift.TProtocol) error { - - var _field *int32 - if v, err := iprot.ReadI32(); err != nil { - return err - } else { - _field = &v - } - p.SuccessCount = _field - return nil -} -func (p *ExportTracesToDatasetResponse) ReadField2(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]*dataset.ItemErrorGroup, 0, size) - values := make([]dataset.ItemErrorGroup, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { - return err - } - p.Errors = _field - return nil -} -func (p *ExportTracesToDatasetResponse) ReadField3(iprot thrift.TProtocol) error { - - var _field *int64 - if v, err := iprot.ReadI64(); err != nil { - return err - } else { - _field = &v - } - p.DatasetID = _field - return nil -} -func (p *ExportTracesToDatasetResponse) ReadField4(iprot thrift.TProtocol) error { +func (p *SpanID) ReadField1(iprot thrift.TProtocol) error { - var _field *string + var _field string if v, err := iprot.ReadString(); err != nil { return err } else { - _field = &v - } - p.DatasetName = _field - return nil -} -func (p *ExportTracesToDatasetResponse) ReadField255(iprot thrift.TProtocol) error { - _field := base.NewBaseResp() - if err := _field.Read(iprot); err != nil { - return err - } - p.BaseResp = _field - return nil -} -func (p *ExportTracesToDatasetResponse) ReadField256(iprot thrift.TProtocol) error { - - var _field *int32 - if v, err := iprot.ReadI32(); err != nil { - return err - } else { - _field = &v + _field = v } - p.Code = _field + p.TraceID = _field return nil } -func (p *ExportTracesToDatasetResponse) ReadField257(iprot thrift.TProtocol) error { +func (p *SpanID) ReadField2(iprot thrift.TProtocol) error { - var _field *string + var _field string if v, err := iprot.ReadString(); err != nil { return err } else { - _field = &v + _field = v } - p.Msg = _field + p.SpanID = _field return nil } -func (p *ExportTracesToDatasetResponse) Write(oprot thrift.TProtocol) (err error) { +func (p *SpanID) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ExportTracesToDatasetResponse"); err != nil { + if err = oprot.WriteStructBegin("SpanID"); err != nil { goto WriteStructBeginError } if p != nil { @@ -14795,26 +14677,6 @@ func (p *ExportTracesToDatasetResponse) Write(oprot thrift.TProtocol) (err error fieldId = 2 goto WriteFieldError } - if err = p.writeField3(oprot); err != nil { - fieldId = 3 - goto WriteFieldError - } - if err = p.writeField4(oprot); err != nil { - fieldId = 4 - goto WriteFieldError - } - if err = p.writeField255(oprot); err != nil { - fieldId = 255 - goto WriteFieldError - } - if err = p.writeField256(oprot); err != nil { - fieldId = 256 - goto WriteFieldError - } - if err = p.writeField257(oprot); err != nil { - fieldId = 257 - goto WriteFieldError - } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -14833,440 +14695,173 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ExportTracesToDatasetResponse) writeField1(oprot thrift.TProtocol) (err error) { - if p.IsSetSuccessCount() { - if err = oprot.WriteFieldBegin("success_count", thrift.I32, 1); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI32(*p.SuccessCount); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) -} -func (p *ExportTracesToDatasetResponse) writeField2(oprot thrift.TProtocol) (err error) { - if p.IsSetErrors() { - if err = oprot.WriteFieldBegin("errors", thrift.LIST, 2); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.Errors)); err != nil { - return err - } - for _, v := range p.Errors { - if err := v.Write(oprot); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } +func (p *SpanID) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("trace_id", thrift.STRING, 1); err != nil { + goto WriteFieldBeginError } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) -} -func (p *ExportTracesToDatasetResponse) writeField3(oprot thrift.TProtocol) (err error) { - if p.IsSetDatasetID() { - if err = oprot.WriteFieldBegin("dataset_id", thrift.I64, 3); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI64(*p.DatasetID); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } + if err := oprot.WriteString(p.TraceID); err != nil { + return err } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) -} -func (p *ExportTracesToDatasetResponse) writeField4(oprot thrift.TProtocol) (err error) { - if p.IsSetDatasetName() { - if err = oprot.WriteFieldBegin("dataset_name", thrift.STRING, 4); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(*p.DatasetName); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ExportTracesToDatasetResponse) writeField255(oprot thrift.TProtocol) (err error) { - if p.IsSetBaseResp() { - if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { - goto WriteFieldBeginError - } - if err := p.BaseResp.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } +func (p *SpanID) writeField2(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("span_id", thrift.STRING, 2); err != nil { + goto WriteFieldBeginError } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) -} -func (p *ExportTracesToDatasetResponse) writeField256(oprot thrift.TProtocol) (err error) { - if p.IsSetCode() { - if err = oprot.WriteFieldBegin("Code", thrift.I32, 256); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI32(*p.Code); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } + if err := oprot.WriteString(p.SpanID); err != nil { + return err } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 256 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 256 end error: ", p), err) -} -func (p *ExportTracesToDatasetResponse) writeField257(oprot thrift.TProtocol) (err error) { - if p.IsSetMsg() { - if err = oprot.WriteFieldBegin("Msg", thrift.STRING, 257); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(*p.Msg); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 257 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 257 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *ExportTracesToDatasetResponse) String() string { +func (p *SpanID) String() string { if p == nil { return "" } - return fmt.Sprintf("ExportTracesToDatasetResponse(%+v)", *p) + return fmt.Sprintf("SpanID(%+v)", *p) } -func (p *ExportTracesToDatasetResponse) DeepEqual(ano *ExportTracesToDatasetResponse) bool { +func (p *SpanID) DeepEqual(ano *SpanID) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.SuccessCount) { - return false - } - if !p.Field2DeepEqual(ano.Errors) { - return false - } - if !p.Field3DeepEqual(ano.DatasetID) { - return false - } - if !p.Field4DeepEqual(ano.DatasetName) { - return false - } - if !p.Field255DeepEqual(ano.BaseResp) { - return false - } - if !p.Field256DeepEqual(ano.Code) { + if !p.Field1DeepEqual(ano.TraceID) { return false } - if !p.Field257DeepEqual(ano.Msg) { + if !p.Field2DeepEqual(ano.SpanID) { return false } return true } -func (p *ExportTracesToDatasetResponse) Field1DeepEqual(src *int32) bool { +func (p *SpanID) Field1DeepEqual(src string) bool { - if p.SuccessCount == src { - return true - } else if p.SuccessCount == nil || src == nil { - return false - } - if *p.SuccessCount != *src { + if strings.Compare(p.TraceID, src) != 0 { return false } return true } -func (p *ExportTracesToDatasetResponse) Field2DeepEqual(src []*dataset.ItemErrorGroup) bool { +func (p *SpanID) Field2DeepEqual(src string) bool { - if len(p.Errors) != len(src) { + if strings.Compare(p.SpanID, src) != 0 { return false } - for i, v := range p.Errors { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } return true } -func (p *ExportTracesToDatasetResponse) Field3DeepEqual(src *int64) bool { - if p.DatasetID == src { - return true - } else if p.DatasetID == nil || src == nil { - return false - } - if *p.DatasetID != *src { - return false - } - return true +type DatasetConfig struct { + // 是否是新增数据集 + IsNewDataset bool `thrift:"is_new_dataset,1,required" frugal:"1,required,bool" form:"is_new_dataset,required" json:"is_new_dataset,required" query:"is_new_dataset,required"` + // 数据集id，新增数据集时可为空 + DatasetID *int64 `thrift:"dataset_id,2,optional" frugal:"2,optional,i64" json:"dataset_id" form:"dataset_id" query:"dataset_id"` + // 数据集名称，选择已有数据集时可为空 + DatasetName *string `thrift:"dataset_name,3,optional" frugal:"3,optional,string" form:"dataset_name" json:"dataset_name,omitempty" query:"dataset_name"` + // 数据集列数据schema + DatasetSchema *dataset0.DatasetSchema `thrift:"dataset_schema,4,optional" frugal:"4,optional,dataset.DatasetSchema" form:"dataset_schema" json:"dataset_schema,omitempty" query:"dataset_schema"` } -func (p *ExportTracesToDatasetResponse) Field4DeepEqual(src *string) bool { - if p.DatasetName == src { - return true - } else if p.DatasetName == nil || src == nil { - return false - } - if strings.Compare(*p.DatasetName, *src) != 0 { - return false - } - return true +func NewDatasetConfig() *DatasetConfig { + return &DatasetConfig{} } -func (p *ExportTracesToDatasetResponse) Field255DeepEqual(src *base.BaseResp) bool { - if !p.BaseResp.DeepEqual(src) { - return false +func (p *DatasetConfig) InitDefault() { +} + +func (p *DatasetConfig) GetIsNewDataset() (v bool) { + if p != nil { + return p.IsNewDataset } - return true + return } -func (p *ExportTracesToDatasetResponse) Field256DeepEqual(src *int32) bool { - if p.Code == src { - return true - } else if p.Code == nil || src == nil { - return false +var DatasetConfig_DatasetID_DEFAULT int64 + +func (p *DatasetConfig) GetDatasetID() (v int64) { + if p == nil { + return } - if *p.Code != *src { - return false + if !p.IsSetDatasetID() { + return DatasetConfig_DatasetID_DEFAULT } - return true + return *p.DatasetID } -func (p *ExportTracesToDatasetResponse) Field257DeepEqual(src *string) bool { - if p.Msg == src { - return true - } else if p.Msg == nil || src == nil { - return false +var DatasetConfig_DatasetName_DEFAULT string + +func (p *DatasetConfig) GetDatasetName() (v string) { + if p == nil { + return } - if strings.Compare(*p.Msg, *src) != 0 { - return false + if !p.IsSetDatasetName() { + return DatasetConfig_DatasetName_DEFAULT } - return true + return *p.DatasetName } -type PreviewExportTracesToDatasetRequest struct { - WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` - SpanIds []*SpanID `thrift:"span_ids,2,required" frugal:"2,required,list" form:"span_ids,required" json:"span_ids,required"` - Category dataset.DatasetCategory `thrift:"category,3,required" frugal:"3,required,DatasetCategory" form:"category,required" json:"category,required"` - Config *DatasetConfig `thrift:"config,4,required" frugal:"4,required,DatasetConfig" form:"config,required" json:"config,required"` - StartTime int64 `thrift:"start_time,5,required" frugal:"5,required,i64" json:"start_time" form:"start_time,required" ` - EndTime int64 `thrift:"end_time,6,required" frugal:"6,required,i64" json:"end_time" form:"end_time,required" ` - PlatformType *common.PlatformType `thrift:"platform_type,7,optional" frugal:"7,optional,string" form:"platform_type" json:"platform_type,omitempty"` - // 导入方式，不填默认为追加 - ExportType dataset0.ExportType `thrift:"export_type,8,required" frugal:"8,required,string" form:"export_type,required" json:"export_type,required"` - FieldMappings []*dataset0.FieldMapping `thrift:"field_mappings,9,optional" frugal:"9,optional,list" form:"field_mappings" json:"field_mappings,omitempty"` - Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"-" json:"-" query:"-"` -} - -func NewPreviewExportTracesToDatasetRequest() *PreviewExportTracesToDatasetRequest { - return &PreviewExportTracesToDatasetRequest{} -} - -func (p *PreviewExportTracesToDatasetRequest) InitDefault() { -} - -func (p *PreviewExportTracesToDatasetRequest) GetWorkspaceID() (v int64) { - if p != nil { - return p.WorkspaceID - } - return -} - -func (p *PreviewExportTracesToDatasetRequest) GetSpanIds() (v []*SpanID) { - if p != nil { - return p.SpanIds - } - return -} - -func (p *PreviewExportTracesToDatasetRequest) GetCategory() (v dataset.DatasetCategory) { - if p != nil { - return p.Category - } - return -} - -var PreviewExportTracesToDatasetRequest_Config_DEFAULT *DatasetConfig - -func (p *PreviewExportTracesToDatasetRequest) GetConfig() (v *DatasetConfig) { - if p == nil { - return - } - if !p.IsSetConfig() { - return PreviewExportTracesToDatasetRequest_Config_DEFAULT - } - return p.Config -} - -func (p *PreviewExportTracesToDatasetRequest) GetStartTime() (v int64) { - if p != nil { - return p.StartTime - } - return -} - -func (p *PreviewExportTracesToDatasetRequest) GetEndTime() (v int64) { - if p != nil { - return p.EndTime - } - return -} - -var PreviewExportTracesToDatasetRequest_PlatformType_DEFAULT common.PlatformType - -func (p *PreviewExportTracesToDatasetRequest) GetPlatformType() (v common.PlatformType) { - if p == nil { - return - } - if !p.IsSetPlatformType() { - return PreviewExportTracesToDatasetRequest_PlatformType_DEFAULT - } - return *p.PlatformType -} - -func (p *PreviewExportTracesToDatasetRequest) GetExportType() (v dataset0.ExportType) { - if p != nil { - return p.ExportType - } - return -} - -var PreviewExportTracesToDatasetRequest_FieldMappings_DEFAULT []*dataset0.FieldMapping - -func (p *PreviewExportTracesToDatasetRequest) GetFieldMappings() (v []*dataset0.FieldMapping) { - if p == nil { - return - } - if !p.IsSetFieldMappings() { - return PreviewExportTracesToDatasetRequest_FieldMappings_DEFAULT - } - return p.FieldMappings -} - -var PreviewExportTracesToDatasetRequest_Base_DEFAULT *base.Base +var DatasetConfig_DatasetSchema_DEFAULT *dataset0.DatasetSchema -func (p *PreviewExportTracesToDatasetRequest) GetBase() (v *base.Base) { +func (p *DatasetConfig) GetDatasetSchema() (v *dataset0.DatasetSchema) { if p == nil { return } - if !p.IsSetBase() { - return PreviewExportTracesToDatasetRequest_Base_DEFAULT + if !p.IsSetDatasetSchema() { + return DatasetConfig_DatasetSchema_DEFAULT } - return p.Base -} -func (p *PreviewExportTracesToDatasetRequest) SetWorkspaceID(val int64) { - p.WorkspaceID = val -} -func (p *PreviewExportTracesToDatasetRequest) SetSpanIds(val []*SpanID) { - p.SpanIds = val -} -func (p *PreviewExportTracesToDatasetRequest) SetCategory(val dataset.DatasetCategory) { - p.Category = val -} -func (p *PreviewExportTracesToDatasetRequest) SetConfig(val *DatasetConfig) { - p.Config = val -} -func (p *PreviewExportTracesToDatasetRequest) SetStartTime(val int64) { - p.StartTime = val -} -func (p *PreviewExportTracesToDatasetRequest) SetEndTime(val int64) { - p.EndTime = val -} -func (p *PreviewExportTracesToDatasetRequest) SetPlatformType(val *common.PlatformType) { - p.PlatformType = val + return p.DatasetSchema } -func (p *PreviewExportTracesToDatasetRequest) SetExportType(val dataset0.ExportType) { - p.ExportType = val +func (p *DatasetConfig) SetIsNewDataset(val bool) { + p.IsNewDataset = val } -func (p *PreviewExportTracesToDatasetRequest) SetFieldMappings(val []*dataset0.FieldMapping) { - p.FieldMappings = val +func (p *DatasetConfig) SetDatasetID(val *int64) { + p.DatasetID = val } -func (p *PreviewExportTracesToDatasetRequest) SetBase(val *base.Base) { - p.Base = val +func (p *DatasetConfig) SetDatasetName(val *string) { + p.DatasetName = val } - -var fieldIDToName_PreviewExportTracesToDatasetRequest = map[int16]string{ - 1: "workspace_id", - 2: "span_ids", - 3: "category", - 4: "config", - 5: "start_time", - 6: "end_time", - 7: "platform_type", - 8: "export_type", - 9: "field_mappings", - 255: "Base", +func (p *DatasetConfig) SetDatasetSchema(val *dataset0.DatasetSchema) { + p.DatasetSchema = val } -func (p *PreviewExportTracesToDatasetRequest) IsSetConfig() bool { - return p.Config != nil +var fieldIDToName_DatasetConfig = map[int16]string{ + 1: "is_new_dataset", + 2: "dataset_id", + 3: "dataset_name", + 4: "dataset_schema", } -func (p *PreviewExportTracesToDatasetRequest) IsSetPlatformType() bool { - return p.PlatformType != nil +func (p *DatasetConfig) IsSetDatasetID() bool { + return p.DatasetID != nil } -func (p *PreviewExportTracesToDatasetRequest) IsSetFieldMappings() bool { - return p.FieldMappings != nil +func (p *DatasetConfig) IsSetDatasetName() bool { + return p.DatasetName != nil } -func (p *PreviewExportTracesToDatasetRequest) IsSetBase() bool { - return p.Base != nil +func (p *DatasetConfig) IsSetDatasetSchema() bool { + return p.DatasetSchema != nil } -func (p *PreviewExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err error) { +func (p *DatasetConfig) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 - var issetWorkspaceID bool = false - var issetSpanIds bool = false - var issetCategory bool = false - var issetConfig bool = false - var issetStartTime bool = false - var issetEndTime bool = false - var issetExportType bool = false + var issetIsNewDataset bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -15283,29 +14878,27 @@ func (p *PreviewExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err switch fieldId { case 1: - if fieldTypeId == thrift.I64 { + if fieldTypeId == thrift.BOOL { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } - issetWorkspaceID = true + issetIsNewDataset = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 2: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.I64 { if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } - issetSpanIds = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 3: - if fieldTypeId == thrift.I32 { + if fieldTypeId == thrift.STRING { if err = p.ReadField3(iprot); err != nil { goto ReadFieldError } - issetCategory = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -15314,58 +14907,6 @@ func (p *PreviewExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err if err = p.ReadField4(iprot); err != nil { goto ReadFieldError } - issetConfig = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 5: - if fieldTypeId == thrift.I64 { - if err = p.ReadField5(iprot); err != nil { - goto ReadFieldError - } - issetStartTime = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 6: - if fieldTypeId == thrift.I64 { - if err = p.ReadField6(iprot); err != nil { - goto ReadFieldError - } - issetEndTime = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 7: - if fieldTypeId == thrift.STRING { - if err = p.ReadField7(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 8: - if fieldTypeId == thrift.STRING { - if err = p.ReadField8(iprot); err != nil { - goto ReadFieldError - } - issetExportType = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 9: - if fieldTypeId == thrift.LIST { - if err = p.ReadField9(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 255: - if fieldTypeId == thrift.STRUCT { - if err = p.ReadField255(iprot); err != nil { - goto ReadFieldError - } } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -15382,190 +14923,73 @@ func (p *PreviewExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err goto ReadStructEndError } - if !issetWorkspaceID { + if !issetIsNewDataset { fieldId = 1 goto RequiredFieldNotSetError } - - if !issetSpanIds { - fieldId = 2 - goto RequiredFieldNotSetError - } - - if !issetCategory { - fieldId = 3 - goto RequiredFieldNotSetError - } - - if !issetConfig { - fieldId = 4 - goto RequiredFieldNotSetError - } - - if !issetStartTime { - fieldId = 5 - goto RequiredFieldNotSetError - } - - if !issetEndTime { - fieldId = 6 - goto RequiredFieldNotSetError - } - - if !issetExportType { - fieldId = 8 - goto RequiredFieldNotSetError - } - return nil -ReadStructBeginError: - return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) -ReadFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) -ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_PreviewExportTracesToDatasetRequest[fieldId]), err) -SkipFieldError: - return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_DatasetConfig[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) ReadFieldEndError: return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_PreviewExportTracesToDatasetRequest[fieldId])) -} - -func (p *PreviewExportTracesToDatasetRequest) ReadField1(iprot thrift.TProtocol) error { - - var _field int64 - if v, err := iprot.ReadI64(); err != nil { - return err - } else { - _field = v - } - p.WorkspaceID = _field - return nil -} -func (p *PreviewExportTracesToDatasetRequest) ReadField2(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]*SpanID, 0, size) - values := make([]SpanID, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { - return err - } - p.SpanIds = _field - return nil + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_DatasetConfig[fieldId])) } -func (p *PreviewExportTracesToDatasetRequest) ReadField3(iprot thrift.TProtocol) error { - var _field dataset.DatasetCategory - if v, err := iprot.ReadI32(); err != nil { - return err - } else { - _field = dataset.DatasetCategory(v) - } - p.Category = _field - return nil -} -func (p *PreviewExportTracesToDatasetRequest) ReadField4(iprot thrift.TProtocol) error { - _field := NewDatasetConfig() - if err := _field.Read(iprot); err != nil { - return err - } - p.Config = _field - return nil -} -func (p *PreviewExportTracesToDatasetRequest) ReadField5(iprot thrift.TProtocol) error { +func (p *DatasetConfig) ReadField1(iprot thrift.TProtocol) error { - var _field int64 - if v, err := iprot.ReadI64(); err != nil { + var _field bool + if v, err := iprot.ReadBool(); err != nil { return err } else { _field = v } - p.StartTime = _field + p.IsNewDataset = _field return nil } -func (p *PreviewExportTracesToDatasetRequest) ReadField6(iprot thrift.TProtocol) error { +func (p *DatasetConfig) ReadField2(iprot thrift.TProtocol) error { - var _field int64 + var _field *int64 if v, err := iprot.ReadI64(); err != nil { return err - } else { - _field = v - } - p.EndTime = _field - return nil -} -func (p *PreviewExportTracesToDatasetRequest) ReadField7(iprot thrift.TProtocol) error { - - var _field *common.PlatformType - if v, err := iprot.ReadString(); err != nil { - return err } else { _field = &v } - p.PlatformType = _field + p.DatasetID = _field return nil } -func (p *PreviewExportTracesToDatasetRequest) ReadField8(iprot thrift.TProtocol) error { +func (p *DatasetConfig) ReadField3(iprot thrift.TProtocol) error { - var _field dataset0.ExportType + var _field *string if v, err := iprot.ReadString(); err != nil { return err } else { - _field = v - } - p.ExportType = _field - return nil -} -func (p *PreviewExportTracesToDatasetRequest) ReadField9(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]*dataset0.FieldMapping, 0, size) - values := make([]dataset0.FieldMapping, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { - return err + _field = &v } - p.FieldMappings = _field + p.DatasetName = _field return nil } -func (p *PreviewExportTracesToDatasetRequest) ReadField255(iprot thrift.TProtocol) error { - _field := base.NewBase() +func (p *DatasetConfig) ReadField4(iprot thrift.TProtocol) error { + _field := dataset0.NewDatasetSchema() if err := _field.Read(iprot); err != nil { return err } - p.Base = _field + p.DatasetSchema = _field return nil } -func (p *PreviewExportTracesToDatasetRequest) Write(oprot thrift.TProtocol) (err error) { +func (p *DatasetConfig) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("PreviewExportTracesToDatasetRequest"); err != nil { + if err = oprot.WriteStructBegin("DatasetConfig"); err != nil { goto WriteStructBeginError } if p != nil { @@ -15585,30 +15009,6 @@ func (p *PreviewExportTracesToDatasetRequest) Write(oprot thrift.TProtocol) (err fieldId = 4 goto WriteFieldError } - if err = p.writeField5(oprot); err != nil { - fieldId = 5 - goto WriteFieldError - } - if err = p.writeField6(oprot); err != nil { - fieldId = 6 - goto WriteFieldError - } - if err = p.writeField7(oprot); err != nil { - fieldId = 7 - goto WriteFieldError - } - if err = p.writeField8(oprot); err != nil { - fieldId = 8 - goto WriteFieldError - } - if err = p.writeField9(oprot); err != nil { - fieldId = 9 - goto WriteFieldError - } - if err = p.writeField255(oprot); err != nil { - fieldId = 255 - goto WriteFieldError - } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -15627,11 +15027,11 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *PreviewExportTracesToDatasetRequest) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { +func (p *DatasetConfig) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("is_new_dataset", thrift.BOOL, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteI64(p.WorkspaceID); err != nil { + if err := oprot.WriteBool(p.IsNewDataset); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -15643,23 +15043,17 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetRequest) writeField2(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("span_ids", thrift.LIST, 2); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.SpanIds)); err != nil { - return err - } - for _, v := range p.SpanIds { - if err := v.Write(oprot); err != nil { +func (p *DatasetConfig) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetDatasetID() { + if err = oprot.WriteFieldBegin("dataset_id", thrift.I64, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.DatasetID); err != nil { return err } - } - if err := oprot.WriteListEnd(); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } return nil WriteFieldBeginError: @@ -15667,15 +15061,17 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetRequest) writeField3(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("category", thrift.I32, 3); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI32(int32(p.Category)); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError +func (p *DatasetConfig) writeField3(oprot thrift.TProtocol) (err error) { + if p.IsSetDatasetName() { + if err = oprot.WriteFieldBegin("dataset_name", thrift.STRING, 3); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.DatasetName); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } return nil WriteFieldBeginError: @@ -15683,15 +15079,17 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetRequest) writeField4(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("config", thrift.STRUCT, 4); err != nil { - goto WriteFieldBeginError - } - if err := p.Config.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError +func (p *DatasetConfig) writeField4(oprot thrift.TProtocol) (err error) { + if p.IsSetDatasetSchema() { + if err = oprot.WriteFieldBegin("dataset_schema", thrift.STRUCT, 4); err != nil { + goto WriteFieldBeginError + } + if err := p.DatasetSchema.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } return nil WriteFieldBeginError: @@ -15699,375 +15097,242 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetRequest) writeField5(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("start_time", thrift.I64, 5); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI64(p.StartTime); err != nil { - return err + +func (p *DatasetConfig) String() string { + if p == nil { + return "" } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) -} -func (p *PreviewExportTracesToDatasetRequest) writeField6(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("end_time", thrift.I64, 6); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI64(p.EndTime); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) -} -func (p *PreviewExportTracesToDatasetRequest) writeField7(oprot thrift.TProtocol) (err error) { - if p.IsSetPlatformType() { - if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 7); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(*p.PlatformType); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 7 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 7 end error: ", p), err) -} -func (p *PreviewExportTracesToDatasetRequest) writeField8(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("export_type", thrift.STRING, 8); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(p.ExportType); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 8 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 8 end error: ", p), err) -} -func (p *PreviewExportTracesToDatasetRequest) writeField9(oprot thrift.TProtocol) (err error) { - if p.IsSetFieldMappings() { - if err = oprot.WriteFieldBegin("field_mappings", thrift.LIST, 9); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FieldMappings)); err != nil { - return err - } - for _, v := range p.FieldMappings { - if err := v.Write(oprot); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 9 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 9 end error: ", p), err) -} -func (p *PreviewExportTracesToDatasetRequest) writeField255(oprot thrift.TProtocol) (err error) { - if p.IsSetBase() { - if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { - goto WriteFieldBeginError - } - if err := p.Base.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) -} - -func (p *PreviewExportTracesToDatasetRequest) String() string { - if p == nil { - return "" - } - return fmt.Sprintf("PreviewExportTracesToDatasetRequest(%+v)", *p) + return fmt.Sprintf("DatasetConfig(%+v)", *p) } -func (p *PreviewExportTracesToDatasetRequest) DeepEqual(ano *PreviewExportTracesToDatasetRequest) bool { +func (p *DatasetConfig) DeepEqual(ano *DatasetConfig) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.WorkspaceID) { - return false - } - if !p.Field2DeepEqual(ano.SpanIds) { - return false - } - if !p.Field3DeepEqual(ano.Category) { - return false - } - if !p.Field4DeepEqual(ano.Config) { - return false - } - if !p.Field5DeepEqual(ano.StartTime) { - return false - } - if !p.Field6DeepEqual(ano.EndTime) { - return false - } - if !p.Field7DeepEqual(ano.PlatformType) { - return false - } - if !p.Field8DeepEqual(ano.ExportType) { - return false - } - if !p.Field9DeepEqual(ano.FieldMappings) { - return false - } - if !p.Field255DeepEqual(ano.Base) { - return false - } - return true -} - -func (p *PreviewExportTracesToDatasetRequest) Field1DeepEqual(src int64) bool { - - if p.WorkspaceID != src { + if !p.Field1DeepEqual(ano.IsNewDataset) { return false } - return true -} -func (p *PreviewExportTracesToDatasetRequest) Field2DeepEqual(src []*SpanID) bool { - - if len(p.SpanIds) != len(src) { + if !p.Field2DeepEqual(ano.DatasetID) { return false } - for i, v := range p.SpanIds { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } - return true -} -func (p *PreviewExportTracesToDatasetRequest) Field3DeepEqual(src dataset.DatasetCategory) bool { - - if p.Category != src { + if !p.Field3DeepEqual(ano.DatasetName) { return false } - return true -} -func (p *PreviewExportTracesToDatasetRequest) Field4DeepEqual(src *DatasetConfig) bool { - - if !p.Config.DeepEqual(src) { + if !p.Field4DeepEqual(ano.DatasetSchema) { return false } return true } -func (p *PreviewExportTracesToDatasetRequest) Field5DeepEqual(src int64) bool { - if p.StartTime != src { - return false - } - return true -} -func (p *PreviewExportTracesToDatasetRequest) Field6DeepEqual(src int64) bool { +func (p *DatasetConfig) Field1DeepEqual(src bool) bool { - if p.EndTime != src { + if p.IsNewDataset != src { return false } return true } -func (p *PreviewExportTracesToDatasetRequest) Field7DeepEqual(src *common.PlatformType) bool { +func (p *DatasetConfig) Field2DeepEqual(src *int64) bool { - if p.PlatformType == src { + if p.DatasetID == src { return true - } else if p.PlatformType == nil || src == nil { + } else if p.DatasetID == nil || src == nil { return false } - if strings.Compare(*p.PlatformType, *src) != 0 { + if *p.DatasetID != *src { return false } return true } -func (p *PreviewExportTracesToDatasetRequest) Field8DeepEqual(src dataset0.ExportType) bool { +func (p *DatasetConfig) Field3DeepEqual(src *string) bool { - if strings.Compare(p.ExportType, src) != 0 { + if p.DatasetName == src { + return true + } else if p.DatasetName == nil || src == nil { return false } - return true -} -func (p *PreviewExportTracesToDatasetRequest) Field9DeepEqual(src []*dataset0.FieldMapping) bool { - - if len(p.FieldMappings) != len(src) { + if strings.Compare(*p.DatasetName, *src) != 0 { return false } - for i, v := range p.FieldMappings { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } return true } -func (p *PreviewExportTracesToDatasetRequest) Field255DeepEqual(src *base.Base) bool { +func (p *DatasetConfig) Field4DeepEqual(src *dataset0.DatasetSchema) bool { - if !p.Base.DeepEqual(src) { + if !p.DatasetSchema.DeepEqual(src) { return false } return true } -type PreviewExportTracesToDatasetResponse struct { - // 预览数据 - Items []*dataset0.Item `thrift:"items,1,optional" frugal:"1,optional,list" form:"items" json:"items,omitempty" query:"items"` - // 概要错误信息 - Errors []*dataset.ItemErrorGroup `thrift:"errors,2,optional" frugal:"2,optional,list" form:"errors" json:"errors,omitempty" query:"errors"` - BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"-" json:"-" query:"-"` +type ExportTracesToDatasetResponse struct { + // 成功导入的数量 + SuccessCount *int32 `thrift:"success_count,1,optional" frugal:"1,optional,i32" form:"success_count" json:"success_count,omitempty" query:"success_count"` + // 错误信息 + Errors []*dataset.ItemErrorGroup `thrift:"errors,2,optional" frugal:"2,optional,list" form:"errors" json:"errors,omitempty" query:"errors"` + // 数据集id + DatasetID *int64 `thrift:"dataset_id,3,optional" frugal:"3,optional,i64" json:"dataset_id" form:"dataset_id" query:"dataset_id"` + // 数据集名称 + DatasetName *string `thrift:"dataset_name,4,optional" frugal:"4,optional,string" form:"dataset_name" json:"dataset_name,omitempty" query:"dataset_name"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"-" json:"-" query:"-"` // 仅供http请求使用; 内部RPC不予使用，统一通过BaseResp获取Code和Msg Code *int32 `thrift:"Code,256,optional" frugal:"256,optional,i32" form:"Code" json:"Code,omitempty" query:"Code"` // 仅供http请求使用; 内部RPC不予使用，统一通过BaseResp获取Code和Msg Msg *string `thrift:"Msg,257,optional" frugal:"257,optional,string" form:"Msg" json:"Msg,omitempty" query:"Msg"` } -func NewPreviewExportTracesToDatasetResponse() *PreviewExportTracesToDatasetResponse { - return &PreviewExportTracesToDatasetResponse{} +func NewExportTracesToDatasetResponse() *ExportTracesToDatasetResponse { + return &ExportTracesToDatasetResponse{} } -func (p *PreviewExportTracesToDatasetResponse) InitDefault() { +func (p *ExportTracesToDatasetResponse) InitDefault() { } -var PreviewExportTracesToDatasetResponse_Items_DEFAULT []*dataset0.Item +var ExportTracesToDatasetResponse_SuccessCount_DEFAULT int32 -func (p *PreviewExportTracesToDatasetResponse) GetItems() (v []*dataset0.Item) { +func (p *ExportTracesToDatasetResponse) GetSuccessCount() (v int32) { if p == nil { return } - if !p.IsSetItems() { - return PreviewExportTracesToDatasetResponse_Items_DEFAULT + if !p.IsSetSuccessCount() { + return ExportTracesToDatasetResponse_SuccessCount_DEFAULT } - return p.Items + return *p.SuccessCount } -var PreviewExportTracesToDatasetResponse_Errors_DEFAULT []*dataset.ItemErrorGroup +var ExportTracesToDatasetResponse_Errors_DEFAULT []*dataset.ItemErrorGroup -func (p *PreviewExportTracesToDatasetResponse) GetErrors() (v []*dataset.ItemErrorGroup) { +func (p *ExportTracesToDatasetResponse) GetErrors() (v []*dataset.ItemErrorGroup) { if p == nil { return } if !p.IsSetErrors() { - return PreviewExportTracesToDatasetResponse_Errors_DEFAULT + return ExportTracesToDatasetResponse_Errors_DEFAULT } return p.Errors } -var PreviewExportTracesToDatasetResponse_BaseResp_DEFAULT *base.BaseResp +var ExportTracesToDatasetResponse_DatasetID_DEFAULT int64 -func (p *PreviewExportTracesToDatasetResponse) GetBaseResp() (v *base.BaseResp) { +func (p *ExportTracesToDatasetResponse) GetDatasetID() (v int64) { if p == nil { return } - if !p.IsSetBaseResp() { - return PreviewExportTracesToDatasetResponse_BaseResp_DEFAULT + if !p.IsSetDatasetID() { + return ExportTracesToDatasetResponse_DatasetID_DEFAULT + } + return *p.DatasetID +} + +var ExportTracesToDatasetResponse_DatasetName_DEFAULT string + +func (p *ExportTracesToDatasetResponse) GetDatasetName() (v string) { + if p == nil { + return + } + if !p.IsSetDatasetName() { + return ExportTracesToDatasetResponse_DatasetName_DEFAULT + } + return *p.DatasetName +} + +var ExportTracesToDatasetResponse_BaseResp_DEFAULT *base.BaseResp + +func (p *ExportTracesToDatasetResponse) GetBaseResp() (v *base.BaseResp) { + if p == nil { + return + } + if !p.IsSetBaseResp() { + return ExportTracesToDatasetResponse_BaseResp_DEFAULT } return p.BaseResp } -var PreviewExportTracesToDatasetResponse_Code_DEFAULT int32 +var ExportTracesToDatasetResponse_Code_DEFAULT int32 -func (p *PreviewExportTracesToDatasetResponse) GetCode() (v int32) { +func (p *ExportTracesToDatasetResponse) GetCode() (v int32) { if p == nil { return } if !p.IsSetCode() { - return PreviewExportTracesToDatasetResponse_Code_DEFAULT + return ExportTracesToDatasetResponse_Code_DEFAULT } return *p.Code } -var PreviewExportTracesToDatasetResponse_Msg_DEFAULT string +var ExportTracesToDatasetResponse_Msg_DEFAULT string -func (p *PreviewExportTracesToDatasetResponse) GetMsg() (v string) { +func (p *ExportTracesToDatasetResponse) GetMsg() (v string) { if p == nil { return } if !p.IsSetMsg() { - return PreviewExportTracesToDatasetResponse_Msg_DEFAULT + return ExportTracesToDatasetResponse_Msg_DEFAULT } return *p.Msg } -func (p *PreviewExportTracesToDatasetResponse) SetItems(val []*dataset0.Item) { - p.Items = val +func (p *ExportTracesToDatasetResponse) SetSuccessCount(val *int32) { + p.SuccessCount = val } -func (p *PreviewExportTracesToDatasetResponse) SetErrors(val []*dataset.ItemErrorGroup) { +func (p *ExportTracesToDatasetResponse) SetErrors(val []*dataset.ItemErrorGroup) { p.Errors = val } -func (p *PreviewExportTracesToDatasetResponse) SetBaseResp(val *base.BaseResp) { +func (p *ExportTracesToDatasetResponse) SetDatasetID(val *int64) { + p.DatasetID = val +} +func (p *ExportTracesToDatasetResponse) SetDatasetName(val *string) { + p.DatasetName = val +} +func (p *ExportTracesToDatasetResponse) SetBaseResp(val *base.BaseResp) { p.BaseResp = val } -func (p *PreviewExportTracesToDatasetResponse) SetCode(val *int32) { +func (p *ExportTracesToDatasetResponse) SetCode(val *int32) { p.Code = val } -func (p *PreviewExportTracesToDatasetResponse) SetMsg(val *string) { +func (p *ExportTracesToDatasetResponse) SetMsg(val *string) { p.Msg = val } -var fieldIDToName_PreviewExportTracesToDatasetResponse = map[int16]string{ - 1: "items", +var fieldIDToName_ExportTracesToDatasetResponse = map[int16]string{ + 1: "success_count", 2: "errors", + 3: "dataset_id", + 4: "dataset_name", 255: "BaseResp", 256: "Code", 257: "Msg", } -func (p *PreviewExportTracesToDatasetResponse) IsSetItems() bool { - return p.Items != nil +func (p *ExportTracesToDatasetResponse) IsSetSuccessCount() bool { + return p.SuccessCount != nil } -func (p *PreviewExportTracesToDatasetResponse) IsSetErrors() bool { +func (p *ExportTracesToDatasetResponse) IsSetErrors() bool { return p.Errors != nil } -func (p *PreviewExportTracesToDatasetResponse) IsSetBaseResp() bool { +func (p *ExportTracesToDatasetResponse) IsSetDatasetID() bool { + return p.DatasetID != nil +} + +func (p *ExportTracesToDatasetResponse) IsSetDatasetName() bool { + return p.DatasetName != nil +} + +func (p *ExportTracesToDatasetResponse) IsSetBaseResp() bool { return p.BaseResp != nil } -func (p *PreviewExportTracesToDatasetResponse) IsSetCode() bool { +func (p *ExportTracesToDatasetResponse) IsSetCode() bool { return p.Code != nil } -func (p *PreviewExportTracesToDatasetResponse) IsSetMsg() bool { +func (p *ExportTracesToDatasetResponse) IsSetMsg() bool { return p.Msg != nil } -func (p *PreviewExportTracesToDatasetResponse) Read(iprot thrift.TProtocol) (err error) { +func (p *ExportTracesToDatasetResponse) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -16086,7 +15351,7 @@ func (p *PreviewExportTracesToDatasetResponse) Read(iprot thrift.TProtocol) (err switch fieldId { case 1: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.I32 { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } @@ -16101,6 +15366,22 @@ func (p *PreviewExportTracesToDatasetResponse) Read(iprot thrift.TProtocol) (err } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } + case 3: + if fieldTypeId == thrift.I64 { + if err = p.ReadField3(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 4: + if fieldTypeId == thrift.STRING { + if err = p.ReadField4(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } case 255: if fieldTypeId == thrift.STRUCT { if err = p.ReadField255(iprot); err != nil { @@ -16144,7 +15425,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_PreviewExportTracesToDatasetResponse[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExportTracesToDatasetResponse[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -16154,30 +15435,18 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *PreviewExportTracesToDatasetResponse) ReadField1(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]*dataset0.Item, 0, size) - values := make([]dataset0.Item, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } +func (p *ExportTracesToDatasetResponse) ReadField1(iprot thrift.TProtocol) error { - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { + var _field *int32 + if v, err := iprot.ReadI32(); err != nil { return err + } else { + _field = &v } - p.Items = _field + p.SuccessCount = _field return nil } -func (p *PreviewExportTracesToDatasetResponse) ReadField2(iprot thrift.TProtocol) error { +func (p *ExportTracesToDatasetResponse) ReadField2(iprot thrift.TProtocol) error { _, size, err := iprot.ReadListBegin() if err != nil { return err @@ -16200,7 +15469,29 @@ func (p *PreviewExportTracesToDatasetResponse) ReadField2(iprot thrift.TProtocol p.Errors = _field return nil } -func (p *PreviewExportTracesToDatasetResponse) ReadField255(iprot thrift.TProtocol) error { +func (p *ExportTracesToDatasetResponse) ReadField3(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v + } + p.DatasetID = _field + return nil +} +func (p *ExportTracesToDatasetResponse) ReadField4(iprot thrift.TProtocol) error { + + var _field *string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.DatasetName = _field + return nil +} +func (p *ExportTracesToDatasetResponse) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBaseResp() if err := _field.Read(iprot); err != nil { return err @@ -16208,7 +15499,7 @@ func (p *PreviewExportTracesToDatasetResponse) ReadField255(iprot thrift.TProtoc p.BaseResp = _field return nil } -func (p *PreviewExportTracesToDatasetResponse) ReadField256(iprot thrift.TProtocol) error { +func (p *ExportTracesToDatasetResponse) ReadField256(iprot thrift.TProtocol) error { var _field *int32 if v, err := iprot.ReadI32(); err != nil { @@ -16219,7 +15510,7 @@ func (p *PreviewExportTracesToDatasetResponse) ReadField256(iprot thrift.TProtoc p.Code = _field return nil } -func (p *PreviewExportTracesToDatasetResponse) ReadField257(iprot thrift.TProtocol) error { +func (p *ExportTracesToDatasetResponse) ReadField257(iprot thrift.TProtocol) error { var _field *string if v, err := iprot.ReadString(); err != nil { @@ -16231,9 +15522,9 @@ func (p *PreviewExportTracesToDatasetResponse) ReadField257(iprot thrift.TProtoc return nil } -func (p *PreviewExportTracesToDatasetResponse) Write(oprot thrift.TProtocol) (err error) { +func (p *ExportTracesToDatasetResponse) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("PreviewExportTracesToDatasetResponse"); err != nil { + if err = oprot.WriteStructBegin("ExportTracesToDatasetResponse"); err != nil { goto WriteStructBeginError } if p != nil { @@ -16245,6 +15536,14 @@ func (p *PreviewExportTracesToDatasetResponse) Write(oprot thrift.TProtocol) (er fieldId = 2 goto WriteFieldError } + if err = p.writeField3(oprot); err != nil { + fieldId = 3 + goto WriteFieldError + } + if err = p.writeField4(oprot); err != nil { + fieldId = 4 + goto WriteFieldError + } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -16275,20 +15574,12 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *PreviewExportTracesToDatasetResponse) writeField1(oprot thrift.TProtocol) (err error) { - if p.IsSetItems() { - if err = oprot.WriteFieldBegin("items", thrift.LIST, 1); err != nil { +func (p *ExportTracesToDatasetResponse) writeField1(oprot thrift.TProtocol) (err error) { + if p.IsSetSuccessCount() { + if err = oprot.WriteFieldBegin("success_count", thrift.I32, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.Items)); err != nil { - return err - } - for _, v := range p.Items { - if err := v.Write(oprot); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { + if err := oprot.WriteI32(*p.SuccessCount); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -16301,7 +15592,7 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetResponse) writeField2(oprot thrift.TProtocol) (err error) { +func (p *ExportTracesToDatasetResponse) writeField2(oprot thrift.TProtocol) (err error) { if p.IsSetErrors() { if err = oprot.WriteFieldBegin("errors", thrift.LIST, 2); err != nil { goto WriteFieldBeginError @@ -16327,7 +15618,43 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetResponse) writeField255(oprot thrift.TProtocol) (err error) { +func (p *ExportTracesToDatasetResponse) writeField3(oprot thrift.TProtocol) (err error) { + if p.IsSetDatasetID() { + if err = oprot.WriteFieldBegin("dataset_id", thrift.I64, 3); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.DatasetID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) +} +func (p *ExportTracesToDatasetResponse) writeField4(oprot thrift.TProtocol) (err error) { + if p.IsSetDatasetName() { + if err = oprot.WriteFieldBegin("dataset_name", thrift.STRING, 4); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.DatasetName); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) +} +func (p *ExportTracesToDatasetResponse) writeField255(oprot thrift.TProtocol) (err error) { if p.IsSetBaseResp() { if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError @@ -16345,7 +15672,7 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetResponse) writeField256(oprot thrift.TProtocol) (err error) { +func (p *ExportTracesToDatasetResponse) writeField256(oprot thrift.TProtocol) (err error) { if p.IsSetCode() { if err = oprot.WriteFieldBegin("Code", thrift.I32, 256); err != nil { goto WriteFieldBeginError @@ -16363,7 +15690,7 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 256 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetResponse) writeField257(oprot thrift.TProtocol) (err error) { +func (p *ExportTracesToDatasetResponse) writeField257(oprot thrift.TProtocol) (err error) { if p.IsSetMsg() { if err = oprot.WriteFieldBegin("Msg", thrift.STRING, 257); err != nil { goto WriteFieldBeginError @@ -16382,26 +15709,32 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 257 end error: ", p), err) } -func (p *PreviewExportTracesToDatasetResponse) String() string { +func (p *ExportTracesToDatasetResponse) String() string { if p == nil { return "" } - return fmt.Sprintf("PreviewExportTracesToDatasetResponse(%+v)", *p) + return fmt.Sprintf("ExportTracesToDatasetResponse(%+v)", *p) } -func (p *PreviewExportTracesToDatasetResponse) DeepEqual(ano *PreviewExportTracesToDatasetResponse) bool { +func (p *ExportTracesToDatasetResponse) DeepEqual(ano *ExportTracesToDatasetResponse) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.Items) { + if !p.Field1DeepEqual(ano.SuccessCount) { return false } if !p.Field2DeepEqual(ano.Errors) { return false } + if !p.Field3DeepEqual(ano.DatasetID) { + return false + } + if !p.Field4DeepEqual(ano.DatasetName) { + return false + } if !p.Field255DeepEqual(ano.BaseResp) { return false } @@ -16414,20 +15747,19 @@ func (p *PreviewExportTracesToDatasetResponse) DeepEqual(ano *PreviewExportTrace return true } -func (p *PreviewExportTracesToDatasetResponse) Field1DeepEqual(src []*dataset0.Item) bool { +func (p *ExportTracesToDatasetResponse) Field1DeepEqual(src *int32) bool { - if len(p.Items) != len(src) { + if p.SuccessCount == src { + return true + } else if p.SuccessCount == nil || src == nil { + return false + } + if *p.SuccessCount != *src { return false } - for i, v := range p.Items { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } return true } -func (p *PreviewExportTracesToDatasetResponse) Field2DeepEqual(src []*dataset.ItemErrorGroup) bool { +func (p *ExportTracesToDatasetResponse) Field2DeepEqual(src []*dataset.ItemErrorGroup) bool { if len(p.Errors) != len(src) { return false @@ -16440,14 +15772,38 @@ func (p *PreviewExportTracesToDatasetResponse) Field2DeepEqual(src []*dataset.It } return true } -func (p *PreviewExportTracesToDatasetResponse) Field255DeepEqual(src *base.BaseResp) bool { +func (p *ExportTracesToDatasetResponse) Field3DeepEqual(src *int64) bool { + + if p.DatasetID == src { + return true + } else if p.DatasetID == nil || src == nil { + return false + } + if *p.DatasetID != *src { + return false + } + return true +} +func (p *ExportTracesToDatasetResponse) Field4DeepEqual(src *string) bool { + + if p.DatasetName == src { + return true + } else if p.DatasetName == nil || src == nil { + return false + } + if strings.Compare(*p.DatasetName, *src) != 0 { + return false + } + return true +} +func (p *ExportTracesToDatasetResponse) Field255DeepEqual(src *base.BaseResp) bool { if !p.BaseResp.DeepEqual(src) { return false } return true } -func (p *PreviewExportTracesToDatasetResponse) Field256DeepEqual(src *int32) bool { +func (p *ExportTracesToDatasetResponse) Field256DeepEqual(src *int32) bool { if p.Code == src { return true @@ -16459,7 +15815,7 @@ func (p *PreviewExportTracesToDatasetResponse) Field256DeepEqual(src *int32) boo } return true } -func (p *PreviewExportTracesToDatasetResponse) Field257DeepEqual(src *string) bool { +func (p *ExportTracesToDatasetResponse) Field257DeepEqual(src *string) bool { if p.Msg == src { return true @@ -16472,138 +15828,228 @@ func (p *PreviewExportTracesToDatasetResponse) Field257DeepEqual(src *string) bo return true } -type ChangeEvaluatorScoreRequest struct { - WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` - AnnotationID string `thrift:"annotation_id,2,required" frugal:"2,required,string" form:"annotation_id,required" json:"annotation_id,required"` - SpanID string `thrift:"span_id,3,required" frugal:"3,required,string" form:"span_id,required" json:"span_id,required"` - StartTime int64 `thrift:"start_time,4,required" frugal:"4,required,i64" json:"start_time" form:"start_time,required" ` - Correction *annotation.Correction `thrift:"correction,5,required" frugal:"5,required,annotation.Correction" form:"correction,required" json:"correction,required"` - PlatformType *common.PlatformType `thrift:"platform_type,6,optional" frugal:"6,optional,string" form:"platform_type" json:"platform_type,omitempty"` - Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` +type PreviewExportTracesToDatasetRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` + SpanIds []*SpanID `thrift:"span_ids,2,required" frugal:"2,required,list" form:"span_ids,required" json:"span_ids,required"` + Category dataset.DatasetCategory `thrift:"category,3,required" frugal:"3,required,DatasetCategory" form:"category,required" json:"category,required"` + Config *DatasetConfig `thrift:"config,4,required" frugal:"4,required,DatasetConfig" form:"config,required" json:"config,required"` + StartTime int64 `thrift:"start_time,5,required" frugal:"5,required,i64" json:"start_time" form:"start_time,required" ` + EndTime int64 `thrift:"end_time,6,required" frugal:"6,required,i64" json:"end_time" form:"end_time,required" ` + PlatformType *common.PlatformType `thrift:"platform_type,7,optional" frugal:"7,optional,string" form:"platform_type" json:"platform_type,omitempty"` + // 导入方式，不填默认为追加 + ExportType dataset0.ExportType `thrift:"export_type,8,required" frugal:"8,required,string" form:"export_type,required" json:"export_type,required"` + FieldMappings []*dataset0.FieldMapping `thrift:"field_mappings,9,optional" frugal:"9,optional,list" form:"field_mappings" json:"field_mappings,omitempty"` + SpanFilters *filter.SpanFilterFields `thrift:"span_filters,10,optional" frugal:"10,optional,filter.SpanFilterFields" form:"span_filters" json:"span_filters,omitempty"` + Limit *int64 `thrift:"limit,11,optional" frugal:"11,optional,i64" form:"limit" json:"limit,string,omitempty"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"-" json:"-" query:"-"` } -func NewChangeEvaluatorScoreRequest() *ChangeEvaluatorScoreRequest { - return &ChangeEvaluatorScoreRequest{} +func NewPreviewExportTracesToDatasetRequest() *PreviewExportTracesToDatasetRequest { + return &PreviewExportTracesToDatasetRequest{} } -func (p *ChangeEvaluatorScoreRequest) InitDefault() { +func (p *PreviewExportTracesToDatasetRequest) InitDefault() { } -func (p *ChangeEvaluatorScoreRequest) GetWorkspaceID() (v int64) { +func (p *PreviewExportTracesToDatasetRequest) GetWorkspaceID() (v int64) { if p != nil { return p.WorkspaceID } return } -func (p *ChangeEvaluatorScoreRequest) GetAnnotationID() (v string) { +func (p *PreviewExportTracesToDatasetRequest) GetSpanIds() (v []*SpanID) { if p != nil { - return p.AnnotationID + return p.SpanIds } return } -func (p *ChangeEvaluatorScoreRequest) GetSpanID() (v string) { +func (p *PreviewExportTracesToDatasetRequest) GetCategory() (v dataset.DatasetCategory) { if p != nil { - return p.SpanID + return p.Category } return } -func (p *ChangeEvaluatorScoreRequest) GetStartTime() (v int64) { +var PreviewExportTracesToDatasetRequest_Config_DEFAULT *DatasetConfig + +func (p *PreviewExportTracesToDatasetRequest) GetConfig() (v *DatasetConfig) { + if p == nil { + return + } + if !p.IsSetConfig() { + return PreviewExportTracesToDatasetRequest_Config_DEFAULT + } + return p.Config +} + +func (p *PreviewExportTracesToDatasetRequest) GetStartTime() (v int64) { if p != nil { return p.StartTime } return } -var ChangeEvaluatorScoreRequest_Correction_DEFAULT *annotation.Correction +func (p *PreviewExportTracesToDatasetRequest) GetEndTime() (v int64) { + if p != nil { + return p.EndTime + } + return +} -func (p *ChangeEvaluatorScoreRequest) GetCorrection() (v *annotation.Correction) { +var PreviewExportTracesToDatasetRequest_PlatformType_DEFAULT common.PlatformType + +func (p *PreviewExportTracesToDatasetRequest) GetPlatformType() (v common.PlatformType) { if p == nil { return } - if !p.IsSetCorrection() { - return ChangeEvaluatorScoreRequest_Correction_DEFAULT + if !p.IsSetPlatformType() { + return PreviewExportTracesToDatasetRequest_PlatformType_DEFAULT } - return p.Correction + return *p.PlatformType } -var ChangeEvaluatorScoreRequest_PlatformType_DEFAULT common.PlatformType +func (p *PreviewExportTracesToDatasetRequest) GetExportType() (v dataset0.ExportType) { + if p != nil { + return p.ExportType + } + return +} -func (p *ChangeEvaluatorScoreRequest) GetPlatformType() (v common.PlatformType) { +var PreviewExportTracesToDatasetRequest_FieldMappings_DEFAULT []*dataset0.FieldMapping + +func (p *PreviewExportTracesToDatasetRequest) GetFieldMappings() (v []*dataset0.FieldMapping) { if p == nil { return } - if !p.IsSetPlatformType() { - return ChangeEvaluatorScoreRequest_PlatformType_DEFAULT + if !p.IsSetFieldMappings() { + return PreviewExportTracesToDatasetRequest_FieldMappings_DEFAULT } - return *p.PlatformType + return p.FieldMappings } -var ChangeEvaluatorScoreRequest_Base_DEFAULT *base.Base +var PreviewExportTracesToDatasetRequest_SpanFilters_DEFAULT *filter.SpanFilterFields -func (p *ChangeEvaluatorScoreRequest) GetBase() (v *base.Base) { +func (p *PreviewExportTracesToDatasetRequest) GetSpanFilters() (v *filter.SpanFilterFields) { + if p == nil { + return + } + if !p.IsSetSpanFilters() { + return PreviewExportTracesToDatasetRequest_SpanFilters_DEFAULT + } + return p.SpanFilters +} + +var PreviewExportTracesToDatasetRequest_Limit_DEFAULT int64 + +func (p *PreviewExportTracesToDatasetRequest) GetLimit() (v int64) { + if p == nil { + return + } + if !p.IsSetLimit() { + return PreviewExportTracesToDatasetRequest_Limit_DEFAULT + } + return *p.Limit +} + +var PreviewExportTracesToDatasetRequest_Base_DEFAULT *base.Base + +func (p *PreviewExportTracesToDatasetRequest) GetBase() (v *base.Base) { if p == nil { return } if !p.IsSetBase() { - return ChangeEvaluatorScoreRequest_Base_DEFAULT + return PreviewExportTracesToDatasetRequest_Base_DEFAULT } return p.Base } -func (p *ChangeEvaluatorScoreRequest) SetWorkspaceID(val int64) { +func (p *PreviewExportTracesToDatasetRequest) SetWorkspaceID(val int64) { p.WorkspaceID = val } -func (p *ChangeEvaluatorScoreRequest) SetAnnotationID(val string) { - p.AnnotationID = val +func (p *PreviewExportTracesToDatasetRequest) SetSpanIds(val []*SpanID) { + p.SpanIds = val } -func (p *ChangeEvaluatorScoreRequest) SetSpanID(val string) { - p.SpanID = val +func (p *PreviewExportTracesToDatasetRequest) SetCategory(val dataset.DatasetCategory) { + p.Category = val } -func (p *ChangeEvaluatorScoreRequest) SetStartTime(val int64) { +func (p *PreviewExportTracesToDatasetRequest) SetConfig(val *DatasetConfig) { + p.Config = val +} +func (p *PreviewExportTracesToDatasetRequest) SetStartTime(val int64) { p.StartTime = val } -func (p *ChangeEvaluatorScoreRequest) SetCorrection(val *annotation.Correction) { - p.Correction = val +func (p *PreviewExportTracesToDatasetRequest) SetEndTime(val int64) { + p.EndTime = val } -func (p *ChangeEvaluatorScoreRequest) SetPlatformType(val *common.PlatformType) { +func (p *PreviewExportTracesToDatasetRequest) SetPlatformType(val *common.PlatformType) { p.PlatformType = val } -func (p *ChangeEvaluatorScoreRequest) SetBase(val *base.Base) { +func (p *PreviewExportTracesToDatasetRequest) SetExportType(val dataset0.ExportType) { + p.ExportType = val +} +func (p *PreviewExportTracesToDatasetRequest) SetFieldMappings(val []*dataset0.FieldMapping) { + p.FieldMappings = val +} +func (p *PreviewExportTracesToDatasetRequest) SetSpanFilters(val *filter.SpanFilterFields) { + p.SpanFilters = val +} +func (p *PreviewExportTracesToDatasetRequest) SetLimit(val *int64) { + p.Limit = val +} +func (p *PreviewExportTracesToDatasetRequest) SetBase(val *base.Base) { p.Base = val } -var fieldIDToName_ChangeEvaluatorScoreRequest = map[int16]string{ +var fieldIDToName_PreviewExportTracesToDatasetRequest = map[int16]string{ 1: "workspace_id", - 2: "annotation_id", - 3: "span_id", - 4: "start_time", - 5: "correction", - 6: "platform_type", + 2: "span_ids", + 3: "category", + 4: "config", + 5: "start_time", + 6: "end_time", + 7: "platform_type", + 8: "export_type", + 9: "field_mappings", + 10: "span_filters", + 11: "limit", 255: "Base", } -func (p *ChangeEvaluatorScoreRequest) IsSetCorrection() bool { - return p.Correction != nil +func (p *PreviewExportTracesToDatasetRequest) IsSetConfig() bool { + return p.Config != nil } -func (p *ChangeEvaluatorScoreRequest) IsSetPlatformType() bool { +func (p *PreviewExportTracesToDatasetRequest) IsSetPlatformType() bool { return p.PlatformType != nil } -func (p *ChangeEvaluatorScoreRequest) IsSetBase() bool { +func (p *PreviewExportTracesToDatasetRequest) IsSetFieldMappings() bool { + return p.FieldMappings != nil +} + +func (p *PreviewExportTracesToDatasetRequest) IsSetSpanFilters() bool { + return p.SpanFilters != nil +} + +func (p *PreviewExportTracesToDatasetRequest) IsSetLimit() bool { + return p.Limit != nil +} + +func (p *PreviewExportTracesToDatasetRequest) IsSetBase() bool { return p.Base != nil } -func (p *ChangeEvaluatorScoreRequest) Read(iprot thrift.TProtocol) (err error) { +func (p *PreviewExportTracesToDatasetRequest) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 var issetWorkspaceID bool = false - var issetAnnotationID bool = false - var issetSpanID bool = false + var issetSpanIds bool = false + var issetCategory bool = false + var issetConfig bool = false var issetStartTime bool = false - var issetCorrection bool = false + var issetEndTime bool = false + var issetExportType bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -16629,101 +16075,153 @@ func (p *ChangeEvaluatorScoreRequest) Read(iprot thrift.TProtocol) (err error) { goto SkipFieldError } case 2: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.LIST { if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } - issetAnnotationID = true + issetSpanIds = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 3: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.I32 { if err = p.ReadField3(iprot); err != nil { goto ReadFieldError } - issetSpanID = true + issetCategory = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 4: - if fieldTypeId == thrift.I64 { + if fieldTypeId == thrift.STRUCT { if err = p.ReadField4(iprot); err != nil { goto ReadFieldError } - issetStartTime = true + issetConfig = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 5: - if fieldTypeId == thrift.STRUCT { + if fieldTypeId == thrift.I64 { if err = p.ReadField5(iprot); err != nil { goto ReadFieldError } - issetCorrection = true + issetStartTime = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 6: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.I64 { if err = p.ReadField6(iprot); err != nil { goto ReadFieldError } + issetEndTime = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - case 255: - if fieldTypeId == thrift.STRUCT { - if err = p.ReadField255(iprot); err != nil { + case 7: + if fieldTypeId == thrift.STRING { + if err = p.ReadField7(iprot); err != nil { goto ReadFieldError } } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - default: - if err = iprot.Skip(fieldTypeId); err != nil { + case 8: + if fieldTypeId == thrift.STRING { + if err = p.ReadField8(iprot); err != nil { + goto ReadFieldError + } + issetExportType = true + } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - } - if err = iprot.ReadFieldEnd(); err != nil { - goto ReadFieldEndError - } - } - if err = iprot.ReadStructEnd(); err != nil { - goto ReadStructEndError - } - - if !issetWorkspaceID { - fieldId = 1 + case 9: + if fieldTypeId == thrift.LIST { + if err = p.ReadField9(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 10: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField10(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 11: + if fieldTypeId == thrift.I64 { + if err = p.ReadField11(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + if !issetWorkspaceID { + fieldId = 1 goto RequiredFieldNotSetError } - if !issetAnnotationID { + if !issetSpanIds { fieldId = 2 goto RequiredFieldNotSetError } - if !issetSpanID { + if !issetCategory { fieldId = 3 goto RequiredFieldNotSetError } - if !issetStartTime { + if !issetConfig { fieldId = 4 goto RequiredFieldNotSetError } - if !issetCorrection { + if !issetStartTime { fieldId = 5 goto RequiredFieldNotSetError } + + if !issetEndTime { + fieldId = 6 + goto RequiredFieldNotSetError + } + + if !issetExportType { + fieldId = 8 + goto RequiredFieldNotSetError + } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ChangeEvaluatorScoreRequest[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_PreviewExportTracesToDatasetRequest[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -16732,10 +16230,10 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ChangeEvaluatorScoreRequest[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_PreviewExportTracesToDatasetRequest[fieldId])) } -func (p *ChangeEvaluatorScoreRequest) ReadField1(iprot thrift.TProtocol) error { +func (p *PreviewExportTracesToDatasetRequest) ReadField1(iprot thrift.TProtocol) error { var _field int64 if v, err := iprot.ReadI64(); err != nil { @@ -16746,29 +16244,49 @@ func (p *ChangeEvaluatorScoreRequest) ReadField1(iprot thrift.TProtocol) error { p.WorkspaceID = _field return nil } -func (p *ChangeEvaluatorScoreRequest) ReadField2(iprot thrift.TProtocol) error { +func (p *PreviewExportTracesToDatasetRequest) ReadField2(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*SpanID, 0, size) + values := make([]SpanID, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() - var _field string - if v, err := iprot.ReadString(); err != nil { + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { return err - } else { - _field = v } - p.AnnotationID = _field + p.SpanIds = _field return nil } -func (p *ChangeEvaluatorScoreRequest) ReadField3(iprot thrift.TProtocol) error { +func (p *PreviewExportTracesToDatasetRequest) ReadField3(iprot thrift.TProtocol) error { - var _field string - if v, err := iprot.ReadString(); err != nil { + var _field dataset.DatasetCategory + if v, err := iprot.ReadI32(); err != nil { return err } else { - _field = v + _field = dataset.DatasetCategory(v) } - p.SpanID = _field + p.Category = _field return nil } -func (p *ChangeEvaluatorScoreRequest) ReadField4(iprot thrift.TProtocol) error { +func (p *PreviewExportTracesToDatasetRequest) ReadField4(iprot thrift.TProtocol) error { + _field := NewDatasetConfig() + if err := _field.Read(iprot); err != nil { + return err + } + p.Config = _field + return nil +} +func (p *PreviewExportTracesToDatasetRequest) ReadField5(iprot thrift.TProtocol) error { var _field int64 if v, err := iprot.ReadI64(); err != nil { @@ -16779,15 +16297,18 @@ func (p *ChangeEvaluatorScoreRequest) ReadField4(iprot thrift.TProtocol) error { p.StartTime = _field return nil } -func (p *ChangeEvaluatorScoreRequest) ReadField5(iprot thrift.TProtocol) error { - _field := annotation.NewCorrection() - if err := _field.Read(iprot); err != nil { +func (p *PreviewExportTracesToDatasetRequest) ReadField6(iprot thrift.TProtocol) error { + + var _field int64 + if v, err := iprot.ReadI64(); err != nil { return err + } else { + _field = v } - p.Correction = _field + p.EndTime = _field return nil } -func (p *ChangeEvaluatorScoreRequest) ReadField6(iprot thrift.TProtocol) error { +func (p *PreviewExportTracesToDatasetRequest) ReadField7(iprot thrift.TProtocol) error { var _field *common.PlatformType if v, err := iprot.ReadString(); err != nil { @@ -16798,7 +16319,60 @@ func (p *ChangeEvaluatorScoreRequest) ReadField6(iprot thrift.TProtocol) error { p.PlatformType = _field return nil } -func (p *ChangeEvaluatorScoreRequest) ReadField255(iprot thrift.TProtocol) error { +func (p *PreviewExportTracesToDatasetRequest) ReadField8(iprot thrift.TProtocol) error { + + var _field dataset0.ExportType + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v + } + p.ExportType = _field + return nil +} +func (p *PreviewExportTracesToDatasetRequest) ReadField9(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*dataset0.FieldMapping, 0, size) + values := make([]dataset0.FieldMapping, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.FieldMappings = _field + return nil +} +func (p *PreviewExportTracesToDatasetRequest) ReadField10(iprot thrift.TProtocol) error { + _field := filter.NewSpanFilterFields() + if err := _field.Read(iprot); err != nil { + return err + } + p.SpanFilters = _field + return nil +} +func (p *PreviewExportTracesToDatasetRequest) ReadField11(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v + } + p.Limit = _field + return nil +} +func (p *PreviewExportTracesToDatasetRequest) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBase() if err := _field.Read(iprot); err != nil { return err @@ -16807,9 +16381,9 @@ func (p *ChangeEvaluatorScoreRequest) ReadField255(iprot thrift.TProtocol) error return nil } -func (p *ChangeEvaluatorScoreRequest) Write(oprot thrift.TProtocol) (err error) { +func (p *PreviewExportTracesToDatasetRequest) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ChangeEvaluatorScoreRequest"); err != nil { + if err = oprot.WriteStructBegin("PreviewExportTracesToDatasetRequest"); err != nil { goto WriteStructBeginError } if p != nil { @@ -16837,6 +16411,26 @@ func (p *ChangeEvaluatorScoreRequest) Write(oprot thrift.TProtocol) (err error) fieldId = 6 goto WriteFieldError } + if err = p.writeField7(oprot); err != nil { + fieldId = 7 + goto WriteFieldError + } + if err = p.writeField8(oprot); err != nil { + fieldId = 8 + goto WriteFieldError + } + if err = p.writeField9(oprot); err != nil { + fieldId = 9 + goto WriteFieldError + } + if err = p.writeField10(oprot); err != nil { + fieldId = 10 + goto WriteFieldError + } + if err = p.writeField11(oprot); err != nil { + fieldId = 11 + goto WriteFieldError + } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -16859,7 +16453,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ChangeEvaluatorScoreRequest) writeField1(oprot thrift.TProtocol) (err error) { +func (p *PreviewExportTracesToDatasetRequest) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { goto WriteFieldBeginError } @@ -16875,11 +16469,19 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ChangeEvaluatorScoreRequest) writeField2(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("annotation_id", thrift.STRING, 2); err != nil { +func (p *PreviewExportTracesToDatasetRequest) writeField2(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("span_ids", thrift.LIST, 2); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteString(p.AnnotationID); err != nil { + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.SpanIds)); err != nil { + return err + } + for _, v := range p.SpanIds { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -16891,11 +16493,11 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *ChangeEvaluatorScoreRequest) writeField3(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("span_id", thrift.STRING, 3); err != nil { +func (p *PreviewExportTracesToDatasetRequest) writeField3(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("category", thrift.I32, 3); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteString(p.SpanID); err != nil { + if err := oprot.WriteI32(int32(p.Category)); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -16907,11 +16509,11 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) } -func (p *ChangeEvaluatorScoreRequest) writeField4(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("start_time", thrift.I64, 4); err != nil { +func (p *PreviewExportTracesToDatasetRequest) writeField4(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("config", thrift.STRUCT, 4); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteI64(p.StartTime); err != nil { + if err := p.Config.Write(oprot); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -16923,11 +16525,11 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) } -func (p *ChangeEvaluatorScoreRequest) writeField5(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("correction", thrift.STRUCT, 5); err != nil { +func (p *PreviewExportTracesToDatasetRequest) writeField5(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("start_time", thrift.I64, 5); err != nil { goto WriteFieldBeginError } - if err := p.Correction.Write(oprot); err != nil { + if err := oprot.WriteI64(p.StartTime); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -16939,9 +16541,25 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) } -func (p *ChangeEvaluatorScoreRequest) writeField6(oprot thrift.TProtocol) (err error) { +func (p *PreviewExportTracesToDatasetRequest) writeField6(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("end_time", thrift.I64, 6); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(p.EndTime); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) +} +func (p *PreviewExportTracesToDatasetRequest) writeField7(oprot thrift.TProtocol) (err error) { if p.IsSetPlatformType() { - if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 6); err != nil { + if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 7); err != nil { goto WriteFieldBeginError } if err := oprot.WriteString(*p.PlatformType); err != nil { @@ -16953,11 +16571,89 @@ func (p *ChangeEvaluatorScoreRequest) writeField6(oprot thrift.TProtocol) (err e } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 7 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 7 end error: ", p), err) } -func (p *ChangeEvaluatorScoreRequest) writeField255(oprot thrift.TProtocol) (err error) { +func (p *PreviewExportTracesToDatasetRequest) writeField8(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("export_type", thrift.STRING, 8); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(p.ExportType); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 8 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 8 end error: ", p), err) +} +func (p *PreviewExportTracesToDatasetRequest) writeField9(oprot thrift.TProtocol) (err error) { + if p.IsSetFieldMappings() { + if err = oprot.WriteFieldBegin("field_mappings", thrift.LIST, 9); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FieldMappings)); err != nil { + return err + } + for _, v := range p.FieldMappings { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 9 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 9 end error: ", p), err) +} +func (p *PreviewExportTracesToDatasetRequest) writeField10(oprot thrift.TProtocol) (err error) { + if p.IsSetSpanFilters() { + if err = oprot.WriteFieldBegin("span_filters", thrift.STRUCT, 10); err != nil { + goto WriteFieldBeginError + } + if err := p.SpanFilters.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 10 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 10 end error: ", p), err) +} +func (p *PreviewExportTracesToDatasetRequest) writeField11(oprot thrift.TProtocol) (err error) { + if p.IsSetLimit() { + if err = oprot.WriteFieldBegin("limit", thrift.I64, 11); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.Limit); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 11 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 11 end error: ", p), err) +} +func (p *PreviewExportTracesToDatasetRequest) writeField255(oprot thrift.TProtocol) (err error) { if p.IsSetBase() { if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError @@ -16976,15 +16672,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *ChangeEvaluatorScoreRequest) String() string { +func (p *PreviewExportTracesToDatasetRequest) String() string { if p == nil { return "" } - return fmt.Sprintf("ChangeEvaluatorScoreRequest(%+v)", *p) + return fmt.Sprintf("PreviewExportTracesToDatasetRequest(%+v)", *p) } -func (p *ChangeEvaluatorScoreRequest) DeepEqual(ano *ChangeEvaluatorScoreRequest) bool { +func (p *PreviewExportTracesToDatasetRequest) DeepEqual(ano *PreviewExportTracesToDatasetRequest) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -16993,19 +16689,34 @@ func (p *ChangeEvaluatorScoreRequest) DeepEqual(ano *ChangeEvaluatorScoreRequest if !p.Field1DeepEqual(ano.WorkspaceID) { return false } - if !p.Field2DeepEqual(ano.AnnotationID) { + if !p.Field2DeepEqual(ano.SpanIds) { return false } - if !p.Field3DeepEqual(ano.SpanID) { + if !p.Field3DeepEqual(ano.Category) { return false } - if !p.Field4DeepEqual(ano.StartTime) { + if !p.Field4DeepEqual(ano.Config) { return false } - if !p.Field5DeepEqual(ano.Correction) { + if !p.Field5DeepEqual(ano.StartTime) { return false } - if !p.Field6DeepEqual(ano.PlatformType) { + if !p.Field6DeepEqual(ano.EndTime) { + return false + } + if !p.Field7DeepEqual(ano.PlatformType) { + return false + } + if !p.Field8DeepEqual(ano.ExportType) { + return false + } + if !p.Field9DeepEqual(ano.FieldMappings) { + return false + } + if !p.Field10DeepEqual(ano.SpanFilters) { + return false + } + if !p.Field11DeepEqual(ano.Limit) { return false } if !p.Field255DeepEqual(ano.Base) { @@ -17014,42 +16725,55 @@ func (p *ChangeEvaluatorScoreRequest) DeepEqual(ano *ChangeEvaluatorScoreRequest return true } -func (p *ChangeEvaluatorScoreRequest) Field1DeepEqual(src int64) bool { +func (p *PreviewExportTracesToDatasetRequest) Field1DeepEqual(src int64) bool { if p.WorkspaceID != src { return false } return true } -func (p *ChangeEvaluatorScoreRequest) Field2DeepEqual(src string) bool { +func (p *PreviewExportTracesToDatasetRequest) Field2DeepEqual(src []*SpanID) bool { - if strings.Compare(p.AnnotationID, src) != 0 { + if len(p.SpanIds) != len(src) { return false } + for i, v := range p.SpanIds { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } return true } -func (p *ChangeEvaluatorScoreRequest) Field3DeepEqual(src string) bool { +func (p *PreviewExportTracesToDatasetRequest) Field3DeepEqual(src dataset.DatasetCategory) bool { - if strings.Compare(p.SpanID, src) != 0 { + if p.Category != src { return false } return true } -func (p *ChangeEvaluatorScoreRequest) Field4DeepEqual(src int64) bool { +func (p *PreviewExportTracesToDatasetRequest) Field4DeepEqual(src *DatasetConfig) bool { + + if !p.Config.DeepEqual(src) { + return false + } + return true +} +func (p *PreviewExportTracesToDatasetRequest) Field5DeepEqual(src int64) bool { if p.StartTime != src { return false } return true } -func (p *ChangeEvaluatorScoreRequest) Field5DeepEqual(src *annotation.Correction) bool { +func (p *PreviewExportTracesToDatasetRequest) Field6DeepEqual(src int64) bool { - if !p.Correction.DeepEqual(src) { + if p.EndTime != src { return false } return true } -func (p *ChangeEvaluatorScoreRequest) Field6DeepEqual(src *common.PlatformType) bool { +func (p *PreviewExportTracesToDatasetRequest) Field7DeepEqual(src *common.PlatformType) bool { if p.PlatformType == src { return true @@ -17061,7 +16785,46 @@ func (p *ChangeEvaluatorScoreRequest) Field6DeepEqual(src *common.PlatformType) } return true } -func (p *ChangeEvaluatorScoreRequest) Field255DeepEqual(src *base.Base) bool { +func (p *PreviewExportTracesToDatasetRequest) Field8DeepEqual(src dataset0.ExportType) bool { + + if strings.Compare(p.ExportType, src) != 0 { + return false + } + return true +} +func (p *PreviewExportTracesToDatasetRequest) Field9DeepEqual(src []*dataset0.FieldMapping) bool { + + if len(p.FieldMappings) != len(src) { + return false + } + for i, v := range p.FieldMappings { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true +} +func (p *PreviewExportTracesToDatasetRequest) Field10DeepEqual(src *filter.SpanFilterFields) bool { + + if !p.SpanFilters.DeepEqual(src) { + return false + } + return true +} +func (p *PreviewExportTracesToDatasetRequest) Field11DeepEqual(src *int64) bool { + + if p.Limit == src { + return true + } else if p.Limit == nil || src == nil { + return false + } + if *p.Limit != *src { + return false + } + return true +} +func (p *PreviewExportTracesToDatasetRequest) Field255DeepEqual(src *base.Base) bool { if !p.Base.DeepEqual(src) { return false @@ -17069,86 +16832,159 @@ func (p *ChangeEvaluatorScoreRequest) Field255DeepEqual(src *base.Base) bool { return true } -type ChangeEvaluatorScoreResponse struct { - Annotation *annotation.Annotation `thrift:"annotation,1,required" frugal:"1,required,annotation.Annotation" form:"annotation,required" json:"annotation,required" query:"annotation,required"` - BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` +type PreviewExportTracesToDatasetResponse struct { + // 预览数据 + Items []*dataset0.Item `thrift:"items,1,optional" frugal:"1,optional,list" form:"items" json:"items,omitempty" query:"items"` + // 概要错误信息 + Errors []*dataset.ItemErrorGroup `thrift:"errors,2,optional" frugal:"2,optional,list" form:"errors" json:"errors,omitempty" query:"errors"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"-" json:"-" query:"-"` + // 仅供http请求使用; 内部RPC不予使用，统一通过BaseResp获取Code和Msg + Code *int32 `thrift:"Code,256,optional" frugal:"256,optional,i32" form:"Code" json:"Code,omitempty" query:"Code"` + // 仅供http请求使用; 内部RPC不予使用，统一通过BaseResp获取Code和Msg + Msg *string `thrift:"Msg,257,optional" frugal:"257,optional,string" form:"Msg" json:"Msg,omitempty" query:"Msg"` } -func NewChangeEvaluatorScoreResponse() *ChangeEvaluatorScoreResponse { - return &ChangeEvaluatorScoreResponse{} +func NewPreviewExportTracesToDatasetResponse() *PreviewExportTracesToDatasetResponse { + return &PreviewExportTracesToDatasetResponse{} } -func (p *ChangeEvaluatorScoreResponse) InitDefault() { +func (p *PreviewExportTracesToDatasetResponse) InitDefault() { } -var ChangeEvaluatorScoreResponse_Annotation_DEFAULT *annotation.Annotation +var PreviewExportTracesToDatasetResponse_Items_DEFAULT []*dataset0.Item -func (p *ChangeEvaluatorScoreResponse) GetAnnotation() (v *annotation.Annotation) { +func (p *PreviewExportTracesToDatasetResponse) GetItems() (v []*dataset0.Item) { if p == nil { return } - if !p.IsSetAnnotation() { - return ChangeEvaluatorScoreResponse_Annotation_DEFAULT + if !p.IsSetItems() { + return PreviewExportTracesToDatasetResponse_Items_DEFAULT } - return p.Annotation + return p.Items } -var ChangeEvaluatorScoreResponse_BaseResp_DEFAULT *base.BaseResp +var PreviewExportTracesToDatasetResponse_Errors_DEFAULT []*dataset.ItemErrorGroup -func (p *ChangeEvaluatorScoreResponse) GetBaseResp() (v *base.BaseResp) { +func (p *PreviewExportTracesToDatasetResponse) GetErrors() (v []*dataset.ItemErrorGroup) { if p == nil { return } - if !p.IsSetBaseResp() { - return ChangeEvaluatorScoreResponse_BaseResp_DEFAULT + if !p.IsSetErrors() { + return PreviewExportTracesToDatasetResponse_Errors_DEFAULT } - return p.BaseResp -} -func (p *ChangeEvaluatorScoreResponse) SetAnnotation(val *annotation.Annotation) { - p.Annotation = val -} -func (p *ChangeEvaluatorScoreResponse) SetBaseResp(val *base.BaseResp) { - p.BaseResp = val -} - -var fieldIDToName_ChangeEvaluatorScoreResponse = map[int16]string{ - 1: "annotation", - 255: "BaseResp", + return p.Errors } -func (p *ChangeEvaluatorScoreResponse) IsSetAnnotation() bool { - return p.Annotation != nil -} +var PreviewExportTracesToDatasetResponse_BaseResp_DEFAULT *base.BaseResp -func (p *ChangeEvaluatorScoreResponse) IsSetBaseResp() bool { - return p.BaseResp != nil +func (p *PreviewExportTracesToDatasetResponse) GetBaseResp() (v *base.BaseResp) { + if p == nil { + return + } + if !p.IsSetBaseResp() { + return PreviewExportTracesToDatasetResponse_BaseResp_DEFAULT + } + return p.BaseResp } -func (p *ChangeEvaluatorScoreResponse) Read(iprot thrift.TProtocol) (err error) { - var fieldTypeId thrift.TType - var fieldId int16 - var issetAnnotation bool = false +var PreviewExportTracesToDatasetResponse_Code_DEFAULT int32 - if _, err = iprot.ReadStructBegin(); err != nil { - goto ReadStructBeginError +func (p *PreviewExportTracesToDatasetResponse) GetCode() (v int32) { + if p == nil { + return + } + if !p.IsSetCode() { + return PreviewExportTracesToDatasetResponse_Code_DEFAULT } + return *p.Code +} - for { - _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() - if err != nil { - goto ReadFieldBeginError - } - if fieldTypeId == thrift.STOP { - break - } +var PreviewExportTracesToDatasetResponse_Msg_DEFAULT string - switch fieldId { - case 1: - if fieldTypeId == thrift.STRUCT { +func (p *PreviewExportTracesToDatasetResponse) GetMsg() (v string) { + if p == nil { + return + } + if !p.IsSetMsg() { + return PreviewExportTracesToDatasetResponse_Msg_DEFAULT + } + return *p.Msg +} +func (p *PreviewExportTracesToDatasetResponse) SetItems(val []*dataset0.Item) { + p.Items = val +} +func (p *PreviewExportTracesToDatasetResponse) SetErrors(val []*dataset.ItemErrorGroup) { + p.Errors = val +} +func (p *PreviewExportTracesToDatasetResponse) SetBaseResp(val *base.BaseResp) { + p.BaseResp = val +} +func (p *PreviewExportTracesToDatasetResponse) SetCode(val *int32) { + p.Code = val +} +func (p *PreviewExportTracesToDatasetResponse) SetMsg(val *string) { + p.Msg = val +} + +var fieldIDToName_PreviewExportTracesToDatasetResponse = map[int16]string{ + 1: "items", + 2: "errors", + 255: "BaseResp", + 256: "Code", + 257: "Msg", +} + +func (p *PreviewExportTracesToDatasetResponse) IsSetItems() bool { + return p.Items != nil +} + +func (p *PreviewExportTracesToDatasetResponse) IsSetErrors() bool { + return p.Errors != nil +} + +func (p *PreviewExportTracesToDatasetResponse) IsSetBaseResp() bool { + return p.BaseResp != nil +} + +func (p *PreviewExportTracesToDatasetResponse) IsSetCode() bool { + return p.Code != nil +} + +func (p *PreviewExportTracesToDatasetResponse) IsSetMsg() bool { + return p.Msg != nil +} + +func (p *PreviewExportTracesToDatasetResponse) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.LIST { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } - issetAnnotation = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 2: + if fieldTypeId == thrift.LIST { + if err = p.ReadField2(iprot); err != nil { + goto ReadFieldError + } } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -17160,6 +16996,22 @@ func (p *ChangeEvaluatorScoreResponse) Read(iprot thrift.TProtocol) (err error) } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } + case 256: + if fieldTypeId == thrift.I32 { + if err = p.ReadField256(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 257: + if fieldTypeId == thrift.STRING { + if err = p.ReadField257(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } default: if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError @@ -17173,17 +17025,13 @@ func (p *ChangeEvaluatorScoreResponse) Read(iprot thrift.TProtocol) (err error) goto ReadStructEndError } - if !issetAnnotation { - fieldId = 1 - goto RequiredFieldNotSetError - } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ChangeEvaluatorScoreResponse[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_PreviewExportTracesToDatasetResponse[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -17191,19 +17039,55 @@ ReadFieldEndError: return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) -RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ChangeEvaluatorScoreResponse[fieldId])) } -func (p *ChangeEvaluatorScoreResponse) ReadField1(iprot thrift.TProtocol) error { - _field := annotation.NewAnnotation() - if err := _field.Read(iprot); err != nil { +func (p *PreviewExportTracesToDatasetResponse) ReadField1(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { return err } - p.Annotation = _field + _field := make([]*dataset0.Item, 0, size) + values := make([]dataset0.Item, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.Items = _field return nil } -func (p *ChangeEvaluatorScoreResponse) ReadField255(iprot thrift.TProtocol) error { +func (p *PreviewExportTracesToDatasetResponse) ReadField2(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*dataset.ItemErrorGroup, 0, size) + values := make([]dataset.ItemErrorGroup, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.Errors = _field + return nil +} +func (p *PreviewExportTracesToDatasetResponse) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBaseResp() if err := _field.Read(iprot); err != nil { return err @@ -17211,10 +17095,32 @@ func (p *ChangeEvaluatorScoreResponse) ReadField255(iprot thrift.TProtocol) erro p.BaseResp = _field return nil } +func (p *PreviewExportTracesToDatasetResponse) ReadField256(iprot thrift.TProtocol) error { -func (p *ChangeEvaluatorScoreResponse) Write(oprot thrift.TProtocol) (err error) { + var _field *int32 + if v, err := iprot.ReadI32(); err != nil { + return err + } else { + _field = &v + } + p.Code = _field + return nil +} +func (p *PreviewExportTracesToDatasetResponse) ReadField257(iprot thrift.TProtocol) error { + + var _field *string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.Msg = _field + return nil +} + +func (p *PreviewExportTracesToDatasetResponse) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ChangeEvaluatorScoreResponse"); err != nil { + if err = oprot.WriteStructBegin("PreviewExportTracesToDatasetResponse"); err != nil { goto WriteStructBeginError } if p != nil { @@ -17222,10 +17128,22 @@ func (p *ChangeEvaluatorScoreResponse) Write(oprot thrift.TProtocol) (err error) fieldId = 1 goto WriteFieldError } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError } + if err = p.writeField256(oprot); err != nil { + fieldId = 256 + goto WriteFieldError + } + if err = p.writeField257(oprot); err != nil { + fieldId = 257 + goto WriteFieldError + } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -17244,15 +17162,25 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ChangeEvaluatorScoreResponse) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("annotation", thrift.STRUCT, 1); err != nil { - goto WriteFieldBeginError - } - if err := p.Annotation.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError +func (p *PreviewExportTracesToDatasetResponse) writeField1(oprot thrift.TProtocol) (err error) { + if p.IsSetItems() { + if err = oprot.WriteFieldBegin("items", thrift.LIST, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.Items)); err != nil { + return err + } + for _, v := range p.Items { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } return nil WriteFieldBeginError: @@ -17260,7 +17188,33 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ChangeEvaluatorScoreResponse) writeField255(oprot thrift.TProtocol) (err error) { +func (p *PreviewExportTracesToDatasetResponse) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetErrors() { + if err = oprot.WriteFieldBegin("errors", thrift.LIST, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.Errors)); err != nil { + return err + } + for _, v := range p.Errors { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} +func (p *PreviewExportTracesToDatasetResponse) writeField255(oprot thrift.TProtocol) (err error) { if p.IsSetBaseResp() { if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError @@ -17278,116 +17232,265 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } +func (p *PreviewExportTracesToDatasetResponse) writeField256(oprot thrift.TProtocol) (err error) { + if p.IsSetCode() { + if err = oprot.WriteFieldBegin("Code", thrift.I32, 256); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI32(*p.Code); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 256 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 256 end error: ", p), err) +} +func (p *PreviewExportTracesToDatasetResponse) writeField257(oprot thrift.TProtocol) (err error) { + if p.IsSetMsg() { + if err = oprot.WriteFieldBegin("Msg", thrift.STRING, 257); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.Msg); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 257 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 257 end error: ", p), err) +} -func (p *ChangeEvaluatorScoreResponse) String() string { - if p == nil { - return "" - } - return fmt.Sprintf("ChangeEvaluatorScoreResponse(%+v)", *p) - -} - -func (p *ChangeEvaluatorScoreResponse) DeepEqual(ano *ChangeEvaluatorScoreResponse) bool { +func (p *PreviewExportTracesToDatasetResponse) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("PreviewExportTracesToDatasetResponse(%+v)", *p) + +} + +func (p *PreviewExportTracesToDatasetResponse) DeepEqual(ano *PreviewExportTracesToDatasetResponse) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.Annotation) { + if !p.Field1DeepEqual(ano.Items) { + return false + } + if !p.Field2DeepEqual(ano.Errors) { return false } if !p.Field255DeepEqual(ano.BaseResp) { return false } + if !p.Field256DeepEqual(ano.Code) { + return false + } + if !p.Field257DeepEqual(ano.Msg) { + return false + } return true } -func (p *ChangeEvaluatorScoreResponse) Field1DeepEqual(src *annotation.Annotation) bool { +func (p *PreviewExportTracesToDatasetResponse) Field1DeepEqual(src []*dataset0.Item) bool { - if !p.Annotation.DeepEqual(src) { + if len(p.Items) != len(src) { + return false + } + for i, v := range p.Items { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true +} +func (p *PreviewExportTracesToDatasetResponse) Field2DeepEqual(src []*dataset.ItemErrorGroup) bool { + + if len(p.Errors) != len(src) { return false } + for i, v := range p.Errors { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } return true } -func (p *ChangeEvaluatorScoreResponse) Field255DeepEqual(src *base.BaseResp) bool { +func (p *PreviewExportTracesToDatasetResponse) Field255DeepEqual(src *base.BaseResp) bool { if !p.BaseResp.DeepEqual(src) { return false } return true } +func (p *PreviewExportTracesToDatasetResponse) Field256DeepEqual(src *int32) bool { -type ListAnnotationEvaluatorsRequest struct { - WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" query:"workspace_id,required" ` - Name *string `thrift:"name,2,optional" frugal:"2,optional,string" json:"name,omitempty" query:"name"` - Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"-" json:"-" query:"-"` + if p.Code == src { + return true + } else if p.Code == nil || src == nil { + return false + } + if *p.Code != *src { + return false + } + return true } +func (p *PreviewExportTracesToDatasetResponse) Field257DeepEqual(src *string) bool { -func NewListAnnotationEvaluatorsRequest() *ListAnnotationEvaluatorsRequest { - return &ListAnnotationEvaluatorsRequest{} + if p.Msg == src { + return true + } else if p.Msg == nil || src == nil { + return false + } + if strings.Compare(*p.Msg, *src) != 0 { + return false + } + return true } -func (p *ListAnnotationEvaluatorsRequest) InitDefault() { +type ChangeEvaluatorScoreRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` + AnnotationID string `thrift:"annotation_id,2,required" frugal:"2,required,string" form:"annotation_id,required" json:"annotation_id,required"` + SpanID string `thrift:"span_id,3,required" frugal:"3,required,string" form:"span_id,required" json:"span_id,required"` + StartTime int64 `thrift:"start_time,4,required" frugal:"4,required,i64" json:"start_time" form:"start_time,required" ` + Correction *annotation.Correction `thrift:"correction,5,required" frugal:"5,required,annotation.Correction" form:"correction,required" json:"correction,required"` + PlatformType *common.PlatformType `thrift:"platform_type,6,optional" frugal:"6,optional,string" form:"platform_type" json:"platform_type,omitempty"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` } -func (p *ListAnnotationEvaluatorsRequest) GetWorkspaceID() (v int64) { +func NewChangeEvaluatorScoreRequest() *ChangeEvaluatorScoreRequest { + return &ChangeEvaluatorScoreRequest{} +} + +func (p *ChangeEvaluatorScoreRequest) InitDefault() { +} + +func (p *ChangeEvaluatorScoreRequest) GetWorkspaceID() (v int64) { if p != nil { return p.WorkspaceID } return } -var ListAnnotationEvaluatorsRequest_Name_DEFAULT string +func (p *ChangeEvaluatorScoreRequest) GetAnnotationID() (v string) { + if p != nil { + return p.AnnotationID + } + return +} -func (p *ListAnnotationEvaluatorsRequest) GetName() (v string) { +func (p *ChangeEvaluatorScoreRequest) GetSpanID() (v string) { + if p != nil { + return p.SpanID + } + return +} + +func (p *ChangeEvaluatorScoreRequest) GetStartTime() (v int64) { + if p != nil { + return p.StartTime + } + return +} + +var ChangeEvaluatorScoreRequest_Correction_DEFAULT *annotation.Correction + +func (p *ChangeEvaluatorScoreRequest) GetCorrection() (v *annotation.Correction) { if p == nil { return } - if !p.IsSetName() { - return ListAnnotationEvaluatorsRequest_Name_DEFAULT + if !p.IsSetCorrection() { + return ChangeEvaluatorScoreRequest_Correction_DEFAULT } - return *p.Name + return p.Correction } -var ListAnnotationEvaluatorsRequest_Base_DEFAULT *base.Base +var ChangeEvaluatorScoreRequest_PlatformType_DEFAULT common.PlatformType -func (p *ListAnnotationEvaluatorsRequest) GetBase() (v *base.Base) { +func (p *ChangeEvaluatorScoreRequest) GetPlatformType() (v common.PlatformType) { + if p == nil { + return + } + if !p.IsSetPlatformType() { + return ChangeEvaluatorScoreRequest_PlatformType_DEFAULT + } + return *p.PlatformType +} + +var ChangeEvaluatorScoreRequest_Base_DEFAULT *base.Base + +func (p *ChangeEvaluatorScoreRequest) GetBase() (v *base.Base) { if p == nil { return } if !p.IsSetBase() { - return ListAnnotationEvaluatorsRequest_Base_DEFAULT + return ChangeEvaluatorScoreRequest_Base_DEFAULT } return p.Base } -func (p *ListAnnotationEvaluatorsRequest) SetWorkspaceID(val int64) { +func (p *ChangeEvaluatorScoreRequest) SetWorkspaceID(val int64) { p.WorkspaceID = val } -func (p *ListAnnotationEvaluatorsRequest) SetName(val *string) { - p.Name = val +func (p *ChangeEvaluatorScoreRequest) SetAnnotationID(val string) { + p.AnnotationID = val } -func (p *ListAnnotationEvaluatorsRequest) SetBase(val *base.Base) { +func (p *ChangeEvaluatorScoreRequest) SetSpanID(val string) { + p.SpanID = val +} +func (p *ChangeEvaluatorScoreRequest) SetStartTime(val int64) { + p.StartTime = val +} +func (p *ChangeEvaluatorScoreRequest) SetCorrection(val *annotation.Correction) { + p.Correction = val +} +func (p *ChangeEvaluatorScoreRequest) SetPlatformType(val *common.PlatformType) { + p.PlatformType = val +} +func (p *ChangeEvaluatorScoreRequest) SetBase(val *base.Base) { p.Base = val } -var fieldIDToName_ListAnnotationEvaluatorsRequest = map[int16]string{ +var fieldIDToName_ChangeEvaluatorScoreRequest = map[int16]string{ 1: "workspace_id", - 2: "name", + 2: "annotation_id", + 3: "span_id", + 4: "start_time", + 5: "correction", + 6: "platform_type", 255: "Base", } -func (p *ListAnnotationEvaluatorsRequest) IsSetName() bool { - return p.Name != nil +func (p *ChangeEvaluatorScoreRequest) IsSetCorrection() bool { + return p.Correction != nil } -func (p *ListAnnotationEvaluatorsRequest) IsSetBase() bool { +func (p *ChangeEvaluatorScoreRequest) IsSetPlatformType() bool { + return p.PlatformType != nil +} + +func (p *ChangeEvaluatorScoreRequest) IsSetBase() bool { return p.Base != nil } -func (p *ListAnnotationEvaluatorsRequest) Read(iprot thrift.TProtocol) (err error) { +func (p *ChangeEvaluatorScoreRequest) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 var issetWorkspaceID bool = false + var issetAnnotationID bool = false + var issetSpanID bool = false + var issetStartTime bool = false + var issetCorrection bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -17417,6 +17520,42 @@ func (p *ListAnnotationEvaluatorsRequest) Read(iprot thrift.TProtocol) (err erro if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } + issetAnnotationID = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 3: + if fieldTypeId == thrift.STRING { + if err = p.ReadField3(iprot); err != nil { + goto ReadFieldError + } + issetSpanID = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 4: + if fieldTypeId == thrift.I64 { + if err = p.ReadField4(iprot); err != nil { + goto ReadFieldError + } + issetStartTime = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 5: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField5(iprot); err != nil { + goto ReadFieldError + } + issetCorrection = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 6: + if fieldTypeId == thrift.STRING { + if err = p.ReadField6(iprot); err != nil { + goto ReadFieldError + } } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -17445,13 +17584,33 @@ func (p *ListAnnotationEvaluatorsRequest) Read(iprot thrift.TProtocol) (err erro fieldId = 1 goto RequiredFieldNotSetError } - return nil + + if !issetAnnotationID { + fieldId = 2 + goto RequiredFieldNotSetError + } + + if !issetSpanID { + fieldId = 3 + goto RequiredFieldNotSetError + } + + if !issetStartTime { + fieldId = 4 + goto RequiredFieldNotSetError + } + + if !issetCorrection { + fieldId = 5 + goto RequiredFieldNotSetError + } + return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListAnnotationEvaluatorsRequest[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ChangeEvaluatorScoreRequest[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -17460,10 +17619,10 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListAnnotationEvaluatorsRequest[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ChangeEvaluatorScoreRequest[fieldId])) } -func (p *ListAnnotationEvaluatorsRequest) ReadField1(iprot thrift.TProtocol) error { +func (p *ChangeEvaluatorScoreRequest) ReadField1(iprot thrift.TProtocol) error { var _field int64 if v, err := iprot.ReadI64(); err != nil { @@ -17474,18 +17633,59 @@ func (p *ListAnnotationEvaluatorsRequest) ReadField1(iprot thrift.TProtocol) err p.WorkspaceID = _field return nil } -func (p *ListAnnotationEvaluatorsRequest) ReadField2(iprot thrift.TProtocol) error { +func (p *ChangeEvaluatorScoreRequest) ReadField2(iprot thrift.TProtocol) error { - var _field *string + var _field string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v + } + p.AnnotationID = _field + return nil +} +func (p *ChangeEvaluatorScoreRequest) ReadField3(iprot thrift.TProtocol) error { + + var _field string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v + } + p.SpanID = _field + return nil +} +func (p *ChangeEvaluatorScoreRequest) ReadField4(iprot thrift.TProtocol) error { + + var _field int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = v + } + p.StartTime = _field + return nil +} +func (p *ChangeEvaluatorScoreRequest) ReadField5(iprot thrift.TProtocol) error { + _field := annotation.NewCorrection() + if err := _field.Read(iprot); err != nil { + return err + } + p.Correction = _field + return nil +} +func (p *ChangeEvaluatorScoreRequest) ReadField6(iprot thrift.TProtocol) error { + + var _field *common.PlatformType if v, err := iprot.ReadString(); err != nil { return err } else { _field = &v } - p.Name = _field + p.PlatformType = _field return nil } -func (p *ListAnnotationEvaluatorsRequest) ReadField255(iprot thrift.TProtocol) error { +func (p *ChangeEvaluatorScoreRequest) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBase() if err := _field.Read(iprot); err != nil { return err @@ -17494,9 +17694,9 @@ func (p *ListAnnotationEvaluatorsRequest) ReadField255(iprot thrift.TProtocol) e return nil } -func (p *ListAnnotationEvaluatorsRequest) Write(oprot thrift.TProtocol) (err error) { +func (p *ChangeEvaluatorScoreRequest) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListAnnotationEvaluatorsRequest"); err != nil { + if err = oprot.WriteStructBegin("ChangeEvaluatorScoreRequest"); err != nil { goto WriteStructBeginError } if p != nil { @@ -17508,6 +17708,22 @@ func (p *ListAnnotationEvaluatorsRequest) Write(oprot thrift.TProtocol) (err err fieldId = 2 goto WriteFieldError } + if err = p.writeField3(oprot); err != nil { + fieldId = 3 + goto WriteFieldError + } + if err = p.writeField4(oprot); err != nil { + fieldId = 4 + goto WriteFieldError + } + if err = p.writeField5(oprot); err != nil { + fieldId = 5 + goto WriteFieldError + } + if err = p.writeField6(oprot); err != nil { + fieldId = 6 + goto WriteFieldError + } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -17530,7 +17746,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ListAnnotationEvaluatorsRequest) writeField1(oprot thrift.TProtocol) (err error) { +func (p *ChangeEvaluatorScoreRequest) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { goto WriteFieldBeginError } @@ -17546,12 +17762,76 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ListAnnotationEvaluatorsRequest) writeField2(oprot thrift.TProtocol) (err error) { - if p.IsSetName() { - if err = oprot.WriteFieldBegin("name", thrift.STRING, 2); err != nil { +func (p *ChangeEvaluatorScoreRequest) writeField2(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("annotation_id", thrift.STRING, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(p.AnnotationID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} +func (p *ChangeEvaluatorScoreRequest) writeField3(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("span_id", thrift.STRING, 3); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(p.SpanID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) +} +func (p *ChangeEvaluatorScoreRequest) writeField4(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("start_time", thrift.I64, 4); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(p.StartTime); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) +} +func (p *ChangeEvaluatorScoreRequest) writeField5(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("correction", thrift.STRUCT, 5); err != nil { + goto WriteFieldBeginError + } + if err := p.Correction.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) +} +func (p *ChangeEvaluatorScoreRequest) writeField6(oprot thrift.TProtocol) (err error) { + if p.IsSetPlatformType() { + if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 6); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteString(*p.Name); err != nil { + if err := oprot.WriteString(*p.PlatformType); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -17560,11 +17840,11 @@ func (p *ListAnnotationEvaluatorsRequest) writeField2(oprot thrift.TProtocol) (e } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) } -func (p *ListAnnotationEvaluatorsRequest) writeField255(oprot thrift.TProtocol) (err error) { +func (p *ChangeEvaluatorScoreRequest) writeField255(oprot thrift.TProtocol) (err error) { if p.IsSetBase() { if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError @@ -17583,15 +17863,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *ListAnnotationEvaluatorsRequest) String() string { +func (p *ChangeEvaluatorScoreRequest) String() string { if p == nil { return "" } - return fmt.Sprintf("ListAnnotationEvaluatorsRequest(%+v)", *p) + return fmt.Sprintf("ChangeEvaluatorScoreRequest(%+v)", *p) } -func (p *ListAnnotationEvaluatorsRequest) DeepEqual(ano *ListAnnotationEvaluatorsRequest) bool { +func (p *ChangeEvaluatorScoreRequest) DeepEqual(ano *ChangeEvaluatorScoreRequest) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -17600,7 +17880,19 @@ func (p *ListAnnotationEvaluatorsRequest) DeepEqual(ano *ListAnnotationEvaluator if !p.Field1DeepEqual(ano.WorkspaceID) { return false } - if !p.Field2DeepEqual(ano.Name) { + if !p.Field2DeepEqual(ano.AnnotationID) { + return false + } + if !p.Field3DeepEqual(ano.SpanID) { + return false + } + if !p.Field4DeepEqual(ano.StartTime) { + return false + } + if !p.Field5DeepEqual(ano.Correction) { + return false + } + if !p.Field6DeepEqual(ano.PlatformType) { return false } if !p.Field255DeepEqual(ano.Base) { @@ -17609,83 +17901,120 @@ func (p *ListAnnotationEvaluatorsRequest) DeepEqual(ano *ListAnnotationEvaluator return true } -func (p *ListAnnotationEvaluatorsRequest) Field1DeepEqual(src int64) bool { +func (p *ChangeEvaluatorScoreRequest) Field1DeepEqual(src int64) bool { if p.WorkspaceID != src { return false } return true } -func (p *ListAnnotationEvaluatorsRequest) Field2DeepEqual(src *string) bool { +func (p *ChangeEvaluatorScoreRequest) Field2DeepEqual(src string) bool { - if p.Name == src { - return true - } else if p.Name == nil || src == nil { + if strings.Compare(p.AnnotationID, src) != 0 { return false } - if strings.Compare(*p.Name, *src) != 0 { + return true +} +func (p *ChangeEvaluatorScoreRequest) Field3DeepEqual(src string) bool { + + if strings.Compare(p.SpanID, src) != 0 { return false } return true } -func (p *ListAnnotationEvaluatorsRequest) Field255DeepEqual(src *base.Base) bool { +func (p *ChangeEvaluatorScoreRequest) Field4DeepEqual(src int64) bool { - if !p.Base.DeepEqual(src) { + if p.StartTime != src { return false } return true } +func (p *ChangeEvaluatorScoreRequest) Field5DeepEqual(src *annotation.Correction) bool { -type ListAnnotationEvaluatorsResponse struct { - Evaluators []*annotation.AnnotationEvaluator `thrift:"evaluators,1,required" frugal:"1,required,list" form:"evaluators,required" json:"evaluators,required" query:"evaluators,required"` - BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` + if !p.Correction.DeepEqual(src) { + return false + } + return true } +func (p *ChangeEvaluatorScoreRequest) Field6DeepEqual(src *common.PlatformType) bool { -func NewListAnnotationEvaluatorsResponse() *ListAnnotationEvaluatorsResponse { - return &ListAnnotationEvaluatorsResponse{} + if p.PlatformType == src { + return true + } else if p.PlatformType == nil || src == nil { + return false + } + if strings.Compare(*p.PlatformType, *src) != 0 { + return false + } + return true } +func (p *ChangeEvaluatorScoreRequest) Field255DeepEqual(src *base.Base) bool { -func (p *ListAnnotationEvaluatorsResponse) InitDefault() { + if !p.Base.DeepEqual(src) { + return false + } + return true } -func (p *ListAnnotationEvaluatorsResponse) GetEvaluators() (v []*annotation.AnnotationEvaluator) { - if p != nil { - return p.Evaluators - } - return +type ChangeEvaluatorScoreResponse struct { + Annotation *annotation.Annotation `thrift:"annotation,1,required" frugal:"1,required,annotation.Annotation" form:"annotation,required" json:"annotation,required" query:"annotation,required"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` } -var ListAnnotationEvaluatorsResponse_BaseResp_DEFAULT *base.BaseResp +func NewChangeEvaluatorScoreResponse() *ChangeEvaluatorScoreResponse { + return &ChangeEvaluatorScoreResponse{} +} -func (p *ListAnnotationEvaluatorsResponse) GetBaseResp() (v *base.BaseResp) { - if p == nil { - return - } +func (p *ChangeEvaluatorScoreResponse) InitDefault() { +} + +var ChangeEvaluatorScoreResponse_Annotation_DEFAULT *annotation.Annotation + +func (p *ChangeEvaluatorScoreResponse) GetAnnotation() (v *annotation.Annotation) { + if p == nil { + return + } + if !p.IsSetAnnotation() { + return ChangeEvaluatorScoreResponse_Annotation_DEFAULT + } + return p.Annotation +} + +var ChangeEvaluatorScoreResponse_BaseResp_DEFAULT *base.BaseResp + +func (p *ChangeEvaluatorScoreResponse) GetBaseResp() (v *base.BaseResp) { + if p == nil { + return + } if !p.IsSetBaseResp() { - return ListAnnotationEvaluatorsResponse_BaseResp_DEFAULT + return ChangeEvaluatorScoreResponse_BaseResp_DEFAULT } return p.BaseResp } -func (p *ListAnnotationEvaluatorsResponse) SetEvaluators(val []*annotation.AnnotationEvaluator) { - p.Evaluators = val +func (p *ChangeEvaluatorScoreResponse) SetAnnotation(val *annotation.Annotation) { + p.Annotation = val } -func (p *ListAnnotationEvaluatorsResponse) SetBaseResp(val *base.BaseResp) { +func (p *ChangeEvaluatorScoreResponse) SetBaseResp(val *base.BaseResp) { p.BaseResp = val } -var fieldIDToName_ListAnnotationEvaluatorsResponse = map[int16]string{ - 1: "evaluators", +var fieldIDToName_ChangeEvaluatorScoreResponse = map[int16]string{ + 1: "annotation", 255: "BaseResp", } -func (p *ListAnnotationEvaluatorsResponse) IsSetBaseResp() bool { +func (p *ChangeEvaluatorScoreResponse) IsSetAnnotation() bool { + return p.Annotation != nil +} + +func (p *ChangeEvaluatorScoreResponse) IsSetBaseResp() bool { return p.BaseResp != nil } -func (p *ListAnnotationEvaluatorsResponse) Read(iprot thrift.TProtocol) (err error) { +func (p *ChangeEvaluatorScoreResponse) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 - var issetEvaluators bool = false + var issetAnnotation bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -17702,11 +18031,11 @@ func (p *ListAnnotationEvaluatorsResponse) Read(iprot thrift.TProtocol) (err err switch fieldId { case 1: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.STRUCT { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } - issetEvaluators = true + issetAnnotation = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -17731,7 +18060,7 @@ func (p *ListAnnotationEvaluatorsResponse) Read(iprot thrift.TProtocol) (err err goto ReadStructEndError } - if !issetEvaluators { + if !issetAnnotation { fieldId = 1 goto RequiredFieldNotSetError } @@ -17741,7 +18070,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListAnnotationEvaluatorsResponse[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ChangeEvaluatorScoreResponse[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -17750,33 +18079,18 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListAnnotationEvaluatorsResponse[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ChangeEvaluatorScoreResponse[fieldId])) } -func (p *ListAnnotationEvaluatorsResponse) ReadField1(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]*annotation.AnnotationEvaluator, 0, size) - values := make([]annotation.AnnotationEvaluator, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { +func (p *ChangeEvaluatorScoreResponse) ReadField1(iprot thrift.TProtocol) error { + _field := annotation.NewAnnotation() + if err := _field.Read(iprot); err != nil { return err } - p.Evaluators = _field + p.Annotation = _field return nil } -func (p *ListAnnotationEvaluatorsResponse) ReadField255(iprot thrift.TProtocol) error { +func (p *ChangeEvaluatorScoreResponse) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBaseResp() if err := _field.Read(iprot); err != nil { return err @@ -17785,9 +18099,9 @@ func (p *ListAnnotationEvaluatorsResponse) ReadField255(iprot thrift.TProtocol) return nil } -func (p *ListAnnotationEvaluatorsResponse) Write(oprot thrift.TProtocol) (err error) { +func (p *ChangeEvaluatorScoreResponse) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListAnnotationEvaluatorsResponse"); err != nil { + if err = oprot.WriteStructBegin("ChangeEvaluatorScoreResponse"); err != nil { goto WriteStructBeginError } if p != nil { @@ -17817,19 +18131,11 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ListAnnotationEvaluatorsResponse) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("evaluators", thrift.LIST, 1); err != nil { +func (p *ChangeEvaluatorScoreResponse) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("annotation", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.Evaluators)); err != nil { - return err - } - for _, v := range p.Evaluators { - if err := v.Write(oprot); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { + if err := p.Annotation.Write(oprot); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -17841,7 +18147,7 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ListAnnotationEvaluatorsResponse) writeField255(oprot thrift.TProtocol) (err error) { +func (p *ChangeEvaluatorScoreResponse) writeField255(oprot thrift.TProtocol) (err error) { if p.IsSetBaseResp() { if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError @@ -17860,21 +18166,21 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *ListAnnotationEvaluatorsResponse) String() string { +func (p *ChangeEvaluatorScoreResponse) String() string { if p == nil { return "" } - return fmt.Sprintf("ListAnnotationEvaluatorsResponse(%+v)", *p) + return fmt.Sprintf("ChangeEvaluatorScoreResponse(%+v)", *p) } -func (p *ListAnnotationEvaluatorsResponse) DeepEqual(ano *ListAnnotationEvaluatorsResponse) bool { +func (p *ChangeEvaluatorScoreResponse) DeepEqual(ano *ChangeEvaluatorScoreResponse) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.Evaluators) { + if !p.Field1DeepEqual(ano.Annotation) { return false } if !p.Field255DeepEqual(ano.BaseResp) { @@ -17883,20 +18189,14 @@ func (p *ListAnnotationEvaluatorsResponse) DeepEqual(ano *ListAnnotationEvaluato return true } -func (p *ListAnnotationEvaluatorsResponse) Field1DeepEqual(src []*annotation.AnnotationEvaluator) bool { +func (p *ChangeEvaluatorScoreResponse) Field1DeepEqual(src *annotation.Annotation) bool { - if len(p.Evaluators) != len(src) { + if !p.Annotation.DeepEqual(src) { return false } - for i, v := range p.Evaluators { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } return true } -func (p *ListAnnotationEvaluatorsResponse) Field255DeepEqual(src *base.BaseResp) bool { +func (p *ChangeEvaluatorScoreResponse) Field255DeepEqual(src *base.BaseResp) bool { if !p.BaseResp.DeepEqual(src) { return false @@ -17904,166 +18204,77 @@ func (p *ListAnnotationEvaluatorsResponse) Field255DeepEqual(src *base.BaseResp) return true } -type ExtractSpanInfoRequest struct { - WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` - TraceID string `thrift:"trace_id,2,required" frugal:"2,required,string" form:"trace_id,required" json:"trace_id,required"` - SpanIds []string `thrift:"span_ids,3,required" frugal:"3,required,list" form:"span_ids,required" json:"span_ids,required"` - StartTime *int64 `thrift:"start_time,4,optional" frugal:"4,optional,i64" json:"start_time" form:"start_time" ` - EndTime *int64 `thrift:"end_time,5,optional" frugal:"5,optional,i64" json:"end_time" form:"end_time" ` - PlatformType *common.PlatformType `thrift:"platform_type,6,optional" frugal:"6,optional,string" form:"platform_type" json:"platform_type,omitempty"` - FieldMappings []*dataset0.FieldMapping `thrift:"field_mappings,7,optional" frugal:"7,optional,list" form:"field_mappings" json:"field_mappings,omitempty" query:"field_mappings"` - Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"-" json:"-" query:"-"` +type ListAnnotationEvaluatorsRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" query:"workspace_id,required" ` + Name *string `thrift:"name,2,optional" frugal:"2,optional,string" json:"name,omitempty" query:"name"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"-" json:"-" query:"-"` } -func NewExtractSpanInfoRequest() *ExtractSpanInfoRequest { - return &ExtractSpanInfoRequest{} +func NewListAnnotationEvaluatorsRequest() *ListAnnotationEvaluatorsRequest { + return &ListAnnotationEvaluatorsRequest{} } -func (p *ExtractSpanInfoRequest) InitDefault() { +func (p *ListAnnotationEvaluatorsRequest) InitDefault() { } -func (p *ExtractSpanInfoRequest) GetWorkspaceID() (v int64) { +func (p *ListAnnotationEvaluatorsRequest) GetWorkspaceID() (v int64) { if p != nil { return p.WorkspaceID } return } -func (p *ExtractSpanInfoRequest) GetTraceID() (v string) { - if p != nil { - return p.TraceID - } - return -} - -func (p *ExtractSpanInfoRequest) GetSpanIds() (v []string) { - if p != nil { - return p.SpanIds - } - return -} - -var ExtractSpanInfoRequest_StartTime_DEFAULT int64 - -func (p *ExtractSpanInfoRequest) GetStartTime() (v int64) { - if p == nil { - return - } - if !p.IsSetStartTime() { - return ExtractSpanInfoRequest_StartTime_DEFAULT - } - return *p.StartTime -} - -var ExtractSpanInfoRequest_EndTime_DEFAULT int64 - -func (p *ExtractSpanInfoRequest) GetEndTime() (v int64) { - if p == nil { - return - } - if !p.IsSetEndTime() { - return ExtractSpanInfoRequest_EndTime_DEFAULT - } - return *p.EndTime -} - -var ExtractSpanInfoRequest_PlatformType_DEFAULT common.PlatformType - -func (p *ExtractSpanInfoRequest) GetPlatformType() (v common.PlatformType) { - if p == nil { - return - } - if !p.IsSetPlatformType() { - return ExtractSpanInfoRequest_PlatformType_DEFAULT - } - return *p.PlatformType -} - -var ExtractSpanInfoRequest_FieldMappings_DEFAULT []*dataset0.FieldMapping +var ListAnnotationEvaluatorsRequest_Name_DEFAULT string -func (p *ExtractSpanInfoRequest) GetFieldMappings() (v []*dataset0.FieldMapping) { +func (p *ListAnnotationEvaluatorsRequest) GetName() (v string) { if p == nil { return } - if !p.IsSetFieldMappings() { - return ExtractSpanInfoRequest_FieldMappings_DEFAULT + if !p.IsSetName() { + return ListAnnotationEvaluatorsRequest_Name_DEFAULT } - return p.FieldMappings + return *p.Name } -var ExtractSpanInfoRequest_Base_DEFAULT *base.Base +var ListAnnotationEvaluatorsRequest_Base_DEFAULT *base.Base -func (p *ExtractSpanInfoRequest) GetBase() (v *base.Base) { +func (p *ListAnnotationEvaluatorsRequest) GetBase() (v *base.Base) { if p == nil { return } if !p.IsSetBase() { - return ExtractSpanInfoRequest_Base_DEFAULT + return ListAnnotationEvaluatorsRequest_Base_DEFAULT } return p.Base } -func (p *ExtractSpanInfoRequest) SetWorkspaceID(val int64) { +func (p *ListAnnotationEvaluatorsRequest) SetWorkspaceID(val int64) { p.WorkspaceID = val } -func (p *ExtractSpanInfoRequest) SetTraceID(val string) { - p.TraceID = val -} -func (p *ExtractSpanInfoRequest) SetSpanIds(val []string) { - p.SpanIds = val -} -func (p *ExtractSpanInfoRequest) SetStartTime(val *int64) { - p.StartTime = val -} -func (p *ExtractSpanInfoRequest) SetEndTime(val *int64) { - p.EndTime = val -} -func (p *ExtractSpanInfoRequest) SetPlatformType(val *common.PlatformType) { - p.PlatformType = val -} -func (p *ExtractSpanInfoRequest) SetFieldMappings(val []*dataset0.FieldMapping) { - p.FieldMappings = val +func (p *ListAnnotationEvaluatorsRequest) SetName(val *string) { + p.Name = val } -func (p *ExtractSpanInfoRequest) SetBase(val *base.Base) { +func (p *ListAnnotationEvaluatorsRequest) SetBase(val *base.Base) { p.Base = val } -var fieldIDToName_ExtractSpanInfoRequest = map[int16]string{ +var fieldIDToName_ListAnnotationEvaluatorsRequest = map[int16]string{ 1: "workspace_id", - 2: "trace_id", - 3: "span_ids", - 4: "start_time", - 5: "end_time", - 6: "platform_type", - 7: "field_mappings", + 2: "name", 255: "Base", } -func (p *ExtractSpanInfoRequest) IsSetStartTime() bool { - return p.StartTime != nil -} - -func (p *ExtractSpanInfoRequest) IsSetEndTime() bool { - return p.EndTime != nil -} - -func (p *ExtractSpanInfoRequest) IsSetPlatformType() bool { - return p.PlatformType != nil -} - -func (p *ExtractSpanInfoRequest) IsSetFieldMappings() bool { - return p.FieldMappings != nil +func (p *ListAnnotationEvaluatorsRequest) IsSetName() bool { + return p.Name != nil } -func (p *ExtractSpanInfoRequest) IsSetBase() bool { +func (p *ListAnnotationEvaluatorsRequest) IsSetBase() bool { return p.Base != nil } -func (p *ExtractSpanInfoRequest) Read(iprot thrift.TProtocol) (err error) { +func (p *ListAnnotationEvaluatorsRequest) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 var issetWorkspaceID bool = false - var issetTraceID bool = false - var issetSpanIds bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -18093,48 +18304,6 @@ func (p *ExtractSpanInfoRequest) Read(iprot thrift.TProtocol) (err error) { if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } - issetTraceID = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 3: - if fieldTypeId == thrift.LIST { - if err = p.ReadField3(iprot); err != nil { - goto ReadFieldError - } - issetSpanIds = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 4: - if fieldTypeId == thrift.I64 { - if err = p.ReadField4(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 5: - if fieldTypeId == thrift.I64 { - if err = p.ReadField5(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 6: - if fieldTypeId == thrift.STRING { - if err = p.ReadField6(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 7: - if fieldTypeId == thrift.LIST { - if err = p.ReadField7(iprot); err != nil { - goto ReadFieldError - } } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -18163,23 +18332,13 @@ func (p *ExtractSpanInfoRequest) Read(iprot thrift.TProtocol) (err error) { fieldId = 1 goto RequiredFieldNotSetError } - - if !issetTraceID { - fieldId = 2 - goto RequiredFieldNotSetError - } - - if !issetSpanIds { - fieldId = 3 - goto RequiredFieldNotSetError - } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExtractSpanInfoRequest[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListAnnotationEvaluatorsRequest[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -18188,10 +18347,10 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ExtractSpanInfoRequest[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListAnnotationEvaluatorsRequest[fieldId])) } -func (p *ExtractSpanInfoRequest) ReadField1(iprot thrift.TProtocol) error { +func (p *ListAnnotationEvaluatorsRequest) ReadField1(iprot thrift.TProtocol) error { var _field int64 if v, err := iprot.ReadI64(); err != nil { @@ -18202,97 +18361,18 @@ func (p *ExtractSpanInfoRequest) ReadField1(iprot thrift.TProtocol) error { p.WorkspaceID = _field return nil } -func (p *ExtractSpanInfoRequest) ReadField2(iprot thrift.TProtocol) error { - - var _field string - if v, err := iprot.ReadString(); err != nil { - return err - } else { - _field = v - } - p.TraceID = _field - return nil -} -func (p *ExtractSpanInfoRequest) ReadField3(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]string, 0, size) - for i := 0; i < size; i++ { - - var _elem string - if v, err := iprot.ReadString(); err != nil { - return err - } else { - _elem = v - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { - return err - } - p.SpanIds = _field - return nil -} -func (p *ExtractSpanInfoRequest) ReadField4(iprot thrift.TProtocol) error { - - var _field *int64 - if v, err := iprot.ReadI64(); err != nil { - return err - } else { - _field = &v - } - p.StartTime = _field - return nil -} -func (p *ExtractSpanInfoRequest) ReadField5(iprot thrift.TProtocol) error { - - var _field *int64 - if v, err := iprot.ReadI64(); err != nil { - return err - } else { - _field = &v - } - p.EndTime = _field - return nil -} -func (p *ExtractSpanInfoRequest) ReadField6(iprot thrift.TProtocol) error { +func (p *ListAnnotationEvaluatorsRequest) ReadField2(iprot thrift.TProtocol) error { - var _field *common.PlatformType + var _field *string if v, err := iprot.ReadString(); err != nil { return err } else { _field = &v } - p.PlatformType = _field - return nil -} -func (p *ExtractSpanInfoRequest) ReadField7(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]*dataset0.FieldMapping, 0, size) - values := make([]dataset0.FieldMapping, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { - return err - } - p.FieldMappings = _field + p.Name = _field return nil } -func (p *ExtractSpanInfoRequest) ReadField255(iprot thrift.TProtocol) error { +func (p *ListAnnotationEvaluatorsRequest) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBase() if err := _field.Read(iprot); err != nil { return err @@ -18301,9 +18381,9 @@ func (p *ExtractSpanInfoRequest) ReadField255(iprot thrift.TProtocol) error { return nil } -func (p *ExtractSpanInfoRequest) Write(oprot thrift.TProtocol) (err error) { +func (p *ListAnnotationEvaluatorsRequest) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ExtractSpanInfoRequest"); err != nil { + if err = oprot.WriteStructBegin("ListAnnotationEvaluatorsRequest"); err != nil { goto WriteStructBeginError } if p != nil { @@ -18315,26 +18395,6 @@ func (p *ExtractSpanInfoRequest) Write(oprot thrift.TProtocol) (err error) { fieldId = 2 goto WriteFieldError } - if err = p.writeField3(oprot); err != nil { - fieldId = 3 - goto WriteFieldError - } - if err = p.writeField4(oprot); err != nil { - fieldId = 4 - goto WriteFieldError - } - if err = p.writeField5(oprot); err != nil { - fieldId = 5 - goto WriteFieldError - } - if err = p.writeField6(oprot); err != nil { - fieldId = 6 - goto WriteFieldError - } - if err = p.writeField7(oprot); err != nil { - fieldId = 7 - goto WriteFieldError - } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -18357,7 +18417,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ExtractSpanInfoRequest) writeField1(oprot thrift.TProtocol) (err error) { +func (p *ListAnnotationEvaluatorsRequest) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { goto WriteFieldBeginError } @@ -18373,52 +18433,12 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ExtractSpanInfoRequest) writeField2(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("trace_id", thrift.STRING, 2); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(p.TraceID); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) -} -func (p *ExtractSpanInfoRequest) writeField3(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("span_ids", thrift.LIST, 3); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteListBegin(thrift.STRING, len(p.SpanIds)); err != nil { - return err - } - for _, v := range p.SpanIds { - if err := oprot.WriteString(v); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) -} -func (p *ExtractSpanInfoRequest) writeField4(oprot thrift.TProtocol) (err error) { - if p.IsSetStartTime() { - if err = oprot.WriteFieldBegin("start_time", thrift.I64, 4); err != nil { +func (p *ListAnnotationEvaluatorsRequest) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetName() { + if err = oprot.WriteFieldBegin("name", thrift.STRING, 2); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteI64(*p.StartTime); err != nil { + if err := oprot.WriteString(*p.Name); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -18427,16 +18447,16 @@ func (p *ExtractSpanInfoRequest) writeField4(oprot thrift.TProtocol) (err error) } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *ExtractSpanInfoRequest) writeField5(oprot thrift.TProtocol) (err error) { - if p.IsSetEndTime() { - if err = oprot.WriteFieldBegin("end_time", thrift.I64, 5); err != nil { +func (p *ListAnnotationEvaluatorsRequest) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBase() { + if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteI64(*p.EndTime); err != nil { + if err := p.Base.Write(oprot); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -18445,82 +18465,20 @@ func (p *ExtractSpanInfoRequest) writeField5(oprot thrift.TProtocol) (err error) } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *ExtractSpanInfoRequest) writeField6(oprot thrift.TProtocol) (err error) { - if p.IsSetPlatformType() { - if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 6); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteString(*p.PlatformType); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } + +func (p *ListAnnotationEvaluatorsRequest) String() string { + if p == nil { + return "" } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) -} -func (p *ExtractSpanInfoRequest) writeField7(oprot thrift.TProtocol) (err error) { - if p.IsSetFieldMappings() { - if err = oprot.WriteFieldBegin("field_mappings", thrift.LIST, 7); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FieldMappings)); err != nil { - return err - } - for _, v := range p.FieldMappings { - if err := v.Write(oprot); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 7 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 7 end error: ", p), err) -} -func (p *ExtractSpanInfoRequest) writeField255(oprot thrift.TProtocol) (err error) { - if p.IsSetBase() { - if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { - goto WriteFieldBeginError - } - if err := p.Base.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) -} - -func (p *ExtractSpanInfoRequest) String() string { - if p == nil { - return "" - } - return fmt.Sprintf("ExtractSpanInfoRequest(%+v)", *p) + return fmt.Sprintf("ListAnnotationEvaluatorsRequest(%+v)", *p) } -func (p *ExtractSpanInfoRequest) DeepEqual(ano *ExtractSpanInfoRequest) bool { +func (p *ListAnnotationEvaluatorsRequest) DeepEqual(ano *ListAnnotationEvaluatorsRequest) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -18529,22 +18487,7 @@ func (p *ExtractSpanInfoRequest) DeepEqual(ano *ExtractSpanInfoRequest) bool { if !p.Field1DeepEqual(ano.WorkspaceID) { return false } - if !p.Field2DeepEqual(ano.TraceID) { - return false - } - if !p.Field3DeepEqual(ano.SpanIds) { - return false - } - if !p.Field4DeepEqual(ano.StartTime) { - return false - } - if !p.Field5DeepEqual(ano.EndTime) { - return false - } - if !p.Field6DeepEqual(ano.PlatformType) { - return false - } - if !p.Field7DeepEqual(ano.FieldMappings) { + if !p.Field2DeepEqual(ano.Name) { return false } if !p.Field255DeepEqual(ano.Base) { @@ -18553,83 +18496,26 @@ func (p *ExtractSpanInfoRequest) DeepEqual(ano *ExtractSpanInfoRequest) bool { return true } -func (p *ExtractSpanInfoRequest) Field1DeepEqual(src int64) bool { +func (p *ListAnnotationEvaluatorsRequest) Field1DeepEqual(src int64) bool { if p.WorkspaceID != src { return false } return true } -func (p *ExtractSpanInfoRequest) Field2DeepEqual(src string) bool { - - if strings.Compare(p.TraceID, src) != 0 { - return false - } - return true -} -func (p *ExtractSpanInfoRequest) Field3DeepEqual(src []string) bool { - - if len(p.SpanIds) != len(src) { - return false - } - for i, v := range p.SpanIds { - _src := src[i] - if strings.Compare(v, _src) != 0 { - return false - } - } - return true -} -func (p *ExtractSpanInfoRequest) Field4DeepEqual(src *int64) bool { - - if p.StartTime == src { - return true - } else if p.StartTime == nil || src == nil { - return false - } - if *p.StartTime != *src { - return false - } - return true -} -func (p *ExtractSpanInfoRequest) Field5DeepEqual(src *int64) bool { - - if p.EndTime == src { - return true - } else if p.EndTime == nil || src == nil { - return false - } - if *p.EndTime != *src { - return false - } - return true -} -func (p *ExtractSpanInfoRequest) Field6DeepEqual(src *common.PlatformType) bool { +func (p *ListAnnotationEvaluatorsRequest) Field2DeepEqual(src *string) bool { - if p.PlatformType == src { + if p.Name == src { return true - } else if p.PlatformType == nil || src == nil { - return false - } - if strings.Compare(*p.PlatformType, *src) != 0 { + } else if p.Name == nil || src == nil { return false } - return true -} -func (p *ExtractSpanInfoRequest) Field7DeepEqual(src []*dataset0.FieldMapping) bool { - - if len(p.FieldMappings) != len(src) { + if strings.Compare(*p.Name, *src) != 0 { return false } - for i, v := range p.FieldMappings { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } return true } -func (p *ExtractSpanInfoRequest) Field255DeepEqual(src *base.Base) bool { +func (p *ListAnnotationEvaluatorsRequest) Field255DeepEqual(src *base.Base) bool { if !p.Base.DeepEqual(src) { return false @@ -18637,48 +18523,56 @@ func (p *ExtractSpanInfoRequest) Field255DeepEqual(src *base.Base) bool { return true } -type SpanInfo struct { - SpanID string `thrift:"span_id,1,required" frugal:"1,required,string" form:"span_id,required" json:"span_id,required" query:"span_id,required"` - FieldList []*dataset0.FieldData `thrift:"field_list,2,required" frugal:"2,required,list" form:"field_list,required" json:"field_list,required" query:"field_list,required"` +type ListAnnotationEvaluatorsResponse struct { + Evaluators []*annotation.AnnotationEvaluator `thrift:"evaluators,1,required" frugal:"1,required,list" form:"evaluators,required" json:"evaluators,required" query:"evaluators,required"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` } -func NewSpanInfo() *SpanInfo { - return &SpanInfo{} +func NewListAnnotationEvaluatorsResponse() *ListAnnotationEvaluatorsResponse { + return &ListAnnotationEvaluatorsResponse{} } -func (p *SpanInfo) InitDefault() { +func (p *ListAnnotationEvaluatorsResponse) InitDefault() { } -func (p *SpanInfo) GetSpanID() (v string) { +func (p *ListAnnotationEvaluatorsResponse) GetEvaluators() (v []*annotation.AnnotationEvaluator) { if p != nil { - return p.SpanID + return p.Evaluators } return } -func (p *SpanInfo) GetFieldList() (v []*dataset0.FieldData) { - if p != nil { - return p.FieldList +var ListAnnotationEvaluatorsResponse_BaseResp_DEFAULT *base.BaseResp + +func (p *ListAnnotationEvaluatorsResponse) GetBaseResp() (v *base.BaseResp) { + if p == nil { + return } - return + if !p.IsSetBaseResp() { + return ListAnnotationEvaluatorsResponse_BaseResp_DEFAULT + } + return p.BaseResp } -func (p *SpanInfo) SetSpanID(val string) { - p.SpanID = val +func (p *ListAnnotationEvaluatorsResponse) SetEvaluators(val []*annotation.AnnotationEvaluator) { + p.Evaluators = val } -func (p *SpanInfo) SetFieldList(val []*dataset0.FieldData) { - p.FieldList = val +func (p *ListAnnotationEvaluatorsResponse) SetBaseResp(val *base.BaseResp) { + p.BaseResp = val } -var fieldIDToName_SpanInfo = map[int16]string{ - 1: "span_id", - 2: "field_list", +var fieldIDToName_ListAnnotationEvaluatorsResponse = map[int16]string{ + 1: "evaluators", + 255: "BaseResp", } -func (p *SpanInfo) Read(iprot thrift.TProtocol) (err error) { +func (p *ListAnnotationEvaluatorsResponse) IsSetBaseResp() bool { + return p.BaseResp != nil +} + +func (p *ListAnnotationEvaluatorsResponse) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 - var issetSpanID bool = false - var issetFieldList bool = false + var issetEvaluators bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -18695,20 +18589,19 @@ func (p *SpanInfo) Read(iprot thrift.TProtocol) (err error) { switch fieldId { case 1: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.LIST { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } - issetSpanID = true + issetEvaluators = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - case 2: - if fieldTypeId == thrift.LIST { - if err = p.ReadField2(iprot); err != nil { + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { goto ReadFieldError } - issetFieldList = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -18725,22 +18618,17 @@ func (p *SpanInfo) Read(iprot thrift.TProtocol) (err error) { goto ReadStructEndError } - if !issetSpanID { + if !issetEvaluators { fieldId = 1 goto RequiredFieldNotSetError } - - if !issetFieldList { - fieldId = 2 - goto RequiredFieldNotSetError - } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SpanInfo[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListAnnotationEvaluatorsResponse[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -18749,27 +18637,16 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_SpanInfo[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListAnnotationEvaluatorsResponse[fieldId])) } -func (p *SpanInfo) ReadField1(iprot thrift.TProtocol) error { - - var _field string - if v, err := iprot.ReadString(); err != nil { - return err - } else { - _field = v - } - p.SpanID = _field - return nil -} -func (p *SpanInfo) ReadField2(iprot thrift.TProtocol) error { +func (p *ListAnnotationEvaluatorsResponse) ReadField1(iprot thrift.TProtocol) error { _, size, err := iprot.ReadListBegin() if err != nil { return err } - _field := make([]*dataset0.FieldData, 0, size) - values := make([]dataset0.FieldData, size) + _field := make([]*annotation.AnnotationEvaluator, 0, size) + values := make([]annotation.AnnotationEvaluator, size) for i := 0; i < size; i++ { _elem := &values[i] _elem.InitDefault() @@ -18783,13 +18660,21 @@ func (p *SpanInfo) ReadField2(iprot thrift.TProtocol) error { if err := iprot.ReadListEnd(); err != nil { return err } - p.FieldList = _field + p.Evaluators = _field + return nil +} +func (p *ListAnnotationEvaluatorsResponse) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBaseResp() + if err := _field.Read(iprot); err != nil { + return err + } + p.BaseResp = _field return nil } -func (p *SpanInfo) Write(oprot thrift.TProtocol) (err error) { +func (p *ListAnnotationEvaluatorsResponse) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("SpanInfo"); err != nil { + if err = oprot.WriteStructBegin("ListAnnotationEvaluatorsResponse"); err != nil { goto WriteStructBeginError } if p != nil { @@ -18797,8 +18682,8 @@ func (p *SpanInfo) Write(oprot thrift.TProtocol) (err error) { fieldId = 1 goto WriteFieldError } - if err = p.writeField2(oprot); err != nil { - fieldId = 2 + if err = p.writeField255(oprot); err != nil { + fieldId = 255 goto WriteFieldError } } @@ -18819,11 +18704,19 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *SpanInfo) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("span_id", thrift.STRING, 1); err != nil { +func (p *ListAnnotationEvaluatorsResponse) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("evaluators", thrift.LIST, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteString(p.SpanID); err != nil { + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.Evaluators)); err != nil { + return err + } + for _, v := range p.Evaluators { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -18835,67 +18728,54 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *SpanInfo) writeField2(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("field_list", thrift.LIST, 2); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FieldList)); err != nil { - return err - } - for _, v := range p.FieldList { - if err := v.Write(oprot); err != nil { +func (p *ListAnnotationEvaluatorsResponse) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBaseResp() { + if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.BaseResp.Write(oprot); err != nil { return err } - } - if err := oprot.WriteListEnd(); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *SpanInfo) String() string { +func (p *ListAnnotationEvaluatorsResponse) String() string { if p == nil { return "" } - return fmt.Sprintf("SpanInfo(%+v)", *p) + return fmt.Sprintf("ListAnnotationEvaluatorsResponse(%+v)", *p) } -func (p *SpanInfo) DeepEqual(ano *SpanInfo) bool { +func (p *ListAnnotationEvaluatorsResponse) DeepEqual(ano *ListAnnotationEvaluatorsResponse) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.SpanID) { + if !p.Field1DeepEqual(ano.Evaluators) { return false } - if !p.Field2DeepEqual(ano.FieldList) { + if !p.Field255DeepEqual(ano.BaseResp) { return false } return true } -func (p *SpanInfo) Field1DeepEqual(src string) bool { - - if strings.Compare(p.SpanID, src) != 0 { - return false - } - return true -} -func (p *SpanInfo) Field2DeepEqual(src []*dataset0.FieldData) bool { +func (p *ListAnnotationEvaluatorsResponse) Field1DeepEqual(src []*annotation.AnnotationEvaluator) bool { - if len(p.FieldList) != len(src) { + if len(p.Evaluators) != len(src) { return false } - for i, v := range p.FieldList { + for i, v := range p.Evaluators { _src := src[i] if !v.DeepEqual(_src) { return false @@ -18903,57 +18783,174 @@ func (p *SpanInfo) Field2DeepEqual(src []*dataset0.FieldData) bool { } return true } +func (p *ListAnnotationEvaluatorsResponse) Field255DeepEqual(src *base.BaseResp) bool { -type ExtractSpanInfoResponse struct { - SpanInfos []*SpanInfo `thrift:"span_infos,1,required" frugal:"1,required,list" form:"span_infos,required" json:"span_infos,required" query:"span_infos,required"` - BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` + if !p.BaseResp.DeepEqual(src) { + return false + } + return true } -func NewExtractSpanInfoResponse() *ExtractSpanInfoResponse { - return &ExtractSpanInfoResponse{} +type ExtractSpanInfoRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` + TraceID string `thrift:"trace_id,2,required" frugal:"2,required,string" form:"trace_id,required" json:"trace_id,required"` + SpanIds []string `thrift:"span_ids,3,required" frugal:"3,required,list" form:"span_ids,required" json:"span_ids,required"` + StartTime *int64 `thrift:"start_time,4,optional" frugal:"4,optional,i64" json:"start_time" form:"start_time" ` + EndTime *int64 `thrift:"end_time,5,optional" frugal:"5,optional,i64" json:"end_time" form:"end_time" ` + PlatformType *common.PlatformType `thrift:"platform_type,6,optional" frugal:"6,optional,string" form:"platform_type" json:"platform_type,omitempty"` + FieldMappings []*dataset0.FieldMapping `thrift:"field_mappings,7,optional" frugal:"7,optional,list" form:"field_mappings" json:"field_mappings,omitempty" query:"field_mappings"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"-" json:"-" query:"-"` } -func (p *ExtractSpanInfoResponse) InitDefault() { +func NewExtractSpanInfoRequest() *ExtractSpanInfoRequest { + return &ExtractSpanInfoRequest{} } -func (p *ExtractSpanInfoResponse) GetSpanInfos() (v []*SpanInfo) { +func (p *ExtractSpanInfoRequest) InitDefault() { +} + +func (p *ExtractSpanInfoRequest) GetWorkspaceID() (v int64) { if p != nil { - return p.SpanInfos + return p.WorkspaceID } return } -var ExtractSpanInfoResponse_BaseResp_DEFAULT *base.BaseResp +func (p *ExtractSpanInfoRequest) GetTraceID() (v string) { + if p != nil { + return p.TraceID + } + return +} -func (p *ExtractSpanInfoResponse) GetBaseResp() (v *base.BaseResp) { +func (p *ExtractSpanInfoRequest) GetSpanIds() (v []string) { + if p != nil { + return p.SpanIds + } + return +} + +var ExtractSpanInfoRequest_StartTime_DEFAULT int64 + +func (p *ExtractSpanInfoRequest) GetStartTime() (v int64) { if p == nil { return } - if !p.IsSetBaseResp() { - return ExtractSpanInfoResponse_BaseResp_DEFAULT + if !p.IsSetStartTime() { + return ExtractSpanInfoRequest_StartTime_DEFAULT } - return p.BaseResp + return *p.StartTime } -func (p *ExtractSpanInfoResponse) SetSpanInfos(val []*SpanInfo) { - p.SpanInfos = val + +var ExtractSpanInfoRequest_EndTime_DEFAULT int64 + +func (p *ExtractSpanInfoRequest) GetEndTime() (v int64) { + if p == nil { + return + } + if !p.IsSetEndTime() { + return ExtractSpanInfoRequest_EndTime_DEFAULT + } + return *p.EndTime } -func (p *ExtractSpanInfoResponse) SetBaseResp(val *base.BaseResp) { - p.BaseResp = val + +var ExtractSpanInfoRequest_PlatformType_DEFAULT common.PlatformType + +func (p *ExtractSpanInfoRequest) GetPlatformType() (v common.PlatformType) { + if p == nil { + return + } + if !p.IsSetPlatformType() { + return ExtractSpanInfoRequest_PlatformType_DEFAULT + } + return *p.PlatformType } -var fieldIDToName_ExtractSpanInfoResponse = map[int16]string{ - 1: "span_infos", - 255: "BaseResp", +var ExtractSpanInfoRequest_FieldMappings_DEFAULT []*dataset0.FieldMapping + +func (p *ExtractSpanInfoRequest) GetFieldMappings() (v []*dataset0.FieldMapping) { + if p == nil { + return + } + if !p.IsSetFieldMappings() { + return ExtractSpanInfoRequest_FieldMappings_DEFAULT + } + return p.FieldMappings } -func (p *ExtractSpanInfoResponse) IsSetBaseResp() bool { - return p.BaseResp != nil +var ExtractSpanInfoRequest_Base_DEFAULT *base.Base + +func (p *ExtractSpanInfoRequest) GetBase() (v *base.Base) { + if p == nil { + return + } + if !p.IsSetBase() { + return ExtractSpanInfoRequest_Base_DEFAULT + } + return p.Base +} +func (p *ExtractSpanInfoRequest) SetWorkspaceID(val int64) { + p.WorkspaceID = val +} +func (p *ExtractSpanInfoRequest) SetTraceID(val string) { + p.TraceID = val +} +func (p *ExtractSpanInfoRequest) SetSpanIds(val []string) { + p.SpanIds = val +} +func (p *ExtractSpanInfoRequest) SetStartTime(val *int64) { + p.StartTime = val +} +func (p *ExtractSpanInfoRequest) SetEndTime(val *int64) { + p.EndTime = val +} +func (p *ExtractSpanInfoRequest) SetPlatformType(val *common.PlatformType) { + p.PlatformType = val +} +func (p *ExtractSpanInfoRequest) SetFieldMappings(val []*dataset0.FieldMapping) { + p.FieldMappings = val +} +func (p *ExtractSpanInfoRequest) SetBase(val *base.Base) { + p.Base = val } -func (p *ExtractSpanInfoResponse) Read(iprot thrift.TProtocol) (err error) { +var fieldIDToName_ExtractSpanInfoRequest = map[int16]string{ + 1: "workspace_id", + 2: "trace_id", + 3: "span_ids", + 4: "start_time", + 5: "end_time", + 6: "platform_type", + 7: "field_mappings", + 255: "Base", +} + +func (p *ExtractSpanInfoRequest) IsSetStartTime() bool { + return p.StartTime != nil +} + +func (p *ExtractSpanInfoRequest) IsSetEndTime() bool { + return p.EndTime != nil +} + +func (p *ExtractSpanInfoRequest) IsSetPlatformType() bool { + return p.PlatformType != nil +} + +func (p *ExtractSpanInfoRequest) IsSetFieldMappings() bool { + return p.FieldMappings != nil +} + +func (p *ExtractSpanInfoRequest) IsSetBase() bool { + return p.Base != nil +} + +func (p *ExtractSpanInfoRequest) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 - var issetSpanInfos bool = false + var issetWorkspaceID bool = false + var issetTraceID bool = false + var issetSpanIds bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -18970,37 +18967,97 @@ func (p *ExtractSpanInfoResponse) Read(iprot thrift.TProtocol) (err error) { switch fieldId { case 1: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.I64 { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } - issetSpanInfos = true + issetWorkspaceID = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - case 255: - if fieldTypeId == thrift.STRUCT { - if err = p.ReadField255(iprot); err != nil { + case 2: + if fieldTypeId == thrift.STRING { + if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } + issetTraceID = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - default: - if err = iprot.Skip(fieldTypeId); err != nil { + case 3: + if fieldTypeId == thrift.LIST { + if err = p.ReadField3(iprot); err != nil { + goto ReadFieldError + } + issetSpanIds = true + } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - } - if err = iprot.ReadFieldEnd(); err != nil { - goto ReadFieldEndError - } - } - if err = iprot.ReadStructEnd(); err != nil { - goto ReadStructEndError - } - - if !issetSpanInfos { - fieldId = 1 + case 4: + if fieldTypeId == thrift.I64 { + if err = p.ReadField4(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 5: + if fieldTypeId == thrift.I64 { + if err = p.ReadField5(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 6: + if fieldTypeId == thrift.STRING { + if err = p.ReadField6(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 7: + if fieldTypeId == thrift.LIST { + if err = p.ReadField7(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + if !issetWorkspaceID { + fieldId = 1 + goto RequiredFieldNotSetError + } + + if !issetTraceID { + fieldId = 2 + goto RequiredFieldNotSetError + } + + if !issetSpanIds { + fieldId = 3 goto RequiredFieldNotSetError } return nil @@ -19009,7 +19066,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExtractSpanInfoResponse[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExtractSpanInfoRequest[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -19018,16 +19075,94 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ExtractSpanInfoResponse[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ExtractSpanInfoRequest[fieldId])) } -func (p *ExtractSpanInfoResponse) ReadField1(iprot thrift.TProtocol) error { +func (p *ExtractSpanInfoRequest) ReadField1(iprot thrift.TProtocol) error { + + var _field int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = v + } + p.WorkspaceID = _field + return nil +} +func (p *ExtractSpanInfoRequest) ReadField2(iprot thrift.TProtocol) error { + + var _field string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v + } + p.TraceID = _field + return nil +} +func (p *ExtractSpanInfoRequest) ReadField3(iprot thrift.TProtocol) error { _, size, err := iprot.ReadListBegin() if err != nil { return err } - _field := make([]*SpanInfo, 0, size) - values := make([]SpanInfo, size) + _field := make([]string, 0, size) + for i := 0; i < size; i++ { + + var _elem string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _elem = v + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.SpanIds = _field + return nil +} +func (p *ExtractSpanInfoRequest) ReadField4(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v + } + p.StartTime = _field + return nil +} +func (p *ExtractSpanInfoRequest) ReadField5(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v + } + p.EndTime = _field + return nil +} +func (p *ExtractSpanInfoRequest) ReadField6(iprot thrift.TProtocol) error { + + var _field *common.PlatformType + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.PlatformType = _field + return nil +} +func (p *ExtractSpanInfoRequest) ReadField7(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*dataset0.FieldMapping, 0, size) + values := make([]dataset0.FieldMapping, size) for i := 0; i < size; i++ { _elem := &values[i] _elem.InitDefault() @@ -19041,21 +19176,21 @@ func (p *ExtractSpanInfoResponse) ReadField1(iprot thrift.TProtocol) error { if err := iprot.ReadListEnd(); err != nil { return err } - p.SpanInfos = _field + p.FieldMappings = _field return nil } -func (p *ExtractSpanInfoResponse) ReadField255(iprot thrift.TProtocol) error { - _field := base.NewBaseResp() +func (p *ExtractSpanInfoRequest) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBase() if err := _field.Read(iprot); err != nil { return err } - p.BaseResp = _field + p.Base = _field return nil } -func (p *ExtractSpanInfoResponse) Write(oprot thrift.TProtocol) (err error) { +func (p *ExtractSpanInfoRequest) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ExtractSpanInfoResponse"); err != nil { + if err = oprot.WriteStructBegin("ExtractSpanInfoRequest"); err != nil { goto WriteStructBeginError } if p != nil { @@ -19063,6 +19198,30 @@ func (p *ExtractSpanInfoResponse) Write(oprot thrift.TProtocol) (err error) { fieldId = 1 goto WriteFieldError } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } + if err = p.writeField3(oprot); err != nil { + fieldId = 3 + goto WriteFieldError + } + if err = p.writeField4(oprot); err != nil { + fieldId = 4 + goto WriteFieldError + } + if err = p.writeField5(oprot); err != nil { + fieldId = 5 + goto WriteFieldError + } + if err = p.writeField6(oprot); err != nil { + fieldId = 6 + goto WriteFieldError + } + if err = p.writeField7(oprot); err != nil { + fieldId = 7 + goto WriteFieldError + } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -19085,15 +19244,47 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ExtractSpanInfoResponse) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("span_infos", thrift.LIST, 1); err != nil { +func (p *ExtractSpanInfoRequest) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.SpanInfos)); err != nil { + if err := oprot.WriteI64(p.WorkspaceID); err != nil { return err } - for _, v := range p.SpanInfos { - if err := v.Write(oprot); err != nil { + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *ExtractSpanInfoRequest) writeField2(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("trace_id", thrift.STRING, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(p.TraceID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} +func (p *ExtractSpanInfoRequest) writeField3(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("span_ids", thrift.LIST, 3); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRING, len(p.SpanIds)); err != nil { + return err + } + for _, v := range p.SpanIds { + if err := oprot.WriteString(v); err != nil { return err } } @@ -19105,16 +19296,16 @@ func (p *ExtractSpanInfoResponse) writeField1(oprot thrift.TProtocol) (err error } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) } -func (p *ExtractSpanInfoResponse) writeField255(oprot thrift.TProtocol) (err error) { - if p.IsSetBaseResp() { - if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { +func (p *ExtractSpanInfoRequest) writeField4(oprot thrift.TProtocol) (err error) { + if p.IsSetStartTime() { + if err = oprot.WriteFieldBegin("start_time", thrift.I64, 4); err != nil { goto WriteFieldBeginError } - if err := p.BaseResp.Write(oprot); err != nil { + if err := oprot.WriteI64(*p.StartTime); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -19123,126 +19314,258 @@ func (p *ExtractSpanInfoResponse) writeField255(oprot thrift.TProtocol) (err err } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) -} - -func (p *ExtractSpanInfoResponse) String() string { - if p == nil { - return "" - } - return fmt.Sprintf("ExtractSpanInfoResponse(%+v)", *p) - + return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) } - -func (p *ExtractSpanInfoResponse) DeepEqual(ano *ExtractSpanInfoResponse) bool { +func (p *ExtractSpanInfoRequest) writeField5(oprot thrift.TProtocol) (err error) { + if p.IsSetEndTime() { + if err = oprot.WriteFieldBegin("end_time", thrift.I64, 5); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.EndTime); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 5 end error: ", p), err) +} +func (p *ExtractSpanInfoRequest) writeField6(oprot thrift.TProtocol) (err error) { + if p.IsSetPlatformType() { + if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 6); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.PlatformType); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 6 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 6 end error: ", p), err) +} +func (p *ExtractSpanInfoRequest) writeField7(oprot thrift.TProtocol) (err error) { + if p.IsSetFieldMappings() { + if err = oprot.WriteFieldBegin("field_mappings", thrift.LIST, 7); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FieldMappings)); err != nil { + return err + } + for _, v := range p.FieldMappings { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 7 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 7 end error: ", p), err) +} +func (p *ExtractSpanInfoRequest) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBase() { + if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.Base.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) +} + +func (p *ExtractSpanInfoRequest) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("ExtractSpanInfoRequest(%+v)", *p) + +} + +func (p *ExtractSpanInfoRequest) DeepEqual(ano *ExtractSpanInfoRequest) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.SpanInfos) { + if !p.Field1DeepEqual(ano.WorkspaceID) { return false } - if !p.Field255DeepEqual(ano.BaseResp) { + if !p.Field2DeepEqual(ano.TraceID) { + return false + } + if !p.Field3DeepEqual(ano.SpanIds) { + return false + } + if !p.Field4DeepEqual(ano.StartTime) { + return false + } + if !p.Field5DeepEqual(ano.EndTime) { + return false + } + if !p.Field6DeepEqual(ano.PlatformType) { + return false + } + if !p.Field7DeepEqual(ano.FieldMappings) { + return false + } + if !p.Field255DeepEqual(ano.Base) { return false } return true } -func (p *ExtractSpanInfoResponse) Field1DeepEqual(src []*SpanInfo) bool { +func (p *ExtractSpanInfoRequest) Field1DeepEqual(src int64) bool { - if len(p.SpanInfos) != len(src) { + if p.WorkspaceID != src { return false } - for i, v := range p.SpanInfos { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } return true } -func (p *ExtractSpanInfoResponse) Field255DeepEqual(src *base.BaseResp) bool { +func (p *ExtractSpanInfoRequest) Field2DeepEqual(src string) bool { - if !p.BaseResp.DeepEqual(src) { + if strings.Compare(p.TraceID, src) != 0 { return false } return true } +func (p *ExtractSpanInfoRequest) Field3DeepEqual(src []string) bool { -type UpsertTrajectoryConfigRequest struct { - WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` - Filters *filter.FilterFields `thrift:"filters,2,optional" frugal:"2,optional,filter.FilterFields" form:"filters" json:"filters,omitempty"` - Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` + if len(p.SpanIds) != len(src) { + return false + } + for i, v := range p.SpanIds { + _src := src[i] + if strings.Compare(v, _src) != 0 { + return false + } + } + return true } +func (p *ExtractSpanInfoRequest) Field4DeepEqual(src *int64) bool { -func NewUpsertTrajectoryConfigRequest() *UpsertTrajectoryConfigRequest { - return &UpsertTrajectoryConfigRequest{} + if p.StartTime == src { + return true + } else if p.StartTime == nil || src == nil { + return false + } + if *p.StartTime != *src { + return false + } + return true } +func (p *ExtractSpanInfoRequest) Field5DeepEqual(src *int64) bool { -func (p *UpsertTrajectoryConfigRequest) InitDefault() { + if p.EndTime == src { + return true + } else if p.EndTime == nil || src == nil { + return false + } + if *p.EndTime != *src { + return false + } + return true } +func (p *ExtractSpanInfoRequest) Field6DeepEqual(src *common.PlatformType) bool { -func (p *UpsertTrajectoryConfigRequest) GetWorkspaceID() (v int64) { - if p != nil { - return p.WorkspaceID + if p.PlatformType == src { + return true + } else if p.PlatformType == nil || src == nil { + return false } - return + if strings.Compare(*p.PlatformType, *src) != 0 { + return false + } + return true } +func (p *ExtractSpanInfoRequest) Field7DeepEqual(src []*dataset0.FieldMapping) bool { -var UpsertTrajectoryConfigRequest_Filters_DEFAULT *filter.FilterFields - -func (p *UpsertTrajectoryConfigRequest) GetFilters() (v *filter.FilterFields) { - if p == nil { - return + if len(p.FieldMappings) != len(src) { + return false } - if !p.IsSetFilters() { - return UpsertTrajectoryConfigRequest_Filters_DEFAULT + for i, v := range p.FieldMappings { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } } - return p.Filters + return true } +func (p *ExtractSpanInfoRequest) Field255DeepEqual(src *base.Base) bool { -var UpsertTrajectoryConfigRequest_Base_DEFAULT *base.Base - -func (p *UpsertTrajectoryConfigRequest) GetBase() (v *base.Base) { - if p == nil { - return - } - if !p.IsSetBase() { - return UpsertTrajectoryConfigRequest_Base_DEFAULT + if !p.Base.DeepEqual(src) { + return false } - return p.Base + return true } -func (p *UpsertTrajectoryConfigRequest) SetWorkspaceID(val int64) { - p.WorkspaceID = val + +type SpanInfo struct { + SpanID string `thrift:"span_id,1,required" frugal:"1,required,string" form:"span_id,required" json:"span_id,required" query:"span_id,required"` + FieldList []*dataset0.FieldData `thrift:"field_list,2,required" frugal:"2,required,list" form:"field_list,required" json:"field_list,required" query:"field_list,required"` } -func (p *UpsertTrajectoryConfigRequest) SetFilters(val *filter.FilterFields) { - p.Filters = val + +func NewSpanInfo() *SpanInfo { + return &SpanInfo{} } -func (p *UpsertTrajectoryConfigRequest) SetBase(val *base.Base) { - p.Base = val + +func (p *SpanInfo) InitDefault() { } -var fieldIDToName_UpsertTrajectoryConfigRequest = map[int16]string{ - 1: "workspace_id", - 2: "filters", - 255: "Base", +func (p *SpanInfo) GetSpanID() (v string) { + if p != nil { + return p.SpanID + } + return } -func (p *UpsertTrajectoryConfigRequest) IsSetFilters() bool { - return p.Filters != nil +func (p *SpanInfo) GetFieldList() (v []*dataset0.FieldData) { + if p != nil { + return p.FieldList + } + return +} +func (p *SpanInfo) SetSpanID(val string) { + p.SpanID = val +} +func (p *SpanInfo) SetFieldList(val []*dataset0.FieldData) { + p.FieldList = val } -func (p *UpsertTrajectoryConfigRequest) IsSetBase() bool { - return p.Base != nil +var fieldIDToName_SpanInfo = map[int16]string{ + 1: "span_id", + 2: "field_list", } -func (p *UpsertTrajectoryConfigRequest) Read(iprot thrift.TProtocol) (err error) { +func (p *SpanInfo) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 - var issetWorkspaceID bool = false + var issetSpanID bool = false + var issetFieldList bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -19259,27 +19582,20 @@ func (p *UpsertTrajectoryConfigRequest) Read(iprot thrift.TProtocol) (err error) switch fieldId { case 1: - if fieldTypeId == thrift.I64 { + if fieldTypeId == thrift.STRING { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } - issetWorkspaceID = true + issetSpanID = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } case 2: - if fieldTypeId == thrift.STRUCT { + if fieldTypeId == thrift.LIST { if err = p.ReadField2(iprot); err != nil { goto ReadFieldError } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 255: - if fieldTypeId == thrift.STRUCT { - if err = p.ReadField255(iprot); err != nil { - goto ReadFieldError - } + issetFieldList = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } @@ -19296,17 +19612,22 @@ func (p *UpsertTrajectoryConfigRequest) Read(iprot thrift.TProtocol) (err error) goto ReadStructEndError } - if !issetWorkspaceID { + if !issetSpanID { fieldId = 1 goto RequiredFieldNotSetError } + + if !issetFieldList { + fieldId = 2 + goto RequiredFieldNotSetError + } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_UpsertTrajectoryConfigRequest[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SpanInfo[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -19315,40 +19636,47 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_UpsertTrajectoryConfigRequest[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_SpanInfo[fieldId])) } -func (p *UpsertTrajectoryConfigRequest) ReadField1(iprot thrift.TProtocol) error { +func (p *SpanInfo) ReadField1(iprot thrift.TProtocol) error { - var _field int64 - if v, err := iprot.ReadI64(); err != nil { + var _field string + if v, err := iprot.ReadString(); err != nil { return err } else { _field = v } - p.WorkspaceID = _field + p.SpanID = _field return nil } -func (p *UpsertTrajectoryConfigRequest) ReadField2(iprot thrift.TProtocol) error { - _field := filter.NewFilterFields() - if err := _field.Read(iprot); err != nil { +func (p *SpanInfo) ReadField2(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { return err } - p.Filters = _field - return nil -} -func (p *UpsertTrajectoryConfigRequest) ReadField255(iprot thrift.TProtocol) error { - _field := base.NewBase() - if err := _field.Read(iprot); err != nil { + _field := make([]*dataset0.FieldData, 0, size) + values := make([]dataset0.FieldData, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { return err } - p.Base = _field + p.FieldList = _field return nil } -func (p *UpsertTrajectoryConfigRequest) Write(oprot thrift.TProtocol) (err error) { +func (p *SpanInfo) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("UpsertTrajectoryConfigRequest"); err != nil { + if err = oprot.WriteStructBegin("SpanInfo"); err != nil { goto WriteStructBeginError } if p != nil { @@ -19360,10 +19688,6 @@ func (p *UpsertTrajectoryConfigRequest) Write(oprot thrift.TProtocol) (err error fieldId = 2 goto WriteFieldError } - if err = p.writeField255(oprot); err != nil { - fieldId = 255 - goto WriteFieldError - } } if err = oprot.WriteFieldStop(); err != nil { goto WriteFieldStopError @@ -19382,11 +19706,11 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *UpsertTrajectoryConfigRequest) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { +func (p *SpanInfo) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("span_id", thrift.STRING, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteI64(p.WorkspaceID); err != nil { + if err := oprot.WriteString(p.SpanID); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -19398,17 +19722,23 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *UpsertTrajectoryConfigRequest) writeField2(oprot thrift.TProtocol) (err error) { - if p.IsSetFilters() { - if err = oprot.WriteFieldBegin("filters", thrift.STRUCT, 2); err != nil { - goto WriteFieldBeginError - } - if err := p.Filters.Write(oprot); err != nil { +func (p *SpanInfo) writeField2(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("field_list", thrift.LIST, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.FieldList)); err != nil { + return err + } + for _, v := range p.FieldList { + if err := v.Write(oprot); err != nil { return err } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } return nil WriteFieldBeginError: @@ -19416,110 +19746,101 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *UpsertTrajectoryConfigRequest) writeField255(oprot thrift.TProtocol) (err error) { - if p.IsSetBase() { - if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { - goto WriteFieldBeginError - } - if err := p.Base.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) -} -func (p *UpsertTrajectoryConfigRequest) String() string { +func (p *SpanInfo) String() string { if p == nil { return "" } - return fmt.Sprintf("UpsertTrajectoryConfigRequest(%+v)", *p) + return fmt.Sprintf("SpanInfo(%+v)", *p) } -func (p *UpsertTrajectoryConfigRequest) DeepEqual(ano *UpsertTrajectoryConfigRequest) bool { +func (p *SpanInfo) DeepEqual(ano *SpanInfo) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.WorkspaceID) { - return false - } - if !p.Field2DeepEqual(ano.Filters) { + if !p.Field1DeepEqual(ano.SpanID) { return false } - if !p.Field255DeepEqual(ano.Base) { + if !p.Field2DeepEqual(ano.FieldList) { return false } return true } -func (p *UpsertTrajectoryConfigRequest) Field1DeepEqual(src int64) bool { +func (p *SpanInfo) Field1DeepEqual(src string) bool { - if p.WorkspaceID != src { + if strings.Compare(p.SpanID, src) != 0 { return false } return true } -func (p *UpsertTrajectoryConfigRequest) Field2DeepEqual(src *filter.FilterFields) bool { +func (p *SpanInfo) Field2DeepEqual(src []*dataset0.FieldData) bool { - if !p.Filters.DeepEqual(src) { + if len(p.FieldList) != len(src) { return false } + for i, v := range p.FieldList { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } return true } -func (p *UpsertTrajectoryConfigRequest) Field255DeepEqual(src *base.Base) bool { - if !p.Base.DeepEqual(src) { - return false - } - return true +type ExtractSpanInfoResponse struct { + SpanInfos []*SpanInfo `thrift:"span_infos,1,required" frugal:"1,required,list" form:"span_infos,required" json:"span_infos,required" query:"span_infos,required"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` } -type UpsertTrajectoryConfigResponse struct { - BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` +func NewExtractSpanInfoResponse() *ExtractSpanInfoResponse { + return &ExtractSpanInfoResponse{} } -func NewUpsertTrajectoryConfigResponse() *UpsertTrajectoryConfigResponse { - return &UpsertTrajectoryConfigResponse{} +func (p *ExtractSpanInfoResponse) InitDefault() { } -func (p *UpsertTrajectoryConfigResponse) InitDefault() { +func (p *ExtractSpanInfoResponse) GetSpanInfos() (v []*SpanInfo) { + if p != nil { + return p.SpanInfos + } + return } -var UpsertTrajectoryConfigResponse_BaseResp_DEFAULT *base.BaseResp +var ExtractSpanInfoResponse_BaseResp_DEFAULT *base.BaseResp -func (p *UpsertTrajectoryConfigResponse) GetBaseResp() (v *base.BaseResp) { +func (p *ExtractSpanInfoResponse) GetBaseResp() (v *base.BaseResp) { if p == nil { return } if !p.IsSetBaseResp() { - return UpsertTrajectoryConfigResponse_BaseResp_DEFAULT + return ExtractSpanInfoResponse_BaseResp_DEFAULT } return p.BaseResp } -func (p *UpsertTrajectoryConfigResponse) SetBaseResp(val *base.BaseResp) { +func (p *ExtractSpanInfoResponse) SetSpanInfos(val []*SpanInfo) { + p.SpanInfos = val +} +func (p *ExtractSpanInfoResponse) SetBaseResp(val *base.BaseResp) { p.BaseResp = val } -var fieldIDToName_UpsertTrajectoryConfigResponse = map[int16]string{ +var fieldIDToName_ExtractSpanInfoResponse = map[int16]string{ + 1: "span_infos", 255: "BaseResp", } -func (p *UpsertTrajectoryConfigResponse) IsSetBaseResp() bool { +func (p *ExtractSpanInfoResponse) IsSetBaseResp() bool { return p.BaseResp != nil } -func (p *UpsertTrajectoryConfigResponse) Read(iprot thrift.TProtocol) (err error) { +func (p *ExtractSpanInfoResponse) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 + var issetSpanInfos bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -19535,6 +19856,15 @@ func (p *UpsertTrajectoryConfigResponse) Read(iprot thrift.TProtocol) (err error } switch fieldId { + case 1: + if fieldTypeId == thrift.LIST { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + issetSpanInfos = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } case 255: if fieldTypeId == thrift.STRUCT { if err = p.ReadField255(iprot); err != nil { @@ -19556,13 +19886,17 @@ func (p *UpsertTrajectoryConfigResponse) Read(iprot thrift.TProtocol) (err error goto ReadStructEndError } + if !issetSpanInfos { + fieldId = 1 + goto RequiredFieldNotSetError + } return nil ReadStructBeginError: return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_UpsertTrajectoryConfigResponse[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExtractSpanInfoResponse[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -19570,9 +19904,34 @@ ReadFieldEndError: return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +RequiredFieldNotSetError: + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ExtractSpanInfoResponse[fieldId])) } -func (p *UpsertTrajectoryConfigResponse) ReadField255(iprot thrift.TProtocol) error { +func (p *ExtractSpanInfoResponse) ReadField1(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*SpanInfo, 0, size) + values := make([]SpanInfo, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.SpanInfos = _field + return nil +} +func (p *ExtractSpanInfoResponse) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBaseResp() if err := _field.Read(iprot); err != nil { return err @@ -19581,12 +19940,16 @@ func (p *UpsertTrajectoryConfigResponse) ReadField255(iprot thrift.TProtocol) er return nil } -func (p *UpsertTrajectoryConfigResponse) Write(oprot thrift.TProtocol) (err error) { +func (p *ExtractSpanInfoResponse) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("UpsertTrajectoryConfigResponse"); err != nil { + if err = oprot.WriteStructBegin("ExtractSpanInfoResponse"); err != nil { goto WriteStructBeginError } if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -19609,46 +19972,86 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *UpsertTrajectoryConfigResponse) writeField255(oprot thrift.TProtocol) (err error) { - if p.IsSetBaseResp() { - if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { - goto WriteFieldBeginError - } - if err := p.BaseResp.Write(oprot); err != nil { +func (p *ExtractSpanInfoResponse) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("span_infos", thrift.LIST, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.SpanInfos)); err != nil { + return err + } + for _, v := range p.SpanInfos { + if err := v.Write(oprot); err != nil { return err } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *ExtractSpanInfoResponse) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBaseResp() { + if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.BaseResp.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *UpsertTrajectoryConfigResponse) String() string { +func (p *ExtractSpanInfoResponse) String() string { if p == nil { return "" } - return fmt.Sprintf("UpsertTrajectoryConfigResponse(%+v)", *p) + return fmt.Sprintf("ExtractSpanInfoResponse(%+v)", *p) } -func (p *UpsertTrajectoryConfigResponse) DeepEqual(ano *UpsertTrajectoryConfigResponse) bool { +func (p *ExtractSpanInfoResponse) DeepEqual(ano *ExtractSpanInfoResponse) bool { if p == ano { return true } else if p == nil || ano == nil { return false } + if !p.Field1DeepEqual(ano.SpanInfos) { + return false + } if !p.Field255DeepEqual(ano.BaseResp) { return false } return true } -func (p *UpsertTrajectoryConfigResponse) Field255DeepEqual(src *base.BaseResp) bool { +func (p *ExtractSpanInfoResponse) Field1DeepEqual(src []*SpanInfo) bool { + + if len(p.SpanInfos) != len(src) { + return false + } + for i, v := range p.SpanInfos { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true +} +func (p *ExtractSpanInfoResponse) Field255DeepEqual(src *base.BaseResp) bool { if !p.BaseResp.DeepEqual(src) { return false @@ -19656,53 +20059,74 @@ func (p *UpsertTrajectoryConfigResponse) Field255DeepEqual(src *base.BaseResp) b return true } -type GetTrajectoryConfigRequest struct { - WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id,string,required" query:"workspace_id,required"` - Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` +type UpsertTrajectoryConfigRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` + Filters *filter.FilterFields `thrift:"filters,2,optional" frugal:"2,optional,filter.FilterFields" form:"filters" json:"filters,omitempty"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` } -func NewGetTrajectoryConfigRequest() *GetTrajectoryConfigRequest { - return &GetTrajectoryConfigRequest{} +func NewUpsertTrajectoryConfigRequest() *UpsertTrajectoryConfigRequest { + return &UpsertTrajectoryConfigRequest{} } -func (p *GetTrajectoryConfigRequest) InitDefault() { +func (p *UpsertTrajectoryConfigRequest) InitDefault() { } -func (p *GetTrajectoryConfigRequest) GetWorkspaceID() (v int64) { +func (p *UpsertTrajectoryConfigRequest) GetWorkspaceID() (v int64) { if p != nil { return p.WorkspaceID } return } -var GetTrajectoryConfigRequest_Base_DEFAULT *base.Base +var UpsertTrajectoryConfigRequest_Filters_DEFAULT *filter.FilterFields -func (p *GetTrajectoryConfigRequest) GetBase() (v *base.Base) { +func (p *UpsertTrajectoryConfigRequest) GetFilters() (v *filter.FilterFields) { + if p == nil { + return + } + if !p.IsSetFilters() { + return UpsertTrajectoryConfigRequest_Filters_DEFAULT + } + return p.Filters +} + +var UpsertTrajectoryConfigRequest_Base_DEFAULT *base.Base + +func (p *UpsertTrajectoryConfigRequest) GetBase() (v *base.Base) { if p == nil { return } if !p.IsSetBase() { - return GetTrajectoryConfigRequest_Base_DEFAULT + return UpsertTrajectoryConfigRequest_Base_DEFAULT } return p.Base } -func (p *GetTrajectoryConfigRequest) SetWorkspaceID(val int64) { +func (p *UpsertTrajectoryConfigRequest) SetWorkspaceID(val int64) { p.WorkspaceID = val } -func (p *GetTrajectoryConfigRequest) SetBase(val *base.Base) { +func (p *UpsertTrajectoryConfigRequest) SetFilters(val *filter.FilterFields) { + p.Filters = val +} +func (p *UpsertTrajectoryConfigRequest) SetBase(val *base.Base) { p.Base = val } -var fieldIDToName_GetTrajectoryConfigRequest = map[int16]string{ +var fieldIDToName_UpsertTrajectoryConfigRequest = map[int16]string{ 1: "workspace_id", + 2: "filters", 255: "Base", } -func (p *GetTrajectoryConfigRequest) IsSetBase() bool { +func (p *UpsertTrajectoryConfigRequest) IsSetFilters() bool { + return p.Filters != nil +} + +func (p *UpsertTrajectoryConfigRequest) IsSetBase() bool { return p.Base != nil } -func (p *GetTrajectoryConfigRequest) Read(iprot thrift.TProtocol) (err error) { +func (p *UpsertTrajectoryConfigRequest) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 var issetWorkspaceID bool = false @@ -19730,6 +20154,14 @@ func (p *GetTrajectoryConfigRequest) Read(iprot thrift.TProtocol) (err error) { } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } + case 2: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField2(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } case 255: if fieldTypeId == thrift.STRUCT { if err = p.ReadField255(iprot); err != nil { @@ -19761,7 +20193,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_GetTrajectoryConfigRequest[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_UpsertTrajectoryConfigRequest[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -19770,10 +20202,10 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_GetTrajectoryConfigRequest[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_UpsertTrajectoryConfigRequest[fieldId])) } -func (p *GetTrajectoryConfigRequest) ReadField1(iprot thrift.TProtocol) error { +func (p *UpsertTrajectoryConfigRequest) ReadField1(iprot thrift.TProtocol) error { var _field int64 if v, err := iprot.ReadI64(); err != nil { @@ -19784,7 +20216,15 @@ func (p *GetTrajectoryConfigRequest) ReadField1(iprot thrift.TProtocol) error { p.WorkspaceID = _field return nil } -func (p *GetTrajectoryConfigRequest) ReadField255(iprot thrift.TProtocol) error { +func (p *UpsertTrajectoryConfigRequest) ReadField2(iprot thrift.TProtocol) error { + _field := filter.NewFilterFields() + if err := _field.Read(iprot); err != nil { + return err + } + p.Filters = _field + return nil +} +func (p *UpsertTrajectoryConfigRequest) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBase() if err := _field.Read(iprot); err != nil { return err @@ -19793,9 +20233,9 @@ func (p *GetTrajectoryConfigRequest) ReadField255(iprot thrift.TProtocol) error return nil } -func (p *GetTrajectoryConfigRequest) Write(oprot thrift.TProtocol) (err error) { +func (p *UpsertTrajectoryConfigRequest) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("GetTrajectoryConfigRequest"); err != nil { + if err = oprot.WriteStructBegin("UpsertTrajectoryConfigRequest"); err != nil { goto WriteStructBeginError } if p != nil { @@ -19803,6 +20243,10 @@ func (p *GetTrajectoryConfigRequest) Write(oprot thrift.TProtocol) (err error) { fieldId = 1 goto WriteFieldError } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -19825,7 +20269,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *GetTrajectoryConfigRequest) writeField1(oprot thrift.TProtocol) (err error) { +func (p *UpsertTrajectoryConfigRequest) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { goto WriteFieldBeginError } @@ -19841,7 +20285,25 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *GetTrajectoryConfigRequest) writeField255(oprot thrift.TProtocol) (err error) { +func (p *UpsertTrajectoryConfigRequest) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetFilters() { + if err = oprot.WriteFieldBegin("filters", thrift.STRUCT, 2); err != nil { + goto WriteFieldBeginError + } + if err := p.Filters.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} +func (p *UpsertTrajectoryConfigRequest) writeField255(oprot thrift.TProtocol) (err error) { if p.IsSetBase() { if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError @@ -19860,15 +20322,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *GetTrajectoryConfigRequest) String() string { +func (p *UpsertTrajectoryConfigRequest) String() string { if p == nil { return "" } - return fmt.Sprintf("GetTrajectoryConfigRequest(%+v)", *p) + return fmt.Sprintf("UpsertTrajectoryConfigRequest(%+v)", *p) } -func (p *GetTrajectoryConfigRequest) DeepEqual(ano *GetTrajectoryConfigRequest) bool { +func (p *UpsertTrajectoryConfigRequest) DeepEqual(ano *UpsertTrajectoryConfigRequest) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -19877,83 +20339,72 @@ func (p *GetTrajectoryConfigRequest) DeepEqual(ano *GetTrajectoryConfigRequest) if !p.Field1DeepEqual(ano.WorkspaceID) { return false } + if !p.Field2DeepEqual(ano.Filters) { + return false + } if !p.Field255DeepEqual(ano.Base) { return false } return true } -func (p *GetTrajectoryConfigRequest) Field1DeepEqual(src int64) bool { +func (p *UpsertTrajectoryConfigRequest) Field1DeepEqual(src int64) bool { if p.WorkspaceID != src { return false } return true } -func (p *GetTrajectoryConfigRequest) Field255DeepEqual(src *base.Base) bool { +func (p *UpsertTrajectoryConfigRequest) Field2DeepEqual(src *filter.FilterFields) bool { - if !p.Base.DeepEqual(src) { + if !p.Filters.DeepEqual(src) { return false } return true } +func (p *UpsertTrajectoryConfigRequest) Field255DeepEqual(src *base.Base) bool { -type GetTrajectoryConfigResponse struct { - Filters *filter.FilterFields `thrift:"filters,1,optional" frugal:"1,optional,filter.FilterFields" form:"filters" json:"filters,omitempty" query:"filters"` - BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` + if !p.Base.DeepEqual(src) { + return false + } + return true } -func NewGetTrajectoryConfigResponse() *GetTrajectoryConfigResponse { - return &GetTrajectoryConfigResponse{} +type UpsertTrajectoryConfigResponse struct { + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` } -func (p *GetTrajectoryConfigResponse) InitDefault() { +func NewUpsertTrajectoryConfigResponse() *UpsertTrajectoryConfigResponse { + return &UpsertTrajectoryConfigResponse{} } -var GetTrajectoryConfigResponse_Filters_DEFAULT *filter.FilterFields - -func (p *GetTrajectoryConfigResponse) GetFilters() (v *filter.FilterFields) { - if p == nil { - return - } - if !p.IsSetFilters() { - return GetTrajectoryConfigResponse_Filters_DEFAULT - } - return p.Filters +func (p *UpsertTrajectoryConfigResponse) InitDefault() { } -var GetTrajectoryConfigResponse_BaseResp_DEFAULT *base.BaseResp +var UpsertTrajectoryConfigResponse_BaseResp_DEFAULT *base.BaseResp -func (p *GetTrajectoryConfigResponse) GetBaseResp() (v *base.BaseResp) { +func (p *UpsertTrajectoryConfigResponse) GetBaseResp() (v *base.BaseResp) { if p == nil { return } if !p.IsSetBaseResp() { - return GetTrajectoryConfigResponse_BaseResp_DEFAULT + return UpsertTrajectoryConfigResponse_BaseResp_DEFAULT } return p.BaseResp } -func (p *GetTrajectoryConfigResponse) SetFilters(val *filter.FilterFields) { - p.Filters = val -} -func (p *GetTrajectoryConfigResponse) SetBaseResp(val *base.BaseResp) { +func (p *UpsertTrajectoryConfigResponse) SetBaseResp(val *base.BaseResp) { p.BaseResp = val } -var fieldIDToName_GetTrajectoryConfigResponse = map[int16]string{ - 1: "filters", +var fieldIDToName_UpsertTrajectoryConfigResponse = map[int16]string{ 255: "BaseResp", } -func (p *GetTrajectoryConfigResponse) IsSetFilters() bool { - return p.Filters != nil -} - -func (p *GetTrajectoryConfigResponse) IsSetBaseResp() bool { +func (p *UpsertTrajectoryConfigResponse) IsSetBaseResp() bool { return p.BaseResp != nil } -func (p *GetTrajectoryConfigResponse) Read(iprot thrift.TProtocol) (err error) { +func (p *UpsertTrajectoryConfigResponse) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -19971,14 +20422,6 @@ func (p *GetTrajectoryConfigResponse) Read(iprot thrift.TProtocol) (err error) { } switch fieldId { - case 1: - if fieldTypeId == thrift.STRUCT { - if err = p.ReadField1(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } case 255: if fieldTypeId == thrift.STRUCT { if err = p.ReadField255(iprot); err != nil { @@ -20006,7 +20449,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_GetTrajectoryConfigResponse[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_UpsertTrajectoryConfigResponse[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -20016,15 +20459,7 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *GetTrajectoryConfigResponse) ReadField1(iprot thrift.TProtocol) error { - _field := filter.NewFilterFields() - if err := _field.Read(iprot); err != nil { - return err - } - p.Filters = _field - return nil -} -func (p *GetTrajectoryConfigResponse) ReadField255(iprot thrift.TProtocol) error { +func (p *UpsertTrajectoryConfigResponse) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBaseResp() if err := _field.Read(iprot); err != nil { return err @@ -20033,16 +20468,12 @@ func (p *GetTrajectoryConfigResponse) ReadField255(iprot thrift.TProtocol) error return nil } -func (p *GetTrajectoryConfigResponse) Write(oprot thrift.TProtocol) (err error) { +func (p *UpsertTrajectoryConfigResponse) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("GetTrajectoryConfigResponse"); err != nil { + if err = oprot.WriteStructBegin("UpsertTrajectoryConfigResponse"); err != nil { goto WriteStructBeginError } if p != nil { - if err = p.writeField1(oprot); err != nil { - fieldId = 1 - goto WriteFieldError - } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -20065,27 +20496,9 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *GetTrajectoryConfigResponse) writeField1(oprot thrift.TProtocol) (err error) { - if p.IsSetFilters() { - if err = oprot.WriteFieldBegin("filters", thrift.STRUCT, 1); err != nil { - goto WriteFieldBeginError - } - if err := p.Filters.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) -} -func (p *GetTrajectoryConfigResponse) writeField255(oprot thrift.TProtocol) (err error) { - if p.IsSetBaseResp() { - if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { +func (p *UpsertTrajectoryConfigResponse) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBaseResp() { + if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError } if err := p.BaseResp.Write(oprot); err != nil { @@ -20102,37 +20515,27 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *GetTrajectoryConfigResponse) String() string { +func (p *UpsertTrajectoryConfigResponse) String() string { if p == nil { return "" } - return fmt.Sprintf("GetTrajectoryConfigResponse(%+v)", *p) + return fmt.Sprintf("UpsertTrajectoryConfigResponse(%+v)", *p) } -func (p *GetTrajectoryConfigResponse) DeepEqual(ano *GetTrajectoryConfigResponse) bool { +func (p *UpsertTrajectoryConfigResponse) DeepEqual(ano *UpsertTrajectoryConfigResponse) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.Filters) { - return false - } if !p.Field255DeepEqual(ano.BaseResp) { return false } return true } -func (p *GetTrajectoryConfigResponse) Field1DeepEqual(src *filter.FilterFields) bool { - - if !p.Filters.DeepEqual(src) { - return false - } - return true -} -func (p *GetTrajectoryConfigResponse) Field255DeepEqual(src *base.BaseResp) bool { +func (p *UpsertTrajectoryConfigResponse) Field255DeepEqual(src *base.BaseResp) bool { if !p.BaseResp.DeepEqual(src) { return false @@ -20140,105 +20543,56 @@ func (p *GetTrajectoryConfigResponse) Field255DeepEqual(src *base.BaseResp) bool return true } -type ListTrajectoryRequest struct { - // 需要准确填写，用于确定查询哪些租户的数据 - PlatformType common.PlatformType `thrift:"platform_type,1,required" frugal:"1,required,string" form:"platform_type,required" json:"platform_type,required"` - WorkspaceID int64 `thrift:"workspace_id,2,required" frugal:"2,required,i64" form:"workspace_id,required" json:"workspace_id,string,required"` - TraceIds []string `thrift:"trace_ids,3,required" frugal:"3,required,list" form:"trace_ids,required" json:"trace_ids,required"` - // ms - StartTime *int64 `thrift:"start_time,4,optional" frugal:"4,optional,i64" json:"start_time" form:"start_time" ` - Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` -} - -func NewListTrajectoryRequest() *ListTrajectoryRequest { - return &ListTrajectoryRequest{} +type GetTrajectoryConfigRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id,string,required" query:"workspace_id,required"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` } -func (p *ListTrajectoryRequest) InitDefault() { +func NewGetTrajectoryConfigRequest() *GetTrajectoryConfigRequest { + return &GetTrajectoryConfigRequest{} } -func (p *ListTrajectoryRequest) GetPlatformType() (v common.PlatformType) { - if p != nil { - return p.PlatformType - } - return +func (p *GetTrajectoryConfigRequest) InitDefault() { } -func (p *ListTrajectoryRequest) GetWorkspaceID() (v int64) { +func (p *GetTrajectoryConfigRequest) GetWorkspaceID() (v int64) { if p != nil { return p.WorkspaceID } return } -func (p *ListTrajectoryRequest) GetTraceIds() (v []string) { - if p != nil { - return p.TraceIds - } - return -} - -var ListTrajectoryRequest_StartTime_DEFAULT int64 - -func (p *ListTrajectoryRequest) GetStartTime() (v int64) { - if p == nil { - return - } - if !p.IsSetStartTime() { - return ListTrajectoryRequest_StartTime_DEFAULT - } - return *p.StartTime -} - -var ListTrajectoryRequest_Base_DEFAULT *base.Base +var GetTrajectoryConfigRequest_Base_DEFAULT *base.Base -func (p *ListTrajectoryRequest) GetBase() (v *base.Base) { +func (p *GetTrajectoryConfigRequest) GetBase() (v *base.Base) { if p == nil { return } if !p.IsSetBase() { - return ListTrajectoryRequest_Base_DEFAULT + return GetTrajectoryConfigRequest_Base_DEFAULT } return p.Base } -func (p *ListTrajectoryRequest) SetPlatformType(val common.PlatformType) { - p.PlatformType = val -} -func (p *ListTrajectoryRequest) SetWorkspaceID(val int64) { +func (p *GetTrajectoryConfigRequest) SetWorkspaceID(val int64) { p.WorkspaceID = val } -func (p *ListTrajectoryRequest) SetTraceIds(val []string) { - p.TraceIds = val -} -func (p *ListTrajectoryRequest) SetStartTime(val *int64) { - p.StartTime = val -} -func (p *ListTrajectoryRequest) SetBase(val *base.Base) { +func (p *GetTrajectoryConfigRequest) SetBase(val *base.Base) { p.Base = val } -var fieldIDToName_ListTrajectoryRequest = map[int16]string{ - 1: "platform_type", - 2: "workspace_id", - 3: "trace_ids", - 4: "start_time", +var fieldIDToName_GetTrajectoryConfigRequest = map[int16]string{ + 1: "workspace_id", 255: "Base", } -func (p *ListTrajectoryRequest) IsSetStartTime() bool { - return p.StartTime != nil -} - -func (p *ListTrajectoryRequest) IsSetBase() bool { +func (p *GetTrajectoryConfigRequest) IsSetBase() bool { return p.Base != nil } -func (p *ListTrajectoryRequest) Read(iprot thrift.TProtocol) (err error) { +func (p *GetTrajectoryConfigRequest) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 - var issetPlatformType bool = false var issetWorkspaceID bool = false - var issetTraceIds bool = false if _, err = iprot.ReadStructBegin(); err != nil { goto ReadStructBeginError @@ -20255,40 +20609,14 @@ func (p *ListTrajectoryRequest) Read(iprot thrift.TProtocol) (err error) { switch fieldId { case 1: - if fieldTypeId == thrift.STRING { - if err = p.ReadField1(iprot); err != nil { - goto ReadFieldError - } - issetPlatformType = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 2: if fieldTypeId == thrift.I64 { - if err = p.ReadField2(iprot); err != nil { + if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } issetWorkspaceID = true } else if err = iprot.Skip(fieldTypeId); err != nil { goto SkipFieldError } - case 3: - if fieldTypeId == thrift.LIST { - if err = p.ReadField3(iprot); err != nil { - goto ReadFieldError - } - issetTraceIds = true - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } - case 4: - if fieldTypeId == thrift.I64 { - if err = p.ReadField4(iprot); err != nil { - goto ReadFieldError - } - } else if err = iprot.Skip(fieldTypeId); err != nil { - goto SkipFieldError - } case 255: if fieldTypeId == thrift.STRUCT { if err = p.ReadField255(iprot); err != nil { @@ -20310,18 +20638,8 @@ func (p *ListTrajectoryRequest) Read(iprot thrift.TProtocol) (err error) { goto ReadStructEndError } - if !issetPlatformType { - fieldId = 1 - goto RequiredFieldNotSetError - } - if !issetWorkspaceID { - fieldId = 2 - goto RequiredFieldNotSetError - } - - if !issetTraceIds { - fieldId = 3 + fieldId = 1 goto RequiredFieldNotSetError } return nil @@ -20330,7 +20648,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListTrajectoryRequest[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_GetTrajectoryConfigRequest[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -20339,21 +20657,10 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) RequiredFieldNotSetError: - return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListTrajectoryRequest[fieldId])) + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_GetTrajectoryConfigRequest[fieldId])) } -func (p *ListTrajectoryRequest) ReadField1(iprot thrift.TProtocol) error { - - var _field common.PlatformType - if v, err := iprot.ReadString(); err != nil { - return err - } else { - _field = v - } - p.PlatformType = _field - return nil -} -func (p *ListTrajectoryRequest) ReadField2(iprot thrift.TProtocol) error { +func (p *GetTrajectoryConfigRequest) ReadField1(iprot thrift.TProtocol) error { var _field int64 if v, err := iprot.ReadI64(); err != nil { @@ -20364,41 +20671,7 @@ func (p *ListTrajectoryRequest) ReadField2(iprot thrift.TProtocol) error { p.WorkspaceID = _field return nil } -func (p *ListTrajectoryRequest) ReadField3(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]string, 0, size) - for i := 0; i < size; i++ { - - var _elem string - if v, err := iprot.ReadString(); err != nil { - return err - } else { - _elem = v - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { - return err - } - p.TraceIds = _field - return nil -} -func (p *ListTrajectoryRequest) ReadField4(iprot thrift.TProtocol) error { - - var _field *int64 - if v, err := iprot.ReadI64(); err != nil { - return err - } else { - _field = &v - } - p.StartTime = _field - return nil -} -func (p *ListTrajectoryRequest) ReadField255(iprot thrift.TProtocol) error { +func (p *GetTrajectoryConfigRequest) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBase() if err := _field.Read(iprot); err != nil { return err @@ -20407,9 +20680,9 @@ func (p *ListTrajectoryRequest) ReadField255(iprot thrift.TProtocol) error { return nil } -func (p *ListTrajectoryRequest) Write(oprot thrift.TProtocol) (err error) { +func (p *GetTrajectoryConfigRequest) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListTrajectoryRequest"); err != nil { + if err = oprot.WriteStructBegin("GetTrajectoryConfigRequest"); err != nil { goto WriteStructBeginError } if p != nil { @@ -20417,18 +20690,6 @@ func (p *ListTrajectoryRequest) Write(oprot thrift.TProtocol) (err error) { fieldId = 1 goto WriteFieldError } - if err = p.writeField2(oprot); err != nil { - fieldId = 2 - goto WriteFieldError - } - if err = p.writeField3(oprot); err != nil { - fieldId = 3 - goto WriteFieldError - } - if err = p.writeField4(oprot); err != nil { - fieldId = 4 - goto WriteFieldError - } if err = p.writeField255(oprot); err != nil { fieldId = 255 goto WriteFieldError @@ -20451,11 +20712,11 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ListTrajectoryRequest) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 1); err != nil { +func (p *GetTrajectoryConfigRequest) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteString(p.PlatformType); err != nil { + if err := oprot.WriteI64(p.WorkspaceID); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -20467,107 +20728,40 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ListTrajectoryRequest) writeField2(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 2); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI64(p.WorkspaceID); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError +func (p *GetTrajectoryConfigRequest) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBase() { + if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.Base.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } return nil WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) -} -func (p *ListTrajectoryRequest) writeField3(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("trace_ids", thrift.LIST, 3); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteListBegin(thrift.STRING, len(p.TraceIds)); err != nil { - return err - } - for _, v := range p.TraceIds { - if err := oprot.WriteString(v); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) -} -func (p *ListTrajectoryRequest) writeField4(oprot thrift.TProtocol) (err error) { - if p.IsSetStartTime() { - if err = oprot.WriteFieldBegin("start_time", thrift.I64, 4); err != nil { - goto WriteFieldBeginError - } - if err := oprot.WriteI64(*p.StartTime); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) -} -func (p *ListTrajectoryRequest) writeField255(oprot thrift.TProtocol) (err error) { - if p.IsSetBase() { - if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { - goto WriteFieldBeginError - } - if err := p.Base.Write(oprot); err != nil { - return err - } - if err = oprot.WriteFieldEnd(); err != nil { - goto WriteFieldEndError - } - } - return nil -WriteFieldBeginError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) -WriteFieldEndError: - return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *ListTrajectoryRequest) String() string { +func (p *GetTrajectoryConfigRequest) String() string { if p == nil { return "" } - return fmt.Sprintf("ListTrajectoryRequest(%+v)", *p) + return fmt.Sprintf("GetTrajectoryConfigRequest(%+v)", *p) } -func (p *ListTrajectoryRequest) DeepEqual(ano *ListTrajectoryRequest) bool { +func (p *GetTrajectoryConfigRequest) DeepEqual(ano *GetTrajectoryConfigRequest) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.PlatformType) { - return false - } - if !p.Field2DeepEqual(ano.WorkspaceID) { - return false - } - if !p.Field3DeepEqual(ano.TraceIds) { - return false - } - if !p.Field4DeepEqual(ano.StartTime) { + if !p.Field1DeepEqual(ano.WorkspaceID) { return false } if !p.Field255DeepEqual(ano.Base) { @@ -20576,46 +20770,14 @@ func (p *ListTrajectoryRequest) DeepEqual(ano *ListTrajectoryRequest) bool { return true } -func (p *ListTrajectoryRequest) Field1DeepEqual(src common.PlatformType) bool { - - if strings.Compare(p.PlatformType, src) != 0 { - return false - } - return true -} -func (p *ListTrajectoryRequest) Field2DeepEqual(src int64) bool { +func (p *GetTrajectoryConfigRequest) Field1DeepEqual(src int64) bool { if p.WorkspaceID != src { return false } return true } -func (p *ListTrajectoryRequest) Field3DeepEqual(src []string) bool { - - if len(p.TraceIds) != len(src) { - return false - } - for i, v := range p.TraceIds { - _src := src[i] - if strings.Compare(v, _src) != 0 { - return false - } - } - return true -} -func (p *ListTrajectoryRequest) Field4DeepEqual(src *int64) bool { - - if p.StartTime == src { - return true - } else if p.StartTime == nil || src == nil { - return false - } - if *p.StartTime != *src { - return false - } - return true -} -func (p *ListTrajectoryRequest) Field255DeepEqual(src *base.Base) bool { +func (p *GetTrajectoryConfigRequest) Field255DeepEqual(src *base.Base) bool { if !p.Base.DeepEqual(src) { return false @@ -20623,62 +20785,62 @@ func (p *ListTrajectoryRequest) Field255DeepEqual(src *base.Base) bool { return true } -type ListTrajectoryResponse struct { - Trajectories []*trajectory.Trajectory `thrift:"trajectories,1,optional" frugal:"1,optional,list" form:"trajectories" json:"trajectories,omitempty" query:"trajectories"` - BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` +type GetTrajectoryConfigResponse struct { + Filters *filter.FilterFields `thrift:"filters,1,optional" frugal:"1,optional,filter.FilterFields" form:"filters" json:"filters,omitempty" query:"filters"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` } -func NewListTrajectoryResponse() *ListTrajectoryResponse { - return &ListTrajectoryResponse{} +func NewGetTrajectoryConfigResponse() *GetTrajectoryConfigResponse { + return &GetTrajectoryConfigResponse{} } -func (p *ListTrajectoryResponse) InitDefault() { +func (p *GetTrajectoryConfigResponse) InitDefault() { } -var ListTrajectoryResponse_Trajectories_DEFAULT []*trajectory.Trajectory +var GetTrajectoryConfigResponse_Filters_DEFAULT *filter.FilterFields -func (p *ListTrajectoryResponse) GetTrajectories() (v []*trajectory.Trajectory) { +func (p *GetTrajectoryConfigResponse) GetFilters() (v *filter.FilterFields) { if p == nil { return } - if !p.IsSetTrajectories() { - return ListTrajectoryResponse_Trajectories_DEFAULT + if !p.IsSetFilters() { + return GetTrajectoryConfigResponse_Filters_DEFAULT } - return p.Trajectories + return p.Filters } -var ListTrajectoryResponse_BaseResp_DEFAULT *base.BaseResp +var GetTrajectoryConfigResponse_BaseResp_DEFAULT *base.BaseResp -func (p *ListTrajectoryResponse) GetBaseResp() (v *base.BaseResp) { +func (p *GetTrajectoryConfigResponse) GetBaseResp() (v *base.BaseResp) { if p == nil { return } if !p.IsSetBaseResp() { - return ListTrajectoryResponse_BaseResp_DEFAULT + return GetTrajectoryConfigResponse_BaseResp_DEFAULT } return p.BaseResp } -func (p *ListTrajectoryResponse) SetTrajectories(val []*trajectory.Trajectory) { - p.Trajectories = val +func (p *GetTrajectoryConfigResponse) SetFilters(val *filter.FilterFields) { + p.Filters = val } -func (p *ListTrajectoryResponse) SetBaseResp(val *base.BaseResp) { +func (p *GetTrajectoryConfigResponse) SetBaseResp(val *base.BaseResp) { p.BaseResp = val } -var fieldIDToName_ListTrajectoryResponse = map[int16]string{ - 1: "trajectories", +var fieldIDToName_GetTrajectoryConfigResponse = map[int16]string{ + 1: "filters", 255: "BaseResp", } -func (p *ListTrajectoryResponse) IsSetTrajectories() bool { - return p.Trajectories != nil +func (p *GetTrajectoryConfigResponse) IsSetFilters() bool { + return p.Filters != nil } -func (p *ListTrajectoryResponse) IsSetBaseResp() bool { +func (p *GetTrajectoryConfigResponse) IsSetBaseResp() bool { return p.BaseResp != nil } -func (p *ListTrajectoryResponse) Read(iprot thrift.TProtocol) (err error) { +func (p *GetTrajectoryConfigResponse) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -20697,7 +20859,7 @@ func (p *ListTrajectoryResponse) Read(iprot thrift.TProtocol) (err error) { switch fieldId { case 1: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.STRUCT { if err = p.ReadField1(iprot); err != nil { goto ReadFieldError } @@ -20731,7 +20893,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListTrajectoryResponse[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_GetTrajectoryConfigResponse[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -20741,30 +20903,15 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *ListTrajectoryResponse) ReadField1(iprot thrift.TProtocol) error { - _, size, err := iprot.ReadListBegin() - if err != nil { - return err - } - _field := make([]*trajectory.Trajectory, 0, size) - values := make([]trajectory.Trajectory, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - - if err := _elem.Read(iprot); err != nil { - return err - } - - _field = append(_field, _elem) - } - if err := iprot.ReadListEnd(); err != nil { +func (p *GetTrajectoryConfigResponse) ReadField1(iprot thrift.TProtocol) error { + _field := filter.NewFilterFields() + if err := _field.Read(iprot); err != nil { return err } - p.Trajectories = _field + p.Filters = _field return nil } -func (p *ListTrajectoryResponse) ReadField255(iprot thrift.TProtocol) error { +func (p *GetTrajectoryConfigResponse) ReadField255(iprot thrift.TProtocol) error { _field := base.NewBaseResp() if err := _field.Read(iprot); err != nil { return err @@ -20773,9 +20920,9 @@ func (p *ListTrajectoryResponse) ReadField255(iprot thrift.TProtocol) error { return nil } -func (p *ListTrajectoryResponse) Write(oprot thrift.TProtocol) (err error) { +func (p *GetTrajectoryConfigResponse) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListTrajectoryResponse"); err != nil { + if err = oprot.WriteStructBegin("GetTrajectoryConfigResponse"); err != nil { goto WriteStructBeginError } if p != nil { @@ -20805,20 +20952,12 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *ListTrajectoryResponse) writeField1(oprot thrift.TProtocol) (err error) { - if p.IsSetTrajectories() { - if err = oprot.WriteFieldBegin("trajectories", thrift.LIST, 1); err != nil { +func (p *GetTrajectoryConfigResponse) writeField1(oprot thrift.TProtocol) (err error) { + if p.IsSetFilters() { + if err = oprot.WriteFieldBegin("filters", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } - if err := oprot.WriteListBegin(thrift.STRUCT, len(p.Trajectories)); err != nil { - return err - } - for _, v := range p.Trajectories { - if err := v.Write(oprot); err != nil { - return err - } - } - if err := oprot.WriteListEnd(); err != nil { + if err := p.Filters.Write(oprot); err != nil { return err } if err = oprot.WriteFieldEnd(); err != nil { @@ -20831,7 +20970,7 @@ WriteFieldBeginError: WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *ListTrajectoryResponse) writeField255(oprot thrift.TProtocol) (err error) { +func (p *GetTrajectoryConfigResponse) writeField255(oprot thrift.TProtocol) (err error) { if p.IsSetBaseResp() { if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { goto WriteFieldBeginError @@ -20850,21 +20989,21 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *ListTrajectoryResponse) String() string { +func (p *GetTrajectoryConfigResponse) String() string { if p == nil { return "" } - return fmt.Sprintf("ListTrajectoryResponse(%+v)", *p) + return fmt.Sprintf("GetTrajectoryConfigResponse(%+v)", *p) } -func (p *ListTrajectoryResponse) DeepEqual(ano *ListTrajectoryResponse) bool { +func (p *GetTrajectoryConfigResponse) DeepEqual(ano *GetTrajectoryConfigResponse) bool { if p == ano { return true } else if p == nil || ano == nil { return false } - if !p.Field1DeepEqual(ano.Trajectories) { + if !p.Field1DeepEqual(ano.Filters) { return false } if !p.Field255DeepEqual(ano.BaseResp) { @@ -20873,20 +21012,14 @@ func (p *ListTrajectoryResponse) DeepEqual(ano *ListTrajectoryResponse) bool { return true } -func (p *ListTrajectoryResponse) Field1DeepEqual(src []*trajectory.Trajectory) bool { +func (p *GetTrajectoryConfigResponse) Field1DeepEqual(src *filter.FilterFields) bool { - if len(p.Trajectories) != len(src) { + if !p.Filters.DeepEqual(src) { return false } - for i, v := range p.Trajectories { - _src := src[i] - if !v.DeepEqual(_src) { - return false - } - } return true } -func (p *ListTrajectoryResponse) Field255DeepEqual(src *base.BaseResp) bool { +func (p *GetTrajectoryConfigResponse) Field255DeepEqual(src *base.BaseResp) bool { if !p.BaseResp.DeepEqual(src) { return false @@ -20894,361 +21027,2620 @@ func (p *ListTrajectoryResponse) Field255DeepEqual(src *base.BaseResp) bool { return true } -type TraceService interface { - ListSpans(ctx context.Context, req *ListSpansRequest) (r *ListSpansResponse, err error) - - ListPreSpan(ctx context.Context, req *ListPreSpanRequest) (r *ListPreSpanResponse, err error) +type ListTrajectoryRequest struct { + // 需要准确填写，用于确定查询哪些租户的数据 + PlatformType common.PlatformType `thrift:"platform_type,1,required" frugal:"1,required,string" form:"platform_type,required" json:"platform_type,required"` + WorkspaceID int64 `thrift:"workspace_id,2,required" frugal:"2,required,i64" form:"workspace_id,required" json:"workspace_id,string,required"` + TraceIds []string `thrift:"trace_ids,3,required" frugal:"3,required,list" form:"trace_ids,required" json:"trace_ids,required"` + // ms + StartTime *int64 `thrift:"start_time,4,optional" frugal:"4,optional,i64" json:"start_time" form:"start_time" ` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` +} - GetTrace(ctx context.Context, req *GetTraceRequest) (r *GetTraceResponse, err error) +func NewListTrajectoryRequest() *ListTrajectoryRequest { + return &ListTrajectoryRequest{} +} - SearchTraceTree(ctx context.Context, req *SearchTraceTreeRequest) (r *SearchTraceTreeResponse, err error) +func (p *ListTrajectoryRequest) InitDefault() { +} - BatchGetTracesAdvanceInfo(ctx context.Context, req *BatchGetTracesAdvanceInfoRequest) (r *BatchGetTracesAdvanceInfoResponse, err error) +func (p *ListTrajectoryRequest) GetPlatformType() (v common.PlatformType) { + if p != nil { + return p.PlatformType + } + return +} - IngestTracesInner(ctx context.Context, req *IngestTracesRequest) (r *IngestTracesResponse, err error) - - GetTracesMetaInfo(ctx context.Context, req *GetTracesMetaInfoRequest) (r *GetTracesMetaInfoResponse, err error) +func (p *ListTrajectoryRequest) GetWorkspaceID() (v int64) { + if p != nil { + return p.WorkspaceID + } + return +} - CreateView(ctx context.Context, req *CreateViewRequest) (r *CreateViewResponse, err error) +func (p *ListTrajectoryRequest) GetTraceIds() (v []string) { + if p != nil { + return p.TraceIds + } + return +} - UpdateView(ctx context.Context, req *UpdateViewRequest) (r *UpdateViewResponse, err error) +var ListTrajectoryRequest_StartTime_DEFAULT int64 - DeleteView(ctx context.Context, req *DeleteViewRequest) (r *DeleteViewResponse, err error) +func (p *ListTrajectoryRequest) GetStartTime() (v int64) { + if p == nil { + return + } + if !p.IsSetStartTime() { + return ListTrajectoryRequest_StartTime_DEFAULT + } + return *p.StartTime +} - ListViews(ctx context.Context, req *ListViewsRequest) (r *ListViewsResponse, err error) +var ListTrajectoryRequest_Base_DEFAULT *base.Base - CreateManualAnnotation(ctx context.Context, req *CreateManualAnnotationRequest) (r *CreateManualAnnotationResponse, err error) +func (p *ListTrajectoryRequest) GetBase() (v *base.Base) { + if p == nil { + return + } + if !p.IsSetBase() { + return ListTrajectoryRequest_Base_DEFAULT + } + return p.Base +} +func (p *ListTrajectoryRequest) SetPlatformType(val common.PlatformType) { + p.PlatformType = val +} +func (p *ListTrajectoryRequest) SetWorkspaceID(val int64) { + p.WorkspaceID = val +} +func (p *ListTrajectoryRequest) SetTraceIds(val []string) { + p.TraceIds = val +} +func (p *ListTrajectoryRequest) SetStartTime(val *int64) { + p.StartTime = val +} +func (p *ListTrajectoryRequest) SetBase(val *base.Base) { + p.Base = val +} - UpdateManualAnnotation(ctx context.Context, req *UpdateManualAnnotationRequest) (r *UpdateManualAnnotationResponse, err error) +var fieldIDToName_ListTrajectoryRequest = map[int16]string{ + 1: "platform_type", + 2: "workspace_id", + 3: "trace_ids", + 4: "start_time", + 255: "Base", +} - DeleteManualAnnotation(ctx context.Context, req *DeleteManualAnnotationRequest) (r *DeleteManualAnnotationResponse, err error) +func (p *ListTrajectoryRequest) IsSetStartTime() bool { + return p.StartTime != nil +} - ListAnnotations(ctx context.Context, req *ListAnnotationsRequest) (r *ListAnnotationsResponse, err error) +func (p *ListTrajectoryRequest) IsSetBase() bool { + return p.Base != nil +} - ExportTracesToDataset(ctx context.Context, req *ExportTracesToDatasetRequest) (r *ExportTracesToDatasetResponse, err error) +func (p *ListTrajectoryRequest) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + var issetPlatformType bool = false + var issetWorkspaceID bool = false + var issetTraceIds bool = false - PreviewExportTracesToDataset(ctx context.Context, req *PreviewExportTracesToDatasetRequest) (r *PreviewExportTracesToDatasetResponse, err error) + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } - ChangeEvaluatorScore(ctx context.Context, req *ChangeEvaluatorScoreRequest) (r *ChangeEvaluatorScoreResponse, err error) + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } - ListAnnotationEvaluators(ctx context.Context, req *ListAnnotationEvaluatorsRequest) (r *ListAnnotationEvaluatorsResponse, err error) + switch fieldId { + case 1: + if fieldTypeId == thrift.STRING { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + issetPlatformType = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 2: + if fieldTypeId == thrift.I64 { + if err = p.ReadField2(iprot); err != nil { + goto ReadFieldError + } + issetWorkspaceID = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 3: + if fieldTypeId == thrift.LIST { + if err = p.ReadField3(iprot); err != nil { + goto ReadFieldError + } + issetTraceIds = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 4: + if fieldTypeId == thrift.I64 { + if err = p.ReadField4(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } - ExtractSpanInfo(ctx context.Context, req *ExtractSpanInfoRequest) (r *ExtractSpanInfoResponse, err error) + if !issetPlatformType { + fieldId = 1 + goto RequiredFieldNotSetError + } - UpsertTrajectoryConfig(ctx context.Context, req *UpsertTrajectoryConfigRequest) (r *UpsertTrajectoryConfigResponse, err error) + if !issetWorkspaceID { + fieldId = 2 + goto RequiredFieldNotSetError + } - GetTrajectoryConfig(ctx context.Context, req *GetTrajectoryConfigRequest) (r *GetTrajectoryConfigResponse, err error) + if !issetTraceIds { + fieldId = 3 + goto RequiredFieldNotSetError + } + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListTrajectoryRequest[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) - ListTrajectory(ctx context.Context, req *ListTrajectoryRequest) (r *ListTrajectoryResponse, err error) +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +RequiredFieldNotSetError: + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListTrajectoryRequest[fieldId])) } -type TraceServiceClient struct { - c thrift.TClient -} +func (p *ListTrajectoryRequest) ReadField1(iprot thrift.TProtocol) error { -func NewTraceServiceClientFactory(t thrift.TTransport, f thrift.TProtocolFactory) *TraceServiceClient { - return &TraceServiceClient{ - c: thrift.NewTStandardClient(f.GetProtocol(t), f.GetProtocol(t)), + var _field common.PlatformType + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v } + p.PlatformType = _field + return nil } +func (p *ListTrajectoryRequest) ReadField2(iprot thrift.TProtocol) error { -func NewTraceServiceClientProtocol(t thrift.TTransport, iprot thrift.TProtocol, oprot thrift.TProtocol) *TraceServiceClient { - return &TraceServiceClient{ - c: thrift.NewTStandardClient(iprot, oprot), + var _field int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = v } + p.WorkspaceID = _field + return nil } - -func NewTraceServiceClient(c thrift.TClient) *TraceServiceClient { - return &TraceServiceClient{ - c: c, +func (p *ListTrajectoryRequest) ReadField3(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err } -} + _field := make([]string, 0, size) + for i := 0; i < size; i++ { -func (p *TraceServiceClient) Client_() thrift.TClient { - return p.c -} + var _elem string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _elem = v + } -func (p *TraceServiceClient) ListSpans(ctx context.Context, req *ListSpansRequest) (r *ListSpansResponse, err error) { - var _args TraceServiceListSpansArgs - _args.Req = req - var _result TraceServiceListSpansResult - if err = p.Client_().Call(ctx, "ListSpans", &_args, &_result); err != nil { - return + _field = append(_field, _elem) } - return _result.GetSuccess(), nil + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.TraceIds = _field + return nil } -func (p *TraceServiceClient) ListPreSpan(ctx context.Context, req *ListPreSpanRequest) (r *ListPreSpanResponse, err error) { - var _args TraceServiceListPreSpanArgs - _args.Req = req - var _result TraceServiceListPreSpanResult - if err = p.Client_().Call(ctx, "ListPreSpan", &_args, &_result); err != nil { - return +func (p *ListTrajectoryRequest) ReadField4(iprot thrift.TProtocol) error { + + var _field *int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = &v } - return _result.GetSuccess(), nil + p.StartTime = _field + return nil } -func (p *TraceServiceClient) GetTrace(ctx context.Context, req *GetTraceRequest) (r *GetTraceResponse, err error) { - var _args TraceServiceGetTraceArgs - _args.Req = req - var _result TraceServiceGetTraceResult - if err = p.Client_().Call(ctx, "GetTrace", &_args, &_result); err != nil { - return +func (p *ListTrajectoryRequest) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBase() + if err := _field.Read(iprot); err != nil { + return err } - return _result.GetSuccess(), nil + p.Base = _field + return nil } -func (p *TraceServiceClient) SearchTraceTree(ctx context.Context, req *SearchTraceTreeRequest) (r *SearchTraceTreeResponse, err error) { - var _args TraceServiceSearchTraceTreeArgs - _args.Req = req - var _result TraceServiceSearchTraceTreeResult - if err = p.Client_().Call(ctx, "SearchTraceTree", &_args, &_result); err != nil { - return + +func (p *ListTrajectoryRequest) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListTrajectoryRequest"); err != nil { + goto WriteStructBeginError } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) BatchGetTracesAdvanceInfo(ctx context.Context, req *BatchGetTracesAdvanceInfoRequest) (r *BatchGetTracesAdvanceInfoResponse, err error) { - var _args TraceServiceBatchGetTracesAdvanceInfoArgs - _args.Req = req - var _result TraceServiceBatchGetTracesAdvanceInfoResult - if err = p.Client_().Call(ctx, "BatchGetTracesAdvanceInfo", &_args, &_result); err != nil { - return + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } + if err = p.writeField3(oprot); err != nil { + fieldId = 3 + goto WriteFieldError + } + if err = p.writeField4(oprot); err != nil { + fieldId = 4 + goto WriteFieldError + } + if err = p.writeField255(oprot); err != nil { + fieldId = 255 + goto WriteFieldError + } } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) IngestTracesInner(ctx context.Context, req *IngestTracesRequest) (r *IngestTracesResponse, err error) { - var _args TraceServiceIngestTracesInnerArgs - _args.Req = req - var _result TraceServiceIngestTracesInnerResult - if err = p.Client_().Call(ctx, "IngestTracesInner", &_args, &_result); err != nil { - return + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) GetTracesMetaInfo(ctx context.Context, req *GetTracesMetaInfoRequest) (r *GetTracesMetaInfoResponse, err error) { - var _args TraceServiceGetTracesMetaInfoArgs - _args.Req = req - var _result TraceServiceGetTracesMetaInfoResult - if err = p.Client_().Call(ctx, "GetTracesMetaInfo", &_args, &_result); err != nil { - return + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError } - return _result.GetSuccess(), nil + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceClient) CreateView(ctx context.Context, req *CreateViewRequest) (r *CreateViewResponse, err error) { - var _args TraceServiceCreateViewArgs - _args.Req = req - var _result TraceServiceCreateViewResult - if err = p.Client_().Call(ctx, "CreateView", &_args, &_result); err != nil { - return + +func (p *ListTrajectoryRequest) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 1); err != nil { + goto WriteFieldBeginError } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) UpdateView(ctx context.Context, req *UpdateViewRequest) (r *UpdateViewResponse, err error) { - var _args TraceServiceUpdateViewArgs - _args.Req = req - var _result TraceServiceUpdateViewResult - if err = p.Client_().Call(ctx, "UpdateView", &_args, &_result); err != nil { - return + if err := oprot.WriteString(p.PlatformType); err != nil { + return err } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) DeleteView(ctx context.Context, req *DeleteViewRequest) (r *DeleteViewResponse, err error) { - var _args TraceServiceDeleteViewArgs - _args.Req = req - var _result TraceServiceDeleteViewResult - if err = p.Client_().Call(ctx, "DeleteView", &_args, &_result); err != nil { - return + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } - return _result.GetSuccess(), nil + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceClient) ListViews(ctx context.Context, req *ListViewsRequest) (r *ListViewsResponse, err error) { - var _args TraceServiceListViewsArgs - _args.Req = req - var _result TraceServiceListViewsResult - if err = p.Client_().Call(ctx, "ListViews", &_args, &_result); err != nil { - return +func (p *ListTrajectoryRequest) writeField2(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 2); err != nil { + goto WriteFieldBeginError } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) CreateManualAnnotation(ctx context.Context, req *CreateManualAnnotationRequest) (r *CreateManualAnnotationResponse, err error) { - var _args TraceServiceCreateManualAnnotationArgs - _args.Req = req - var _result TraceServiceCreateManualAnnotationResult - if err = p.Client_().Call(ctx, "CreateManualAnnotation", &_args, &_result); err != nil { - return + if err := oprot.WriteI64(p.WorkspaceID); err != nil { + return err } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) UpdateManualAnnotation(ctx context.Context, req *UpdateManualAnnotationRequest) (r *UpdateManualAnnotationResponse, err error) { - var _args TraceServiceUpdateManualAnnotationArgs - _args.Req = req - var _result TraceServiceUpdateManualAnnotationResult - if err = p.Client_().Call(ctx, "UpdateManualAnnotation", &_args, &_result); err != nil { - return + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } - return _result.GetSuccess(), nil + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) } -func (p *TraceServiceClient) DeleteManualAnnotation(ctx context.Context, req *DeleteManualAnnotationRequest) (r *DeleteManualAnnotationResponse, err error) { - var _args TraceServiceDeleteManualAnnotationArgs - _args.Req = req - var _result TraceServiceDeleteManualAnnotationResult - if err = p.Client_().Call(ctx, "DeleteManualAnnotation", &_args, &_result); err != nil { - return +func (p *ListTrajectoryRequest) writeField3(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("trace_ids", thrift.LIST, 3); err != nil { + goto WriteFieldBeginError } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) ListAnnotations(ctx context.Context, req *ListAnnotationsRequest) (r *ListAnnotationsResponse, err error) { - var _args TraceServiceListAnnotationsArgs - _args.Req = req - var _result TraceServiceListAnnotationsResult - if err = p.Client_().Call(ctx, "ListAnnotations", &_args, &_result); err != nil { - return + if err := oprot.WriteListBegin(thrift.STRING, len(p.TraceIds)); err != nil { + return err } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) ExportTracesToDataset(ctx context.Context, req *ExportTracesToDatasetRequest) (r *ExportTracesToDatasetResponse, err error) { - var _args TraceServiceExportTracesToDatasetArgs - _args.Req = req - var _result TraceServiceExportTracesToDatasetResult - if err = p.Client_().Call(ctx, "ExportTracesToDataset", &_args, &_result); err != nil { - return + for _, v := range p.TraceIds { + if err := oprot.WriteString(v); err != nil { + return err + } } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) PreviewExportTracesToDataset(ctx context.Context, req *PreviewExportTracesToDatasetRequest) (r *PreviewExportTracesToDatasetResponse, err error) { - var _args TraceServicePreviewExportTracesToDatasetArgs - _args.Req = req - var _result TraceServicePreviewExportTracesToDatasetResult - if err = p.Client_().Call(ctx, "PreviewExportTracesToDataset", &_args, &_result); err != nil { - return + if err := oprot.WriteListEnd(); err != nil { + return err } - return _result.GetSuccess(), nil -} -func (p *TraceServiceClient) ChangeEvaluatorScore(ctx context.Context, req *ChangeEvaluatorScoreRequest) (r *ChangeEvaluatorScoreResponse, err error) { - var _args TraceServiceChangeEvaluatorScoreArgs - _args.Req = req - var _result TraceServiceChangeEvaluatorScoreResult - if err = p.Client_().Call(ctx, "ChangeEvaluatorScore", &_args, &_result); err != nil { - return + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } - return _result.GetSuccess(), nil + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) } -func (p *TraceServiceClient) ListAnnotationEvaluators(ctx context.Context, req *ListAnnotationEvaluatorsRequest) (r *ListAnnotationEvaluatorsResponse, err error) { - var _args TraceServiceListAnnotationEvaluatorsArgs - _args.Req = req - var _result TraceServiceListAnnotationEvaluatorsResult - if err = p.Client_().Call(ctx, "ListAnnotationEvaluators", &_args, &_result); err != nil { - return +func (p *ListTrajectoryRequest) writeField4(oprot thrift.TProtocol) (err error) { + if p.IsSetStartTime() { + if err = oprot.WriteFieldBegin("start_time", thrift.I64, 4); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(*p.StartTime); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } - return _result.GetSuccess(), nil + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err) } -func (p *TraceServiceClient) ExtractSpanInfo(ctx context.Context, req *ExtractSpanInfoRequest) (r *ExtractSpanInfoResponse, err error) { - var _args TraceServiceExtractSpanInfoArgs - _args.Req = req - var _result TraceServiceExtractSpanInfoResult - if err = p.Client_().Call(ctx, "ExtractSpanInfo", &_args, &_result); err != nil { - return +func (p *ListTrajectoryRequest) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBase() { + if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.Base.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } } - return _result.GetSuccess(), nil + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) } -func (p *TraceServiceClient) UpsertTrajectoryConfig(ctx context.Context, req *UpsertTrajectoryConfigRequest) (r *UpsertTrajectoryConfigResponse, err error) { - var _args TraceServiceUpsertTrajectoryConfigArgs + +func (p *ListTrajectoryRequest) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("ListTrajectoryRequest(%+v)", *p) + +} + +func (p *ListTrajectoryRequest) DeepEqual(ano *ListTrajectoryRequest) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.PlatformType) { + return false + } + if !p.Field2DeepEqual(ano.WorkspaceID) { + return false + } + if !p.Field3DeepEqual(ano.TraceIds) { + return false + } + if !p.Field4DeepEqual(ano.StartTime) { + return false + } + if !p.Field255DeepEqual(ano.Base) { + return false + } + return true +} + +func (p *ListTrajectoryRequest) Field1DeepEqual(src common.PlatformType) bool { + + if strings.Compare(p.PlatformType, src) != 0 { + return false + } + return true +} +func (p *ListTrajectoryRequest) Field2DeepEqual(src int64) bool { + + if p.WorkspaceID != src { + return false + } + return true +} +func (p *ListTrajectoryRequest) Field3DeepEqual(src []string) bool { + + if len(p.TraceIds) != len(src) { + return false + } + for i, v := range p.TraceIds { + _src := src[i] + if strings.Compare(v, _src) != 0 { + return false + } + } + return true +} +func (p *ListTrajectoryRequest) Field4DeepEqual(src *int64) bool { + + if p.StartTime == src { + return true + } else if p.StartTime == nil || src == nil { + return false + } + if *p.StartTime != *src { + return false + } + return true +} +func (p *ListTrajectoryRequest) Field255DeepEqual(src *base.Base) bool { + + if !p.Base.DeepEqual(src) { + return false + } + return true +} + +type ListTrajectoryResponse struct { + Trajectories []*trajectory.Trajectory `thrift:"trajectories,1,optional" frugal:"1,optional,list" form:"trajectories" json:"trajectories,omitempty" query:"trajectories"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` +} + +func NewListTrajectoryResponse() *ListTrajectoryResponse { + return &ListTrajectoryResponse{} +} + +func (p *ListTrajectoryResponse) InitDefault() { +} + +var ListTrajectoryResponse_Trajectories_DEFAULT []*trajectory.Trajectory + +func (p *ListTrajectoryResponse) GetTrajectories() (v []*trajectory.Trajectory) { + if p == nil { + return + } + if !p.IsSetTrajectories() { + return ListTrajectoryResponse_Trajectories_DEFAULT + } + return p.Trajectories +} + +var ListTrajectoryResponse_BaseResp_DEFAULT *base.BaseResp + +func (p *ListTrajectoryResponse) GetBaseResp() (v *base.BaseResp) { + if p == nil { + return + } + if !p.IsSetBaseResp() { + return ListTrajectoryResponse_BaseResp_DEFAULT + } + return p.BaseResp +} +func (p *ListTrajectoryResponse) SetTrajectories(val []*trajectory.Trajectory) { + p.Trajectories = val +} +func (p *ListTrajectoryResponse) SetBaseResp(val *base.BaseResp) { + p.BaseResp = val +} + +var fieldIDToName_ListTrajectoryResponse = map[int16]string{ + 1: "trajectories", + 255: "BaseResp", +} + +func (p *ListTrajectoryResponse) IsSetTrajectories() bool { + return p.Trajectories != nil +} + +func (p *ListTrajectoryResponse) IsSetBaseResp() bool { + return p.BaseResp != nil +} + +func (p *ListTrajectoryResponse) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.LIST { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListTrajectoryResponse[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +} + +func (p *ListTrajectoryResponse) ReadField1(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*trajectory.Trajectory, 0, size) + values := make([]trajectory.Trajectory, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.Trajectories = _field + return nil +} +func (p *ListTrajectoryResponse) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBaseResp() + if err := _field.Read(iprot); err != nil { + return err + } + p.BaseResp = _field + return nil +} + +func (p *ListTrajectoryResponse) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListTrajectoryResponse"); err != nil { + goto WriteStructBeginError + } + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + if err = p.writeField255(oprot); err != nil { + fieldId = 255 + goto WriteFieldError + } + } + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError + } + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) +} + +func (p *ListTrajectoryResponse) writeField1(oprot thrift.TProtocol) (err error) { + if p.IsSetTrajectories() { + if err = oprot.WriteFieldBegin("trajectories", thrift.LIST, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.Trajectories)); err != nil { + return err + } + for _, v := range p.Trajectories { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *ListTrajectoryResponse) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBaseResp() { + if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.BaseResp.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) +} + +func (p *ListTrajectoryResponse) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("ListTrajectoryResponse(%+v)", *p) + +} + +func (p *ListTrajectoryResponse) DeepEqual(ano *ListTrajectoryResponse) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.Trajectories) { + return false + } + if !p.Field255DeepEqual(ano.BaseResp) { + return false + } + return true +} + +func (p *ListTrajectoryResponse) Field1DeepEqual(src []*trajectory.Trajectory) bool { + + if len(p.Trajectories) != len(src) { + return false + } + for i, v := range p.Trajectories { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true +} +func (p *ListTrajectoryResponse) Field255DeepEqual(src *base.BaseResp) bool { + + if !p.BaseResp.DeepEqual(src) { + return false + } + return true +} + +type ListMetadataRequest struct { + WorkspaceID int64 `thrift:"workspace_id,1,required" frugal:"1,required,i64" json:"workspace_id" form:"workspace_id,required" ` + PlatformType *common.PlatformType `thrift:"platform_type,2,optional" frugal:"2,optional,string" form:"platform_type" json:"platform_type,omitempty"` + SpanListType *common.SpanListType `thrift:"span_list_type,3,optional" frugal:"3,optional,string" form:"span_list_type" json:"span_list_type,omitempty"` + Base *base.Base `thrift:"Base,255,optional" frugal:"255,optional,base.Base" form:"Base" json:"Base,omitempty" query:"Base"` +} + +func NewListMetadataRequest() *ListMetadataRequest { + return &ListMetadataRequest{} +} + +func (p *ListMetadataRequest) InitDefault() { +} + +func (p *ListMetadataRequest) GetWorkspaceID() (v int64) { + if p != nil { + return p.WorkspaceID + } + return +} + +var ListMetadataRequest_PlatformType_DEFAULT common.PlatformType + +func (p *ListMetadataRequest) GetPlatformType() (v common.PlatformType) { + if p == nil { + return + } + if !p.IsSetPlatformType() { + return ListMetadataRequest_PlatformType_DEFAULT + } + return *p.PlatformType +} + +var ListMetadataRequest_SpanListType_DEFAULT common.SpanListType + +func (p *ListMetadataRequest) GetSpanListType() (v common.SpanListType) { + if p == nil { + return + } + if !p.IsSetSpanListType() { + return ListMetadataRequest_SpanListType_DEFAULT + } + return *p.SpanListType +} + +var ListMetadataRequest_Base_DEFAULT *base.Base + +func (p *ListMetadataRequest) GetBase() (v *base.Base) { + if p == nil { + return + } + if !p.IsSetBase() { + return ListMetadataRequest_Base_DEFAULT + } + return p.Base +} +func (p *ListMetadataRequest) SetWorkspaceID(val int64) { + p.WorkspaceID = val +} +func (p *ListMetadataRequest) SetPlatformType(val *common.PlatformType) { + p.PlatformType = val +} +func (p *ListMetadataRequest) SetSpanListType(val *common.SpanListType) { + p.SpanListType = val +} +func (p *ListMetadataRequest) SetBase(val *base.Base) { + p.Base = val +} + +var fieldIDToName_ListMetadataRequest = map[int16]string{ + 1: "workspace_id", + 2: "platform_type", + 3: "span_list_type", + 255: "Base", +} + +func (p *ListMetadataRequest) IsSetPlatformType() bool { + return p.PlatformType != nil +} + +func (p *ListMetadataRequest) IsSetSpanListType() bool { + return p.SpanListType != nil +} + +func (p *ListMetadataRequest) IsSetBase() bool { + return p.Base != nil +} + +func (p *ListMetadataRequest) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + var issetWorkspaceID bool = false + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.I64 { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + issetWorkspaceID = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 2: + if fieldTypeId == thrift.STRING { + if err = p.ReadField2(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 3: + if fieldTypeId == thrift.STRING { + if err = p.ReadField3(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + if !issetWorkspaceID { + fieldId = 1 + goto RequiredFieldNotSetError + } + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListMetadataRequest[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +RequiredFieldNotSetError: + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListMetadataRequest[fieldId])) +} + +func (p *ListMetadataRequest) ReadField1(iprot thrift.TProtocol) error { + + var _field int64 + if v, err := iprot.ReadI64(); err != nil { + return err + } else { + _field = v + } + p.WorkspaceID = _field + return nil +} +func (p *ListMetadataRequest) ReadField2(iprot thrift.TProtocol) error { + + var _field *common.PlatformType + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.PlatformType = _field + return nil +} +func (p *ListMetadataRequest) ReadField3(iprot thrift.TProtocol) error { + + var _field *common.SpanListType + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = &v + } + p.SpanListType = _field + return nil +} +func (p *ListMetadataRequest) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBase() + if err := _field.Read(iprot); err != nil { + return err + } + p.Base = _field + return nil +} + +func (p *ListMetadataRequest) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListMetadataRequest"); err != nil { + goto WriteStructBeginError + } + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } + if err = p.writeField3(oprot); err != nil { + fieldId = 3 + goto WriteFieldError + } + if err = p.writeField255(oprot); err != nil { + fieldId = 255 + goto WriteFieldError + } + } + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError + } + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) +} + +func (p *ListMetadataRequest) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("workspace_id", thrift.I64, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteI64(p.WorkspaceID); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *ListMetadataRequest) writeField2(oprot thrift.TProtocol) (err error) { + if p.IsSetPlatformType() { + if err = oprot.WriteFieldBegin("platform_type", thrift.STRING, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.PlatformType); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} +func (p *ListMetadataRequest) writeField3(oprot thrift.TProtocol) (err error) { + if p.IsSetSpanListType() { + if err = oprot.WriteFieldBegin("span_list_type", thrift.STRING, 3); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(*p.SpanListType); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 3 end error: ", p), err) +} +func (p *ListMetadataRequest) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBase() { + if err = oprot.WriteFieldBegin("Base", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.Base.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) +} + +func (p *ListMetadataRequest) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("ListMetadataRequest(%+v)", *p) + +} + +func (p *ListMetadataRequest) DeepEqual(ano *ListMetadataRequest) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.WorkspaceID) { + return false + } + if !p.Field2DeepEqual(ano.PlatformType) { + return false + } + if !p.Field3DeepEqual(ano.SpanListType) { + return false + } + if !p.Field255DeepEqual(ano.Base) { + return false + } + return true +} + +func (p *ListMetadataRequest) Field1DeepEqual(src int64) bool { + + if p.WorkspaceID != src { + return false + } + return true +} +func (p *ListMetadataRequest) Field2DeepEqual(src *common.PlatformType) bool { + + if p.PlatformType == src { + return true + } else if p.PlatformType == nil || src == nil { + return false + } + if strings.Compare(*p.PlatformType, *src) != 0 { + return false + } + return true +} +func (p *ListMetadataRequest) Field3DeepEqual(src *common.SpanListType) bool { + + if p.SpanListType == src { + return true + } else if p.SpanListType == nil || src == nil { + return false + } + if strings.Compare(*p.SpanListType, *src) != 0 { + return false + } + return true +} +func (p *ListMetadataRequest) Field255DeepEqual(src *base.Base) bool { + + if !p.Base.DeepEqual(src) { + return false + } + return true +} + +type MetadataItemInfo struct { + Key string `thrift:"key,1,required" frugal:"1,required,string" form:"key,required" json:"key,required" query:"key,required"` + ValueType span.MetadataValueType `thrift:"value_type,2,required" frugal:"2,required,string" form:"value_type,required" json:"value_type,required" query:"value_type,required"` +} + +func NewMetadataItemInfo() *MetadataItemInfo { + return &MetadataItemInfo{} +} + +func (p *MetadataItemInfo) InitDefault() { +} + +func (p *MetadataItemInfo) GetKey() (v string) { + if p != nil { + return p.Key + } + return +} + +func (p *MetadataItemInfo) GetValueType() (v span.MetadataValueType) { + if p != nil { + return p.ValueType + } + return +} +func (p *MetadataItemInfo) SetKey(val string) { + p.Key = val +} +func (p *MetadataItemInfo) SetValueType(val span.MetadataValueType) { + p.ValueType = val +} + +var fieldIDToName_MetadataItemInfo = map[int16]string{ + 1: "key", + 2: "value_type", +} + +func (p *MetadataItemInfo) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + var issetKey bool = false + var issetValueType bool = false + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.STRING { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + issetKey = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 2: + if fieldTypeId == thrift.STRING { + if err = p.ReadField2(iprot); err != nil { + goto ReadFieldError + } + issetValueType = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + if !issetKey { + fieldId = 1 + goto RequiredFieldNotSetError + } + + if !issetValueType { + fieldId = 2 + goto RequiredFieldNotSetError + } + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_MetadataItemInfo[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +RequiredFieldNotSetError: + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_MetadataItemInfo[fieldId])) +} + +func (p *MetadataItemInfo) ReadField1(iprot thrift.TProtocol) error { + + var _field string + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v + } + p.Key = _field + return nil +} +func (p *MetadataItemInfo) ReadField2(iprot thrift.TProtocol) error { + + var _field span.MetadataValueType + if v, err := iprot.ReadString(); err != nil { + return err + } else { + _field = v + } + p.ValueType = _field + return nil +} + +func (p *MetadataItemInfo) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("MetadataItemInfo"); err != nil { + goto WriteStructBeginError + } + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + if err = p.writeField2(oprot); err != nil { + fieldId = 2 + goto WriteFieldError + } + } + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError + } + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) +} + +func (p *MetadataItemInfo) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("key", thrift.STRING, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(p.Key); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *MetadataItemInfo) writeField2(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("value_type", thrift.STRING, 2); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteString(p.ValueType); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 2 end error: ", p), err) +} + +func (p *MetadataItemInfo) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("MetadataItemInfo(%+v)", *p) + +} + +func (p *MetadataItemInfo) DeepEqual(ano *MetadataItemInfo) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.Key) { + return false + } + if !p.Field2DeepEqual(ano.ValueType) { + return false + } + return true +} + +func (p *MetadataItemInfo) Field1DeepEqual(src string) bool { + + if strings.Compare(p.Key, src) != 0 { + return false + } + return true +} +func (p *MetadataItemInfo) Field2DeepEqual(src span.MetadataValueType) bool { + + if strings.Compare(p.ValueType, src) != 0 { + return false + } + return true +} + +type ListMetadataResponse struct { + MetadataItemList []*MetadataItemInfo `thrift:"metadataItemList,1,required" frugal:"1,required,list" form:"metadataItemList,required" json:"metadataItemList,required" query:"metadataItemList,required"` + BaseResp *base.BaseResp `thrift:"BaseResp,255,optional" frugal:"255,optional,base.BaseResp" form:"BaseResp" json:"BaseResp,omitempty" query:"BaseResp"` +} + +func NewListMetadataResponse() *ListMetadataResponse { + return &ListMetadataResponse{} +} + +func (p *ListMetadataResponse) InitDefault() { +} + +func (p *ListMetadataResponse) GetMetadataItemList() (v []*MetadataItemInfo) { + if p != nil { + return p.MetadataItemList + } + return +} + +var ListMetadataResponse_BaseResp_DEFAULT *base.BaseResp + +func (p *ListMetadataResponse) GetBaseResp() (v *base.BaseResp) { + if p == nil { + return + } + if !p.IsSetBaseResp() { + return ListMetadataResponse_BaseResp_DEFAULT + } + return p.BaseResp +} +func (p *ListMetadataResponse) SetMetadataItemList(val []*MetadataItemInfo) { + p.MetadataItemList = val +} +func (p *ListMetadataResponse) SetBaseResp(val *base.BaseResp) { + p.BaseResp = val +} + +var fieldIDToName_ListMetadataResponse = map[int16]string{ + 1: "metadataItemList", + 255: "BaseResp", +} + +func (p *ListMetadataResponse) IsSetBaseResp() bool { + return p.BaseResp != nil +} + +func (p *ListMetadataResponse) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + var issetMetadataItemList bool = false + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.LIST { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + issetMetadataItemList = true + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + case 255: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField255(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + if !issetMetadataItemList { + fieldId = 1 + goto RequiredFieldNotSetError + } + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListMetadataResponse[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +RequiredFieldNotSetError: + return thrift.NewTProtocolExceptionWithType(thrift.INVALID_DATA, fmt.Errorf("required field %s is not set", fieldIDToName_ListMetadataResponse[fieldId])) +} + +func (p *ListMetadataResponse) ReadField1(iprot thrift.TProtocol) error { + _, size, err := iprot.ReadListBegin() + if err != nil { + return err + } + _field := make([]*MetadataItemInfo, 0, size) + values := make([]MetadataItemInfo, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + + if err := _elem.Read(iprot); err != nil { + return err + } + + _field = append(_field, _elem) + } + if err := iprot.ReadListEnd(); err != nil { + return err + } + p.MetadataItemList = _field + return nil +} +func (p *ListMetadataResponse) ReadField255(iprot thrift.TProtocol) error { + _field := base.NewBaseResp() + if err := _field.Read(iprot); err != nil { + return err + } + p.BaseResp = _field + return nil +} + +func (p *ListMetadataResponse) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListMetadataResponse"); err != nil { + goto WriteStructBeginError + } + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + if err = p.writeField255(oprot); err != nil { + fieldId = 255 + goto WriteFieldError + } + } + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError + } + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) +} + +func (p *ListMetadataResponse) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("metadataItemList", thrift.LIST, 1); err != nil { + goto WriteFieldBeginError + } + if err := oprot.WriteListBegin(thrift.STRUCT, len(p.MetadataItemList)); err != nil { + return err + } + for _, v := range p.MetadataItemList { + if err := v.Write(oprot); err != nil { + return err + } + } + if err := oprot.WriteListEnd(); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} +func (p *ListMetadataResponse) writeField255(oprot thrift.TProtocol) (err error) { + if p.IsSetBaseResp() { + if err = oprot.WriteFieldBegin("BaseResp", thrift.STRUCT, 255); err != nil { + goto WriteFieldBeginError + } + if err := p.BaseResp.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 255 end error: ", p), err) +} + +func (p *ListMetadataResponse) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("ListMetadataResponse(%+v)", *p) + +} + +func (p *ListMetadataResponse) DeepEqual(ano *ListMetadataResponse) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.MetadataItemList) { + return false + } + if !p.Field255DeepEqual(ano.BaseResp) { + return false + } + return true +} + +func (p *ListMetadataResponse) Field1DeepEqual(src []*MetadataItemInfo) bool { + + if len(p.MetadataItemList) != len(src) { + return false + } + for i, v := range p.MetadataItemList { + _src := src[i] + if !v.DeepEqual(_src) { + return false + } + } + return true +} +func (p *ListMetadataResponse) Field255DeepEqual(src *base.BaseResp) bool { + + if !p.BaseResp.DeepEqual(src) { + return false + } + return true +} + +type TraceService interface { + ListSpans(ctx context.Context, req *ListSpansRequest) (r *ListSpansResponse, err error) + + ListPreSpan(ctx context.Context, req *ListPreSpanRequest) (r *ListPreSpanResponse, err error) + + GetTrace(ctx context.Context, req *GetTraceRequest) (r *GetTraceResponse, err error) + + SearchTraceTree(ctx context.Context, req *SearchTraceTreeRequest) (r *SearchTraceTreeResponse, err error) + + BatchGetTracesAdvanceInfo(ctx context.Context, req *BatchGetTracesAdvanceInfoRequest) (r *BatchGetTracesAdvanceInfoResponse, err error) + + IngestTracesInner(ctx context.Context, req *IngestTracesRequest) (r *IngestTracesResponse, err error) + + GetTracesMetaInfo(ctx context.Context, req *GetTracesMetaInfoRequest) (r *GetTracesMetaInfoResponse, err error) + + CreateView(ctx context.Context, req *CreateViewRequest) (r *CreateViewResponse, err error) + + UpdateView(ctx context.Context, req *UpdateViewRequest) (r *UpdateViewResponse, err error) + + DeleteView(ctx context.Context, req *DeleteViewRequest) (r *DeleteViewResponse, err error) + + ListViews(ctx context.Context, req *ListViewsRequest) (r *ListViewsResponse, err error) + + CreateManualAnnotation(ctx context.Context, req *CreateManualAnnotationRequest) (r *CreateManualAnnotationResponse, err error) + + UpdateManualAnnotation(ctx context.Context, req *UpdateManualAnnotationRequest) (r *UpdateManualAnnotationResponse, err error) + + DeleteManualAnnotation(ctx context.Context, req *DeleteManualAnnotationRequest) (r *DeleteManualAnnotationResponse, err error) + + ListAnnotations(ctx context.Context, req *ListAnnotationsRequest) (r *ListAnnotationsResponse, err error) + + ListWorkspaceAnnotations(ctx context.Context, req *ListWorkspaceAnnotationsRequest) (r *ListWorkspaceAnnotationsResponse, err error) + + ExportTracesToDataset(ctx context.Context, req *ExportTracesToDatasetRequest) (r *ExportTracesToDatasetResponse, err error) + + PreviewExportTracesToDataset(ctx context.Context, req *PreviewExportTracesToDatasetRequest) (r *PreviewExportTracesToDatasetResponse, err error) + + ChangeEvaluatorScore(ctx context.Context, req *ChangeEvaluatorScoreRequest) (r *ChangeEvaluatorScoreResponse, err error) + + ListAnnotationEvaluators(ctx context.Context, req *ListAnnotationEvaluatorsRequest) (r *ListAnnotationEvaluatorsResponse, err error) + + ExtractSpanInfo(ctx context.Context, req *ExtractSpanInfoRequest) (r *ExtractSpanInfoResponse, err error) + + UpsertTrajectoryConfig(ctx context.Context, req *UpsertTrajectoryConfigRequest) (r *UpsertTrajectoryConfigResponse, err error) + + GetTrajectoryConfig(ctx context.Context, req *GetTrajectoryConfigRequest) (r *GetTrajectoryConfigResponse, err error) + + ListTrajectory(ctx context.Context, req *ListTrajectoryRequest) (r *ListTrajectoryResponse, err error) + + ListMetadata(ctx context.Context, req *ListMetadataRequest) (r *ListMetadataResponse, err error) +} + +type TraceServiceClient struct { + c thrift.TClient +} + +func NewTraceServiceClientFactory(t thrift.TTransport, f thrift.TProtocolFactory) *TraceServiceClient { + return &TraceServiceClient{ + c: thrift.NewTStandardClient(f.GetProtocol(t), f.GetProtocol(t)), + } +} + +func NewTraceServiceClientProtocol(t thrift.TTransport, iprot thrift.TProtocol, oprot thrift.TProtocol) *TraceServiceClient { + return &TraceServiceClient{ + c: thrift.NewTStandardClient(iprot, oprot), + } +} + +func NewTraceServiceClient(c thrift.TClient) *TraceServiceClient { + return &TraceServiceClient{ + c: c, + } +} + +func (p *TraceServiceClient) Client_() thrift.TClient { + return p.c +} + +func (p *TraceServiceClient) ListSpans(ctx context.Context, req *ListSpansRequest) (r *ListSpansResponse, err error) { + var _args TraceServiceListSpansArgs + _args.Req = req + var _result TraceServiceListSpansResult + if err = p.Client_().Call(ctx, "ListSpans", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ListPreSpan(ctx context.Context, req *ListPreSpanRequest) (r *ListPreSpanResponse, err error) { + var _args TraceServiceListPreSpanArgs + _args.Req = req + var _result TraceServiceListPreSpanResult + if err = p.Client_().Call(ctx, "ListPreSpan", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) GetTrace(ctx context.Context, req *GetTraceRequest) (r *GetTraceResponse, err error) { + var _args TraceServiceGetTraceArgs + _args.Req = req + var _result TraceServiceGetTraceResult + if err = p.Client_().Call(ctx, "GetTrace", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) SearchTraceTree(ctx context.Context, req *SearchTraceTreeRequest) (r *SearchTraceTreeResponse, err error) { + var _args TraceServiceSearchTraceTreeArgs + _args.Req = req + var _result TraceServiceSearchTraceTreeResult + if err = p.Client_().Call(ctx, "SearchTraceTree", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) BatchGetTracesAdvanceInfo(ctx context.Context, req *BatchGetTracesAdvanceInfoRequest) (r *BatchGetTracesAdvanceInfoResponse, err error) { + var _args TraceServiceBatchGetTracesAdvanceInfoArgs + _args.Req = req + var _result TraceServiceBatchGetTracesAdvanceInfoResult + if err = p.Client_().Call(ctx, "BatchGetTracesAdvanceInfo", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) IngestTracesInner(ctx context.Context, req *IngestTracesRequest) (r *IngestTracesResponse, err error) { + var _args TraceServiceIngestTracesInnerArgs + _args.Req = req + var _result TraceServiceIngestTracesInnerResult + if err = p.Client_().Call(ctx, "IngestTracesInner", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) GetTracesMetaInfo(ctx context.Context, req *GetTracesMetaInfoRequest) (r *GetTracesMetaInfoResponse, err error) { + var _args TraceServiceGetTracesMetaInfoArgs + _args.Req = req + var _result TraceServiceGetTracesMetaInfoResult + if err = p.Client_().Call(ctx, "GetTracesMetaInfo", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) CreateView(ctx context.Context, req *CreateViewRequest) (r *CreateViewResponse, err error) { + var _args TraceServiceCreateViewArgs + _args.Req = req + var _result TraceServiceCreateViewResult + if err = p.Client_().Call(ctx, "CreateView", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) UpdateView(ctx context.Context, req *UpdateViewRequest) (r *UpdateViewResponse, err error) { + var _args TraceServiceUpdateViewArgs + _args.Req = req + var _result TraceServiceUpdateViewResult + if err = p.Client_().Call(ctx, "UpdateView", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) DeleteView(ctx context.Context, req *DeleteViewRequest) (r *DeleteViewResponse, err error) { + var _args TraceServiceDeleteViewArgs + _args.Req = req + var _result TraceServiceDeleteViewResult + if err = p.Client_().Call(ctx, "DeleteView", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ListViews(ctx context.Context, req *ListViewsRequest) (r *ListViewsResponse, err error) { + var _args TraceServiceListViewsArgs + _args.Req = req + var _result TraceServiceListViewsResult + if err = p.Client_().Call(ctx, "ListViews", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) CreateManualAnnotation(ctx context.Context, req *CreateManualAnnotationRequest) (r *CreateManualAnnotationResponse, err error) { + var _args TraceServiceCreateManualAnnotationArgs + _args.Req = req + var _result TraceServiceCreateManualAnnotationResult + if err = p.Client_().Call(ctx, "CreateManualAnnotation", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) UpdateManualAnnotation(ctx context.Context, req *UpdateManualAnnotationRequest) (r *UpdateManualAnnotationResponse, err error) { + var _args TraceServiceUpdateManualAnnotationArgs + _args.Req = req + var _result TraceServiceUpdateManualAnnotationResult + if err = p.Client_().Call(ctx, "UpdateManualAnnotation", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) DeleteManualAnnotation(ctx context.Context, req *DeleteManualAnnotationRequest) (r *DeleteManualAnnotationResponse, err error) { + var _args TraceServiceDeleteManualAnnotationArgs + _args.Req = req + var _result TraceServiceDeleteManualAnnotationResult + if err = p.Client_().Call(ctx, "DeleteManualAnnotation", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ListAnnotations(ctx context.Context, req *ListAnnotationsRequest) (r *ListAnnotationsResponse, err error) { + var _args TraceServiceListAnnotationsArgs + _args.Req = req + var _result TraceServiceListAnnotationsResult + if err = p.Client_().Call(ctx, "ListAnnotations", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ListWorkspaceAnnotations(ctx context.Context, req *ListWorkspaceAnnotationsRequest) (r *ListWorkspaceAnnotationsResponse, err error) { + var _args TraceServiceListWorkspaceAnnotationsArgs + _args.Req = req + var _result TraceServiceListWorkspaceAnnotationsResult + if err = p.Client_().Call(ctx, "ListWorkspaceAnnotations", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ExportTracesToDataset(ctx context.Context, req *ExportTracesToDatasetRequest) (r *ExportTracesToDatasetResponse, err error) { + var _args TraceServiceExportTracesToDatasetArgs + _args.Req = req + var _result TraceServiceExportTracesToDatasetResult + if err = p.Client_().Call(ctx, "ExportTracesToDataset", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) PreviewExportTracesToDataset(ctx context.Context, req *PreviewExportTracesToDatasetRequest) (r *PreviewExportTracesToDatasetResponse, err error) { + var _args TraceServicePreviewExportTracesToDatasetArgs + _args.Req = req + var _result TraceServicePreviewExportTracesToDatasetResult + if err = p.Client_().Call(ctx, "PreviewExportTracesToDataset", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ChangeEvaluatorScore(ctx context.Context, req *ChangeEvaluatorScoreRequest) (r *ChangeEvaluatorScoreResponse, err error) { + var _args TraceServiceChangeEvaluatorScoreArgs + _args.Req = req + var _result TraceServiceChangeEvaluatorScoreResult + if err = p.Client_().Call(ctx, "ChangeEvaluatorScore", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ListAnnotationEvaluators(ctx context.Context, req *ListAnnotationEvaluatorsRequest) (r *ListAnnotationEvaluatorsResponse, err error) { + var _args TraceServiceListAnnotationEvaluatorsArgs + _args.Req = req + var _result TraceServiceListAnnotationEvaluatorsResult + if err = p.Client_().Call(ctx, "ListAnnotationEvaluators", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ExtractSpanInfo(ctx context.Context, req *ExtractSpanInfoRequest) (r *ExtractSpanInfoResponse, err error) { + var _args TraceServiceExtractSpanInfoArgs + _args.Req = req + var _result TraceServiceExtractSpanInfoResult + if err = p.Client_().Call(ctx, "ExtractSpanInfo", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) UpsertTrajectoryConfig(ctx context.Context, req *UpsertTrajectoryConfigRequest) (r *UpsertTrajectoryConfigResponse, err error) { + var _args TraceServiceUpsertTrajectoryConfigArgs _args.Req = req var _result TraceServiceUpsertTrajectoryConfigResult if err = p.Client_().Call(ctx, "UpsertTrajectoryConfig", &_args, &_result); err != nil { return } - return _result.GetSuccess(), nil + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) GetTrajectoryConfig(ctx context.Context, req *GetTrajectoryConfigRequest) (r *GetTrajectoryConfigResponse, err error) { + var _args TraceServiceGetTrajectoryConfigArgs + _args.Req = req + var _result TraceServiceGetTrajectoryConfigResult + if err = p.Client_().Call(ctx, "GetTrajectoryConfig", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ListTrajectory(ctx context.Context, req *ListTrajectoryRequest) (r *ListTrajectoryResponse, err error) { + var _args TraceServiceListTrajectoryArgs + _args.Req = req + var _result TraceServiceListTrajectoryResult + if err = p.Client_().Call(ctx, "ListTrajectory", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} +func (p *TraceServiceClient) ListMetadata(ctx context.Context, req *ListMetadataRequest) (r *ListMetadataResponse, err error) { + var _args TraceServiceListMetadataArgs + _args.Req = req + var _result TraceServiceListMetadataResult + if err = p.Client_().Call(ctx, "ListMetadata", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} + +type TraceServiceProcessor struct { + processorMap map[string]thrift.TProcessorFunction + handler TraceService +} + +func (p *TraceServiceProcessor) AddToProcessorMap(key string, processor thrift.TProcessorFunction) { + p.processorMap[key] = processor +} + +func (p *TraceServiceProcessor) GetProcessorFunction(key string) (processor thrift.TProcessorFunction, ok bool) { + processor, ok = p.processorMap[key] + return processor, ok +} + +func (p *TraceServiceProcessor) ProcessorMap() map[string]thrift.TProcessorFunction { + return p.processorMap +} + +func NewTraceServiceProcessor(handler TraceService) *TraceServiceProcessor { + self := &TraceServiceProcessor{handler: handler, processorMap: make(map[string]thrift.TProcessorFunction)} + self.AddToProcessorMap("ListSpans", &traceServiceProcessorListSpans{handler: handler}) + self.AddToProcessorMap("ListPreSpan", &traceServiceProcessorListPreSpan{handler: handler}) + self.AddToProcessorMap("GetTrace", &traceServiceProcessorGetTrace{handler: handler}) + self.AddToProcessorMap("SearchTraceTree", &traceServiceProcessorSearchTraceTree{handler: handler}) + self.AddToProcessorMap("BatchGetTracesAdvanceInfo", &traceServiceProcessorBatchGetTracesAdvanceInfo{handler: handler}) + self.AddToProcessorMap("IngestTracesInner", &traceServiceProcessorIngestTracesInner{handler: handler}) + self.AddToProcessorMap("GetTracesMetaInfo", &traceServiceProcessorGetTracesMetaInfo{handler: handler}) + self.AddToProcessorMap("CreateView", &traceServiceProcessorCreateView{handler: handler}) + self.AddToProcessorMap("UpdateView", &traceServiceProcessorUpdateView{handler: handler}) + self.AddToProcessorMap("DeleteView", &traceServiceProcessorDeleteView{handler: handler}) + self.AddToProcessorMap("ListViews", &traceServiceProcessorListViews{handler: handler}) + self.AddToProcessorMap("CreateManualAnnotation", &traceServiceProcessorCreateManualAnnotation{handler: handler}) + self.AddToProcessorMap("UpdateManualAnnotation", &traceServiceProcessorUpdateManualAnnotation{handler: handler}) + self.AddToProcessorMap("DeleteManualAnnotation", &traceServiceProcessorDeleteManualAnnotation{handler: handler}) + self.AddToProcessorMap("ListAnnotations", &traceServiceProcessorListAnnotations{handler: handler}) + self.AddToProcessorMap("ListWorkspaceAnnotations", &traceServiceProcessorListWorkspaceAnnotations{handler: handler}) + self.AddToProcessorMap("ExportTracesToDataset", &traceServiceProcessorExportTracesToDataset{handler: handler}) + self.AddToProcessorMap("PreviewExportTracesToDataset", &traceServiceProcessorPreviewExportTracesToDataset{handler: handler}) + self.AddToProcessorMap("ChangeEvaluatorScore", &traceServiceProcessorChangeEvaluatorScore{handler: handler}) + self.AddToProcessorMap("ListAnnotationEvaluators", &traceServiceProcessorListAnnotationEvaluators{handler: handler}) + self.AddToProcessorMap("ExtractSpanInfo", &traceServiceProcessorExtractSpanInfo{handler: handler}) + self.AddToProcessorMap("UpsertTrajectoryConfig", &traceServiceProcessorUpsertTrajectoryConfig{handler: handler}) + self.AddToProcessorMap("GetTrajectoryConfig", &traceServiceProcessorGetTrajectoryConfig{handler: handler}) + self.AddToProcessorMap("ListTrajectory", &traceServiceProcessorListTrajectory{handler: handler}) + self.AddToProcessorMap("ListMetadata", &traceServiceProcessorListMetadata{handler: handler}) + return self +} +func (p *TraceServiceProcessor) Process(ctx context.Context, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + name, _, seqId, err := iprot.ReadMessageBegin() + if err != nil { + return false, err + } + if processor, ok := p.GetProcessorFunction(name); ok { + return processor.Process(ctx, seqId, iprot, oprot) + } + iprot.Skip(thrift.STRUCT) + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.UNKNOWN_METHOD, "Unknown function "+name) + oprot.WriteMessageBegin(name, thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, x +} + +type traceServiceProcessorListSpans struct { + handler TraceService +} + +func (p *traceServiceProcessorListSpans) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceListSpansArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("ListSpans", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceListSpansResult{} + var retval *ListSpansResponse + if retval, err2 = p.handler.ListSpans(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListSpans: "+err2.Error()) + oprot.WriteMessageBegin("ListSpans", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("ListSpans", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err +} + +type traceServiceProcessorListPreSpan struct { + handler TraceService +} + +func (p *traceServiceProcessorListPreSpan) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceListPreSpanArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("ListPreSpan", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceListPreSpanResult{} + var retval *ListPreSpanResponse + if retval, err2 = p.handler.ListPreSpan(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListPreSpan: "+err2.Error()) + oprot.WriteMessageBegin("ListPreSpan", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("ListPreSpan", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err +} + +type traceServiceProcessorGetTrace struct { + handler TraceService +} + +func (p *traceServiceProcessorGetTrace) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceGetTraceArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("GetTrace", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceGetTraceResult{} + var retval *GetTraceResponse + if retval, err2 = p.handler.GetTrace(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing GetTrace: "+err2.Error()) + oprot.WriteMessageBegin("GetTrace", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("GetTrace", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err +} + +type traceServiceProcessorSearchTraceTree struct { + handler TraceService +} + +func (p *traceServiceProcessorSearchTraceTree) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceSearchTraceTreeArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("SearchTraceTree", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceSearchTraceTreeResult{} + var retval *SearchTraceTreeResponse + if retval, err2 = p.handler.SearchTraceTree(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing SearchTraceTree: "+err2.Error()) + oprot.WriteMessageBegin("SearchTraceTree", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("SearchTraceTree", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err +} + +type traceServiceProcessorBatchGetTracesAdvanceInfo struct { + handler TraceService +} + +func (p *traceServiceProcessorBatchGetTracesAdvanceInfo) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceBatchGetTracesAdvanceInfoArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("BatchGetTracesAdvanceInfo", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceBatchGetTracesAdvanceInfoResult{} + var retval *BatchGetTracesAdvanceInfoResponse + if retval, err2 = p.handler.BatchGetTracesAdvanceInfo(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing BatchGetTracesAdvanceInfo: "+err2.Error()) + oprot.WriteMessageBegin("BatchGetTracesAdvanceInfo", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("BatchGetTracesAdvanceInfo", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err +} + +type traceServiceProcessorIngestTracesInner struct { + handler TraceService +} + +func (p *traceServiceProcessorIngestTracesInner) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceIngestTracesInnerArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("IngestTracesInner", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceIngestTracesInnerResult{} + var retval *IngestTracesResponse + if retval, err2 = p.handler.IngestTracesInner(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing IngestTracesInner: "+err2.Error()) + oprot.WriteMessageBegin("IngestTracesInner", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("IngestTracesInner", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err } -func (p *TraceServiceClient) GetTrajectoryConfig(ctx context.Context, req *GetTrajectoryConfigRequest) (r *GetTrajectoryConfigResponse, err error) { - var _args TraceServiceGetTrajectoryConfigArgs - _args.Req = req - var _result TraceServiceGetTrajectoryConfigResult - if err = p.Client_().Call(ctx, "GetTrajectoryConfig", &_args, &_result); err != nil { + +type traceServiceProcessorGetTracesMetaInfo struct { + handler TraceService +} + +func (p *traceServiceProcessorGetTracesMetaInfo) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceGetTracesMetaInfoArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("GetTracesMetaInfo", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceGetTracesMetaInfoResult{} + var retval *GetTracesMetaInfoResponse + if retval, err2 = p.handler.GetTracesMetaInfo(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing GetTracesMetaInfo: "+err2.Error()) + oprot.WriteMessageBegin("GetTracesMetaInfo", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("GetTracesMetaInfo", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { return } - return _result.GetSuccess(), nil + return true, err } -func (p *TraceServiceClient) ListTrajectory(ctx context.Context, req *ListTrajectoryRequest) (r *ListTrajectoryResponse, err error) { - var _args TraceServiceListTrajectoryArgs - _args.Req = req - var _result TraceServiceListTrajectoryResult - if err = p.Client_().Call(ctx, "ListTrajectory", &_args, &_result); err != nil { + +type traceServiceProcessorCreateView struct { + handler TraceService +} + +func (p *traceServiceProcessorCreateView) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceCreateViewArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("CreateView", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceCreateViewResult{} + var retval *CreateViewResponse + if retval, err2 = p.handler.CreateView(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing CreateView: "+err2.Error()) + oprot.WriteMessageBegin("CreateView", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("CreateView", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err +} + +type traceServiceProcessorUpdateView struct { + handler TraceService +} + +func (p *traceServiceProcessorUpdateView) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceUpdateViewArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("UpdateView", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceUpdateViewResult{} + var retval *UpdateViewResponse + if retval, err2 = p.handler.UpdateView(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing UpdateView: "+err2.Error()) + oprot.WriteMessageBegin("UpdateView", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("UpdateView", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err +} + +type traceServiceProcessorDeleteView struct { + handler TraceService +} + +func (p *traceServiceProcessorDeleteView) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceDeleteViewArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("DeleteView", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceDeleteViewResult{} + var retval *DeleteViewResponse + if retval, err2 = p.handler.DeleteView(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing DeleteView: "+err2.Error()) + oprot.WriteMessageBegin("DeleteView", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("DeleteView", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err +} + +type traceServiceProcessorListViews struct { + handler TraceService +} + +func (p *traceServiceProcessorListViews) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceListViewsArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("ListViews", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return false, err + } + + iprot.ReadMessageEnd() + var err2 error + result := TraceServiceListViewsResult{} + var retval *ListViewsResponse + if retval, err2 = p.handler.ListViews(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListViews: "+err2.Error()) + oprot.WriteMessageBegin("ListViews", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("ListViews", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { return } - return _result.GetSuccess(), nil -} - -type TraceServiceProcessor struct { - processorMap map[string]thrift.TProcessorFunction - handler TraceService -} - -func (p *TraceServiceProcessor) AddToProcessorMap(key string, processor thrift.TProcessorFunction) { - p.processorMap[key] = processor -} - -func (p *TraceServiceProcessor) GetProcessorFunction(key string) (processor thrift.TProcessorFunction, ok bool) { - processor, ok = p.processorMap[key] - return processor, ok + return true, err } -func (p *TraceServiceProcessor) ProcessorMap() map[string]thrift.TProcessorFunction { - return p.processorMap +type traceServiceProcessorCreateManualAnnotation struct { + handler TraceService } -func NewTraceServiceProcessor(handler TraceService) *TraceServiceProcessor { - self := &TraceServiceProcessor{handler: handler, processorMap: make(map[string]thrift.TProcessorFunction)} - self.AddToProcessorMap("ListSpans", &traceServiceProcessorListSpans{handler: handler}) - self.AddToProcessorMap("ListPreSpan", &traceServiceProcessorListPreSpan{handler: handler}) - self.AddToProcessorMap("GetTrace", &traceServiceProcessorGetTrace{handler: handler}) - self.AddToProcessorMap("SearchTraceTree", &traceServiceProcessorSearchTraceTree{handler: handler}) - self.AddToProcessorMap("BatchGetTracesAdvanceInfo", &traceServiceProcessorBatchGetTracesAdvanceInfo{handler: handler}) - self.AddToProcessorMap("IngestTracesInner", &traceServiceProcessorIngestTracesInner{handler: handler}) - self.AddToProcessorMap("GetTracesMetaInfo", &traceServiceProcessorGetTracesMetaInfo{handler: handler}) - self.AddToProcessorMap("CreateView", &traceServiceProcessorCreateView{handler: handler}) - self.AddToProcessorMap("UpdateView", &traceServiceProcessorUpdateView{handler: handler}) - self.AddToProcessorMap("DeleteView", &traceServiceProcessorDeleteView{handler: handler}) - self.AddToProcessorMap("ListViews", &traceServiceProcessorListViews{handler: handler}) - self.AddToProcessorMap("CreateManualAnnotation", &traceServiceProcessorCreateManualAnnotation{handler: handler}) - self.AddToProcessorMap("UpdateManualAnnotation", &traceServiceProcessorUpdateManualAnnotation{handler: handler}) - self.AddToProcessorMap("DeleteManualAnnotation", &traceServiceProcessorDeleteManualAnnotation{handler: handler}) - self.AddToProcessorMap("ListAnnotations", &traceServiceProcessorListAnnotations{handler: handler}) - self.AddToProcessorMap("ExportTracesToDataset", &traceServiceProcessorExportTracesToDataset{handler: handler}) - self.AddToProcessorMap("PreviewExportTracesToDataset", &traceServiceProcessorPreviewExportTracesToDataset{handler: handler}) - self.AddToProcessorMap("ChangeEvaluatorScore", &traceServiceProcessorChangeEvaluatorScore{handler: handler}) - self.AddToProcessorMap("ListAnnotationEvaluators", &traceServiceProcessorListAnnotationEvaluators{handler: handler}) - self.AddToProcessorMap("ExtractSpanInfo", &traceServiceProcessorExtractSpanInfo{handler: handler}) - self.AddToProcessorMap("UpsertTrajectoryConfig", &traceServiceProcessorUpsertTrajectoryConfig{handler: handler}) - self.AddToProcessorMap("GetTrajectoryConfig", &traceServiceProcessorGetTrajectoryConfig{handler: handler}) - self.AddToProcessorMap("ListTrajectory", &traceServiceProcessorListTrajectory{handler: handler}) - return self -} -func (p *TraceServiceProcessor) Process(ctx context.Context, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - name, _, seqId, err := iprot.ReadMessageBegin() - if err != nil { +func (p *traceServiceProcessorCreateManualAnnotation) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceCreateManualAnnotationArgs{} + if err = args.Read(iprot); err != nil { + iprot.ReadMessageEnd() + x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) + oprot.WriteMessageBegin("CreateManualAnnotation", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) return false, err } - if processor, ok := p.GetProcessorFunction(name); ok { - return processor.Process(ctx, seqId, iprot, oprot) - } - iprot.Skip(thrift.STRUCT) + iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.UNKNOWN_METHOD, "Unknown function "+name) - oprot.WriteMessageBegin(name, thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, x + var err2 error + result := TraceServiceCreateManualAnnotationResult{} + var retval *CreateManualAnnotationResponse + if retval, err2 = p.handler.CreateManualAnnotation(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing CreateManualAnnotation: "+err2.Error()) + oprot.WriteMessageBegin("CreateManualAnnotation", thrift.EXCEPTION, seqId) + x.Write(oprot) + oprot.WriteMessageEnd() + oprot.Flush(ctx) + return true, err2 + } else { + result.Success = retval + } + if err2 = oprot.WriteMessageBegin("CreateManualAnnotation", thrift.REPLY, seqId); err2 != nil { + err = err2 + } + if err2 = result.Write(oprot); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { + err = err2 + } + if err2 = oprot.Flush(ctx); err == nil && err2 != nil { + err = err2 + } + if err != nil { + return + } + return true, err } -type traceServiceProcessorListSpans struct { +type traceServiceProcessorUpdateManualAnnotation struct { handler TraceService } -func (p *traceServiceProcessorListSpans) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceListSpansArgs{} +func (p *traceServiceProcessorUpdateManualAnnotation) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceUpdateManualAnnotationArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ListSpans", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("UpdateManualAnnotation", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21257,11 +23649,11 @@ func (p *traceServiceProcessorListSpans) Process(ctx context.Context, seqId int3 iprot.ReadMessageEnd() var err2 error - result := TraceServiceListSpansResult{} - var retval *ListSpansResponse - if retval, err2 = p.handler.ListSpans(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListSpans: "+err2.Error()) - oprot.WriteMessageBegin("ListSpans", thrift.EXCEPTION, seqId) + result := TraceServiceUpdateManualAnnotationResult{} + var retval *UpdateManualAnnotationResponse + if retval, err2 = p.handler.UpdateManualAnnotation(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing UpdateManualAnnotation: "+err2.Error()) + oprot.WriteMessageBegin("UpdateManualAnnotation", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21269,7 +23661,7 @@ func (p *traceServiceProcessorListSpans) Process(ctx context.Context, seqId int3 } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("ListSpans", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("UpdateManualAnnotation", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21287,16 +23679,16 @@ func (p *traceServiceProcessorListSpans) Process(ctx context.Context, seqId int3 return true, err } -type traceServiceProcessorListPreSpan struct { +type traceServiceProcessorDeleteManualAnnotation struct { handler TraceService } -func (p *traceServiceProcessorListPreSpan) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceListPreSpanArgs{} +func (p *traceServiceProcessorDeleteManualAnnotation) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceDeleteManualAnnotationArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ListPreSpan", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("DeleteManualAnnotation", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21305,11 +23697,11 @@ func (p *traceServiceProcessorListPreSpan) Process(ctx context.Context, seqId in iprot.ReadMessageEnd() var err2 error - result := TraceServiceListPreSpanResult{} - var retval *ListPreSpanResponse - if retval, err2 = p.handler.ListPreSpan(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListPreSpan: "+err2.Error()) - oprot.WriteMessageBegin("ListPreSpan", thrift.EXCEPTION, seqId) + result := TraceServiceDeleteManualAnnotationResult{} + var retval *DeleteManualAnnotationResponse + if retval, err2 = p.handler.DeleteManualAnnotation(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing DeleteManualAnnotation: "+err2.Error()) + oprot.WriteMessageBegin("DeleteManualAnnotation", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21317,7 +23709,7 @@ func (p *traceServiceProcessorListPreSpan) Process(ctx context.Context, seqId in } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("ListPreSpan", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("DeleteManualAnnotation", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21335,16 +23727,16 @@ func (p *traceServiceProcessorListPreSpan) Process(ctx context.Context, seqId in return true, err } -type traceServiceProcessorGetTrace struct { +type traceServiceProcessorListAnnotations struct { handler TraceService } -func (p *traceServiceProcessorGetTrace) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceGetTraceArgs{} +func (p *traceServiceProcessorListAnnotations) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceListAnnotationsArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("GetTrace", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("ListAnnotations", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21353,11 +23745,11 @@ func (p *traceServiceProcessorGetTrace) Process(ctx context.Context, seqId int32 iprot.ReadMessageEnd() var err2 error - result := TraceServiceGetTraceResult{} - var retval *GetTraceResponse - if retval, err2 = p.handler.GetTrace(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing GetTrace: "+err2.Error()) - oprot.WriteMessageBegin("GetTrace", thrift.EXCEPTION, seqId) + result := TraceServiceListAnnotationsResult{} + var retval *ListAnnotationsResponse + if retval, err2 = p.handler.ListAnnotations(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListAnnotations: "+err2.Error()) + oprot.WriteMessageBegin("ListAnnotations", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21365,7 +23757,7 @@ func (p *traceServiceProcessorGetTrace) Process(ctx context.Context, seqId int32 } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("GetTrace", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("ListAnnotations", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21383,16 +23775,16 @@ func (p *traceServiceProcessorGetTrace) Process(ctx context.Context, seqId int32 return true, err } -type traceServiceProcessorSearchTraceTree struct { +type traceServiceProcessorListWorkspaceAnnotations struct { handler TraceService } -func (p *traceServiceProcessorSearchTraceTree) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceSearchTraceTreeArgs{} +func (p *traceServiceProcessorListWorkspaceAnnotations) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceListWorkspaceAnnotationsArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("SearchTraceTree", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("ListWorkspaceAnnotations", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21401,11 +23793,11 @@ func (p *traceServiceProcessorSearchTraceTree) Process(ctx context.Context, seqI iprot.ReadMessageEnd() var err2 error - result := TraceServiceSearchTraceTreeResult{} - var retval *SearchTraceTreeResponse - if retval, err2 = p.handler.SearchTraceTree(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing SearchTraceTree: "+err2.Error()) - oprot.WriteMessageBegin("SearchTraceTree", thrift.EXCEPTION, seqId) + result := TraceServiceListWorkspaceAnnotationsResult{} + var retval *ListWorkspaceAnnotationsResponse + if retval, err2 = p.handler.ListWorkspaceAnnotations(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListWorkspaceAnnotations: "+err2.Error()) + oprot.WriteMessageBegin("ListWorkspaceAnnotations", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21413,7 +23805,7 @@ func (p *traceServiceProcessorSearchTraceTree) Process(ctx context.Context, seqI } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("SearchTraceTree", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("ListWorkspaceAnnotations", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21431,16 +23823,16 @@ func (p *traceServiceProcessorSearchTraceTree) Process(ctx context.Context, seqI return true, err } -type traceServiceProcessorBatchGetTracesAdvanceInfo struct { +type traceServiceProcessorExportTracesToDataset struct { handler TraceService } -func (p *traceServiceProcessorBatchGetTracesAdvanceInfo) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceBatchGetTracesAdvanceInfoArgs{} +func (p *traceServiceProcessorExportTracesToDataset) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceExportTracesToDatasetArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("BatchGetTracesAdvanceInfo", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("ExportTracesToDataset", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21449,11 +23841,11 @@ func (p *traceServiceProcessorBatchGetTracesAdvanceInfo) Process(ctx context.Con iprot.ReadMessageEnd() var err2 error - result := TraceServiceBatchGetTracesAdvanceInfoResult{} - var retval *BatchGetTracesAdvanceInfoResponse - if retval, err2 = p.handler.BatchGetTracesAdvanceInfo(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing BatchGetTracesAdvanceInfo: "+err2.Error()) - oprot.WriteMessageBegin("BatchGetTracesAdvanceInfo", thrift.EXCEPTION, seqId) + result := TraceServiceExportTracesToDatasetResult{} + var retval *ExportTracesToDatasetResponse + if retval, err2 = p.handler.ExportTracesToDataset(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ExportTracesToDataset: "+err2.Error()) + oprot.WriteMessageBegin("ExportTracesToDataset", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21461,7 +23853,7 @@ func (p *traceServiceProcessorBatchGetTracesAdvanceInfo) Process(ctx context.Con } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("BatchGetTracesAdvanceInfo", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("ExportTracesToDataset", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21479,16 +23871,16 @@ func (p *traceServiceProcessorBatchGetTracesAdvanceInfo) Process(ctx context.Con return true, err } -type traceServiceProcessorIngestTracesInner struct { +type traceServiceProcessorPreviewExportTracesToDataset struct { handler TraceService } -func (p *traceServiceProcessorIngestTracesInner) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceIngestTracesInnerArgs{} +func (p *traceServiceProcessorPreviewExportTracesToDataset) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServicePreviewExportTracesToDatasetArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("IngestTracesInner", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("PreviewExportTracesToDataset", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21497,11 +23889,11 @@ func (p *traceServiceProcessorIngestTracesInner) Process(ctx context.Context, se iprot.ReadMessageEnd() var err2 error - result := TraceServiceIngestTracesInnerResult{} - var retval *IngestTracesResponse - if retval, err2 = p.handler.IngestTracesInner(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing IngestTracesInner: "+err2.Error()) - oprot.WriteMessageBegin("IngestTracesInner", thrift.EXCEPTION, seqId) + result := TraceServicePreviewExportTracesToDatasetResult{} + var retval *PreviewExportTracesToDatasetResponse + if retval, err2 = p.handler.PreviewExportTracesToDataset(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing PreviewExportTracesToDataset: "+err2.Error()) + oprot.WriteMessageBegin("PreviewExportTracesToDataset", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21509,7 +23901,7 @@ func (p *traceServiceProcessorIngestTracesInner) Process(ctx context.Context, se } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("IngestTracesInner", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("PreviewExportTracesToDataset", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21527,16 +23919,16 @@ func (p *traceServiceProcessorIngestTracesInner) Process(ctx context.Context, se return true, err } -type traceServiceProcessorGetTracesMetaInfo struct { +type traceServiceProcessorChangeEvaluatorScore struct { handler TraceService } -func (p *traceServiceProcessorGetTracesMetaInfo) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceGetTracesMetaInfoArgs{} +func (p *traceServiceProcessorChangeEvaluatorScore) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceChangeEvaluatorScoreArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("GetTracesMetaInfo", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("ChangeEvaluatorScore", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21545,11 +23937,11 @@ func (p *traceServiceProcessorGetTracesMetaInfo) Process(ctx context.Context, se iprot.ReadMessageEnd() var err2 error - result := TraceServiceGetTracesMetaInfoResult{} - var retval *GetTracesMetaInfoResponse - if retval, err2 = p.handler.GetTracesMetaInfo(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing GetTracesMetaInfo: "+err2.Error()) - oprot.WriteMessageBegin("GetTracesMetaInfo", thrift.EXCEPTION, seqId) + result := TraceServiceChangeEvaluatorScoreResult{} + var retval *ChangeEvaluatorScoreResponse + if retval, err2 = p.handler.ChangeEvaluatorScore(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ChangeEvaluatorScore: "+err2.Error()) + oprot.WriteMessageBegin("ChangeEvaluatorScore", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21557,7 +23949,7 @@ func (p *traceServiceProcessorGetTracesMetaInfo) Process(ctx context.Context, se } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("GetTracesMetaInfo", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("ChangeEvaluatorScore", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21575,16 +23967,16 @@ func (p *traceServiceProcessorGetTracesMetaInfo) Process(ctx context.Context, se return true, err } -type traceServiceProcessorCreateView struct { +type traceServiceProcessorListAnnotationEvaluators struct { handler TraceService } -func (p *traceServiceProcessorCreateView) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceCreateViewArgs{} +func (p *traceServiceProcessorListAnnotationEvaluators) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceListAnnotationEvaluatorsArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("CreateView", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("ListAnnotationEvaluators", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21593,11 +23985,11 @@ func (p *traceServiceProcessorCreateView) Process(ctx context.Context, seqId int iprot.ReadMessageEnd() var err2 error - result := TraceServiceCreateViewResult{} - var retval *CreateViewResponse - if retval, err2 = p.handler.CreateView(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing CreateView: "+err2.Error()) - oprot.WriteMessageBegin("CreateView", thrift.EXCEPTION, seqId) + result := TraceServiceListAnnotationEvaluatorsResult{} + var retval *ListAnnotationEvaluatorsResponse + if retval, err2 = p.handler.ListAnnotationEvaluators(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListAnnotationEvaluators: "+err2.Error()) + oprot.WriteMessageBegin("ListAnnotationEvaluators", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21605,7 +23997,7 @@ func (p *traceServiceProcessorCreateView) Process(ctx context.Context, seqId int } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("CreateView", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("ListAnnotationEvaluators", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21623,16 +24015,16 @@ func (p *traceServiceProcessorCreateView) Process(ctx context.Context, seqId int return true, err } -type traceServiceProcessorUpdateView struct { +type traceServiceProcessorExtractSpanInfo struct { handler TraceService } -func (p *traceServiceProcessorUpdateView) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceUpdateViewArgs{} +func (p *traceServiceProcessorExtractSpanInfo) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceExtractSpanInfoArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("UpdateView", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("ExtractSpanInfo", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21641,11 +24033,11 @@ func (p *traceServiceProcessorUpdateView) Process(ctx context.Context, seqId int iprot.ReadMessageEnd() var err2 error - result := TraceServiceUpdateViewResult{} - var retval *UpdateViewResponse - if retval, err2 = p.handler.UpdateView(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing UpdateView: "+err2.Error()) - oprot.WriteMessageBegin("UpdateView", thrift.EXCEPTION, seqId) + result := TraceServiceExtractSpanInfoResult{} + var retval *ExtractSpanInfoResponse + if retval, err2 = p.handler.ExtractSpanInfo(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ExtractSpanInfo: "+err2.Error()) + oprot.WriteMessageBegin("ExtractSpanInfo", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21653,7 +24045,7 @@ func (p *traceServiceProcessorUpdateView) Process(ctx context.Context, seqId int } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("UpdateView", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("ExtractSpanInfo", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21671,16 +24063,16 @@ func (p *traceServiceProcessorUpdateView) Process(ctx context.Context, seqId int return true, err } -type traceServiceProcessorDeleteView struct { +type traceServiceProcessorUpsertTrajectoryConfig struct { handler TraceService } -func (p *traceServiceProcessorDeleteView) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceDeleteViewArgs{} +func (p *traceServiceProcessorUpsertTrajectoryConfig) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceUpsertTrajectoryConfigArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("DeleteView", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("UpsertTrajectoryConfig", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21689,11 +24081,11 @@ func (p *traceServiceProcessorDeleteView) Process(ctx context.Context, seqId int iprot.ReadMessageEnd() var err2 error - result := TraceServiceDeleteViewResult{} - var retval *DeleteViewResponse - if retval, err2 = p.handler.DeleteView(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing DeleteView: "+err2.Error()) - oprot.WriteMessageBegin("DeleteView", thrift.EXCEPTION, seqId) + result := TraceServiceUpsertTrajectoryConfigResult{} + var retval *UpsertTrajectoryConfigResponse + if retval, err2 = p.handler.UpsertTrajectoryConfig(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing UpsertTrajectoryConfig: "+err2.Error()) + oprot.WriteMessageBegin("UpsertTrajectoryConfig", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21701,7 +24093,7 @@ func (p *traceServiceProcessorDeleteView) Process(ctx context.Context, seqId int } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("DeleteView", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("UpsertTrajectoryConfig", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21719,16 +24111,16 @@ func (p *traceServiceProcessorDeleteView) Process(ctx context.Context, seqId int return true, err } -type traceServiceProcessorListViews struct { +type traceServiceProcessorGetTrajectoryConfig struct { handler TraceService } -func (p *traceServiceProcessorListViews) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceListViewsArgs{} +func (p *traceServiceProcessorGetTrajectoryConfig) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceGetTrajectoryConfigArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ListViews", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("GetTrajectoryConfig", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21737,11 +24129,11 @@ func (p *traceServiceProcessorListViews) Process(ctx context.Context, seqId int3 iprot.ReadMessageEnd() var err2 error - result := TraceServiceListViewsResult{} - var retval *ListViewsResponse - if retval, err2 = p.handler.ListViews(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListViews: "+err2.Error()) - oprot.WriteMessageBegin("ListViews", thrift.EXCEPTION, seqId) + result := TraceServiceGetTrajectoryConfigResult{} + var retval *GetTrajectoryConfigResponse + if retval, err2 = p.handler.GetTrajectoryConfig(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing GetTrajectoryConfig: "+err2.Error()) + oprot.WriteMessageBegin("GetTrajectoryConfig", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21749,7 +24141,7 @@ func (p *traceServiceProcessorListViews) Process(ctx context.Context, seqId int3 } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("ListViews", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("GetTrajectoryConfig", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21767,16 +24159,16 @@ func (p *traceServiceProcessorListViews) Process(ctx context.Context, seqId int3 return true, err } -type traceServiceProcessorCreateManualAnnotation struct { +type traceServiceProcessorListTrajectory struct { handler TraceService } -func (p *traceServiceProcessorCreateManualAnnotation) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceCreateManualAnnotationArgs{} +func (p *traceServiceProcessorListTrajectory) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceListTrajectoryArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("CreateManualAnnotation", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("ListTrajectory", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21785,11 +24177,11 @@ func (p *traceServiceProcessorCreateManualAnnotation) Process(ctx context.Contex iprot.ReadMessageEnd() var err2 error - result := TraceServiceCreateManualAnnotationResult{} - var retval *CreateManualAnnotationResponse - if retval, err2 = p.handler.CreateManualAnnotation(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing CreateManualAnnotation: "+err2.Error()) - oprot.WriteMessageBegin("CreateManualAnnotation", thrift.EXCEPTION, seqId) + result := TraceServiceListTrajectoryResult{} + var retval *ListTrajectoryResponse + if retval, err2 = p.handler.ListTrajectory(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListTrajectory: "+err2.Error()) + oprot.WriteMessageBegin("ListTrajectory", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21797,7 +24189,7 @@ func (p *traceServiceProcessorCreateManualAnnotation) Process(ctx context.Contex } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("CreateManualAnnotation", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("ListTrajectory", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21815,16 +24207,16 @@ func (p *traceServiceProcessorCreateManualAnnotation) Process(ctx context.Contex return true, err } -type traceServiceProcessorUpdateManualAnnotation struct { +type traceServiceProcessorListMetadata struct { handler TraceService } -func (p *traceServiceProcessorUpdateManualAnnotation) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceUpdateManualAnnotationArgs{} +func (p *traceServiceProcessorListMetadata) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { + args := TraceServiceListMetadataArgs{} if err = args.Read(iprot); err != nil { iprot.ReadMessageEnd() x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("UpdateManualAnnotation", thrift.EXCEPTION, seqId) + oprot.WriteMessageBegin("ListMetadata", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21833,11 +24225,11 @@ func (p *traceServiceProcessorUpdateManualAnnotation) Process(ctx context.Contex iprot.ReadMessageEnd() var err2 error - result := TraceServiceUpdateManualAnnotationResult{} - var retval *UpdateManualAnnotationResponse - if retval, err2 = p.handler.UpdateManualAnnotation(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing UpdateManualAnnotation: "+err2.Error()) - oprot.WriteMessageBegin("UpdateManualAnnotation", thrift.EXCEPTION, seqId) + result := TraceServiceListMetadataResult{} + var retval *ListMetadataResponse + if retval, err2 = p.handler.ListMetadata(ctx, args.Req); err2 != nil { + x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListMetadata: "+err2.Error()) + oprot.WriteMessageBegin("ListMetadata", thrift.EXCEPTION, seqId) x.Write(oprot) oprot.WriteMessageEnd() oprot.Flush(ctx) @@ -21845,7 +24237,7 @@ func (p *traceServiceProcessorUpdateManualAnnotation) Process(ctx context.Contex } else { result.Success = retval } - if err2 = oprot.WriteMessageBegin("UpdateManualAnnotation", thrift.REPLY, seqId); err2 != nil { + if err2 = oprot.WriteMessageBegin("ListMetadata", thrift.REPLY, seqId); err2 != nil { err = err2 } if err2 = result.Write(oprot); err == nil && err2 != nil { @@ -21860,524 +24252,732 @@ func (p *traceServiceProcessorUpdateManualAnnotation) Process(ctx context.Contex if err != nil { return } - return true, err + return true, err +} + +type TraceServiceListSpansArgs struct { + Req *ListSpansRequest `thrift:"req,1" frugal:"1,default,ListSpansRequest"` +} + +func NewTraceServiceListSpansArgs() *TraceServiceListSpansArgs { + return &TraceServiceListSpansArgs{} +} + +func (p *TraceServiceListSpansArgs) InitDefault() { +} + +var TraceServiceListSpansArgs_Req_DEFAULT *ListSpansRequest + +func (p *TraceServiceListSpansArgs) GetReq() (v *ListSpansRequest) { + if p == nil { + return + } + if !p.IsSetReq() { + return TraceServiceListSpansArgs_Req_DEFAULT + } + return p.Req +} +func (p *TraceServiceListSpansArgs) SetReq(val *ListSpansRequest) { + p.Req = val +} + +var fieldIDToName_TraceServiceListSpansArgs = map[int16]string{ + 1: "req", +} + +func (p *TraceServiceListSpansArgs) IsSetReq() bool { + return p.Req != nil +} + +func (p *TraceServiceListSpansArgs) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError + } + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } + } + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError + } + + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListSpansArgs[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +} + +func (p *TraceServiceListSpansArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewListSpansRequest() + if err := _field.Read(iprot); err != nil { + return err + } + p.Req = _field + return nil +} + +func (p *TraceServiceListSpansArgs) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListSpans_args"); err != nil { + goto WriteStructBeginError + } + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } + } + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError + } + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) +} + +func (p *TraceServiceListSpansArgs) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { + goto WriteFieldBeginError + } + if err := p.Req.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} + +func (p *TraceServiceListSpansArgs) String() string { + if p == nil { + return "" + } + return fmt.Sprintf("TraceServiceListSpansArgs(%+v)", *p) + +} + +func (p *TraceServiceListSpansArgs) DeepEqual(ano *TraceServiceListSpansArgs) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false + } + if !p.Field1DeepEqual(ano.Req) { + return false + } + return true +} + +func (p *TraceServiceListSpansArgs) Field1DeepEqual(src *ListSpansRequest) bool { + + if !p.Req.DeepEqual(src) { + return false + } + return true } -type traceServiceProcessorDeleteManualAnnotation struct { - handler TraceService +type TraceServiceListSpansResult struct { + Success *ListSpansResponse `thrift:"success,0,optional" frugal:"0,optional,ListSpansResponse"` } -func (p *traceServiceProcessorDeleteManualAnnotation) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceDeleteManualAnnotationArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("DeleteManualAnnotation", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err - } +func NewTraceServiceListSpansResult() *TraceServiceListSpansResult { + return &TraceServiceListSpansResult{} +} - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceDeleteManualAnnotationResult{} - var retval *DeleteManualAnnotationResponse - if retval, err2 = p.handler.DeleteManualAnnotation(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing DeleteManualAnnotation: "+err2.Error()) - oprot.WriteMessageBegin("DeleteManualAnnotation", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval - } - if err2 = oprot.WriteMessageBegin("DeleteManualAnnotation", thrift.REPLY, seqId); err2 != nil { - err = err2 - } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 - } - if err != nil { +func (p *TraceServiceListSpansResult) InitDefault() { +} + +var TraceServiceListSpansResult_Success_DEFAULT *ListSpansResponse + +func (p *TraceServiceListSpansResult) GetSuccess() (v *ListSpansResponse) { + if p == nil { return } - return true, err + if !p.IsSetSuccess() { + return TraceServiceListSpansResult_Success_DEFAULT + } + return p.Success +} +func (p *TraceServiceListSpansResult) SetSuccess(x interface{}) { + p.Success = x.(*ListSpansResponse) } -type traceServiceProcessorListAnnotations struct { - handler TraceService +var fieldIDToName_TraceServiceListSpansResult = map[int16]string{ + 0: "success", } -func (p *traceServiceProcessorListAnnotations) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceListAnnotationsArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ListAnnotations", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err +func (p *TraceServiceListSpansResult) IsSetSuccess() bool { + return p.Success != nil +} + +func (p *TraceServiceListSpansResult) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError } - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceListAnnotationsResult{} - var retval *ListAnnotationsResponse - if retval, err2 = p.handler.ListAnnotations(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListAnnotations: "+err2.Error()) - oprot.WriteMessageBegin("ListAnnotations", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 0: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField0(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } } - if err2 = oprot.WriteMessageBegin("ListAnnotations", thrift.REPLY, seqId); err2 != nil { - err = err2 + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 + + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListSpansResult[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +} + +func (p *TraceServiceListSpansResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewListSpansResponse() + if err := _field.Read(iprot); err != nil { + return err } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 + p.Success = _field + return nil +} + +func (p *TraceServiceListSpansResult) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListSpans_result"); err != nil { + goto WriteStructBeginError } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 + if p != nil { + if err = p.writeField0(oprot); err != nil { + fieldId = 0 + goto WriteFieldError + } } - if err != nil { - return + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError } - return true, err + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -type traceServiceProcessorExportTracesToDataset struct { - handler TraceService +func (p *TraceServiceListSpansResult) writeField0(oprot thrift.TProtocol) (err error) { + if p.IsSetSuccess() { + if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { + goto WriteFieldBeginError + } + if err := p.Success.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 0 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *traceServiceProcessorExportTracesToDataset) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceExportTracesToDatasetArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ExportTracesToDataset", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err +func (p *TraceServiceListSpansResult) String() string { + if p == nil { + return "" } + return fmt.Sprintf("TraceServiceListSpansResult(%+v)", *p) - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceExportTracesToDatasetResult{} - var retval *ExportTracesToDatasetResponse - if retval, err2 = p.handler.ExportTracesToDataset(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ExportTracesToDataset: "+err2.Error()) - oprot.WriteMessageBegin("ExportTracesToDataset", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval - } - if err2 = oprot.WriteMessageBegin("ExportTracesToDataset", thrift.REPLY, seqId); err2 != nil { - err = err2 - } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 +} + +func (p *TraceServiceListSpansResult) DeepEqual(ano *TraceServiceListSpansResult) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 + if !p.Field0DeepEqual(ano.Success) { + return false } - if err != nil { - return + return true +} + +func (p *TraceServiceListSpansResult) Field0DeepEqual(src *ListSpansResponse) bool { + + if !p.Success.DeepEqual(src) { + return false } - return true, err + return true } -type traceServiceProcessorPreviewExportTracesToDataset struct { - handler TraceService +type TraceServiceListPreSpanArgs struct { + Req *ListPreSpanRequest `thrift:"req,1" frugal:"1,default,ListPreSpanRequest"` } -func (p *traceServiceProcessorPreviewExportTracesToDataset) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServicePreviewExportTracesToDatasetArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("PreviewExportTracesToDataset", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err - } +func NewTraceServiceListPreSpanArgs() *TraceServiceListPreSpanArgs { + return &TraceServiceListPreSpanArgs{} +} - iprot.ReadMessageEnd() - var err2 error - result := TraceServicePreviewExportTracesToDatasetResult{} - var retval *PreviewExportTracesToDatasetResponse - if retval, err2 = p.handler.PreviewExportTracesToDataset(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing PreviewExportTracesToDataset: "+err2.Error()) - oprot.WriteMessageBegin("PreviewExportTracesToDataset", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval - } - if err2 = oprot.WriteMessageBegin("PreviewExportTracesToDataset", thrift.REPLY, seqId); err2 != nil { - err = err2 - } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 - } - if err != nil { +func (p *TraceServiceListPreSpanArgs) InitDefault() { +} + +var TraceServiceListPreSpanArgs_Req_DEFAULT *ListPreSpanRequest + +func (p *TraceServiceListPreSpanArgs) GetReq() (v *ListPreSpanRequest) { + if p == nil { return } - return true, err + if !p.IsSetReq() { + return TraceServiceListPreSpanArgs_Req_DEFAULT + } + return p.Req +} +func (p *TraceServiceListPreSpanArgs) SetReq(val *ListPreSpanRequest) { + p.Req = val } -type traceServiceProcessorChangeEvaluatorScore struct { - handler TraceService +var fieldIDToName_TraceServiceListPreSpanArgs = map[int16]string{ + 1: "req", } -func (p *traceServiceProcessorChangeEvaluatorScore) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceChangeEvaluatorScoreArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ChangeEvaluatorScore", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err +func (p *TraceServiceListPreSpanArgs) IsSetReq() bool { + return p.Req != nil +} + +func (p *TraceServiceListPreSpanArgs) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError } - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceChangeEvaluatorScoreResult{} - var retval *ChangeEvaluatorScoreResponse - if retval, err2 = p.handler.ChangeEvaluatorScore(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ChangeEvaluatorScore: "+err2.Error()) - oprot.WriteMessageBegin("ChangeEvaluatorScore", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 1: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField1(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } } - if err2 = oprot.WriteMessageBegin("ChangeEvaluatorScore", thrift.REPLY, seqId); err2 != nil { - err = err2 + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 + + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListPreSpanArgs[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) +} + +func (p *TraceServiceListPreSpanArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewListPreSpanRequest() + if err := _field.Read(iprot); err != nil { + return err } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 + p.Req = _field + return nil +} + +func (p *TraceServiceListPreSpanArgs) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListPreSpan_args"); err != nil { + goto WriteStructBeginError } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 + if p != nil { + if err = p.writeField1(oprot); err != nil { + fieldId = 1 + goto WriteFieldError + } } - if err != nil { - return + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError } - return true, err -} - -type traceServiceProcessorListAnnotationEvaluators struct { - handler TraceService + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError + } + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *traceServiceProcessorListAnnotationEvaluators) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceListAnnotationEvaluatorsArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ListAnnotationEvaluators", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err +func (p *TraceServiceListPreSpanArgs) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { + goto WriteFieldBeginError } - - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceListAnnotationEvaluatorsResult{} - var retval *ListAnnotationEvaluatorsResponse - if retval, err2 = p.handler.ListAnnotationEvaluators(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListAnnotationEvaluators: "+err2.Error()) - oprot.WriteMessageBegin("ListAnnotationEvaluators", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval + if err := p.Req.Write(oprot); err != nil { + return err } - if err2 = oprot.WriteMessageBegin("ListAnnotationEvaluators", thrift.REPLY, seqId); err2 != nil { - err = err2 + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) +} + +func (p *TraceServiceListPreSpanArgs) String() string { + if p == nil { + return "" } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 + return fmt.Sprintf("TraceServiceListPreSpanArgs(%+v)", *p) + +} + +func (p *TraceServiceListPreSpanArgs) DeepEqual(ano *TraceServiceListPreSpanArgs) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 + if !p.Field1DeepEqual(ano.Req) { + return false } - if err != nil { - return + return true +} + +func (p *TraceServiceListPreSpanArgs) Field1DeepEqual(src *ListPreSpanRequest) bool { + + if !p.Req.DeepEqual(src) { + return false } - return true, err + return true } -type traceServiceProcessorExtractSpanInfo struct { - handler TraceService +type TraceServiceListPreSpanResult struct { + Success *ListPreSpanResponse `thrift:"success,0,optional" frugal:"0,optional,ListPreSpanResponse"` } -func (p *traceServiceProcessorExtractSpanInfo) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceExtractSpanInfoArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ExtractSpanInfo", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err - } +func NewTraceServiceListPreSpanResult() *TraceServiceListPreSpanResult { + return &TraceServiceListPreSpanResult{} +} - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceExtractSpanInfoResult{} - var retval *ExtractSpanInfoResponse - if retval, err2 = p.handler.ExtractSpanInfo(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ExtractSpanInfo: "+err2.Error()) - oprot.WriteMessageBegin("ExtractSpanInfo", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval - } - if err2 = oprot.WriteMessageBegin("ExtractSpanInfo", thrift.REPLY, seqId); err2 != nil { - err = err2 - } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 - } - if err != nil { +func (p *TraceServiceListPreSpanResult) InitDefault() { +} + +var TraceServiceListPreSpanResult_Success_DEFAULT *ListPreSpanResponse + +func (p *TraceServiceListPreSpanResult) GetSuccess() (v *ListPreSpanResponse) { + if p == nil { return } - return true, err + if !p.IsSetSuccess() { + return TraceServiceListPreSpanResult_Success_DEFAULT + } + return p.Success +} +func (p *TraceServiceListPreSpanResult) SetSuccess(x interface{}) { + p.Success = x.(*ListPreSpanResponse) } -type traceServiceProcessorUpsertTrajectoryConfig struct { - handler TraceService +var fieldIDToName_TraceServiceListPreSpanResult = map[int16]string{ + 0: "success", } -func (p *traceServiceProcessorUpsertTrajectoryConfig) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceUpsertTrajectoryConfigArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("UpsertTrajectoryConfig", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err - } +func (p *TraceServiceListPreSpanResult) IsSetSuccess() bool { + return p.Success != nil +} - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceUpsertTrajectoryConfigResult{} - var retval *UpsertTrajectoryConfigResponse - if retval, err2 = p.handler.UpsertTrajectoryConfig(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing UpsertTrajectoryConfig: "+err2.Error()) - oprot.WriteMessageBegin("UpsertTrajectoryConfig", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval - } - if err2 = oprot.WriteMessageBegin("UpsertTrajectoryConfig", thrift.REPLY, seqId); err2 != nil { - err = err2 - } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 +func (p *TraceServiceListPreSpanResult) Read(iprot thrift.TProtocol) (err error) { + var fieldTypeId thrift.TType + var fieldId int16 + + if _, err = iprot.ReadStructBegin(); err != nil { + goto ReadStructBeginError } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 + + for { + _, fieldTypeId, fieldId, err = iprot.ReadFieldBegin() + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + + switch fieldId { + case 0: + if fieldTypeId == thrift.STRUCT { + if err = p.ReadField0(iprot); err != nil { + goto ReadFieldError + } + } else if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + default: + if err = iprot.Skip(fieldTypeId); err != nil { + goto SkipFieldError + } + } + if err = iprot.ReadFieldEnd(); err != nil { + goto ReadFieldEndError + } } - if err != nil { - return + if err = iprot.ReadStructEnd(); err != nil { + goto ReadStructEndError } - return true, err -} -type traceServiceProcessorGetTrajectoryConfig struct { - handler TraceService + return nil +ReadStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T read struct begin error: ", p), err) +ReadFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListPreSpanResult[fieldId]), err) +SkipFieldError: + return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) + +ReadFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T read field end error", p), err) +ReadStructEndError: + return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *traceServiceProcessorGetTrajectoryConfig) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceGetTrajectoryConfigArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("GetTrajectoryConfig", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err +func (p *TraceServiceListPreSpanResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewListPreSpanResponse() + if err := _field.Read(iprot); err != nil { + return err } + p.Success = _field + return nil +} - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceGetTrajectoryConfigResult{} - var retval *GetTrajectoryConfigResponse - if retval, err2 = p.handler.GetTrajectoryConfig(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing GetTrajectoryConfig: "+err2.Error()) - oprot.WriteMessageBegin("GetTrajectoryConfig", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval - } - if err2 = oprot.WriteMessageBegin("GetTrajectoryConfig", thrift.REPLY, seqId); err2 != nil { - err = err2 - } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 +func (p *TraceServiceListPreSpanResult) Write(oprot thrift.TProtocol) (err error) { + var fieldId int16 + if err = oprot.WriteStructBegin("ListPreSpan_result"); err != nil { + goto WriteStructBeginError } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 + if p != nil { + if err = p.writeField0(oprot); err != nil { + fieldId = 0 + goto WriteFieldError + } } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 + if err = oprot.WriteFieldStop(); err != nil { + goto WriteFieldStopError } - if err != nil { - return + if err = oprot.WriteStructEnd(); err != nil { + goto WriteStructEndError } - return true, err + return nil +WriteStructBeginError: + return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) +WriteFieldError: + return thrift.PrependError(fmt.Sprintf("%T write field %d error: ", p, fieldId), err) +WriteFieldStopError: + return thrift.PrependError(fmt.Sprintf("%T write field stop error: ", p), err) +WriteStructEndError: + return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -type traceServiceProcessorListTrajectory struct { - handler TraceService +func (p *TraceServiceListPreSpanResult) writeField0(oprot thrift.TProtocol) (err error) { + if p.IsSetSuccess() { + if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { + goto WriteFieldBeginError + } + if err := p.Success.Write(oprot); err != nil { + return err + } + if err = oprot.WriteFieldEnd(); err != nil { + goto WriteFieldEndError + } + } + return nil +WriteFieldBeginError: + return thrift.PrependError(fmt.Sprintf("%T write field 0 begin error: ", p), err) +WriteFieldEndError: + return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *traceServiceProcessorListTrajectory) Process(ctx context.Context, seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) { - args := TraceServiceListTrajectoryArgs{} - if err = args.Read(iprot); err != nil { - iprot.ReadMessageEnd() - x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error()) - oprot.WriteMessageBegin("ListTrajectory", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return false, err +func (p *TraceServiceListPreSpanResult) String() string { + if p == nil { + return "" } + return fmt.Sprintf("TraceServiceListPreSpanResult(%+v)", *p) - iprot.ReadMessageEnd() - var err2 error - result := TraceServiceListTrajectoryResult{} - var retval *ListTrajectoryResponse - if retval, err2 = p.handler.ListTrajectory(ctx, args.Req); err2 != nil { - x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ListTrajectory: "+err2.Error()) - oprot.WriteMessageBegin("ListTrajectory", thrift.EXCEPTION, seqId) - x.Write(oprot) - oprot.WriteMessageEnd() - oprot.Flush(ctx) - return true, err2 - } else { - result.Success = retval - } - if err2 = oprot.WriteMessageBegin("ListTrajectory", thrift.REPLY, seqId); err2 != nil { - err = err2 - } - if err2 = result.Write(oprot); err == nil && err2 != nil { - err = err2 - } - if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil { - err = err2 +} + +func (p *TraceServiceListPreSpanResult) DeepEqual(ano *TraceServiceListPreSpanResult) bool { + if p == ano { + return true + } else if p == nil || ano == nil { + return false } - if err2 = oprot.Flush(ctx); err == nil && err2 != nil { - err = err2 + if !p.Field0DeepEqual(ano.Success) { + return false } - if err != nil { - return + return true +} + +func (p *TraceServiceListPreSpanResult) Field0DeepEqual(src *ListPreSpanResponse) bool { + + if !p.Success.DeepEqual(src) { + return false } - return true, err + return true } -type TraceServiceListSpansArgs struct { - Req *ListSpansRequest `thrift:"req,1" frugal:"1,default,ListSpansRequest"` +type TraceServiceGetTraceArgs struct { + Req *GetTraceRequest `thrift:"req,1" frugal:"1,default,GetTraceRequest"` } -func NewTraceServiceListSpansArgs() *TraceServiceListSpansArgs { - return &TraceServiceListSpansArgs{} +func NewTraceServiceGetTraceArgs() *TraceServiceGetTraceArgs { + return &TraceServiceGetTraceArgs{} } -func (p *TraceServiceListSpansArgs) InitDefault() { +func (p *TraceServiceGetTraceArgs) InitDefault() { } -var TraceServiceListSpansArgs_Req_DEFAULT *ListSpansRequest +var TraceServiceGetTraceArgs_Req_DEFAULT *GetTraceRequest -func (p *TraceServiceListSpansArgs) GetReq() (v *ListSpansRequest) { +func (p *TraceServiceGetTraceArgs) GetReq() (v *GetTraceRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceListSpansArgs_Req_DEFAULT + return TraceServiceGetTraceArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceListSpansArgs) SetReq(val *ListSpansRequest) { +func (p *TraceServiceGetTraceArgs) SetReq(val *GetTraceRequest) { p.Req = val } -var fieldIDToName_TraceServiceListSpansArgs = map[int16]string{ +var fieldIDToName_TraceServiceGetTraceArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceListSpansArgs) IsSetReq() bool { +func (p *TraceServiceGetTraceArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceListSpansArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTraceArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -22422,7 +25022,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListSpansArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTraceArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -22432,8 +25032,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListSpansArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewListSpansRequest() +func (p *TraceServiceGetTraceArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewGetTraceRequest() if err := _field.Read(iprot); err != nil { return err } @@ -22441,9 +25041,9 @@ func (p *TraceServiceListSpansArgs) ReadField1(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceListSpansArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTraceArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListSpans_args"); err != nil { + if err = oprot.WriteStructBegin("GetTrace_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -22469,7 +25069,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListSpansArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTraceArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -22486,15 +25086,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceListSpansArgs) String() string { +func (p *TraceServiceGetTraceArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListSpansArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceGetTraceArgs(%+v)", *p) } -func (p *TraceServiceListSpansArgs) DeepEqual(ano *TraceServiceListSpansArgs) bool { +func (p *TraceServiceGetTraceArgs) DeepEqual(ano *TraceServiceGetTraceArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -22506,7 +25106,7 @@ func (p *TraceServiceListSpansArgs) DeepEqual(ano *TraceServiceListSpansArgs) bo return true } -func (p *TraceServiceListSpansArgs) Field1DeepEqual(src *ListSpansRequest) bool { +func (p *TraceServiceGetTraceArgs) Field1DeepEqual(src *GetTraceRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -22514,41 +25114,41 @@ func (p *TraceServiceListSpansArgs) Field1DeepEqual(src *ListSpansRequest) bool return true } -type TraceServiceListSpansResult struct { - Success *ListSpansResponse `thrift:"success,0,optional" frugal:"0,optional,ListSpansResponse"` +type TraceServiceGetTraceResult struct { + Success *GetTraceResponse `thrift:"success,0,optional" frugal:"0,optional,GetTraceResponse"` } -func NewTraceServiceListSpansResult() *TraceServiceListSpansResult { - return &TraceServiceListSpansResult{} +func NewTraceServiceGetTraceResult() *TraceServiceGetTraceResult { + return &TraceServiceGetTraceResult{} } -func (p *TraceServiceListSpansResult) InitDefault() { +func (p *TraceServiceGetTraceResult) InitDefault() { } -var TraceServiceListSpansResult_Success_DEFAULT *ListSpansResponse +var TraceServiceGetTraceResult_Success_DEFAULT *GetTraceResponse -func (p *TraceServiceListSpansResult) GetSuccess() (v *ListSpansResponse) { +func (p *TraceServiceGetTraceResult) GetSuccess() (v *GetTraceResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceListSpansResult_Success_DEFAULT + return TraceServiceGetTraceResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceListSpansResult) SetSuccess(x interface{}) { - p.Success = x.(*ListSpansResponse) +func (p *TraceServiceGetTraceResult) SetSuccess(x interface{}) { + p.Success = x.(*GetTraceResponse) } -var fieldIDToName_TraceServiceListSpansResult = map[int16]string{ +var fieldIDToName_TraceServiceGetTraceResult = map[int16]string{ 0: "success", } -func (p *TraceServiceListSpansResult) IsSetSuccess() bool { +func (p *TraceServiceGetTraceResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceListSpansResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTraceResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -22593,7 +25193,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListSpansResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTraceResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -22603,8 +25203,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListSpansResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewListSpansResponse() +func (p *TraceServiceGetTraceResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewGetTraceResponse() if err := _field.Read(iprot); err != nil { return err } @@ -22612,9 +25212,9 @@ func (p *TraceServiceListSpansResult) ReadField0(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceListSpansResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTraceResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListSpans_result"); err != nil { + if err = oprot.WriteStructBegin("GetTrace_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -22640,7 +25240,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListSpansResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTraceResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -22659,15 +25259,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceListSpansResult) String() string { +func (p *TraceServiceGetTraceResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListSpansResult(%+v)", *p) + return fmt.Sprintf("TraceServiceGetTraceResult(%+v)", *p) } -func (p *TraceServiceListSpansResult) DeepEqual(ano *TraceServiceListSpansResult) bool { +func (p *TraceServiceGetTraceResult) DeepEqual(ano *TraceServiceGetTraceResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -22679,7 +25279,7 @@ func (p *TraceServiceListSpansResult) DeepEqual(ano *TraceServiceListSpansResult return true } -func (p *TraceServiceListSpansResult) Field0DeepEqual(src *ListSpansResponse) bool { +func (p *TraceServiceGetTraceResult) Field0DeepEqual(src *GetTraceResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -22687,41 +25287,41 @@ func (p *TraceServiceListSpansResult) Field0DeepEqual(src *ListSpansResponse) bo return true } -type TraceServiceListPreSpanArgs struct { - Req *ListPreSpanRequest `thrift:"req,1" frugal:"1,default,ListPreSpanRequest"` +type TraceServiceSearchTraceTreeArgs struct { + Req *SearchTraceTreeRequest `thrift:"req,1" frugal:"1,default,SearchTraceTreeRequest"` } -func NewTraceServiceListPreSpanArgs() *TraceServiceListPreSpanArgs { - return &TraceServiceListPreSpanArgs{} +func NewTraceServiceSearchTraceTreeArgs() *TraceServiceSearchTraceTreeArgs { + return &TraceServiceSearchTraceTreeArgs{} } -func (p *TraceServiceListPreSpanArgs) InitDefault() { +func (p *TraceServiceSearchTraceTreeArgs) InitDefault() { } -var TraceServiceListPreSpanArgs_Req_DEFAULT *ListPreSpanRequest +var TraceServiceSearchTraceTreeArgs_Req_DEFAULT *SearchTraceTreeRequest -func (p *TraceServiceListPreSpanArgs) GetReq() (v *ListPreSpanRequest) { +func (p *TraceServiceSearchTraceTreeArgs) GetReq() (v *SearchTraceTreeRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceListPreSpanArgs_Req_DEFAULT + return TraceServiceSearchTraceTreeArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceListPreSpanArgs) SetReq(val *ListPreSpanRequest) { +func (p *TraceServiceSearchTraceTreeArgs) SetReq(val *SearchTraceTreeRequest) { p.Req = val } -var fieldIDToName_TraceServiceListPreSpanArgs = map[int16]string{ +var fieldIDToName_TraceServiceSearchTraceTreeArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceListPreSpanArgs) IsSetReq() bool { +func (p *TraceServiceSearchTraceTreeArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceListPreSpanArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceSearchTraceTreeArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -22766,7 +25366,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListPreSpanArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceSearchTraceTreeArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -22776,8 +25376,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListPreSpanArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewListPreSpanRequest() +func (p *TraceServiceSearchTraceTreeArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewSearchTraceTreeRequest() if err := _field.Read(iprot); err != nil { return err } @@ -22785,9 +25385,9 @@ func (p *TraceServiceListPreSpanArgs) ReadField1(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceListPreSpanArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceSearchTraceTreeArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListPreSpan_args"); err != nil { + if err = oprot.WriteStructBegin("SearchTraceTree_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -22813,7 +25413,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListPreSpanArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceSearchTraceTreeArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -22830,15 +25430,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceListPreSpanArgs) String() string { +func (p *TraceServiceSearchTraceTreeArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListPreSpanArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceSearchTraceTreeArgs(%+v)", *p) } -func (p *TraceServiceListPreSpanArgs) DeepEqual(ano *TraceServiceListPreSpanArgs) bool { +func (p *TraceServiceSearchTraceTreeArgs) DeepEqual(ano *TraceServiceSearchTraceTreeArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -22850,7 +25450,7 @@ func (p *TraceServiceListPreSpanArgs) DeepEqual(ano *TraceServiceListPreSpanArgs return true } -func (p *TraceServiceListPreSpanArgs) Field1DeepEqual(src *ListPreSpanRequest) bool { +func (p *TraceServiceSearchTraceTreeArgs) Field1DeepEqual(src *SearchTraceTreeRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -22858,41 +25458,41 @@ func (p *TraceServiceListPreSpanArgs) Field1DeepEqual(src *ListPreSpanRequest) b return true } -type TraceServiceListPreSpanResult struct { - Success *ListPreSpanResponse `thrift:"success,0,optional" frugal:"0,optional,ListPreSpanResponse"` +type TraceServiceSearchTraceTreeResult struct { + Success *SearchTraceTreeResponse `thrift:"success,0,optional" frugal:"0,optional,SearchTraceTreeResponse"` } -func NewTraceServiceListPreSpanResult() *TraceServiceListPreSpanResult { - return &TraceServiceListPreSpanResult{} +func NewTraceServiceSearchTraceTreeResult() *TraceServiceSearchTraceTreeResult { + return &TraceServiceSearchTraceTreeResult{} } -func (p *TraceServiceListPreSpanResult) InitDefault() { +func (p *TraceServiceSearchTraceTreeResult) InitDefault() { } -var TraceServiceListPreSpanResult_Success_DEFAULT *ListPreSpanResponse +var TraceServiceSearchTraceTreeResult_Success_DEFAULT *SearchTraceTreeResponse -func (p *TraceServiceListPreSpanResult) GetSuccess() (v *ListPreSpanResponse) { +func (p *TraceServiceSearchTraceTreeResult) GetSuccess() (v *SearchTraceTreeResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceListPreSpanResult_Success_DEFAULT + return TraceServiceSearchTraceTreeResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceListPreSpanResult) SetSuccess(x interface{}) { - p.Success = x.(*ListPreSpanResponse) +func (p *TraceServiceSearchTraceTreeResult) SetSuccess(x interface{}) { + p.Success = x.(*SearchTraceTreeResponse) } -var fieldIDToName_TraceServiceListPreSpanResult = map[int16]string{ +var fieldIDToName_TraceServiceSearchTraceTreeResult = map[int16]string{ 0: "success", } -func (p *TraceServiceListPreSpanResult) IsSetSuccess() bool { +func (p *TraceServiceSearchTraceTreeResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceListPreSpanResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceSearchTraceTreeResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -22937,7 +25537,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListPreSpanResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceSearchTraceTreeResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -22947,8 +25547,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListPreSpanResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewListPreSpanResponse() +func (p *TraceServiceSearchTraceTreeResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewSearchTraceTreeResponse() if err := _field.Read(iprot); err != nil { return err } @@ -22956,9 +25556,9 @@ func (p *TraceServiceListPreSpanResult) ReadField0(iprot thrift.TProtocol) error return nil } -func (p *TraceServiceListPreSpanResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceSearchTraceTreeResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListPreSpan_result"); err != nil { + if err = oprot.WriteStructBegin("SearchTraceTree_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -22984,7 +25584,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListPreSpanResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceSearchTraceTreeResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -23003,15 +25603,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceListPreSpanResult) String() string { +func (p *TraceServiceSearchTraceTreeResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListPreSpanResult(%+v)", *p) + return fmt.Sprintf("TraceServiceSearchTraceTreeResult(%+v)", *p) } -func (p *TraceServiceListPreSpanResult) DeepEqual(ano *TraceServiceListPreSpanResult) bool { +func (p *TraceServiceSearchTraceTreeResult) DeepEqual(ano *TraceServiceSearchTraceTreeResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -23023,7 +25623,7 @@ func (p *TraceServiceListPreSpanResult) DeepEqual(ano *TraceServiceListPreSpanRe return true } -func (p *TraceServiceListPreSpanResult) Field0DeepEqual(src *ListPreSpanResponse) bool { +func (p *TraceServiceSearchTraceTreeResult) Field0DeepEqual(src *SearchTraceTreeResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -23031,41 +25631,41 @@ func (p *TraceServiceListPreSpanResult) Field0DeepEqual(src *ListPreSpanResponse return true } -type TraceServiceGetTraceArgs struct { - Req *GetTraceRequest `thrift:"req,1" frugal:"1,default,GetTraceRequest"` +type TraceServiceBatchGetTracesAdvanceInfoArgs struct { + Req *BatchGetTracesAdvanceInfoRequest `thrift:"req,1" frugal:"1,default,BatchGetTracesAdvanceInfoRequest"` } -func NewTraceServiceGetTraceArgs() *TraceServiceGetTraceArgs { - return &TraceServiceGetTraceArgs{} +func NewTraceServiceBatchGetTracesAdvanceInfoArgs() *TraceServiceBatchGetTracesAdvanceInfoArgs { + return &TraceServiceBatchGetTracesAdvanceInfoArgs{} } -func (p *TraceServiceGetTraceArgs) InitDefault() { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) InitDefault() { } -var TraceServiceGetTraceArgs_Req_DEFAULT *GetTraceRequest +var TraceServiceBatchGetTracesAdvanceInfoArgs_Req_DEFAULT *BatchGetTracesAdvanceInfoRequest -func (p *TraceServiceGetTraceArgs) GetReq() (v *GetTraceRequest) { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) GetReq() (v *BatchGetTracesAdvanceInfoRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceGetTraceArgs_Req_DEFAULT + return TraceServiceBatchGetTracesAdvanceInfoArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceGetTraceArgs) SetReq(val *GetTraceRequest) { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) SetReq(val *BatchGetTracesAdvanceInfoRequest) { p.Req = val } -var fieldIDToName_TraceServiceGetTraceArgs = map[int16]string{ +var fieldIDToName_TraceServiceBatchGetTracesAdvanceInfoArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceGetTraceArgs) IsSetReq() bool { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceGetTraceArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -23110,7 +25710,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTraceArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceBatchGetTracesAdvanceInfoArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -23120,8 +25720,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceGetTraceArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewGetTraceRequest() +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewBatchGetTracesAdvanceInfoRequest() if err := _field.Read(iprot); err != nil { return err } @@ -23129,9 +25729,9 @@ func (p *TraceServiceGetTraceArgs) ReadField1(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceGetTraceArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("GetTrace_args"); err != nil { + if err = oprot.WriteStructBegin("BatchGetTracesAdvanceInfo_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -23157,7 +25757,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceGetTraceArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -23174,15 +25774,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceGetTraceArgs) String() string { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceGetTraceArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceBatchGetTracesAdvanceInfoArgs(%+v)", *p) } -func (p *TraceServiceGetTraceArgs) DeepEqual(ano *TraceServiceGetTraceArgs) bool { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) DeepEqual(ano *TraceServiceBatchGetTracesAdvanceInfoArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -23194,7 +25794,7 @@ func (p *TraceServiceGetTraceArgs) DeepEqual(ano *TraceServiceGetTraceArgs) bool return true } -func (p *TraceServiceGetTraceArgs) Field1DeepEqual(src *GetTraceRequest) bool { +func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) Field1DeepEqual(src *BatchGetTracesAdvanceInfoRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -23202,41 +25802,41 @@ func (p *TraceServiceGetTraceArgs) Field1DeepEqual(src *GetTraceRequest) bool { return true } -type TraceServiceGetTraceResult struct { - Success *GetTraceResponse `thrift:"success,0,optional" frugal:"0,optional,GetTraceResponse"` +type TraceServiceBatchGetTracesAdvanceInfoResult struct { + Success *BatchGetTracesAdvanceInfoResponse `thrift:"success,0,optional" frugal:"0,optional,BatchGetTracesAdvanceInfoResponse"` } -func NewTraceServiceGetTraceResult() *TraceServiceGetTraceResult { - return &TraceServiceGetTraceResult{} +func NewTraceServiceBatchGetTracesAdvanceInfoResult() *TraceServiceBatchGetTracesAdvanceInfoResult { + return &TraceServiceBatchGetTracesAdvanceInfoResult{} } -func (p *TraceServiceGetTraceResult) InitDefault() { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) InitDefault() { } -var TraceServiceGetTraceResult_Success_DEFAULT *GetTraceResponse +var TraceServiceBatchGetTracesAdvanceInfoResult_Success_DEFAULT *BatchGetTracesAdvanceInfoResponse -func (p *TraceServiceGetTraceResult) GetSuccess() (v *GetTraceResponse) { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) GetSuccess() (v *BatchGetTracesAdvanceInfoResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceGetTraceResult_Success_DEFAULT + return TraceServiceBatchGetTracesAdvanceInfoResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceGetTraceResult) SetSuccess(x interface{}) { - p.Success = x.(*GetTraceResponse) +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) SetSuccess(x interface{}) { + p.Success = x.(*BatchGetTracesAdvanceInfoResponse) } -var fieldIDToName_TraceServiceGetTraceResult = map[int16]string{ +var fieldIDToName_TraceServiceBatchGetTracesAdvanceInfoResult = map[int16]string{ 0: "success", } -func (p *TraceServiceGetTraceResult) IsSetSuccess() bool { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceGetTraceResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -23281,7 +25881,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTraceResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceBatchGetTracesAdvanceInfoResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -23291,8 +25891,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceGetTraceResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewGetTraceResponse() +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewBatchGetTracesAdvanceInfoResponse() if err := _field.Read(iprot); err != nil { return err } @@ -23300,9 +25900,9 @@ func (p *TraceServiceGetTraceResult) ReadField0(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceGetTraceResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("GetTrace_result"); err != nil { + if err = oprot.WriteStructBegin("BatchGetTracesAdvanceInfo_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -23328,7 +25928,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceGetTraceResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -23347,15 +25947,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceGetTraceResult) String() string { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceGetTraceResult(%+v)", *p) + return fmt.Sprintf("TraceServiceBatchGetTracesAdvanceInfoResult(%+v)", *p) } -func (p *TraceServiceGetTraceResult) DeepEqual(ano *TraceServiceGetTraceResult) bool { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) DeepEqual(ano *TraceServiceBatchGetTracesAdvanceInfoResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -23367,7 +25967,7 @@ func (p *TraceServiceGetTraceResult) DeepEqual(ano *TraceServiceGetTraceResult) return true } -func (p *TraceServiceGetTraceResult) Field0DeepEqual(src *GetTraceResponse) bool { +func (p *TraceServiceBatchGetTracesAdvanceInfoResult) Field0DeepEqual(src *BatchGetTracesAdvanceInfoResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -23375,41 +25975,41 @@ func (p *TraceServiceGetTraceResult) Field0DeepEqual(src *GetTraceResponse) bool return true } -type TraceServiceSearchTraceTreeArgs struct { - Req *SearchTraceTreeRequest `thrift:"req,1" frugal:"1,default,SearchTraceTreeRequest"` +type TraceServiceIngestTracesInnerArgs struct { + Req *IngestTracesRequest `thrift:"req,1" frugal:"1,default,IngestTracesRequest"` } -func NewTraceServiceSearchTraceTreeArgs() *TraceServiceSearchTraceTreeArgs { - return &TraceServiceSearchTraceTreeArgs{} +func NewTraceServiceIngestTracesInnerArgs() *TraceServiceIngestTracesInnerArgs { + return &TraceServiceIngestTracesInnerArgs{} } -func (p *TraceServiceSearchTraceTreeArgs) InitDefault() { +func (p *TraceServiceIngestTracesInnerArgs) InitDefault() { } -var TraceServiceSearchTraceTreeArgs_Req_DEFAULT *SearchTraceTreeRequest +var TraceServiceIngestTracesInnerArgs_Req_DEFAULT *IngestTracesRequest -func (p *TraceServiceSearchTraceTreeArgs) GetReq() (v *SearchTraceTreeRequest) { +func (p *TraceServiceIngestTracesInnerArgs) GetReq() (v *IngestTracesRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceSearchTraceTreeArgs_Req_DEFAULT + return TraceServiceIngestTracesInnerArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceSearchTraceTreeArgs) SetReq(val *SearchTraceTreeRequest) { +func (p *TraceServiceIngestTracesInnerArgs) SetReq(val *IngestTracesRequest) { p.Req = val } -var fieldIDToName_TraceServiceSearchTraceTreeArgs = map[int16]string{ +var fieldIDToName_TraceServiceIngestTracesInnerArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceSearchTraceTreeArgs) IsSetReq() bool { +func (p *TraceServiceIngestTracesInnerArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceSearchTraceTreeArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceIngestTracesInnerArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -23454,7 +26054,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceSearchTraceTreeArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceIngestTracesInnerArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -23464,8 +26064,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceSearchTraceTreeArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewSearchTraceTreeRequest() +func (p *TraceServiceIngestTracesInnerArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewIngestTracesRequest() if err := _field.Read(iprot); err != nil { return err } @@ -23473,9 +26073,9 @@ func (p *TraceServiceSearchTraceTreeArgs) ReadField1(iprot thrift.TProtocol) err return nil } -func (p *TraceServiceSearchTraceTreeArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceIngestTracesInnerArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("SearchTraceTree_args"); err != nil { + if err = oprot.WriteStructBegin("IngestTracesInner_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -23501,7 +26101,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceSearchTraceTreeArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceIngestTracesInnerArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -23518,15 +26118,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceSearchTraceTreeArgs) String() string { +func (p *TraceServiceIngestTracesInnerArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceSearchTraceTreeArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceIngestTracesInnerArgs(%+v)", *p) } -func (p *TraceServiceSearchTraceTreeArgs) DeepEqual(ano *TraceServiceSearchTraceTreeArgs) bool { +func (p *TraceServiceIngestTracesInnerArgs) DeepEqual(ano *TraceServiceIngestTracesInnerArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -23538,7 +26138,7 @@ func (p *TraceServiceSearchTraceTreeArgs) DeepEqual(ano *TraceServiceSearchTrace return true } -func (p *TraceServiceSearchTraceTreeArgs) Field1DeepEqual(src *SearchTraceTreeRequest) bool { +func (p *TraceServiceIngestTracesInnerArgs) Field1DeepEqual(src *IngestTracesRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -23546,41 +26146,41 @@ func (p *TraceServiceSearchTraceTreeArgs) Field1DeepEqual(src *SearchTraceTreeRe return true } -type TraceServiceSearchTraceTreeResult struct { - Success *SearchTraceTreeResponse `thrift:"success,0,optional" frugal:"0,optional,SearchTraceTreeResponse"` +type TraceServiceIngestTracesInnerResult struct { + Success *IngestTracesResponse `thrift:"success,0,optional" frugal:"0,optional,IngestTracesResponse"` } -func NewTraceServiceSearchTraceTreeResult() *TraceServiceSearchTraceTreeResult { - return &TraceServiceSearchTraceTreeResult{} +func NewTraceServiceIngestTracesInnerResult() *TraceServiceIngestTracesInnerResult { + return &TraceServiceIngestTracesInnerResult{} } -func (p *TraceServiceSearchTraceTreeResult) InitDefault() { +func (p *TraceServiceIngestTracesInnerResult) InitDefault() { } -var TraceServiceSearchTraceTreeResult_Success_DEFAULT *SearchTraceTreeResponse +var TraceServiceIngestTracesInnerResult_Success_DEFAULT *IngestTracesResponse -func (p *TraceServiceSearchTraceTreeResult) GetSuccess() (v *SearchTraceTreeResponse) { +func (p *TraceServiceIngestTracesInnerResult) GetSuccess() (v *IngestTracesResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceSearchTraceTreeResult_Success_DEFAULT + return TraceServiceIngestTracesInnerResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceSearchTraceTreeResult) SetSuccess(x interface{}) { - p.Success = x.(*SearchTraceTreeResponse) +func (p *TraceServiceIngestTracesInnerResult) SetSuccess(x interface{}) { + p.Success = x.(*IngestTracesResponse) } -var fieldIDToName_TraceServiceSearchTraceTreeResult = map[int16]string{ +var fieldIDToName_TraceServiceIngestTracesInnerResult = map[int16]string{ 0: "success", } -func (p *TraceServiceSearchTraceTreeResult) IsSetSuccess() bool { +func (p *TraceServiceIngestTracesInnerResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceSearchTraceTreeResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceIngestTracesInnerResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -23625,7 +26225,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceSearchTraceTreeResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceIngestTracesInnerResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -23635,8 +26235,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceSearchTraceTreeResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewSearchTraceTreeResponse() +func (p *TraceServiceIngestTracesInnerResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewIngestTracesResponse() if err := _field.Read(iprot); err != nil { return err } @@ -23644,9 +26244,9 @@ func (p *TraceServiceSearchTraceTreeResult) ReadField0(iprot thrift.TProtocol) e return nil } -func (p *TraceServiceSearchTraceTreeResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceIngestTracesInnerResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("SearchTraceTree_result"); err != nil { + if err = oprot.WriteStructBegin("IngestTracesInner_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -23672,7 +26272,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceSearchTraceTreeResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceIngestTracesInnerResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -23691,15 +26291,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceSearchTraceTreeResult) String() string { +func (p *TraceServiceIngestTracesInnerResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceSearchTraceTreeResult(%+v)", *p) + return fmt.Sprintf("TraceServiceIngestTracesInnerResult(%+v)", *p) } -func (p *TraceServiceSearchTraceTreeResult) DeepEqual(ano *TraceServiceSearchTraceTreeResult) bool { +func (p *TraceServiceIngestTracesInnerResult) DeepEqual(ano *TraceServiceIngestTracesInnerResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -23711,7 +26311,7 @@ func (p *TraceServiceSearchTraceTreeResult) DeepEqual(ano *TraceServiceSearchTra return true } -func (p *TraceServiceSearchTraceTreeResult) Field0DeepEqual(src *SearchTraceTreeResponse) bool { +func (p *TraceServiceIngestTracesInnerResult) Field0DeepEqual(src *IngestTracesResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -23719,41 +26319,41 @@ func (p *TraceServiceSearchTraceTreeResult) Field0DeepEqual(src *SearchTraceTree return true } -type TraceServiceBatchGetTracesAdvanceInfoArgs struct { - Req *BatchGetTracesAdvanceInfoRequest `thrift:"req,1" frugal:"1,default,BatchGetTracesAdvanceInfoRequest"` +type TraceServiceGetTracesMetaInfoArgs struct { + Req *GetTracesMetaInfoRequest `thrift:"req,1" frugal:"1,default,GetTracesMetaInfoRequest"` } -func NewTraceServiceBatchGetTracesAdvanceInfoArgs() *TraceServiceBatchGetTracesAdvanceInfoArgs { - return &TraceServiceBatchGetTracesAdvanceInfoArgs{} +func NewTraceServiceGetTracesMetaInfoArgs() *TraceServiceGetTracesMetaInfoArgs { + return &TraceServiceGetTracesMetaInfoArgs{} } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) InitDefault() { +func (p *TraceServiceGetTracesMetaInfoArgs) InitDefault() { } -var TraceServiceBatchGetTracesAdvanceInfoArgs_Req_DEFAULT *BatchGetTracesAdvanceInfoRequest +var TraceServiceGetTracesMetaInfoArgs_Req_DEFAULT *GetTracesMetaInfoRequest -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) GetReq() (v *BatchGetTracesAdvanceInfoRequest) { +func (p *TraceServiceGetTracesMetaInfoArgs) GetReq() (v *GetTracesMetaInfoRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceBatchGetTracesAdvanceInfoArgs_Req_DEFAULT + return TraceServiceGetTracesMetaInfoArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) SetReq(val *BatchGetTracesAdvanceInfoRequest) { +func (p *TraceServiceGetTracesMetaInfoArgs) SetReq(val *GetTracesMetaInfoRequest) { p.Req = val } -var fieldIDToName_TraceServiceBatchGetTracesAdvanceInfoArgs = map[int16]string{ +var fieldIDToName_TraceServiceGetTracesMetaInfoArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) IsSetReq() bool { +func (p *TraceServiceGetTracesMetaInfoArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTracesMetaInfoArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -23798,7 +26398,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceBatchGetTracesAdvanceInfoArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTracesMetaInfoArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -23808,8 +26408,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewBatchGetTracesAdvanceInfoRequest() +func (p *TraceServiceGetTracesMetaInfoArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewGetTracesMetaInfoRequest() if err := _field.Read(iprot); err != nil { return err } @@ -23817,9 +26417,9 @@ func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) ReadField1(iprot thrift.TPro return nil } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTracesMetaInfoArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("BatchGetTracesAdvanceInfo_args"); err != nil { + if err = oprot.WriteStructBegin("GetTracesMetaInfo_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -23845,7 +26445,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTracesMetaInfoArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -23862,15 +26462,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) String() string { +func (p *TraceServiceGetTracesMetaInfoArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceBatchGetTracesAdvanceInfoArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceGetTracesMetaInfoArgs(%+v)", *p) } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) DeepEqual(ano *TraceServiceBatchGetTracesAdvanceInfoArgs) bool { +func (p *TraceServiceGetTracesMetaInfoArgs) DeepEqual(ano *TraceServiceGetTracesMetaInfoArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -23882,7 +26482,7 @@ func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) DeepEqual(ano *TraceServiceB return true } -func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) Field1DeepEqual(src *BatchGetTracesAdvanceInfoRequest) bool { +func (p *TraceServiceGetTracesMetaInfoArgs) Field1DeepEqual(src *GetTracesMetaInfoRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -23890,41 +26490,41 @@ func (p *TraceServiceBatchGetTracesAdvanceInfoArgs) Field1DeepEqual(src *BatchGe return true } -type TraceServiceBatchGetTracesAdvanceInfoResult struct { - Success *BatchGetTracesAdvanceInfoResponse `thrift:"success,0,optional" frugal:"0,optional,BatchGetTracesAdvanceInfoResponse"` +type TraceServiceGetTracesMetaInfoResult struct { + Success *GetTracesMetaInfoResponse `thrift:"success,0,optional" frugal:"0,optional,GetTracesMetaInfoResponse"` } -func NewTraceServiceBatchGetTracesAdvanceInfoResult() *TraceServiceBatchGetTracesAdvanceInfoResult { - return &TraceServiceBatchGetTracesAdvanceInfoResult{} +func NewTraceServiceGetTracesMetaInfoResult() *TraceServiceGetTracesMetaInfoResult { + return &TraceServiceGetTracesMetaInfoResult{} } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) InitDefault() { +func (p *TraceServiceGetTracesMetaInfoResult) InitDefault() { } -var TraceServiceBatchGetTracesAdvanceInfoResult_Success_DEFAULT *BatchGetTracesAdvanceInfoResponse +var TraceServiceGetTracesMetaInfoResult_Success_DEFAULT *GetTracesMetaInfoResponse -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) GetSuccess() (v *BatchGetTracesAdvanceInfoResponse) { +func (p *TraceServiceGetTracesMetaInfoResult) GetSuccess() (v *GetTracesMetaInfoResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceBatchGetTracesAdvanceInfoResult_Success_DEFAULT + return TraceServiceGetTracesMetaInfoResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) SetSuccess(x interface{}) { - p.Success = x.(*BatchGetTracesAdvanceInfoResponse) +func (p *TraceServiceGetTracesMetaInfoResult) SetSuccess(x interface{}) { + p.Success = x.(*GetTracesMetaInfoResponse) } -var fieldIDToName_TraceServiceBatchGetTracesAdvanceInfoResult = map[int16]string{ +var fieldIDToName_TraceServiceGetTracesMetaInfoResult = map[int16]string{ 0: "success", } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) IsSetSuccess() bool { +func (p *TraceServiceGetTracesMetaInfoResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTracesMetaInfoResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -23969,7 +26569,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceBatchGetTracesAdvanceInfoResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTracesMetaInfoResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -23979,8 +26579,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewBatchGetTracesAdvanceInfoResponse() +func (p *TraceServiceGetTracesMetaInfoResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewGetTracesMetaInfoResponse() if err := _field.Read(iprot); err != nil { return err } @@ -23988,9 +26588,9 @@ func (p *TraceServiceBatchGetTracesAdvanceInfoResult) ReadField0(iprot thrift.TP return nil } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTracesMetaInfoResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("BatchGetTracesAdvanceInfo_result"); err != nil { + if err = oprot.WriteStructBegin("GetTracesMetaInfo_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -24016,7 +26616,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTracesMetaInfoResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -24035,15 +26635,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) String() string { +func (p *TraceServiceGetTracesMetaInfoResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceBatchGetTracesAdvanceInfoResult(%+v)", *p) + return fmt.Sprintf("TraceServiceGetTracesMetaInfoResult(%+v)", *p) } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) DeepEqual(ano *TraceServiceBatchGetTracesAdvanceInfoResult) bool { +func (p *TraceServiceGetTracesMetaInfoResult) DeepEqual(ano *TraceServiceGetTracesMetaInfoResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -24055,7 +26655,7 @@ func (p *TraceServiceBatchGetTracesAdvanceInfoResult) DeepEqual(ano *TraceServic return true } -func (p *TraceServiceBatchGetTracesAdvanceInfoResult) Field0DeepEqual(src *BatchGetTracesAdvanceInfoResponse) bool { +func (p *TraceServiceGetTracesMetaInfoResult) Field0DeepEqual(src *GetTracesMetaInfoResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -24063,41 +26663,41 @@ func (p *TraceServiceBatchGetTracesAdvanceInfoResult) Field0DeepEqual(src *Batch return true } -type TraceServiceIngestTracesInnerArgs struct { - Req *IngestTracesRequest `thrift:"req,1" frugal:"1,default,IngestTracesRequest"` +type TraceServiceCreateViewArgs struct { + Req *CreateViewRequest `thrift:"req,1" frugal:"1,default,CreateViewRequest"` } -func NewTraceServiceIngestTracesInnerArgs() *TraceServiceIngestTracesInnerArgs { - return &TraceServiceIngestTracesInnerArgs{} +func NewTraceServiceCreateViewArgs() *TraceServiceCreateViewArgs { + return &TraceServiceCreateViewArgs{} } -func (p *TraceServiceIngestTracesInnerArgs) InitDefault() { +func (p *TraceServiceCreateViewArgs) InitDefault() { } -var TraceServiceIngestTracesInnerArgs_Req_DEFAULT *IngestTracesRequest +var TraceServiceCreateViewArgs_Req_DEFAULT *CreateViewRequest -func (p *TraceServiceIngestTracesInnerArgs) GetReq() (v *IngestTracesRequest) { +func (p *TraceServiceCreateViewArgs) GetReq() (v *CreateViewRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceIngestTracesInnerArgs_Req_DEFAULT + return TraceServiceCreateViewArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceIngestTracesInnerArgs) SetReq(val *IngestTracesRequest) { +func (p *TraceServiceCreateViewArgs) SetReq(val *CreateViewRequest) { p.Req = val } -var fieldIDToName_TraceServiceIngestTracesInnerArgs = map[int16]string{ +var fieldIDToName_TraceServiceCreateViewArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceIngestTracesInnerArgs) IsSetReq() bool { +func (p *TraceServiceCreateViewArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceIngestTracesInnerArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateViewArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -24142,7 +26742,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceIngestTracesInnerArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceCreateViewArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -24152,8 +26752,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceIngestTracesInnerArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewIngestTracesRequest() +func (p *TraceServiceCreateViewArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewCreateViewRequest() if err := _field.Read(iprot); err != nil { return err } @@ -24161,9 +26761,9 @@ func (p *TraceServiceIngestTracesInnerArgs) ReadField1(iprot thrift.TProtocol) e return nil } -func (p *TraceServiceIngestTracesInnerArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateViewArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("IngestTracesInner_args"); err != nil { + if err = oprot.WriteStructBegin("CreateView_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -24189,7 +26789,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceIngestTracesInnerArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateViewArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -24206,15 +26806,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceIngestTracesInnerArgs) String() string { +func (p *TraceServiceCreateViewArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceIngestTracesInnerArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceCreateViewArgs(%+v)", *p) } -func (p *TraceServiceIngestTracesInnerArgs) DeepEqual(ano *TraceServiceIngestTracesInnerArgs) bool { +func (p *TraceServiceCreateViewArgs) DeepEqual(ano *TraceServiceCreateViewArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -24226,7 +26826,7 @@ func (p *TraceServiceIngestTracesInnerArgs) DeepEqual(ano *TraceServiceIngestTra return true } -func (p *TraceServiceIngestTracesInnerArgs) Field1DeepEqual(src *IngestTracesRequest) bool { +func (p *TraceServiceCreateViewArgs) Field1DeepEqual(src *CreateViewRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -24234,41 +26834,41 @@ func (p *TraceServiceIngestTracesInnerArgs) Field1DeepEqual(src *IngestTracesReq return true } -type TraceServiceIngestTracesInnerResult struct { - Success *IngestTracesResponse `thrift:"success,0,optional" frugal:"0,optional,IngestTracesResponse"` +type TraceServiceCreateViewResult struct { + Success *CreateViewResponse `thrift:"success,0,optional" frugal:"0,optional,CreateViewResponse"` } -func NewTraceServiceIngestTracesInnerResult() *TraceServiceIngestTracesInnerResult { - return &TraceServiceIngestTracesInnerResult{} +func NewTraceServiceCreateViewResult() *TraceServiceCreateViewResult { + return &TraceServiceCreateViewResult{} } -func (p *TraceServiceIngestTracesInnerResult) InitDefault() { +func (p *TraceServiceCreateViewResult) InitDefault() { } -var TraceServiceIngestTracesInnerResult_Success_DEFAULT *IngestTracesResponse +var TraceServiceCreateViewResult_Success_DEFAULT *CreateViewResponse -func (p *TraceServiceIngestTracesInnerResult) GetSuccess() (v *IngestTracesResponse) { +func (p *TraceServiceCreateViewResult) GetSuccess() (v *CreateViewResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceIngestTracesInnerResult_Success_DEFAULT + return TraceServiceCreateViewResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceIngestTracesInnerResult) SetSuccess(x interface{}) { - p.Success = x.(*IngestTracesResponse) +func (p *TraceServiceCreateViewResult) SetSuccess(x interface{}) { + p.Success = x.(*CreateViewResponse) } -var fieldIDToName_TraceServiceIngestTracesInnerResult = map[int16]string{ +var fieldIDToName_TraceServiceCreateViewResult = map[int16]string{ 0: "success", } -func (p *TraceServiceIngestTracesInnerResult) IsSetSuccess() bool { +func (p *TraceServiceCreateViewResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceIngestTracesInnerResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateViewResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -24313,7 +26913,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceIngestTracesInnerResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceCreateViewResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -24323,8 +26923,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceIngestTracesInnerResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewIngestTracesResponse() +func (p *TraceServiceCreateViewResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewCreateViewResponse() if err := _field.Read(iprot); err != nil { return err } @@ -24332,9 +26932,9 @@ func (p *TraceServiceIngestTracesInnerResult) ReadField0(iprot thrift.TProtocol) return nil } -func (p *TraceServiceIngestTracesInnerResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateViewResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("IngestTracesInner_result"); err != nil { + if err = oprot.WriteStructBegin("CreateView_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -24360,7 +26960,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceIngestTracesInnerResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateViewResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -24379,15 +26979,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceIngestTracesInnerResult) String() string { +func (p *TraceServiceCreateViewResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceIngestTracesInnerResult(%+v)", *p) + return fmt.Sprintf("TraceServiceCreateViewResult(%+v)", *p) } -func (p *TraceServiceIngestTracesInnerResult) DeepEqual(ano *TraceServiceIngestTracesInnerResult) bool { +func (p *TraceServiceCreateViewResult) DeepEqual(ano *TraceServiceCreateViewResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -24399,7 +26999,7 @@ func (p *TraceServiceIngestTracesInnerResult) DeepEqual(ano *TraceServiceIngestT return true } -func (p *TraceServiceIngestTracesInnerResult) Field0DeepEqual(src *IngestTracesResponse) bool { +func (p *TraceServiceCreateViewResult) Field0DeepEqual(src *CreateViewResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -24407,41 +27007,41 @@ func (p *TraceServiceIngestTracesInnerResult) Field0DeepEqual(src *IngestTracesR return true } -type TraceServiceGetTracesMetaInfoArgs struct { - Req *GetTracesMetaInfoRequest `thrift:"req,1" frugal:"1,default,GetTracesMetaInfoRequest"` +type TraceServiceUpdateViewArgs struct { + Req *UpdateViewRequest `thrift:"req,1" frugal:"1,default,UpdateViewRequest"` } -func NewTraceServiceGetTracesMetaInfoArgs() *TraceServiceGetTracesMetaInfoArgs { - return &TraceServiceGetTracesMetaInfoArgs{} +func NewTraceServiceUpdateViewArgs() *TraceServiceUpdateViewArgs { + return &TraceServiceUpdateViewArgs{} } -func (p *TraceServiceGetTracesMetaInfoArgs) InitDefault() { +func (p *TraceServiceUpdateViewArgs) InitDefault() { } -var TraceServiceGetTracesMetaInfoArgs_Req_DEFAULT *GetTracesMetaInfoRequest +var TraceServiceUpdateViewArgs_Req_DEFAULT *UpdateViewRequest -func (p *TraceServiceGetTracesMetaInfoArgs) GetReq() (v *GetTracesMetaInfoRequest) { +func (p *TraceServiceUpdateViewArgs) GetReq() (v *UpdateViewRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceGetTracesMetaInfoArgs_Req_DEFAULT + return TraceServiceUpdateViewArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceGetTracesMetaInfoArgs) SetReq(val *GetTracesMetaInfoRequest) { +func (p *TraceServiceUpdateViewArgs) SetReq(val *UpdateViewRequest) { p.Req = val } -var fieldIDToName_TraceServiceGetTracesMetaInfoArgs = map[int16]string{ +var fieldIDToName_TraceServiceUpdateViewArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceGetTracesMetaInfoArgs) IsSetReq() bool { +func (p *TraceServiceUpdateViewArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceGetTracesMetaInfoArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateViewArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -24486,7 +27086,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTracesMetaInfoArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpdateViewArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -24496,8 +27096,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceGetTracesMetaInfoArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewGetTracesMetaInfoRequest() +func (p *TraceServiceUpdateViewArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewUpdateViewRequest() if err := _field.Read(iprot); err != nil { return err } @@ -24505,9 +27105,9 @@ func (p *TraceServiceGetTracesMetaInfoArgs) ReadField1(iprot thrift.TProtocol) e return nil } -func (p *TraceServiceGetTracesMetaInfoArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateViewArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("GetTracesMetaInfo_args"); err != nil { + if err = oprot.WriteStructBegin("UpdateView_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -24533,7 +27133,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceGetTracesMetaInfoArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateViewArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -24550,15 +27150,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceGetTracesMetaInfoArgs) String() string { +func (p *TraceServiceUpdateViewArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceGetTracesMetaInfoArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceUpdateViewArgs(%+v)", *p) } -func (p *TraceServiceGetTracesMetaInfoArgs) DeepEqual(ano *TraceServiceGetTracesMetaInfoArgs) bool { +func (p *TraceServiceUpdateViewArgs) DeepEqual(ano *TraceServiceUpdateViewArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -24570,7 +27170,7 @@ func (p *TraceServiceGetTracesMetaInfoArgs) DeepEqual(ano *TraceServiceGetTraces return true } -func (p *TraceServiceGetTracesMetaInfoArgs) Field1DeepEqual(src *GetTracesMetaInfoRequest) bool { +func (p *TraceServiceUpdateViewArgs) Field1DeepEqual(src *UpdateViewRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -24578,41 +27178,41 @@ func (p *TraceServiceGetTracesMetaInfoArgs) Field1DeepEqual(src *GetTracesMetaIn return true } -type TraceServiceGetTracesMetaInfoResult struct { - Success *GetTracesMetaInfoResponse `thrift:"success,0,optional" frugal:"0,optional,GetTracesMetaInfoResponse"` +type TraceServiceUpdateViewResult struct { + Success *UpdateViewResponse `thrift:"success,0,optional" frugal:"0,optional,UpdateViewResponse"` } -func NewTraceServiceGetTracesMetaInfoResult() *TraceServiceGetTracesMetaInfoResult { - return &TraceServiceGetTracesMetaInfoResult{} +func NewTraceServiceUpdateViewResult() *TraceServiceUpdateViewResult { + return &TraceServiceUpdateViewResult{} } -func (p *TraceServiceGetTracesMetaInfoResult) InitDefault() { +func (p *TraceServiceUpdateViewResult) InitDefault() { } -var TraceServiceGetTracesMetaInfoResult_Success_DEFAULT *GetTracesMetaInfoResponse +var TraceServiceUpdateViewResult_Success_DEFAULT *UpdateViewResponse -func (p *TraceServiceGetTracesMetaInfoResult) GetSuccess() (v *GetTracesMetaInfoResponse) { +func (p *TraceServiceUpdateViewResult) GetSuccess() (v *UpdateViewResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceGetTracesMetaInfoResult_Success_DEFAULT + return TraceServiceUpdateViewResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceGetTracesMetaInfoResult) SetSuccess(x interface{}) { - p.Success = x.(*GetTracesMetaInfoResponse) +func (p *TraceServiceUpdateViewResult) SetSuccess(x interface{}) { + p.Success = x.(*UpdateViewResponse) } -var fieldIDToName_TraceServiceGetTracesMetaInfoResult = map[int16]string{ +var fieldIDToName_TraceServiceUpdateViewResult = map[int16]string{ 0: "success", } -func (p *TraceServiceGetTracesMetaInfoResult) IsSetSuccess() bool { +func (p *TraceServiceUpdateViewResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceGetTracesMetaInfoResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateViewResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -24657,7 +27257,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTracesMetaInfoResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpdateViewResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -24667,8 +27267,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceGetTracesMetaInfoResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewGetTracesMetaInfoResponse() +func (p *TraceServiceUpdateViewResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewUpdateViewResponse() if err := _field.Read(iprot); err != nil { return err } @@ -24676,9 +27276,9 @@ func (p *TraceServiceGetTracesMetaInfoResult) ReadField0(iprot thrift.TProtocol) return nil } -func (p *TraceServiceGetTracesMetaInfoResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateViewResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("GetTracesMetaInfo_result"); err != nil { + if err = oprot.WriteStructBegin("UpdateView_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -24704,7 +27304,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceGetTracesMetaInfoResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateViewResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -24723,15 +27323,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceGetTracesMetaInfoResult) String() string { +func (p *TraceServiceUpdateViewResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceGetTracesMetaInfoResult(%+v)", *p) + return fmt.Sprintf("TraceServiceUpdateViewResult(%+v)", *p) } -func (p *TraceServiceGetTracesMetaInfoResult) DeepEqual(ano *TraceServiceGetTracesMetaInfoResult) bool { +func (p *TraceServiceUpdateViewResult) DeepEqual(ano *TraceServiceUpdateViewResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -24743,7 +27343,7 @@ func (p *TraceServiceGetTracesMetaInfoResult) DeepEqual(ano *TraceServiceGetTrac return true } -func (p *TraceServiceGetTracesMetaInfoResult) Field0DeepEqual(src *GetTracesMetaInfoResponse) bool { +func (p *TraceServiceUpdateViewResult) Field0DeepEqual(src *UpdateViewResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -24751,41 +27351,41 @@ func (p *TraceServiceGetTracesMetaInfoResult) Field0DeepEqual(src *GetTracesMeta return true } -type TraceServiceCreateViewArgs struct { - Req *CreateViewRequest `thrift:"req,1" frugal:"1,default,CreateViewRequest"` +type TraceServiceDeleteViewArgs struct { + Req *DeleteViewRequest `thrift:"req,1" frugal:"1,default,DeleteViewRequest"` } -func NewTraceServiceCreateViewArgs() *TraceServiceCreateViewArgs { - return &TraceServiceCreateViewArgs{} +func NewTraceServiceDeleteViewArgs() *TraceServiceDeleteViewArgs { + return &TraceServiceDeleteViewArgs{} } -func (p *TraceServiceCreateViewArgs) InitDefault() { +func (p *TraceServiceDeleteViewArgs) InitDefault() { } -var TraceServiceCreateViewArgs_Req_DEFAULT *CreateViewRequest +var TraceServiceDeleteViewArgs_Req_DEFAULT *DeleteViewRequest -func (p *TraceServiceCreateViewArgs) GetReq() (v *CreateViewRequest) { +func (p *TraceServiceDeleteViewArgs) GetReq() (v *DeleteViewRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceCreateViewArgs_Req_DEFAULT + return TraceServiceDeleteViewArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceCreateViewArgs) SetReq(val *CreateViewRequest) { +func (p *TraceServiceDeleteViewArgs) SetReq(val *DeleteViewRequest) { p.Req = val } -var fieldIDToName_TraceServiceCreateViewArgs = map[int16]string{ +var fieldIDToName_TraceServiceDeleteViewArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceCreateViewArgs) IsSetReq() bool { +func (p *TraceServiceDeleteViewArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceCreateViewArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteViewArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -24830,7 +27430,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceCreateViewArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceDeleteViewArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -24840,8 +27440,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceCreateViewArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewCreateViewRequest() +func (p *TraceServiceDeleteViewArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewDeleteViewRequest() if err := _field.Read(iprot); err != nil { return err } @@ -24849,9 +27449,9 @@ func (p *TraceServiceCreateViewArgs) ReadField1(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceCreateViewArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteViewArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("CreateView_args"); err != nil { + if err = oprot.WriteStructBegin("DeleteView_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -24877,7 +27477,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceCreateViewArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteViewArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -24894,15 +27494,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceCreateViewArgs) String() string { +func (p *TraceServiceDeleteViewArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceCreateViewArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceDeleteViewArgs(%+v)", *p) } -func (p *TraceServiceCreateViewArgs) DeepEqual(ano *TraceServiceCreateViewArgs) bool { +func (p *TraceServiceDeleteViewArgs) DeepEqual(ano *TraceServiceDeleteViewArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -24914,7 +27514,7 @@ func (p *TraceServiceCreateViewArgs) DeepEqual(ano *TraceServiceCreateViewArgs) return true } -func (p *TraceServiceCreateViewArgs) Field1DeepEqual(src *CreateViewRequest) bool { +func (p *TraceServiceDeleteViewArgs) Field1DeepEqual(src *DeleteViewRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -24922,41 +27522,41 @@ func (p *TraceServiceCreateViewArgs) Field1DeepEqual(src *CreateViewRequest) boo return true } -type TraceServiceCreateViewResult struct { - Success *CreateViewResponse `thrift:"success,0,optional" frugal:"0,optional,CreateViewResponse"` +type TraceServiceDeleteViewResult struct { + Success *DeleteViewResponse `thrift:"success,0,optional" frugal:"0,optional,DeleteViewResponse"` } -func NewTraceServiceCreateViewResult() *TraceServiceCreateViewResult { - return &TraceServiceCreateViewResult{} +func NewTraceServiceDeleteViewResult() *TraceServiceDeleteViewResult { + return &TraceServiceDeleteViewResult{} } -func (p *TraceServiceCreateViewResult) InitDefault() { +func (p *TraceServiceDeleteViewResult) InitDefault() { } -var TraceServiceCreateViewResult_Success_DEFAULT *CreateViewResponse +var TraceServiceDeleteViewResult_Success_DEFAULT *DeleteViewResponse -func (p *TraceServiceCreateViewResult) GetSuccess() (v *CreateViewResponse) { +func (p *TraceServiceDeleteViewResult) GetSuccess() (v *DeleteViewResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceCreateViewResult_Success_DEFAULT + return TraceServiceDeleteViewResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceCreateViewResult) SetSuccess(x interface{}) { - p.Success = x.(*CreateViewResponse) +func (p *TraceServiceDeleteViewResult) SetSuccess(x interface{}) { + p.Success = x.(*DeleteViewResponse) } -var fieldIDToName_TraceServiceCreateViewResult = map[int16]string{ +var fieldIDToName_TraceServiceDeleteViewResult = map[int16]string{ 0: "success", } -func (p *TraceServiceCreateViewResult) IsSetSuccess() bool { +func (p *TraceServiceDeleteViewResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceCreateViewResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteViewResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -25001,7 +27601,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceCreateViewResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceDeleteViewResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -25011,8 +27611,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceCreateViewResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewCreateViewResponse() +func (p *TraceServiceDeleteViewResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewDeleteViewResponse() if err := _field.Read(iprot); err != nil { return err } @@ -25020,9 +27620,9 @@ func (p *TraceServiceCreateViewResult) ReadField0(iprot thrift.TProtocol) error return nil } -func (p *TraceServiceCreateViewResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteViewResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("CreateView_result"); err != nil { + if err = oprot.WriteStructBegin("DeleteView_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -25048,7 +27648,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceCreateViewResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteViewResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -25067,15 +27667,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceCreateViewResult) String() string { +func (p *TraceServiceDeleteViewResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceCreateViewResult(%+v)", *p) + return fmt.Sprintf("TraceServiceDeleteViewResult(%+v)", *p) } -func (p *TraceServiceCreateViewResult) DeepEqual(ano *TraceServiceCreateViewResult) bool { +func (p *TraceServiceDeleteViewResult) DeepEqual(ano *TraceServiceDeleteViewResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -25087,7 +27687,7 @@ func (p *TraceServiceCreateViewResult) DeepEqual(ano *TraceServiceCreateViewResu return true } -func (p *TraceServiceCreateViewResult) Field0DeepEqual(src *CreateViewResponse) bool { +func (p *TraceServiceDeleteViewResult) Field0DeepEqual(src *DeleteViewResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -25095,41 +27695,41 @@ func (p *TraceServiceCreateViewResult) Field0DeepEqual(src *CreateViewResponse) return true } -type TraceServiceUpdateViewArgs struct { - Req *UpdateViewRequest `thrift:"req,1" frugal:"1,default,UpdateViewRequest"` +type TraceServiceListViewsArgs struct { + Req *ListViewsRequest `thrift:"req,1" frugal:"1,default,ListViewsRequest"` } -func NewTraceServiceUpdateViewArgs() *TraceServiceUpdateViewArgs { - return &TraceServiceUpdateViewArgs{} +func NewTraceServiceListViewsArgs() *TraceServiceListViewsArgs { + return &TraceServiceListViewsArgs{} } -func (p *TraceServiceUpdateViewArgs) InitDefault() { +func (p *TraceServiceListViewsArgs) InitDefault() { } -var TraceServiceUpdateViewArgs_Req_DEFAULT *UpdateViewRequest +var TraceServiceListViewsArgs_Req_DEFAULT *ListViewsRequest -func (p *TraceServiceUpdateViewArgs) GetReq() (v *UpdateViewRequest) { +func (p *TraceServiceListViewsArgs) GetReq() (v *ListViewsRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceUpdateViewArgs_Req_DEFAULT + return TraceServiceListViewsArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceUpdateViewArgs) SetReq(val *UpdateViewRequest) { +func (p *TraceServiceListViewsArgs) SetReq(val *ListViewsRequest) { p.Req = val } -var fieldIDToName_TraceServiceUpdateViewArgs = map[int16]string{ +var fieldIDToName_TraceServiceListViewsArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceUpdateViewArgs) IsSetReq() bool { +func (p *TraceServiceListViewsArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceUpdateViewArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListViewsArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -25174,7 +27774,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpdateViewArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListViewsArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -25184,8 +27784,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceUpdateViewArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewUpdateViewRequest() +func (p *TraceServiceListViewsArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewListViewsRequest() if err := _field.Read(iprot); err != nil { return err } @@ -25193,9 +27793,9 @@ func (p *TraceServiceUpdateViewArgs) ReadField1(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceUpdateViewArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListViewsArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("UpdateView_args"); err != nil { + if err = oprot.WriteStructBegin("ListViews_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -25221,7 +27821,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceUpdateViewArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListViewsArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -25238,15 +27838,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceUpdateViewArgs) String() string { +func (p *TraceServiceListViewsArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceUpdateViewArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceListViewsArgs(%+v)", *p) } -func (p *TraceServiceUpdateViewArgs) DeepEqual(ano *TraceServiceUpdateViewArgs) bool { +func (p *TraceServiceListViewsArgs) DeepEqual(ano *TraceServiceListViewsArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -25258,7 +27858,7 @@ func (p *TraceServiceUpdateViewArgs) DeepEqual(ano *TraceServiceUpdateViewArgs) return true } -func (p *TraceServiceUpdateViewArgs) Field1DeepEqual(src *UpdateViewRequest) bool { +func (p *TraceServiceListViewsArgs) Field1DeepEqual(src *ListViewsRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -25266,41 +27866,41 @@ func (p *TraceServiceUpdateViewArgs) Field1DeepEqual(src *UpdateViewRequest) boo return true } -type TraceServiceUpdateViewResult struct { - Success *UpdateViewResponse `thrift:"success,0,optional" frugal:"0,optional,UpdateViewResponse"` +type TraceServiceListViewsResult struct { + Success *ListViewsResponse `thrift:"success,0,optional" frugal:"0,optional,ListViewsResponse"` } -func NewTraceServiceUpdateViewResult() *TraceServiceUpdateViewResult { - return &TraceServiceUpdateViewResult{} +func NewTraceServiceListViewsResult() *TraceServiceListViewsResult { + return &TraceServiceListViewsResult{} } -func (p *TraceServiceUpdateViewResult) InitDefault() { +func (p *TraceServiceListViewsResult) InitDefault() { } -var TraceServiceUpdateViewResult_Success_DEFAULT *UpdateViewResponse +var TraceServiceListViewsResult_Success_DEFAULT *ListViewsResponse -func (p *TraceServiceUpdateViewResult) GetSuccess() (v *UpdateViewResponse) { +func (p *TraceServiceListViewsResult) GetSuccess() (v *ListViewsResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceUpdateViewResult_Success_DEFAULT + return TraceServiceListViewsResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceUpdateViewResult) SetSuccess(x interface{}) { - p.Success = x.(*UpdateViewResponse) +func (p *TraceServiceListViewsResult) SetSuccess(x interface{}) { + p.Success = x.(*ListViewsResponse) } -var fieldIDToName_TraceServiceUpdateViewResult = map[int16]string{ +var fieldIDToName_TraceServiceListViewsResult = map[int16]string{ 0: "success", } -func (p *TraceServiceUpdateViewResult) IsSetSuccess() bool { +func (p *TraceServiceListViewsResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceUpdateViewResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListViewsResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -25345,7 +27945,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpdateViewResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListViewsResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -25355,8 +27955,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceUpdateViewResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewUpdateViewResponse() +func (p *TraceServiceListViewsResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewListViewsResponse() if err := _field.Read(iprot); err != nil { return err } @@ -25364,9 +27964,9 @@ func (p *TraceServiceUpdateViewResult) ReadField0(iprot thrift.TProtocol) error return nil } -func (p *TraceServiceUpdateViewResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListViewsResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("UpdateView_result"); err != nil { + if err = oprot.WriteStructBegin("ListViews_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -25392,7 +27992,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceUpdateViewResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListViewsResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -25411,15 +28011,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceUpdateViewResult) String() string { +func (p *TraceServiceListViewsResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceUpdateViewResult(%+v)", *p) + return fmt.Sprintf("TraceServiceListViewsResult(%+v)", *p) } -func (p *TraceServiceUpdateViewResult) DeepEqual(ano *TraceServiceUpdateViewResult) bool { +func (p *TraceServiceListViewsResult) DeepEqual(ano *TraceServiceListViewsResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -25431,7 +28031,7 @@ func (p *TraceServiceUpdateViewResult) DeepEqual(ano *TraceServiceUpdateViewResu return true } -func (p *TraceServiceUpdateViewResult) Field0DeepEqual(src *UpdateViewResponse) bool { +func (p *TraceServiceListViewsResult) Field0DeepEqual(src *ListViewsResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -25439,41 +28039,41 @@ func (p *TraceServiceUpdateViewResult) Field0DeepEqual(src *UpdateViewResponse) return true } -type TraceServiceDeleteViewArgs struct { - Req *DeleteViewRequest `thrift:"req,1" frugal:"1,default,DeleteViewRequest"` +type TraceServiceCreateManualAnnotationArgs struct { + Req *CreateManualAnnotationRequest `thrift:"req,1" frugal:"1,default,CreateManualAnnotationRequest"` } -func NewTraceServiceDeleteViewArgs() *TraceServiceDeleteViewArgs { - return &TraceServiceDeleteViewArgs{} +func NewTraceServiceCreateManualAnnotationArgs() *TraceServiceCreateManualAnnotationArgs { + return &TraceServiceCreateManualAnnotationArgs{} } -func (p *TraceServiceDeleteViewArgs) InitDefault() { +func (p *TraceServiceCreateManualAnnotationArgs) InitDefault() { } -var TraceServiceDeleteViewArgs_Req_DEFAULT *DeleteViewRequest +var TraceServiceCreateManualAnnotationArgs_Req_DEFAULT *CreateManualAnnotationRequest -func (p *TraceServiceDeleteViewArgs) GetReq() (v *DeleteViewRequest) { +func (p *TraceServiceCreateManualAnnotationArgs) GetReq() (v *CreateManualAnnotationRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceDeleteViewArgs_Req_DEFAULT + return TraceServiceCreateManualAnnotationArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceDeleteViewArgs) SetReq(val *DeleteViewRequest) { +func (p *TraceServiceCreateManualAnnotationArgs) SetReq(val *CreateManualAnnotationRequest) { p.Req = val } -var fieldIDToName_TraceServiceDeleteViewArgs = map[int16]string{ +var fieldIDToName_TraceServiceCreateManualAnnotationArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceDeleteViewArgs) IsSetReq() bool { +func (p *TraceServiceCreateManualAnnotationArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceDeleteViewArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateManualAnnotationArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -25518,7 +28118,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceDeleteViewArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceCreateManualAnnotationArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -25528,8 +28128,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceDeleteViewArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewDeleteViewRequest() +func (p *TraceServiceCreateManualAnnotationArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewCreateManualAnnotationRequest() if err := _field.Read(iprot); err != nil { return err } @@ -25537,9 +28137,9 @@ func (p *TraceServiceDeleteViewArgs) ReadField1(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceDeleteViewArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateManualAnnotationArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("DeleteView_args"); err != nil { + if err = oprot.WriteStructBegin("CreateManualAnnotation_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -25565,7 +28165,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceDeleteViewArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateManualAnnotationArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -25582,15 +28182,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceDeleteViewArgs) String() string { +func (p *TraceServiceCreateManualAnnotationArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceDeleteViewArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceCreateManualAnnotationArgs(%+v)", *p) } -func (p *TraceServiceDeleteViewArgs) DeepEqual(ano *TraceServiceDeleteViewArgs) bool { +func (p *TraceServiceCreateManualAnnotationArgs) DeepEqual(ano *TraceServiceCreateManualAnnotationArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -25602,7 +28202,7 @@ func (p *TraceServiceDeleteViewArgs) DeepEqual(ano *TraceServiceDeleteViewArgs) return true } -func (p *TraceServiceDeleteViewArgs) Field1DeepEqual(src *DeleteViewRequest) bool { +func (p *TraceServiceCreateManualAnnotationArgs) Field1DeepEqual(src *CreateManualAnnotationRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -25610,41 +28210,41 @@ func (p *TraceServiceDeleteViewArgs) Field1DeepEqual(src *DeleteViewRequest) boo return true } -type TraceServiceDeleteViewResult struct { - Success *DeleteViewResponse `thrift:"success,0,optional" frugal:"0,optional,DeleteViewResponse"` +type TraceServiceCreateManualAnnotationResult struct { + Success *CreateManualAnnotationResponse `thrift:"success,0,optional" frugal:"0,optional,CreateManualAnnotationResponse"` } -func NewTraceServiceDeleteViewResult() *TraceServiceDeleteViewResult { - return &TraceServiceDeleteViewResult{} +func NewTraceServiceCreateManualAnnotationResult() *TraceServiceCreateManualAnnotationResult { + return &TraceServiceCreateManualAnnotationResult{} } -func (p *TraceServiceDeleteViewResult) InitDefault() { +func (p *TraceServiceCreateManualAnnotationResult) InitDefault() { } -var TraceServiceDeleteViewResult_Success_DEFAULT *DeleteViewResponse +var TraceServiceCreateManualAnnotationResult_Success_DEFAULT *CreateManualAnnotationResponse -func (p *TraceServiceDeleteViewResult) GetSuccess() (v *DeleteViewResponse) { +func (p *TraceServiceCreateManualAnnotationResult) GetSuccess() (v *CreateManualAnnotationResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceDeleteViewResult_Success_DEFAULT + return TraceServiceCreateManualAnnotationResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceDeleteViewResult) SetSuccess(x interface{}) { - p.Success = x.(*DeleteViewResponse) +func (p *TraceServiceCreateManualAnnotationResult) SetSuccess(x interface{}) { + p.Success = x.(*CreateManualAnnotationResponse) } -var fieldIDToName_TraceServiceDeleteViewResult = map[int16]string{ +var fieldIDToName_TraceServiceCreateManualAnnotationResult = map[int16]string{ 0: "success", } -func (p *TraceServiceDeleteViewResult) IsSetSuccess() bool { +func (p *TraceServiceCreateManualAnnotationResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceDeleteViewResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateManualAnnotationResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -25689,7 +28289,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceDeleteViewResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceCreateManualAnnotationResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -25699,8 +28299,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceDeleteViewResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewDeleteViewResponse() +func (p *TraceServiceCreateManualAnnotationResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewCreateManualAnnotationResponse() if err := _field.Read(iprot); err != nil { return err } @@ -25708,9 +28308,9 @@ func (p *TraceServiceDeleteViewResult) ReadField0(iprot thrift.TProtocol) error return nil } -func (p *TraceServiceDeleteViewResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateManualAnnotationResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("DeleteView_result"); err != nil { + if err = oprot.WriteStructBegin("CreateManualAnnotation_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -25736,7 +28336,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceDeleteViewResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceCreateManualAnnotationResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -25755,15 +28355,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceDeleteViewResult) String() string { +func (p *TraceServiceCreateManualAnnotationResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceDeleteViewResult(%+v)", *p) + return fmt.Sprintf("TraceServiceCreateManualAnnotationResult(%+v)", *p) } -func (p *TraceServiceDeleteViewResult) DeepEqual(ano *TraceServiceDeleteViewResult) bool { +func (p *TraceServiceCreateManualAnnotationResult) DeepEqual(ano *TraceServiceCreateManualAnnotationResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -25775,7 +28375,7 @@ func (p *TraceServiceDeleteViewResult) DeepEqual(ano *TraceServiceDeleteViewResu return true } -func (p *TraceServiceDeleteViewResult) Field0DeepEqual(src *DeleteViewResponse) bool { +func (p *TraceServiceCreateManualAnnotationResult) Field0DeepEqual(src *CreateManualAnnotationResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -25783,41 +28383,41 @@ func (p *TraceServiceDeleteViewResult) Field0DeepEqual(src *DeleteViewResponse) return true } -type TraceServiceListViewsArgs struct { - Req *ListViewsRequest `thrift:"req,1" frugal:"1,default,ListViewsRequest"` +type TraceServiceUpdateManualAnnotationArgs struct { + Req *UpdateManualAnnotationRequest `thrift:"req,1" frugal:"1,default,UpdateManualAnnotationRequest"` } -func NewTraceServiceListViewsArgs() *TraceServiceListViewsArgs { - return &TraceServiceListViewsArgs{} +func NewTraceServiceUpdateManualAnnotationArgs() *TraceServiceUpdateManualAnnotationArgs { + return &TraceServiceUpdateManualAnnotationArgs{} } -func (p *TraceServiceListViewsArgs) InitDefault() { +func (p *TraceServiceUpdateManualAnnotationArgs) InitDefault() { } -var TraceServiceListViewsArgs_Req_DEFAULT *ListViewsRequest +var TraceServiceUpdateManualAnnotationArgs_Req_DEFAULT *UpdateManualAnnotationRequest -func (p *TraceServiceListViewsArgs) GetReq() (v *ListViewsRequest) { +func (p *TraceServiceUpdateManualAnnotationArgs) GetReq() (v *UpdateManualAnnotationRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceListViewsArgs_Req_DEFAULT + return TraceServiceUpdateManualAnnotationArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceListViewsArgs) SetReq(val *ListViewsRequest) { +func (p *TraceServiceUpdateManualAnnotationArgs) SetReq(val *UpdateManualAnnotationRequest) { p.Req = val } -var fieldIDToName_TraceServiceListViewsArgs = map[int16]string{ +var fieldIDToName_TraceServiceUpdateManualAnnotationArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceListViewsArgs) IsSetReq() bool { +func (p *TraceServiceUpdateManualAnnotationArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceListViewsArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateManualAnnotationArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -25862,7 +28462,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListViewsArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpdateManualAnnotationArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -25872,8 +28472,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListViewsArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewListViewsRequest() +func (p *TraceServiceUpdateManualAnnotationArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewUpdateManualAnnotationRequest() if err := _field.Read(iprot); err != nil { return err } @@ -25881,9 +28481,9 @@ func (p *TraceServiceListViewsArgs) ReadField1(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceListViewsArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateManualAnnotationArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListViews_args"); err != nil { + if err = oprot.WriteStructBegin("UpdateManualAnnotation_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -25909,7 +28509,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListViewsArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateManualAnnotationArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -25926,15 +28526,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceListViewsArgs) String() string { +func (p *TraceServiceUpdateManualAnnotationArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListViewsArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceUpdateManualAnnotationArgs(%+v)", *p) } -func (p *TraceServiceListViewsArgs) DeepEqual(ano *TraceServiceListViewsArgs) bool { +func (p *TraceServiceUpdateManualAnnotationArgs) DeepEqual(ano *TraceServiceUpdateManualAnnotationArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -25946,7 +28546,7 @@ func (p *TraceServiceListViewsArgs) DeepEqual(ano *TraceServiceListViewsArgs) bo return true } -func (p *TraceServiceListViewsArgs) Field1DeepEqual(src *ListViewsRequest) bool { +func (p *TraceServiceUpdateManualAnnotationArgs) Field1DeepEqual(src *UpdateManualAnnotationRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -25954,41 +28554,41 @@ func (p *TraceServiceListViewsArgs) Field1DeepEqual(src *ListViewsRequest) bool return true } -type TraceServiceListViewsResult struct { - Success *ListViewsResponse `thrift:"success,0,optional" frugal:"0,optional,ListViewsResponse"` +type TraceServiceUpdateManualAnnotationResult struct { + Success *UpdateManualAnnotationResponse `thrift:"success,0,optional" frugal:"0,optional,UpdateManualAnnotationResponse"` } -func NewTraceServiceListViewsResult() *TraceServiceListViewsResult { - return &TraceServiceListViewsResult{} +func NewTraceServiceUpdateManualAnnotationResult() *TraceServiceUpdateManualAnnotationResult { + return &TraceServiceUpdateManualAnnotationResult{} } -func (p *TraceServiceListViewsResult) InitDefault() { +func (p *TraceServiceUpdateManualAnnotationResult) InitDefault() { } -var TraceServiceListViewsResult_Success_DEFAULT *ListViewsResponse +var TraceServiceUpdateManualAnnotationResult_Success_DEFAULT *UpdateManualAnnotationResponse -func (p *TraceServiceListViewsResult) GetSuccess() (v *ListViewsResponse) { +func (p *TraceServiceUpdateManualAnnotationResult) GetSuccess() (v *UpdateManualAnnotationResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceListViewsResult_Success_DEFAULT + return TraceServiceUpdateManualAnnotationResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceListViewsResult) SetSuccess(x interface{}) { - p.Success = x.(*ListViewsResponse) +func (p *TraceServiceUpdateManualAnnotationResult) SetSuccess(x interface{}) { + p.Success = x.(*UpdateManualAnnotationResponse) } -var fieldIDToName_TraceServiceListViewsResult = map[int16]string{ +var fieldIDToName_TraceServiceUpdateManualAnnotationResult = map[int16]string{ 0: "success", } -func (p *TraceServiceListViewsResult) IsSetSuccess() bool { +func (p *TraceServiceUpdateManualAnnotationResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceListViewsResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateManualAnnotationResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -26033,7 +28633,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListViewsResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpdateManualAnnotationResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -26043,8 +28643,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListViewsResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewListViewsResponse() +func (p *TraceServiceUpdateManualAnnotationResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewUpdateManualAnnotationResponse() if err := _field.Read(iprot); err != nil { return err } @@ -26052,9 +28652,9 @@ func (p *TraceServiceListViewsResult) ReadField0(iprot thrift.TProtocol) error { return nil } -func (p *TraceServiceListViewsResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateManualAnnotationResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListViews_result"); err != nil { + if err = oprot.WriteStructBegin("UpdateManualAnnotation_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -26080,7 +28680,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListViewsResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpdateManualAnnotationResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -26099,15 +28699,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceListViewsResult) String() string { +func (p *TraceServiceUpdateManualAnnotationResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListViewsResult(%+v)", *p) + return fmt.Sprintf("TraceServiceUpdateManualAnnotationResult(%+v)", *p) } -func (p *TraceServiceListViewsResult) DeepEqual(ano *TraceServiceListViewsResult) bool { +func (p *TraceServiceUpdateManualAnnotationResult) DeepEqual(ano *TraceServiceUpdateManualAnnotationResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -26119,7 +28719,7 @@ func (p *TraceServiceListViewsResult) DeepEqual(ano *TraceServiceListViewsResult return true } -func (p *TraceServiceListViewsResult) Field0DeepEqual(src *ListViewsResponse) bool { +func (p *TraceServiceUpdateManualAnnotationResult) Field0DeepEqual(src *UpdateManualAnnotationResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -26127,41 +28727,41 @@ func (p *TraceServiceListViewsResult) Field0DeepEqual(src *ListViewsResponse) bo return true } -type TraceServiceCreateManualAnnotationArgs struct { - Req *CreateManualAnnotationRequest `thrift:"req,1" frugal:"1,default,CreateManualAnnotationRequest"` +type TraceServiceDeleteManualAnnotationArgs struct { + Req *DeleteManualAnnotationRequest `thrift:"req,1" frugal:"1,default,DeleteManualAnnotationRequest"` } -func NewTraceServiceCreateManualAnnotationArgs() *TraceServiceCreateManualAnnotationArgs { - return &TraceServiceCreateManualAnnotationArgs{} +func NewTraceServiceDeleteManualAnnotationArgs() *TraceServiceDeleteManualAnnotationArgs { + return &TraceServiceDeleteManualAnnotationArgs{} } -func (p *TraceServiceCreateManualAnnotationArgs) InitDefault() { +func (p *TraceServiceDeleteManualAnnotationArgs) InitDefault() { } -var TraceServiceCreateManualAnnotationArgs_Req_DEFAULT *CreateManualAnnotationRequest +var TraceServiceDeleteManualAnnotationArgs_Req_DEFAULT *DeleteManualAnnotationRequest -func (p *TraceServiceCreateManualAnnotationArgs) GetReq() (v *CreateManualAnnotationRequest) { +func (p *TraceServiceDeleteManualAnnotationArgs) GetReq() (v *DeleteManualAnnotationRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceCreateManualAnnotationArgs_Req_DEFAULT + return TraceServiceDeleteManualAnnotationArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceCreateManualAnnotationArgs) SetReq(val *CreateManualAnnotationRequest) { +func (p *TraceServiceDeleteManualAnnotationArgs) SetReq(val *DeleteManualAnnotationRequest) { p.Req = val } -var fieldIDToName_TraceServiceCreateManualAnnotationArgs = map[int16]string{ +var fieldIDToName_TraceServiceDeleteManualAnnotationArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceCreateManualAnnotationArgs) IsSetReq() bool { +func (p *TraceServiceDeleteManualAnnotationArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceCreateManualAnnotationArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteManualAnnotationArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -26206,7 +28806,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceCreateManualAnnotationArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceDeleteManualAnnotationArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -26216,8 +28816,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceCreateManualAnnotationArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewCreateManualAnnotationRequest() +func (p *TraceServiceDeleteManualAnnotationArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewDeleteManualAnnotationRequest() if err := _field.Read(iprot); err != nil { return err } @@ -26225,9 +28825,9 @@ func (p *TraceServiceCreateManualAnnotationArgs) ReadField1(iprot thrift.TProtoc return nil } -func (p *TraceServiceCreateManualAnnotationArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteManualAnnotationArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("CreateManualAnnotation_args"); err != nil { + if err = oprot.WriteStructBegin("DeleteManualAnnotation_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -26253,7 +28853,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceCreateManualAnnotationArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteManualAnnotationArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -26270,15 +28870,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceCreateManualAnnotationArgs) String() string { +func (p *TraceServiceDeleteManualAnnotationArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceCreateManualAnnotationArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceDeleteManualAnnotationArgs(%+v)", *p) } -func (p *TraceServiceCreateManualAnnotationArgs) DeepEqual(ano *TraceServiceCreateManualAnnotationArgs) bool { +func (p *TraceServiceDeleteManualAnnotationArgs) DeepEqual(ano *TraceServiceDeleteManualAnnotationArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -26290,7 +28890,7 @@ func (p *TraceServiceCreateManualAnnotationArgs) DeepEqual(ano *TraceServiceCrea return true } -func (p *TraceServiceCreateManualAnnotationArgs) Field1DeepEqual(src *CreateManualAnnotationRequest) bool { +func (p *TraceServiceDeleteManualAnnotationArgs) Field1DeepEqual(src *DeleteManualAnnotationRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -26298,41 +28898,41 @@ func (p *TraceServiceCreateManualAnnotationArgs) Field1DeepEqual(src *CreateManu return true } -type TraceServiceCreateManualAnnotationResult struct { - Success *CreateManualAnnotationResponse `thrift:"success,0,optional" frugal:"0,optional,CreateManualAnnotationResponse"` +type TraceServiceDeleteManualAnnotationResult struct { + Success *DeleteManualAnnotationResponse `thrift:"success,0,optional" frugal:"0,optional,DeleteManualAnnotationResponse"` } -func NewTraceServiceCreateManualAnnotationResult() *TraceServiceCreateManualAnnotationResult { - return &TraceServiceCreateManualAnnotationResult{} +func NewTraceServiceDeleteManualAnnotationResult() *TraceServiceDeleteManualAnnotationResult { + return &TraceServiceDeleteManualAnnotationResult{} } -func (p *TraceServiceCreateManualAnnotationResult) InitDefault() { +func (p *TraceServiceDeleteManualAnnotationResult) InitDefault() { } -var TraceServiceCreateManualAnnotationResult_Success_DEFAULT *CreateManualAnnotationResponse +var TraceServiceDeleteManualAnnotationResult_Success_DEFAULT *DeleteManualAnnotationResponse -func (p *TraceServiceCreateManualAnnotationResult) GetSuccess() (v *CreateManualAnnotationResponse) { +func (p *TraceServiceDeleteManualAnnotationResult) GetSuccess() (v *DeleteManualAnnotationResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceCreateManualAnnotationResult_Success_DEFAULT + return TraceServiceDeleteManualAnnotationResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceCreateManualAnnotationResult) SetSuccess(x interface{}) { - p.Success = x.(*CreateManualAnnotationResponse) +func (p *TraceServiceDeleteManualAnnotationResult) SetSuccess(x interface{}) { + p.Success = x.(*DeleteManualAnnotationResponse) } -var fieldIDToName_TraceServiceCreateManualAnnotationResult = map[int16]string{ +var fieldIDToName_TraceServiceDeleteManualAnnotationResult = map[int16]string{ 0: "success", } -func (p *TraceServiceCreateManualAnnotationResult) IsSetSuccess() bool { +func (p *TraceServiceDeleteManualAnnotationResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceCreateManualAnnotationResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteManualAnnotationResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -26377,7 +28977,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceCreateManualAnnotationResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceDeleteManualAnnotationResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -26387,8 +28987,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceCreateManualAnnotationResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewCreateManualAnnotationResponse() +func (p *TraceServiceDeleteManualAnnotationResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewDeleteManualAnnotationResponse() if err := _field.Read(iprot); err != nil { return err } @@ -26396,9 +28996,9 @@ func (p *TraceServiceCreateManualAnnotationResult) ReadField0(iprot thrift.TProt return nil } -func (p *TraceServiceCreateManualAnnotationResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteManualAnnotationResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("CreateManualAnnotation_result"); err != nil { + if err = oprot.WriteStructBegin("DeleteManualAnnotation_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -26424,7 +29024,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceCreateManualAnnotationResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceDeleteManualAnnotationResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -26443,15 +29043,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceCreateManualAnnotationResult) String() string { +func (p *TraceServiceDeleteManualAnnotationResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceCreateManualAnnotationResult(%+v)", *p) + return fmt.Sprintf("TraceServiceDeleteManualAnnotationResult(%+v)", *p) } -func (p *TraceServiceCreateManualAnnotationResult) DeepEqual(ano *TraceServiceCreateManualAnnotationResult) bool { +func (p *TraceServiceDeleteManualAnnotationResult) DeepEqual(ano *TraceServiceDeleteManualAnnotationResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -26463,7 +29063,7 @@ func (p *TraceServiceCreateManualAnnotationResult) DeepEqual(ano *TraceServiceCr return true } -func (p *TraceServiceCreateManualAnnotationResult) Field0DeepEqual(src *CreateManualAnnotationResponse) bool { +func (p *TraceServiceDeleteManualAnnotationResult) Field0DeepEqual(src *DeleteManualAnnotationResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -26471,41 +29071,41 @@ func (p *TraceServiceCreateManualAnnotationResult) Field0DeepEqual(src *CreateMa return true } -type TraceServiceUpdateManualAnnotationArgs struct { - Req *UpdateManualAnnotationRequest `thrift:"req,1" frugal:"1,default,UpdateManualAnnotationRequest"` +type TraceServiceListAnnotationsArgs struct { + Req *ListAnnotationsRequest `thrift:"req,1" frugal:"1,default,ListAnnotationsRequest"` } -func NewTraceServiceUpdateManualAnnotationArgs() *TraceServiceUpdateManualAnnotationArgs { - return &TraceServiceUpdateManualAnnotationArgs{} +func NewTraceServiceListAnnotationsArgs() *TraceServiceListAnnotationsArgs { + return &TraceServiceListAnnotationsArgs{} } -func (p *TraceServiceUpdateManualAnnotationArgs) InitDefault() { +func (p *TraceServiceListAnnotationsArgs) InitDefault() { } -var TraceServiceUpdateManualAnnotationArgs_Req_DEFAULT *UpdateManualAnnotationRequest +var TraceServiceListAnnotationsArgs_Req_DEFAULT *ListAnnotationsRequest -func (p *TraceServiceUpdateManualAnnotationArgs) GetReq() (v *UpdateManualAnnotationRequest) { +func (p *TraceServiceListAnnotationsArgs) GetReq() (v *ListAnnotationsRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceUpdateManualAnnotationArgs_Req_DEFAULT + return TraceServiceListAnnotationsArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceUpdateManualAnnotationArgs) SetReq(val *UpdateManualAnnotationRequest) { +func (p *TraceServiceListAnnotationsArgs) SetReq(val *ListAnnotationsRequest) { p.Req = val } -var fieldIDToName_TraceServiceUpdateManualAnnotationArgs = map[int16]string{ +var fieldIDToName_TraceServiceListAnnotationsArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceUpdateManualAnnotationArgs) IsSetReq() bool { +func (p *TraceServiceListAnnotationsArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceUpdateManualAnnotationArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationsArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -26550,7 +29150,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpdateManualAnnotationArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationsArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -26560,8 +29160,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceUpdateManualAnnotationArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewUpdateManualAnnotationRequest() +func (p *TraceServiceListAnnotationsArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewListAnnotationsRequest() if err := _field.Read(iprot); err != nil { return err } @@ -26569,9 +29169,9 @@ func (p *TraceServiceUpdateManualAnnotationArgs) ReadField1(iprot thrift.TProtoc return nil } -func (p *TraceServiceUpdateManualAnnotationArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationsArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("UpdateManualAnnotation_args"); err != nil { + if err = oprot.WriteStructBegin("ListAnnotations_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -26597,7 +29197,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceUpdateManualAnnotationArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationsArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -26614,15 +29214,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceUpdateManualAnnotationArgs) String() string { +func (p *TraceServiceListAnnotationsArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceUpdateManualAnnotationArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceListAnnotationsArgs(%+v)", *p) } -func (p *TraceServiceUpdateManualAnnotationArgs) DeepEqual(ano *TraceServiceUpdateManualAnnotationArgs) bool { +func (p *TraceServiceListAnnotationsArgs) DeepEqual(ano *TraceServiceListAnnotationsArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -26634,7 +29234,7 @@ func (p *TraceServiceUpdateManualAnnotationArgs) DeepEqual(ano *TraceServiceUpda return true } -func (p *TraceServiceUpdateManualAnnotationArgs) Field1DeepEqual(src *UpdateManualAnnotationRequest) bool { +func (p *TraceServiceListAnnotationsArgs) Field1DeepEqual(src *ListAnnotationsRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -26642,41 +29242,41 @@ func (p *TraceServiceUpdateManualAnnotationArgs) Field1DeepEqual(src *UpdateManu return true } -type TraceServiceUpdateManualAnnotationResult struct { - Success *UpdateManualAnnotationResponse `thrift:"success,0,optional" frugal:"0,optional,UpdateManualAnnotationResponse"` +type TraceServiceListAnnotationsResult struct { + Success *ListAnnotationsResponse `thrift:"success,0,optional" frugal:"0,optional,ListAnnotationsResponse"` } -func NewTraceServiceUpdateManualAnnotationResult() *TraceServiceUpdateManualAnnotationResult { - return &TraceServiceUpdateManualAnnotationResult{} +func NewTraceServiceListAnnotationsResult() *TraceServiceListAnnotationsResult { + return &TraceServiceListAnnotationsResult{} } -func (p *TraceServiceUpdateManualAnnotationResult) InitDefault() { +func (p *TraceServiceListAnnotationsResult) InitDefault() { } -var TraceServiceUpdateManualAnnotationResult_Success_DEFAULT *UpdateManualAnnotationResponse +var TraceServiceListAnnotationsResult_Success_DEFAULT *ListAnnotationsResponse -func (p *TraceServiceUpdateManualAnnotationResult) GetSuccess() (v *UpdateManualAnnotationResponse) { +func (p *TraceServiceListAnnotationsResult) GetSuccess() (v *ListAnnotationsResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceUpdateManualAnnotationResult_Success_DEFAULT + return TraceServiceListAnnotationsResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceUpdateManualAnnotationResult) SetSuccess(x interface{}) { - p.Success = x.(*UpdateManualAnnotationResponse) +func (p *TraceServiceListAnnotationsResult) SetSuccess(x interface{}) { + p.Success = x.(*ListAnnotationsResponse) } -var fieldIDToName_TraceServiceUpdateManualAnnotationResult = map[int16]string{ +var fieldIDToName_TraceServiceListAnnotationsResult = map[int16]string{ 0: "success", } -func (p *TraceServiceUpdateManualAnnotationResult) IsSetSuccess() bool { +func (p *TraceServiceListAnnotationsResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceUpdateManualAnnotationResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationsResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -26721,7 +29321,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpdateManualAnnotationResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationsResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -26731,8 +29331,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceUpdateManualAnnotationResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewUpdateManualAnnotationResponse() +func (p *TraceServiceListAnnotationsResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewListAnnotationsResponse() if err := _field.Read(iprot); err != nil { return err } @@ -26740,9 +29340,9 @@ func (p *TraceServiceUpdateManualAnnotationResult) ReadField0(iprot thrift.TProt return nil } -func (p *TraceServiceUpdateManualAnnotationResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationsResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("UpdateManualAnnotation_result"); err != nil { + if err = oprot.WriteStructBegin("ListAnnotations_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -26768,7 +29368,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceUpdateManualAnnotationResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationsResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -26787,15 +29387,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceUpdateManualAnnotationResult) String() string { +func (p *TraceServiceListAnnotationsResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceUpdateManualAnnotationResult(%+v)", *p) + return fmt.Sprintf("TraceServiceListAnnotationsResult(%+v)", *p) } -func (p *TraceServiceUpdateManualAnnotationResult) DeepEqual(ano *TraceServiceUpdateManualAnnotationResult) bool { +func (p *TraceServiceListAnnotationsResult) DeepEqual(ano *TraceServiceListAnnotationsResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -26807,7 +29407,7 @@ func (p *TraceServiceUpdateManualAnnotationResult) DeepEqual(ano *TraceServiceUp return true } -func (p *TraceServiceUpdateManualAnnotationResult) Field0DeepEqual(src *UpdateManualAnnotationResponse) bool { +func (p *TraceServiceListAnnotationsResult) Field0DeepEqual(src *ListAnnotationsResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -26815,41 +29415,41 @@ func (p *TraceServiceUpdateManualAnnotationResult) Field0DeepEqual(src *UpdateMa return true } -type TraceServiceDeleteManualAnnotationArgs struct { - Req *DeleteManualAnnotationRequest `thrift:"req,1" frugal:"1,default,DeleteManualAnnotationRequest"` +type TraceServiceListWorkspaceAnnotationsArgs struct { + Req *ListWorkspaceAnnotationsRequest `thrift:"req,1" frugal:"1,default,ListWorkspaceAnnotationsRequest"` } -func NewTraceServiceDeleteManualAnnotationArgs() *TraceServiceDeleteManualAnnotationArgs { - return &TraceServiceDeleteManualAnnotationArgs{} +func NewTraceServiceListWorkspaceAnnotationsArgs() *TraceServiceListWorkspaceAnnotationsArgs { + return &TraceServiceListWorkspaceAnnotationsArgs{} } -func (p *TraceServiceDeleteManualAnnotationArgs) InitDefault() { +func (p *TraceServiceListWorkspaceAnnotationsArgs) InitDefault() { } -var TraceServiceDeleteManualAnnotationArgs_Req_DEFAULT *DeleteManualAnnotationRequest +var TraceServiceListWorkspaceAnnotationsArgs_Req_DEFAULT *ListWorkspaceAnnotationsRequest -func (p *TraceServiceDeleteManualAnnotationArgs) GetReq() (v *DeleteManualAnnotationRequest) { +func (p *TraceServiceListWorkspaceAnnotationsArgs) GetReq() (v *ListWorkspaceAnnotationsRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceDeleteManualAnnotationArgs_Req_DEFAULT + return TraceServiceListWorkspaceAnnotationsArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceDeleteManualAnnotationArgs) SetReq(val *DeleteManualAnnotationRequest) { +func (p *TraceServiceListWorkspaceAnnotationsArgs) SetReq(val *ListWorkspaceAnnotationsRequest) { p.Req = val } -var fieldIDToName_TraceServiceDeleteManualAnnotationArgs = map[int16]string{ +var fieldIDToName_TraceServiceListWorkspaceAnnotationsArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceDeleteManualAnnotationArgs) IsSetReq() bool { +func (p *TraceServiceListWorkspaceAnnotationsArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceDeleteManualAnnotationArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListWorkspaceAnnotationsArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -26894,7 +29494,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceDeleteManualAnnotationArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListWorkspaceAnnotationsArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -26904,8 +29504,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceDeleteManualAnnotationArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewDeleteManualAnnotationRequest() +func (p *TraceServiceListWorkspaceAnnotationsArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewListWorkspaceAnnotationsRequest() if err := _field.Read(iprot); err != nil { return err } @@ -26913,9 +29513,9 @@ func (p *TraceServiceDeleteManualAnnotationArgs) ReadField1(iprot thrift.TProtoc return nil } -func (p *TraceServiceDeleteManualAnnotationArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListWorkspaceAnnotationsArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("DeleteManualAnnotation_args"); err != nil { + if err = oprot.WriteStructBegin("ListWorkspaceAnnotations_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -26941,7 +29541,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceDeleteManualAnnotationArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListWorkspaceAnnotationsArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -26958,15 +29558,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceDeleteManualAnnotationArgs) String() string { +func (p *TraceServiceListWorkspaceAnnotationsArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceDeleteManualAnnotationArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceListWorkspaceAnnotationsArgs(%+v)", *p) } -func (p *TraceServiceDeleteManualAnnotationArgs) DeepEqual(ano *TraceServiceDeleteManualAnnotationArgs) bool { +func (p *TraceServiceListWorkspaceAnnotationsArgs) DeepEqual(ano *TraceServiceListWorkspaceAnnotationsArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -26978,7 +29578,7 @@ func (p *TraceServiceDeleteManualAnnotationArgs) DeepEqual(ano *TraceServiceDele return true } -func (p *TraceServiceDeleteManualAnnotationArgs) Field1DeepEqual(src *DeleteManualAnnotationRequest) bool { +func (p *TraceServiceListWorkspaceAnnotationsArgs) Field1DeepEqual(src *ListWorkspaceAnnotationsRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -26986,41 +29586,41 @@ func (p *TraceServiceDeleteManualAnnotationArgs) Field1DeepEqual(src *DeleteManu return true } -type TraceServiceDeleteManualAnnotationResult struct { - Success *DeleteManualAnnotationResponse `thrift:"success,0,optional" frugal:"0,optional,DeleteManualAnnotationResponse"` +type TraceServiceListWorkspaceAnnotationsResult struct { + Success *ListWorkspaceAnnotationsResponse `thrift:"success,0,optional" frugal:"0,optional,ListWorkspaceAnnotationsResponse"` } -func NewTraceServiceDeleteManualAnnotationResult() *TraceServiceDeleteManualAnnotationResult { - return &TraceServiceDeleteManualAnnotationResult{} +func NewTraceServiceListWorkspaceAnnotationsResult() *TraceServiceListWorkspaceAnnotationsResult { + return &TraceServiceListWorkspaceAnnotationsResult{} } -func (p *TraceServiceDeleteManualAnnotationResult) InitDefault() { +func (p *TraceServiceListWorkspaceAnnotationsResult) InitDefault() { } -var TraceServiceDeleteManualAnnotationResult_Success_DEFAULT *DeleteManualAnnotationResponse +var TraceServiceListWorkspaceAnnotationsResult_Success_DEFAULT *ListWorkspaceAnnotationsResponse -func (p *TraceServiceDeleteManualAnnotationResult) GetSuccess() (v *DeleteManualAnnotationResponse) { +func (p *TraceServiceListWorkspaceAnnotationsResult) GetSuccess() (v *ListWorkspaceAnnotationsResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceDeleteManualAnnotationResult_Success_DEFAULT + return TraceServiceListWorkspaceAnnotationsResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceDeleteManualAnnotationResult) SetSuccess(x interface{}) { - p.Success = x.(*DeleteManualAnnotationResponse) +func (p *TraceServiceListWorkspaceAnnotationsResult) SetSuccess(x interface{}) { + p.Success = x.(*ListWorkspaceAnnotationsResponse) } -var fieldIDToName_TraceServiceDeleteManualAnnotationResult = map[int16]string{ +var fieldIDToName_TraceServiceListWorkspaceAnnotationsResult = map[int16]string{ 0: "success", } -func (p *TraceServiceDeleteManualAnnotationResult) IsSetSuccess() bool { +func (p *TraceServiceListWorkspaceAnnotationsResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceDeleteManualAnnotationResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListWorkspaceAnnotationsResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -27065,7 +29665,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceDeleteManualAnnotationResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListWorkspaceAnnotationsResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -27075,8 +29675,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceDeleteManualAnnotationResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewDeleteManualAnnotationResponse() +func (p *TraceServiceListWorkspaceAnnotationsResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewListWorkspaceAnnotationsResponse() if err := _field.Read(iprot); err != nil { return err } @@ -27084,9 +29684,9 @@ func (p *TraceServiceDeleteManualAnnotationResult) ReadField0(iprot thrift.TProt return nil } -func (p *TraceServiceDeleteManualAnnotationResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListWorkspaceAnnotationsResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("DeleteManualAnnotation_result"); err != nil { + if err = oprot.WriteStructBegin("ListWorkspaceAnnotations_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -27112,7 +29712,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceDeleteManualAnnotationResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListWorkspaceAnnotationsResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -27131,15 +29731,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceDeleteManualAnnotationResult) String() string { +func (p *TraceServiceListWorkspaceAnnotationsResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceDeleteManualAnnotationResult(%+v)", *p) + return fmt.Sprintf("TraceServiceListWorkspaceAnnotationsResult(%+v)", *p) } -func (p *TraceServiceDeleteManualAnnotationResult) DeepEqual(ano *TraceServiceDeleteManualAnnotationResult) bool { +func (p *TraceServiceListWorkspaceAnnotationsResult) DeepEqual(ano *TraceServiceListWorkspaceAnnotationsResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -27151,7 +29751,7 @@ func (p *TraceServiceDeleteManualAnnotationResult) DeepEqual(ano *TraceServiceDe return true } -func (p *TraceServiceDeleteManualAnnotationResult) Field0DeepEqual(src *DeleteManualAnnotationResponse) bool { +func (p *TraceServiceListWorkspaceAnnotationsResult) Field0DeepEqual(src *ListWorkspaceAnnotationsResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -27159,41 +29759,41 @@ func (p *TraceServiceDeleteManualAnnotationResult) Field0DeepEqual(src *DeleteMa return true } -type TraceServiceListAnnotationsArgs struct { - Req *ListAnnotationsRequest `thrift:"req,1" frugal:"1,default,ListAnnotationsRequest"` +type TraceServiceExportTracesToDatasetArgs struct { + Req *ExportTracesToDatasetRequest `thrift:"Req,1" frugal:"1,default,ExportTracesToDatasetRequest"` } -func NewTraceServiceListAnnotationsArgs() *TraceServiceListAnnotationsArgs { - return &TraceServiceListAnnotationsArgs{} +func NewTraceServiceExportTracesToDatasetArgs() *TraceServiceExportTracesToDatasetArgs { + return &TraceServiceExportTracesToDatasetArgs{} } -func (p *TraceServiceListAnnotationsArgs) InitDefault() { +func (p *TraceServiceExportTracesToDatasetArgs) InitDefault() { } -var TraceServiceListAnnotationsArgs_Req_DEFAULT *ListAnnotationsRequest +var TraceServiceExportTracesToDatasetArgs_Req_DEFAULT *ExportTracesToDatasetRequest -func (p *TraceServiceListAnnotationsArgs) GetReq() (v *ListAnnotationsRequest) { +func (p *TraceServiceExportTracesToDatasetArgs) GetReq() (v *ExportTracesToDatasetRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceListAnnotationsArgs_Req_DEFAULT + return TraceServiceExportTracesToDatasetArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceListAnnotationsArgs) SetReq(val *ListAnnotationsRequest) { +func (p *TraceServiceExportTracesToDatasetArgs) SetReq(val *ExportTracesToDatasetRequest) { p.Req = val } -var fieldIDToName_TraceServiceListAnnotationsArgs = map[int16]string{ - 1: "req", +var fieldIDToName_TraceServiceExportTracesToDatasetArgs = map[int16]string{ + 1: "Req", } -func (p *TraceServiceListAnnotationsArgs) IsSetReq() bool { +func (p *TraceServiceExportTracesToDatasetArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceListAnnotationsArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceExportTracesToDatasetArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -27238,7 +29838,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationsArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceExportTracesToDatasetArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -27248,8 +29848,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListAnnotationsArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewListAnnotationsRequest() +func (p *TraceServiceExportTracesToDatasetArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewExportTracesToDatasetRequest() if err := _field.Read(iprot); err != nil { return err } @@ -27257,9 +29857,9 @@ func (p *TraceServiceListAnnotationsArgs) ReadField1(iprot thrift.TProtocol) err return nil } -func (p *TraceServiceListAnnotationsArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceExportTracesToDatasetArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListAnnotations_args"); err != nil { + if err = oprot.WriteStructBegin("ExportTracesToDataset_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -27285,8 +29885,8 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListAnnotationsArgs) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { +func (p *TraceServiceExportTracesToDatasetArgs) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("Req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } if err := p.Req.Write(oprot); err != nil { @@ -27302,15 +29902,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceListAnnotationsArgs) String() string { +func (p *TraceServiceExportTracesToDatasetArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListAnnotationsArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceExportTracesToDatasetArgs(%+v)", *p) } -func (p *TraceServiceListAnnotationsArgs) DeepEqual(ano *TraceServiceListAnnotationsArgs) bool { +func (p *TraceServiceExportTracesToDatasetArgs) DeepEqual(ano *TraceServiceExportTracesToDatasetArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -27322,7 +29922,7 @@ func (p *TraceServiceListAnnotationsArgs) DeepEqual(ano *TraceServiceListAnnotat return true } -func (p *TraceServiceListAnnotationsArgs) Field1DeepEqual(src *ListAnnotationsRequest) bool { +func (p *TraceServiceExportTracesToDatasetArgs) Field1DeepEqual(src *ExportTracesToDatasetRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -27330,41 +29930,41 @@ func (p *TraceServiceListAnnotationsArgs) Field1DeepEqual(src *ListAnnotationsRe return true } -type TraceServiceListAnnotationsResult struct { - Success *ListAnnotationsResponse `thrift:"success,0,optional" frugal:"0,optional,ListAnnotationsResponse"` +type TraceServiceExportTracesToDatasetResult struct { + Success *ExportTracesToDatasetResponse `thrift:"success,0,optional" frugal:"0,optional,ExportTracesToDatasetResponse"` } -func NewTraceServiceListAnnotationsResult() *TraceServiceListAnnotationsResult { - return &TraceServiceListAnnotationsResult{} +func NewTraceServiceExportTracesToDatasetResult() *TraceServiceExportTracesToDatasetResult { + return &TraceServiceExportTracesToDatasetResult{} } -func (p *TraceServiceListAnnotationsResult) InitDefault() { +func (p *TraceServiceExportTracesToDatasetResult) InitDefault() { } -var TraceServiceListAnnotationsResult_Success_DEFAULT *ListAnnotationsResponse +var TraceServiceExportTracesToDatasetResult_Success_DEFAULT *ExportTracesToDatasetResponse -func (p *TraceServiceListAnnotationsResult) GetSuccess() (v *ListAnnotationsResponse) { +func (p *TraceServiceExportTracesToDatasetResult) GetSuccess() (v *ExportTracesToDatasetResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceListAnnotationsResult_Success_DEFAULT + return TraceServiceExportTracesToDatasetResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceListAnnotationsResult) SetSuccess(x interface{}) { - p.Success = x.(*ListAnnotationsResponse) +func (p *TraceServiceExportTracesToDatasetResult) SetSuccess(x interface{}) { + p.Success = x.(*ExportTracesToDatasetResponse) } -var fieldIDToName_TraceServiceListAnnotationsResult = map[int16]string{ +var fieldIDToName_TraceServiceExportTracesToDatasetResult = map[int16]string{ 0: "success", } -func (p *TraceServiceListAnnotationsResult) IsSetSuccess() bool { +func (p *TraceServiceExportTracesToDatasetResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceListAnnotationsResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceExportTracesToDatasetResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -27409,7 +30009,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationsResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceExportTracesToDatasetResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -27419,8 +30019,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListAnnotationsResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewListAnnotationsResponse() +func (p *TraceServiceExportTracesToDatasetResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewExportTracesToDatasetResponse() if err := _field.Read(iprot); err != nil { return err } @@ -27428,9 +30028,9 @@ func (p *TraceServiceListAnnotationsResult) ReadField0(iprot thrift.TProtocol) e return nil } -func (p *TraceServiceListAnnotationsResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceExportTracesToDatasetResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListAnnotations_result"); err != nil { + if err = oprot.WriteStructBegin("ExportTracesToDataset_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -27456,7 +30056,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListAnnotationsResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceExportTracesToDatasetResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -27475,15 +30075,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceListAnnotationsResult) String() string { +func (p *TraceServiceExportTracesToDatasetResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListAnnotationsResult(%+v)", *p) + return fmt.Sprintf("TraceServiceExportTracesToDatasetResult(%+v)", *p) } -func (p *TraceServiceListAnnotationsResult) DeepEqual(ano *TraceServiceListAnnotationsResult) bool { +func (p *TraceServiceExportTracesToDatasetResult) DeepEqual(ano *TraceServiceExportTracesToDatasetResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -27495,7 +30095,7 @@ func (p *TraceServiceListAnnotationsResult) DeepEqual(ano *TraceServiceListAnnot return true } -func (p *TraceServiceListAnnotationsResult) Field0DeepEqual(src *ListAnnotationsResponse) bool { +func (p *TraceServiceExportTracesToDatasetResult) Field0DeepEqual(src *ExportTracesToDatasetResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -27503,41 +30103,41 @@ func (p *TraceServiceListAnnotationsResult) Field0DeepEqual(src *ListAnnotations return true } -type TraceServiceExportTracesToDatasetArgs struct { - Req *ExportTracesToDatasetRequest `thrift:"Req,1" frugal:"1,default,ExportTracesToDatasetRequest"` +type TraceServicePreviewExportTracesToDatasetArgs struct { + Req *PreviewExportTracesToDatasetRequest `thrift:"Req,1" frugal:"1,default,PreviewExportTracesToDatasetRequest"` } -func NewTraceServiceExportTracesToDatasetArgs() *TraceServiceExportTracesToDatasetArgs { - return &TraceServiceExportTracesToDatasetArgs{} +func NewTraceServicePreviewExportTracesToDatasetArgs() *TraceServicePreviewExportTracesToDatasetArgs { + return &TraceServicePreviewExportTracesToDatasetArgs{} } -func (p *TraceServiceExportTracesToDatasetArgs) InitDefault() { +func (p *TraceServicePreviewExportTracesToDatasetArgs) InitDefault() { } -var TraceServiceExportTracesToDatasetArgs_Req_DEFAULT *ExportTracesToDatasetRequest +var TraceServicePreviewExportTracesToDatasetArgs_Req_DEFAULT *PreviewExportTracesToDatasetRequest -func (p *TraceServiceExportTracesToDatasetArgs) GetReq() (v *ExportTracesToDatasetRequest) { +func (p *TraceServicePreviewExportTracesToDatasetArgs) GetReq() (v *PreviewExportTracesToDatasetRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceExportTracesToDatasetArgs_Req_DEFAULT + return TraceServicePreviewExportTracesToDatasetArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceExportTracesToDatasetArgs) SetReq(val *ExportTracesToDatasetRequest) { +func (p *TraceServicePreviewExportTracesToDatasetArgs) SetReq(val *PreviewExportTracesToDatasetRequest) { p.Req = val } -var fieldIDToName_TraceServiceExportTracesToDatasetArgs = map[int16]string{ +var fieldIDToName_TraceServicePreviewExportTracesToDatasetArgs = map[int16]string{ 1: "Req", } -func (p *TraceServiceExportTracesToDatasetArgs) IsSetReq() bool { +func (p *TraceServicePreviewExportTracesToDatasetArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceExportTracesToDatasetArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServicePreviewExportTracesToDatasetArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -27582,7 +30182,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceExportTracesToDatasetArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServicePreviewExportTracesToDatasetArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -27592,8 +30192,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceExportTracesToDatasetArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewExportTracesToDatasetRequest() +func (p *TraceServicePreviewExportTracesToDatasetArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewPreviewExportTracesToDatasetRequest() if err := _field.Read(iprot); err != nil { return err } @@ -27601,9 +30201,9 @@ func (p *TraceServiceExportTracesToDatasetArgs) ReadField1(iprot thrift.TProtoco return nil } -func (p *TraceServiceExportTracesToDatasetArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServicePreviewExportTracesToDatasetArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ExportTracesToDataset_args"); err != nil { + if err = oprot.WriteStructBegin("PreviewExportTracesToDataset_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -27629,7 +30229,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceExportTracesToDatasetArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServicePreviewExportTracesToDatasetArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("Req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -27646,15 +30246,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceExportTracesToDatasetArgs) String() string { +func (p *TraceServicePreviewExportTracesToDatasetArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceExportTracesToDatasetArgs(%+v)", *p) + return fmt.Sprintf("TraceServicePreviewExportTracesToDatasetArgs(%+v)", *p) } -func (p *TraceServiceExportTracesToDatasetArgs) DeepEqual(ano *TraceServiceExportTracesToDatasetArgs) bool { +func (p *TraceServicePreviewExportTracesToDatasetArgs) DeepEqual(ano *TraceServicePreviewExportTracesToDatasetArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -27666,7 +30266,7 @@ func (p *TraceServiceExportTracesToDatasetArgs) DeepEqual(ano *TraceServiceExpor return true } -func (p *TraceServiceExportTracesToDatasetArgs) Field1DeepEqual(src *ExportTracesToDatasetRequest) bool { +func (p *TraceServicePreviewExportTracesToDatasetArgs) Field1DeepEqual(src *PreviewExportTracesToDatasetRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -27674,41 +30274,41 @@ func (p *TraceServiceExportTracesToDatasetArgs) Field1DeepEqual(src *ExportTrace return true } -type TraceServiceExportTracesToDatasetResult struct { - Success *ExportTracesToDatasetResponse `thrift:"success,0,optional" frugal:"0,optional,ExportTracesToDatasetResponse"` +type TraceServicePreviewExportTracesToDatasetResult struct { + Success *PreviewExportTracesToDatasetResponse `thrift:"success,0,optional" frugal:"0,optional,PreviewExportTracesToDatasetResponse"` } -func NewTraceServiceExportTracesToDatasetResult() *TraceServiceExportTracesToDatasetResult { - return &TraceServiceExportTracesToDatasetResult{} +func NewTraceServicePreviewExportTracesToDatasetResult() *TraceServicePreviewExportTracesToDatasetResult { + return &TraceServicePreviewExportTracesToDatasetResult{} } -func (p *TraceServiceExportTracesToDatasetResult) InitDefault() { +func (p *TraceServicePreviewExportTracesToDatasetResult) InitDefault() { } -var TraceServiceExportTracesToDatasetResult_Success_DEFAULT *ExportTracesToDatasetResponse +var TraceServicePreviewExportTracesToDatasetResult_Success_DEFAULT *PreviewExportTracesToDatasetResponse -func (p *TraceServiceExportTracesToDatasetResult) GetSuccess() (v *ExportTracesToDatasetResponse) { +func (p *TraceServicePreviewExportTracesToDatasetResult) GetSuccess() (v *PreviewExportTracesToDatasetResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceExportTracesToDatasetResult_Success_DEFAULT + return TraceServicePreviewExportTracesToDatasetResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceExportTracesToDatasetResult) SetSuccess(x interface{}) { - p.Success = x.(*ExportTracesToDatasetResponse) +func (p *TraceServicePreviewExportTracesToDatasetResult) SetSuccess(x interface{}) { + p.Success = x.(*PreviewExportTracesToDatasetResponse) } -var fieldIDToName_TraceServiceExportTracesToDatasetResult = map[int16]string{ +var fieldIDToName_TraceServicePreviewExportTracesToDatasetResult = map[int16]string{ 0: "success", } -func (p *TraceServiceExportTracesToDatasetResult) IsSetSuccess() bool { +func (p *TraceServicePreviewExportTracesToDatasetResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceExportTracesToDatasetResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServicePreviewExportTracesToDatasetResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -27753,7 +30353,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceExportTracesToDatasetResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServicePreviewExportTracesToDatasetResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -27763,8 +30363,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceExportTracesToDatasetResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewExportTracesToDatasetResponse() +func (p *TraceServicePreviewExportTracesToDatasetResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewPreviewExportTracesToDatasetResponse() if err := _field.Read(iprot); err != nil { return err } @@ -27772,9 +30372,9 @@ func (p *TraceServiceExportTracesToDatasetResult) ReadField0(iprot thrift.TProto return nil } -func (p *TraceServiceExportTracesToDatasetResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServicePreviewExportTracesToDatasetResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ExportTracesToDataset_result"); err != nil { + if err = oprot.WriteStructBegin("PreviewExportTracesToDataset_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -27800,7 +30400,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceExportTracesToDatasetResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServicePreviewExportTracesToDatasetResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -27819,15 +30419,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceExportTracesToDatasetResult) String() string { +func (p *TraceServicePreviewExportTracesToDatasetResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceExportTracesToDatasetResult(%+v)", *p) + return fmt.Sprintf("TraceServicePreviewExportTracesToDatasetResult(%+v)", *p) } -func (p *TraceServiceExportTracesToDatasetResult) DeepEqual(ano *TraceServiceExportTracesToDatasetResult) bool { +func (p *TraceServicePreviewExportTracesToDatasetResult) DeepEqual(ano *TraceServicePreviewExportTracesToDatasetResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -27839,7 +30439,7 @@ func (p *TraceServiceExportTracesToDatasetResult) DeepEqual(ano *TraceServiceExp return true } -func (p *TraceServiceExportTracesToDatasetResult) Field0DeepEqual(src *ExportTracesToDatasetResponse) bool { +func (p *TraceServicePreviewExportTracesToDatasetResult) Field0DeepEqual(src *PreviewExportTracesToDatasetResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -27847,41 +30447,41 @@ func (p *TraceServiceExportTracesToDatasetResult) Field0DeepEqual(src *ExportTra return true } -type TraceServicePreviewExportTracesToDatasetArgs struct { - Req *PreviewExportTracesToDatasetRequest `thrift:"Req,1" frugal:"1,default,PreviewExportTracesToDatasetRequest"` +type TraceServiceChangeEvaluatorScoreArgs struct { + Req *ChangeEvaluatorScoreRequest `thrift:"req,1" frugal:"1,default,ChangeEvaluatorScoreRequest"` } -func NewTraceServicePreviewExportTracesToDatasetArgs() *TraceServicePreviewExportTracesToDatasetArgs { - return &TraceServicePreviewExportTracesToDatasetArgs{} +func NewTraceServiceChangeEvaluatorScoreArgs() *TraceServiceChangeEvaluatorScoreArgs { + return &TraceServiceChangeEvaluatorScoreArgs{} } -func (p *TraceServicePreviewExportTracesToDatasetArgs) InitDefault() { +func (p *TraceServiceChangeEvaluatorScoreArgs) InitDefault() { } -var TraceServicePreviewExportTracesToDatasetArgs_Req_DEFAULT *PreviewExportTracesToDatasetRequest +var TraceServiceChangeEvaluatorScoreArgs_Req_DEFAULT *ChangeEvaluatorScoreRequest -func (p *TraceServicePreviewExportTracesToDatasetArgs) GetReq() (v *PreviewExportTracesToDatasetRequest) { +func (p *TraceServiceChangeEvaluatorScoreArgs) GetReq() (v *ChangeEvaluatorScoreRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServicePreviewExportTracesToDatasetArgs_Req_DEFAULT + return TraceServiceChangeEvaluatorScoreArgs_Req_DEFAULT } return p.Req } -func (p *TraceServicePreviewExportTracesToDatasetArgs) SetReq(val *PreviewExportTracesToDatasetRequest) { +func (p *TraceServiceChangeEvaluatorScoreArgs) SetReq(val *ChangeEvaluatorScoreRequest) { p.Req = val } -var fieldIDToName_TraceServicePreviewExportTracesToDatasetArgs = map[int16]string{ - 1: "Req", +var fieldIDToName_TraceServiceChangeEvaluatorScoreArgs = map[int16]string{ + 1: "req", } -func (p *TraceServicePreviewExportTracesToDatasetArgs) IsSetReq() bool { +func (p *TraceServiceChangeEvaluatorScoreArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServicePreviewExportTracesToDatasetArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceChangeEvaluatorScoreArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -27926,7 +30526,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServicePreviewExportTracesToDatasetArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceChangeEvaluatorScoreArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -27936,8 +30536,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServicePreviewExportTracesToDatasetArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewPreviewExportTracesToDatasetRequest() +func (p *TraceServiceChangeEvaluatorScoreArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewChangeEvaluatorScoreRequest() if err := _field.Read(iprot); err != nil { return err } @@ -27945,9 +30545,9 @@ func (p *TraceServicePreviewExportTracesToDatasetArgs) ReadField1(iprot thrift.T return nil } -func (p *TraceServicePreviewExportTracesToDatasetArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceChangeEvaluatorScoreArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("PreviewExportTracesToDataset_args"); err != nil { + if err = oprot.WriteStructBegin("ChangeEvaluatorScore_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -27973,8 +30573,8 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServicePreviewExportTracesToDatasetArgs) writeField1(oprot thrift.TProtocol) (err error) { - if err = oprot.WriteFieldBegin("Req", thrift.STRUCT, 1); err != nil { +func (p *TraceServiceChangeEvaluatorScoreArgs) writeField1(oprot thrift.TProtocol) (err error) { + if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } if err := p.Req.Write(oprot); err != nil { @@ -27990,15 +30590,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServicePreviewExportTracesToDatasetArgs) String() string { +func (p *TraceServiceChangeEvaluatorScoreArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServicePreviewExportTracesToDatasetArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceChangeEvaluatorScoreArgs(%+v)", *p) } -func (p *TraceServicePreviewExportTracesToDatasetArgs) DeepEqual(ano *TraceServicePreviewExportTracesToDatasetArgs) bool { +func (p *TraceServiceChangeEvaluatorScoreArgs) DeepEqual(ano *TraceServiceChangeEvaluatorScoreArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -28010,7 +30610,7 @@ func (p *TraceServicePreviewExportTracesToDatasetArgs) DeepEqual(ano *TraceServi return true } -func (p *TraceServicePreviewExportTracesToDatasetArgs) Field1DeepEqual(src *PreviewExportTracesToDatasetRequest) bool { +func (p *TraceServiceChangeEvaluatorScoreArgs) Field1DeepEqual(src *ChangeEvaluatorScoreRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -28018,41 +30618,41 @@ func (p *TraceServicePreviewExportTracesToDatasetArgs) Field1DeepEqual(src *Prev return true } -type TraceServicePreviewExportTracesToDatasetResult struct { - Success *PreviewExportTracesToDatasetResponse `thrift:"success,0,optional" frugal:"0,optional,PreviewExportTracesToDatasetResponse"` +type TraceServiceChangeEvaluatorScoreResult struct { + Success *ChangeEvaluatorScoreResponse `thrift:"success,0,optional" frugal:"0,optional,ChangeEvaluatorScoreResponse"` } -func NewTraceServicePreviewExportTracesToDatasetResult() *TraceServicePreviewExportTracesToDatasetResult { - return &TraceServicePreviewExportTracesToDatasetResult{} +func NewTraceServiceChangeEvaluatorScoreResult() *TraceServiceChangeEvaluatorScoreResult { + return &TraceServiceChangeEvaluatorScoreResult{} } -func (p *TraceServicePreviewExportTracesToDatasetResult) InitDefault() { +func (p *TraceServiceChangeEvaluatorScoreResult) InitDefault() { } -var TraceServicePreviewExportTracesToDatasetResult_Success_DEFAULT *PreviewExportTracesToDatasetResponse +var TraceServiceChangeEvaluatorScoreResult_Success_DEFAULT *ChangeEvaluatorScoreResponse -func (p *TraceServicePreviewExportTracesToDatasetResult) GetSuccess() (v *PreviewExportTracesToDatasetResponse) { +func (p *TraceServiceChangeEvaluatorScoreResult) GetSuccess() (v *ChangeEvaluatorScoreResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServicePreviewExportTracesToDatasetResult_Success_DEFAULT + return TraceServiceChangeEvaluatorScoreResult_Success_DEFAULT } return p.Success } -func (p *TraceServicePreviewExportTracesToDatasetResult) SetSuccess(x interface{}) { - p.Success = x.(*PreviewExportTracesToDatasetResponse) +func (p *TraceServiceChangeEvaluatorScoreResult) SetSuccess(x interface{}) { + p.Success = x.(*ChangeEvaluatorScoreResponse) } -var fieldIDToName_TraceServicePreviewExportTracesToDatasetResult = map[int16]string{ +var fieldIDToName_TraceServiceChangeEvaluatorScoreResult = map[int16]string{ 0: "success", } -func (p *TraceServicePreviewExportTracesToDatasetResult) IsSetSuccess() bool { +func (p *TraceServiceChangeEvaluatorScoreResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServicePreviewExportTracesToDatasetResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceChangeEvaluatorScoreResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -28097,7 +30697,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServicePreviewExportTracesToDatasetResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceChangeEvaluatorScoreResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -28106,9 +30706,9 @@ ReadFieldEndError: ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } - -func (p *TraceServicePreviewExportTracesToDatasetResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewPreviewExportTracesToDatasetResponse() + +func (p *TraceServiceChangeEvaluatorScoreResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewChangeEvaluatorScoreResponse() if err := _field.Read(iprot); err != nil { return err } @@ -28116,9 +30716,9 @@ func (p *TraceServicePreviewExportTracesToDatasetResult) ReadField0(iprot thrift return nil } -func (p *TraceServicePreviewExportTracesToDatasetResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceChangeEvaluatorScoreResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("PreviewExportTracesToDataset_result"); err != nil { + if err = oprot.WriteStructBegin("ChangeEvaluatorScore_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -28144,7 +30744,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServicePreviewExportTracesToDatasetResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceChangeEvaluatorScoreResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -28163,15 +30763,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServicePreviewExportTracesToDatasetResult) String() string { +func (p *TraceServiceChangeEvaluatorScoreResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServicePreviewExportTracesToDatasetResult(%+v)", *p) + return fmt.Sprintf("TraceServiceChangeEvaluatorScoreResult(%+v)", *p) } -func (p *TraceServicePreviewExportTracesToDatasetResult) DeepEqual(ano *TraceServicePreviewExportTracesToDatasetResult) bool { +func (p *TraceServiceChangeEvaluatorScoreResult) DeepEqual(ano *TraceServiceChangeEvaluatorScoreResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -28183,7 +30783,7 @@ func (p *TraceServicePreviewExportTracesToDatasetResult) DeepEqual(ano *TraceSer return true } -func (p *TraceServicePreviewExportTracesToDatasetResult) Field0DeepEqual(src *PreviewExportTracesToDatasetResponse) bool { +func (p *TraceServiceChangeEvaluatorScoreResult) Field0DeepEqual(src *ChangeEvaluatorScoreResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -28191,41 +30791,41 @@ func (p *TraceServicePreviewExportTracesToDatasetResult) Field0DeepEqual(src *Pr return true } -type TraceServiceChangeEvaluatorScoreArgs struct { - Req *ChangeEvaluatorScoreRequest `thrift:"req,1" frugal:"1,default,ChangeEvaluatorScoreRequest"` +type TraceServiceListAnnotationEvaluatorsArgs struct { + Req *ListAnnotationEvaluatorsRequest `thrift:"req,1" frugal:"1,default,ListAnnotationEvaluatorsRequest"` } -func NewTraceServiceChangeEvaluatorScoreArgs() *TraceServiceChangeEvaluatorScoreArgs { - return &TraceServiceChangeEvaluatorScoreArgs{} +func NewTraceServiceListAnnotationEvaluatorsArgs() *TraceServiceListAnnotationEvaluatorsArgs { + return &TraceServiceListAnnotationEvaluatorsArgs{} } -func (p *TraceServiceChangeEvaluatorScoreArgs) InitDefault() { +func (p *TraceServiceListAnnotationEvaluatorsArgs) InitDefault() { } -var TraceServiceChangeEvaluatorScoreArgs_Req_DEFAULT *ChangeEvaluatorScoreRequest +var TraceServiceListAnnotationEvaluatorsArgs_Req_DEFAULT *ListAnnotationEvaluatorsRequest -func (p *TraceServiceChangeEvaluatorScoreArgs) GetReq() (v *ChangeEvaluatorScoreRequest) { +func (p *TraceServiceListAnnotationEvaluatorsArgs) GetReq() (v *ListAnnotationEvaluatorsRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceChangeEvaluatorScoreArgs_Req_DEFAULT + return TraceServiceListAnnotationEvaluatorsArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceChangeEvaluatorScoreArgs) SetReq(val *ChangeEvaluatorScoreRequest) { +func (p *TraceServiceListAnnotationEvaluatorsArgs) SetReq(val *ListAnnotationEvaluatorsRequest) { p.Req = val } -var fieldIDToName_TraceServiceChangeEvaluatorScoreArgs = map[int16]string{ +var fieldIDToName_TraceServiceListAnnotationEvaluatorsArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceChangeEvaluatorScoreArgs) IsSetReq() bool { +func (p *TraceServiceListAnnotationEvaluatorsArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceChangeEvaluatorScoreArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationEvaluatorsArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -28270,7 +30870,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceChangeEvaluatorScoreArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationEvaluatorsArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -28280,8 +30880,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceChangeEvaluatorScoreArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewChangeEvaluatorScoreRequest() +func (p *TraceServiceListAnnotationEvaluatorsArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewListAnnotationEvaluatorsRequest() if err := _field.Read(iprot); err != nil { return err } @@ -28289,9 +30889,9 @@ func (p *TraceServiceChangeEvaluatorScoreArgs) ReadField1(iprot thrift.TProtocol return nil } -func (p *TraceServiceChangeEvaluatorScoreArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationEvaluatorsArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ChangeEvaluatorScore_args"); err != nil { + if err = oprot.WriteStructBegin("ListAnnotationEvaluators_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -28317,7 +30917,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceChangeEvaluatorScoreArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationEvaluatorsArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -28334,15 +30934,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceChangeEvaluatorScoreArgs) String() string { +func (p *TraceServiceListAnnotationEvaluatorsArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceChangeEvaluatorScoreArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceListAnnotationEvaluatorsArgs(%+v)", *p) } -func (p *TraceServiceChangeEvaluatorScoreArgs) DeepEqual(ano *TraceServiceChangeEvaluatorScoreArgs) bool { +func (p *TraceServiceListAnnotationEvaluatorsArgs) DeepEqual(ano *TraceServiceListAnnotationEvaluatorsArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -28354,7 +30954,7 @@ func (p *TraceServiceChangeEvaluatorScoreArgs) DeepEqual(ano *TraceServiceChange return true } -func (p *TraceServiceChangeEvaluatorScoreArgs) Field1DeepEqual(src *ChangeEvaluatorScoreRequest) bool { +func (p *TraceServiceListAnnotationEvaluatorsArgs) Field1DeepEqual(src *ListAnnotationEvaluatorsRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -28362,41 +30962,41 @@ func (p *TraceServiceChangeEvaluatorScoreArgs) Field1DeepEqual(src *ChangeEvalua return true } -type TraceServiceChangeEvaluatorScoreResult struct { - Success *ChangeEvaluatorScoreResponse `thrift:"success,0,optional" frugal:"0,optional,ChangeEvaluatorScoreResponse"` +type TraceServiceListAnnotationEvaluatorsResult struct { + Success *ListAnnotationEvaluatorsResponse `thrift:"success,0,optional" frugal:"0,optional,ListAnnotationEvaluatorsResponse"` } -func NewTraceServiceChangeEvaluatorScoreResult() *TraceServiceChangeEvaluatorScoreResult { - return &TraceServiceChangeEvaluatorScoreResult{} +func NewTraceServiceListAnnotationEvaluatorsResult() *TraceServiceListAnnotationEvaluatorsResult { + return &TraceServiceListAnnotationEvaluatorsResult{} } -func (p *TraceServiceChangeEvaluatorScoreResult) InitDefault() { +func (p *TraceServiceListAnnotationEvaluatorsResult) InitDefault() { } -var TraceServiceChangeEvaluatorScoreResult_Success_DEFAULT *ChangeEvaluatorScoreResponse +var TraceServiceListAnnotationEvaluatorsResult_Success_DEFAULT *ListAnnotationEvaluatorsResponse -func (p *TraceServiceChangeEvaluatorScoreResult) GetSuccess() (v *ChangeEvaluatorScoreResponse) { +func (p *TraceServiceListAnnotationEvaluatorsResult) GetSuccess() (v *ListAnnotationEvaluatorsResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceChangeEvaluatorScoreResult_Success_DEFAULT + return TraceServiceListAnnotationEvaluatorsResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceChangeEvaluatorScoreResult) SetSuccess(x interface{}) { - p.Success = x.(*ChangeEvaluatorScoreResponse) +func (p *TraceServiceListAnnotationEvaluatorsResult) SetSuccess(x interface{}) { + p.Success = x.(*ListAnnotationEvaluatorsResponse) } -var fieldIDToName_TraceServiceChangeEvaluatorScoreResult = map[int16]string{ +var fieldIDToName_TraceServiceListAnnotationEvaluatorsResult = map[int16]string{ 0: "success", } -func (p *TraceServiceChangeEvaluatorScoreResult) IsSetSuccess() bool { +func (p *TraceServiceListAnnotationEvaluatorsResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceChangeEvaluatorScoreResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationEvaluatorsResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -28441,7 +31041,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceChangeEvaluatorScoreResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationEvaluatorsResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -28451,8 +31051,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceChangeEvaluatorScoreResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewChangeEvaluatorScoreResponse() +func (p *TraceServiceListAnnotationEvaluatorsResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewListAnnotationEvaluatorsResponse() if err := _field.Read(iprot); err != nil { return err } @@ -28460,9 +31060,9 @@ func (p *TraceServiceChangeEvaluatorScoreResult) ReadField0(iprot thrift.TProtoc return nil } -func (p *TraceServiceChangeEvaluatorScoreResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationEvaluatorsResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ChangeEvaluatorScore_result"); err != nil { + if err = oprot.WriteStructBegin("ListAnnotationEvaluators_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -28488,7 +31088,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceChangeEvaluatorScoreResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListAnnotationEvaluatorsResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -28507,15 +31107,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceChangeEvaluatorScoreResult) String() string { +func (p *TraceServiceListAnnotationEvaluatorsResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceChangeEvaluatorScoreResult(%+v)", *p) + return fmt.Sprintf("TraceServiceListAnnotationEvaluatorsResult(%+v)", *p) } -func (p *TraceServiceChangeEvaluatorScoreResult) DeepEqual(ano *TraceServiceChangeEvaluatorScoreResult) bool { +func (p *TraceServiceListAnnotationEvaluatorsResult) DeepEqual(ano *TraceServiceListAnnotationEvaluatorsResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -28527,7 +31127,7 @@ func (p *TraceServiceChangeEvaluatorScoreResult) DeepEqual(ano *TraceServiceChan return true } -func (p *TraceServiceChangeEvaluatorScoreResult) Field0DeepEqual(src *ChangeEvaluatorScoreResponse) bool { +func (p *TraceServiceListAnnotationEvaluatorsResult) Field0DeepEqual(src *ListAnnotationEvaluatorsResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -28535,41 +31135,41 @@ func (p *TraceServiceChangeEvaluatorScoreResult) Field0DeepEqual(src *ChangeEval return true } -type TraceServiceListAnnotationEvaluatorsArgs struct { - Req *ListAnnotationEvaluatorsRequest `thrift:"req,1" frugal:"1,default,ListAnnotationEvaluatorsRequest"` +type TraceServiceExtractSpanInfoArgs struct { + Req *ExtractSpanInfoRequest `thrift:"req,1" frugal:"1,default,ExtractSpanInfoRequest"` } -func NewTraceServiceListAnnotationEvaluatorsArgs() *TraceServiceListAnnotationEvaluatorsArgs { - return &TraceServiceListAnnotationEvaluatorsArgs{} +func NewTraceServiceExtractSpanInfoArgs() *TraceServiceExtractSpanInfoArgs { + return &TraceServiceExtractSpanInfoArgs{} } -func (p *TraceServiceListAnnotationEvaluatorsArgs) InitDefault() { +func (p *TraceServiceExtractSpanInfoArgs) InitDefault() { } -var TraceServiceListAnnotationEvaluatorsArgs_Req_DEFAULT *ListAnnotationEvaluatorsRequest +var TraceServiceExtractSpanInfoArgs_Req_DEFAULT *ExtractSpanInfoRequest -func (p *TraceServiceListAnnotationEvaluatorsArgs) GetReq() (v *ListAnnotationEvaluatorsRequest) { +func (p *TraceServiceExtractSpanInfoArgs) GetReq() (v *ExtractSpanInfoRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceListAnnotationEvaluatorsArgs_Req_DEFAULT + return TraceServiceExtractSpanInfoArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceListAnnotationEvaluatorsArgs) SetReq(val *ListAnnotationEvaluatorsRequest) { +func (p *TraceServiceExtractSpanInfoArgs) SetReq(val *ExtractSpanInfoRequest) { p.Req = val } -var fieldIDToName_TraceServiceListAnnotationEvaluatorsArgs = map[int16]string{ +var fieldIDToName_TraceServiceExtractSpanInfoArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceListAnnotationEvaluatorsArgs) IsSetReq() bool { +func (p *TraceServiceExtractSpanInfoArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceListAnnotationEvaluatorsArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceExtractSpanInfoArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -28614,7 +31214,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationEvaluatorsArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceExtractSpanInfoArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -28624,8 +31224,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListAnnotationEvaluatorsArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewListAnnotationEvaluatorsRequest() +func (p *TraceServiceExtractSpanInfoArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewExtractSpanInfoRequest() if err := _field.Read(iprot); err != nil { return err } @@ -28633,9 +31233,9 @@ func (p *TraceServiceListAnnotationEvaluatorsArgs) ReadField1(iprot thrift.TProt return nil } -func (p *TraceServiceListAnnotationEvaluatorsArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceExtractSpanInfoArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListAnnotationEvaluators_args"); err != nil { + if err = oprot.WriteStructBegin("ExtractSpanInfo_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -28661,7 +31261,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListAnnotationEvaluatorsArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceExtractSpanInfoArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -28678,15 +31278,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceListAnnotationEvaluatorsArgs) String() string { +func (p *TraceServiceExtractSpanInfoArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListAnnotationEvaluatorsArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceExtractSpanInfoArgs(%+v)", *p) } -func (p *TraceServiceListAnnotationEvaluatorsArgs) DeepEqual(ano *TraceServiceListAnnotationEvaluatorsArgs) bool { +func (p *TraceServiceExtractSpanInfoArgs) DeepEqual(ano *TraceServiceExtractSpanInfoArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -28698,7 +31298,7 @@ func (p *TraceServiceListAnnotationEvaluatorsArgs) DeepEqual(ano *TraceServiceLi return true } -func (p *TraceServiceListAnnotationEvaluatorsArgs) Field1DeepEqual(src *ListAnnotationEvaluatorsRequest) bool { +func (p *TraceServiceExtractSpanInfoArgs) Field1DeepEqual(src *ExtractSpanInfoRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -28706,41 +31306,41 @@ func (p *TraceServiceListAnnotationEvaluatorsArgs) Field1DeepEqual(src *ListAnno return true } -type TraceServiceListAnnotationEvaluatorsResult struct { - Success *ListAnnotationEvaluatorsResponse `thrift:"success,0,optional" frugal:"0,optional,ListAnnotationEvaluatorsResponse"` +type TraceServiceExtractSpanInfoResult struct { + Success *ExtractSpanInfoResponse `thrift:"success,0,optional" frugal:"0,optional,ExtractSpanInfoResponse"` } -func NewTraceServiceListAnnotationEvaluatorsResult() *TraceServiceListAnnotationEvaluatorsResult { - return &TraceServiceListAnnotationEvaluatorsResult{} +func NewTraceServiceExtractSpanInfoResult() *TraceServiceExtractSpanInfoResult { + return &TraceServiceExtractSpanInfoResult{} } -func (p *TraceServiceListAnnotationEvaluatorsResult) InitDefault() { +func (p *TraceServiceExtractSpanInfoResult) InitDefault() { } -var TraceServiceListAnnotationEvaluatorsResult_Success_DEFAULT *ListAnnotationEvaluatorsResponse +var TraceServiceExtractSpanInfoResult_Success_DEFAULT *ExtractSpanInfoResponse -func (p *TraceServiceListAnnotationEvaluatorsResult) GetSuccess() (v *ListAnnotationEvaluatorsResponse) { +func (p *TraceServiceExtractSpanInfoResult) GetSuccess() (v *ExtractSpanInfoResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceListAnnotationEvaluatorsResult_Success_DEFAULT + return TraceServiceExtractSpanInfoResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceListAnnotationEvaluatorsResult) SetSuccess(x interface{}) { - p.Success = x.(*ListAnnotationEvaluatorsResponse) +func (p *TraceServiceExtractSpanInfoResult) SetSuccess(x interface{}) { + p.Success = x.(*ExtractSpanInfoResponse) } -var fieldIDToName_TraceServiceListAnnotationEvaluatorsResult = map[int16]string{ +var fieldIDToName_TraceServiceExtractSpanInfoResult = map[int16]string{ 0: "success", } -func (p *TraceServiceListAnnotationEvaluatorsResult) IsSetSuccess() bool { +func (p *TraceServiceExtractSpanInfoResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceListAnnotationEvaluatorsResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceExtractSpanInfoResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -28785,7 +31385,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationEvaluatorsResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceExtractSpanInfoResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -28795,8 +31395,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListAnnotationEvaluatorsResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewListAnnotationEvaluatorsResponse() +func (p *TraceServiceExtractSpanInfoResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewExtractSpanInfoResponse() if err := _field.Read(iprot); err != nil { return err } @@ -28804,9 +31404,9 @@ func (p *TraceServiceListAnnotationEvaluatorsResult) ReadField0(iprot thrift.TPr return nil } -func (p *TraceServiceListAnnotationEvaluatorsResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceExtractSpanInfoResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListAnnotationEvaluators_result"); err != nil { + if err = oprot.WriteStructBegin("ExtractSpanInfo_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -28832,7 +31432,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListAnnotationEvaluatorsResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceExtractSpanInfoResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -28851,15 +31451,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceListAnnotationEvaluatorsResult) String() string { +func (p *TraceServiceExtractSpanInfoResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListAnnotationEvaluatorsResult(%+v)", *p) + return fmt.Sprintf("TraceServiceExtractSpanInfoResult(%+v)", *p) } -func (p *TraceServiceListAnnotationEvaluatorsResult) DeepEqual(ano *TraceServiceListAnnotationEvaluatorsResult) bool { +func (p *TraceServiceExtractSpanInfoResult) DeepEqual(ano *TraceServiceExtractSpanInfoResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -28871,7 +31471,7 @@ func (p *TraceServiceListAnnotationEvaluatorsResult) DeepEqual(ano *TraceService return true } -func (p *TraceServiceListAnnotationEvaluatorsResult) Field0DeepEqual(src *ListAnnotationEvaluatorsResponse) bool { +func (p *TraceServiceExtractSpanInfoResult) Field0DeepEqual(src *ExtractSpanInfoResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -28879,41 +31479,41 @@ func (p *TraceServiceListAnnotationEvaluatorsResult) Field0DeepEqual(src *ListAn return true } -type TraceServiceExtractSpanInfoArgs struct { - Req *ExtractSpanInfoRequest `thrift:"req,1" frugal:"1,default,ExtractSpanInfoRequest"` +type TraceServiceUpsertTrajectoryConfigArgs struct { + Req *UpsertTrajectoryConfigRequest `thrift:"req,1" frugal:"1,default,UpsertTrajectoryConfigRequest"` } -func NewTraceServiceExtractSpanInfoArgs() *TraceServiceExtractSpanInfoArgs { - return &TraceServiceExtractSpanInfoArgs{} +func NewTraceServiceUpsertTrajectoryConfigArgs() *TraceServiceUpsertTrajectoryConfigArgs { + return &TraceServiceUpsertTrajectoryConfigArgs{} } -func (p *TraceServiceExtractSpanInfoArgs) InitDefault() { +func (p *TraceServiceUpsertTrajectoryConfigArgs) InitDefault() { } -var TraceServiceExtractSpanInfoArgs_Req_DEFAULT *ExtractSpanInfoRequest +var TraceServiceUpsertTrajectoryConfigArgs_Req_DEFAULT *UpsertTrajectoryConfigRequest -func (p *TraceServiceExtractSpanInfoArgs) GetReq() (v *ExtractSpanInfoRequest) { +func (p *TraceServiceUpsertTrajectoryConfigArgs) GetReq() (v *UpsertTrajectoryConfigRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceExtractSpanInfoArgs_Req_DEFAULT + return TraceServiceUpsertTrajectoryConfigArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceExtractSpanInfoArgs) SetReq(val *ExtractSpanInfoRequest) { +func (p *TraceServiceUpsertTrajectoryConfigArgs) SetReq(val *UpsertTrajectoryConfigRequest) { p.Req = val } -var fieldIDToName_TraceServiceExtractSpanInfoArgs = map[int16]string{ +var fieldIDToName_TraceServiceUpsertTrajectoryConfigArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceExtractSpanInfoArgs) IsSetReq() bool { +func (p *TraceServiceUpsertTrajectoryConfigArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceExtractSpanInfoArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpsertTrajectoryConfigArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -28958,7 +31558,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceExtractSpanInfoArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpsertTrajectoryConfigArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -28968,8 +31568,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceExtractSpanInfoArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewExtractSpanInfoRequest() +func (p *TraceServiceUpsertTrajectoryConfigArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewUpsertTrajectoryConfigRequest() if err := _field.Read(iprot); err != nil { return err } @@ -28977,9 +31577,9 @@ func (p *TraceServiceExtractSpanInfoArgs) ReadField1(iprot thrift.TProtocol) err return nil } -func (p *TraceServiceExtractSpanInfoArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpsertTrajectoryConfigArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ExtractSpanInfo_args"); err != nil { + if err = oprot.WriteStructBegin("UpsertTrajectoryConfig_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -29005,7 +31605,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceExtractSpanInfoArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpsertTrajectoryConfigArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -29022,15 +31622,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceExtractSpanInfoArgs) String() string { +func (p *TraceServiceUpsertTrajectoryConfigArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceExtractSpanInfoArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceUpsertTrajectoryConfigArgs(%+v)", *p) } -func (p *TraceServiceExtractSpanInfoArgs) DeepEqual(ano *TraceServiceExtractSpanInfoArgs) bool { +func (p *TraceServiceUpsertTrajectoryConfigArgs) DeepEqual(ano *TraceServiceUpsertTrajectoryConfigArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -29042,7 +31642,7 @@ func (p *TraceServiceExtractSpanInfoArgs) DeepEqual(ano *TraceServiceExtractSpan return true } -func (p *TraceServiceExtractSpanInfoArgs) Field1DeepEqual(src *ExtractSpanInfoRequest) bool { +func (p *TraceServiceUpsertTrajectoryConfigArgs) Field1DeepEqual(src *UpsertTrajectoryConfigRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -29050,41 +31650,41 @@ func (p *TraceServiceExtractSpanInfoArgs) Field1DeepEqual(src *ExtractSpanInfoRe return true } -type TraceServiceExtractSpanInfoResult struct { - Success *ExtractSpanInfoResponse `thrift:"success,0,optional" frugal:"0,optional,ExtractSpanInfoResponse"` +type TraceServiceUpsertTrajectoryConfigResult struct { + Success *UpsertTrajectoryConfigResponse `thrift:"success,0,optional" frugal:"0,optional,UpsertTrajectoryConfigResponse"` } -func NewTraceServiceExtractSpanInfoResult() *TraceServiceExtractSpanInfoResult { - return &TraceServiceExtractSpanInfoResult{} +func NewTraceServiceUpsertTrajectoryConfigResult() *TraceServiceUpsertTrajectoryConfigResult { + return &TraceServiceUpsertTrajectoryConfigResult{} } -func (p *TraceServiceExtractSpanInfoResult) InitDefault() { +func (p *TraceServiceUpsertTrajectoryConfigResult) InitDefault() { } -var TraceServiceExtractSpanInfoResult_Success_DEFAULT *ExtractSpanInfoResponse +var TraceServiceUpsertTrajectoryConfigResult_Success_DEFAULT *UpsertTrajectoryConfigResponse -func (p *TraceServiceExtractSpanInfoResult) GetSuccess() (v *ExtractSpanInfoResponse) { +func (p *TraceServiceUpsertTrajectoryConfigResult) GetSuccess() (v *UpsertTrajectoryConfigResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceExtractSpanInfoResult_Success_DEFAULT + return TraceServiceUpsertTrajectoryConfigResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceExtractSpanInfoResult) SetSuccess(x interface{}) { - p.Success = x.(*ExtractSpanInfoResponse) +func (p *TraceServiceUpsertTrajectoryConfigResult) SetSuccess(x interface{}) { + p.Success = x.(*UpsertTrajectoryConfigResponse) } -var fieldIDToName_TraceServiceExtractSpanInfoResult = map[int16]string{ +var fieldIDToName_TraceServiceUpsertTrajectoryConfigResult = map[int16]string{ 0: "success", } -func (p *TraceServiceExtractSpanInfoResult) IsSetSuccess() bool { +func (p *TraceServiceUpsertTrajectoryConfigResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceExtractSpanInfoResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpsertTrajectoryConfigResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -29129,7 +31729,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceExtractSpanInfoResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpsertTrajectoryConfigResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -29139,8 +31739,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceExtractSpanInfoResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewExtractSpanInfoResponse() +func (p *TraceServiceUpsertTrajectoryConfigResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewUpsertTrajectoryConfigResponse() if err := _field.Read(iprot); err != nil { return err } @@ -29148,9 +31748,9 @@ func (p *TraceServiceExtractSpanInfoResult) ReadField0(iprot thrift.TProtocol) e return nil } -func (p *TraceServiceExtractSpanInfoResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpsertTrajectoryConfigResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ExtractSpanInfo_result"); err != nil { + if err = oprot.WriteStructBegin("UpsertTrajectoryConfig_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -29176,7 +31776,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceExtractSpanInfoResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceUpsertTrajectoryConfigResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -29195,15 +31795,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceExtractSpanInfoResult) String() string { +func (p *TraceServiceUpsertTrajectoryConfigResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceExtractSpanInfoResult(%+v)", *p) + return fmt.Sprintf("TraceServiceUpsertTrajectoryConfigResult(%+v)", *p) } -func (p *TraceServiceExtractSpanInfoResult) DeepEqual(ano *TraceServiceExtractSpanInfoResult) bool { +func (p *TraceServiceUpsertTrajectoryConfigResult) DeepEqual(ano *TraceServiceUpsertTrajectoryConfigResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -29215,7 +31815,7 @@ func (p *TraceServiceExtractSpanInfoResult) DeepEqual(ano *TraceServiceExtractSp return true } -func (p *TraceServiceExtractSpanInfoResult) Field0DeepEqual(src *ExtractSpanInfoResponse) bool { +func (p *TraceServiceUpsertTrajectoryConfigResult) Field0DeepEqual(src *UpsertTrajectoryConfigResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -29223,41 +31823,41 @@ func (p *TraceServiceExtractSpanInfoResult) Field0DeepEqual(src *ExtractSpanInfo return true } -type TraceServiceUpsertTrajectoryConfigArgs struct { - Req *UpsertTrajectoryConfigRequest `thrift:"req,1" frugal:"1,default,UpsertTrajectoryConfigRequest"` +type TraceServiceGetTrajectoryConfigArgs struct { + Req *GetTrajectoryConfigRequest `thrift:"req,1" frugal:"1,default,GetTrajectoryConfigRequest"` } -func NewTraceServiceUpsertTrajectoryConfigArgs() *TraceServiceUpsertTrajectoryConfigArgs { - return &TraceServiceUpsertTrajectoryConfigArgs{} +func NewTraceServiceGetTrajectoryConfigArgs() *TraceServiceGetTrajectoryConfigArgs { + return &TraceServiceGetTrajectoryConfigArgs{} } -func (p *TraceServiceUpsertTrajectoryConfigArgs) InitDefault() { +func (p *TraceServiceGetTrajectoryConfigArgs) InitDefault() { } -var TraceServiceUpsertTrajectoryConfigArgs_Req_DEFAULT *UpsertTrajectoryConfigRequest +var TraceServiceGetTrajectoryConfigArgs_Req_DEFAULT *GetTrajectoryConfigRequest -func (p *TraceServiceUpsertTrajectoryConfigArgs) GetReq() (v *UpsertTrajectoryConfigRequest) { +func (p *TraceServiceGetTrajectoryConfigArgs) GetReq() (v *GetTrajectoryConfigRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceUpsertTrajectoryConfigArgs_Req_DEFAULT + return TraceServiceGetTrajectoryConfigArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceUpsertTrajectoryConfigArgs) SetReq(val *UpsertTrajectoryConfigRequest) { +func (p *TraceServiceGetTrajectoryConfigArgs) SetReq(val *GetTrajectoryConfigRequest) { p.Req = val } -var fieldIDToName_TraceServiceUpsertTrajectoryConfigArgs = map[int16]string{ +var fieldIDToName_TraceServiceGetTrajectoryConfigArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceUpsertTrajectoryConfigArgs) IsSetReq() bool { +func (p *TraceServiceGetTrajectoryConfigArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceUpsertTrajectoryConfigArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTrajectoryConfigArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -29302,7 +31902,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpsertTrajectoryConfigArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTrajectoryConfigArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -29312,8 +31912,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceUpsertTrajectoryConfigArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewUpsertTrajectoryConfigRequest() +func (p *TraceServiceGetTrajectoryConfigArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewGetTrajectoryConfigRequest() if err := _field.Read(iprot); err != nil { return err } @@ -29321,9 +31921,9 @@ func (p *TraceServiceUpsertTrajectoryConfigArgs) ReadField1(iprot thrift.TProtoc return nil } -func (p *TraceServiceUpsertTrajectoryConfigArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTrajectoryConfigArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("UpsertTrajectoryConfig_args"); err != nil { + if err = oprot.WriteStructBegin("GetTrajectoryConfig_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -29349,7 +31949,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceUpsertTrajectoryConfigArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTrajectoryConfigArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -29366,15 +31966,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceUpsertTrajectoryConfigArgs) String() string { +func (p *TraceServiceGetTrajectoryConfigArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceUpsertTrajectoryConfigArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceGetTrajectoryConfigArgs(%+v)", *p) } -func (p *TraceServiceUpsertTrajectoryConfigArgs) DeepEqual(ano *TraceServiceUpsertTrajectoryConfigArgs) bool { +func (p *TraceServiceGetTrajectoryConfigArgs) DeepEqual(ano *TraceServiceGetTrajectoryConfigArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -29386,7 +31986,7 @@ func (p *TraceServiceUpsertTrajectoryConfigArgs) DeepEqual(ano *TraceServiceUpse return true } -func (p *TraceServiceUpsertTrajectoryConfigArgs) Field1DeepEqual(src *UpsertTrajectoryConfigRequest) bool { +func (p *TraceServiceGetTrajectoryConfigArgs) Field1DeepEqual(src *GetTrajectoryConfigRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -29394,41 +31994,41 @@ func (p *TraceServiceUpsertTrajectoryConfigArgs) Field1DeepEqual(src *UpsertTraj return true } -type TraceServiceUpsertTrajectoryConfigResult struct { - Success *UpsertTrajectoryConfigResponse `thrift:"success,0,optional" frugal:"0,optional,UpsertTrajectoryConfigResponse"` +type TraceServiceGetTrajectoryConfigResult struct { + Success *GetTrajectoryConfigResponse `thrift:"success,0,optional" frugal:"0,optional,GetTrajectoryConfigResponse"` } -func NewTraceServiceUpsertTrajectoryConfigResult() *TraceServiceUpsertTrajectoryConfigResult { - return &TraceServiceUpsertTrajectoryConfigResult{} +func NewTraceServiceGetTrajectoryConfigResult() *TraceServiceGetTrajectoryConfigResult { + return &TraceServiceGetTrajectoryConfigResult{} } -func (p *TraceServiceUpsertTrajectoryConfigResult) InitDefault() { +func (p *TraceServiceGetTrajectoryConfigResult) InitDefault() { } -var TraceServiceUpsertTrajectoryConfigResult_Success_DEFAULT *UpsertTrajectoryConfigResponse +var TraceServiceGetTrajectoryConfigResult_Success_DEFAULT *GetTrajectoryConfigResponse -func (p *TraceServiceUpsertTrajectoryConfigResult) GetSuccess() (v *UpsertTrajectoryConfigResponse) { +func (p *TraceServiceGetTrajectoryConfigResult) GetSuccess() (v *GetTrajectoryConfigResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceUpsertTrajectoryConfigResult_Success_DEFAULT + return TraceServiceGetTrajectoryConfigResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceUpsertTrajectoryConfigResult) SetSuccess(x interface{}) { - p.Success = x.(*UpsertTrajectoryConfigResponse) +func (p *TraceServiceGetTrajectoryConfigResult) SetSuccess(x interface{}) { + p.Success = x.(*GetTrajectoryConfigResponse) } -var fieldIDToName_TraceServiceUpsertTrajectoryConfigResult = map[int16]string{ +var fieldIDToName_TraceServiceGetTrajectoryConfigResult = map[int16]string{ 0: "success", } -func (p *TraceServiceUpsertTrajectoryConfigResult) IsSetSuccess() bool { +func (p *TraceServiceGetTrajectoryConfigResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceUpsertTrajectoryConfigResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTrajectoryConfigResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -29473,7 +32073,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceUpsertTrajectoryConfigResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTrajectoryConfigResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -29483,8 +32083,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceUpsertTrajectoryConfigResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewUpsertTrajectoryConfigResponse() +func (p *TraceServiceGetTrajectoryConfigResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewGetTrajectoryConfigResponse() if err := _field.Read(iprot); err != nil { return err } @@ -29492,9 +32092,9 @@ func (p *TraceServiceUpsertTrajectoryConfigResult) ReadField0(iprot thrift.TProt return nil } -func (p *TraceServiceUpsertTrajectoryConfigResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTrajectoryConfigResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("UpsertTrajectoryConfig_result"); err != nil { + if err = oprot.WriteStructBegin("GetTrajectoryConfig_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -29520,7 +32120,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceUpsertTrajectoryConfigResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceGetTrajectoryConfigResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -29539,15 +32139,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceUpsertTrajectoryConfigResult) String() string { +func (p *TraceServiceGetTrajectoryConfigResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceUpsertTrajectoryConfigResult(%+v)", *p) + return fmt.Sprintf("TraceServiceGetTrajectoryConfigResult(%+v)", *p) } -func (p *TraceServiceUpsertTrajectoryConfigResult) DeepEqual(ano *TraceServiceUpsertTrajectoryConfigResult) bool { +func (p *TraceServiceGetTrajectoryConfigResult) DeepEqual(ano *TraceServiceGetTrajectoryConfigResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -29559,7 +32159,7 @@ func (p *TraceServiceUpsertTrajectoryConfigResult) DeepEqual(ano *TraceServiceUp return true } -func (p *TraceServiceUpsertTrajectoryConfigResult) Field0DeepEqual(src *UpsertTrajectoryConfigResponse) bool { +func (p *TraceServiceGetTrajectoryConfigResult) Field0DeepEqual(src *GetTrajectoryConfigResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -29567,41 +32167,41 @@ func (p *TraceServiceUpsertTrajectoryConfigResult) Field0DeepEqual(src *UpsertTr return true } -type TraceServiceGetTrajectoryConfigArgs struct { - Req *GetTrajectoryConfigRequest `thrift:"req,1" frugal:"1,default,GetTrajectoryConfigRequest"` +type TraceServiceListTrajectoryArgs struct { + Req *ListTrajectoryRequest `thrift:"req,1" frugal:"1,default,ListTrajectoryRequest"` } -func NewTraceServiceGetTrajectoryConfigArgs() *TraceServiceGetTrajectoryConfigArgs { - return &TraceServiceGetTrajectoryConfigArgs{} +func NewTraceServiceListTrajectoryArgs() *TraceServiceListTrajectoryArgs { + return &TraceServiceListTrajectoryArgs{} } -func (p *TraceServiceGetTrajectoryConfigArgs) InitDefault() { +func (p *TraceServiceListTrajectoryArgs) InitDefault() { } -var TraceServiceGetTrajectoryConfigArgs_Req_DEFAULT *GetTrajectoryConfigRequest +var TraceServiceListTrajectoryArgs_Req_DEFAULT *ListTrajectoryRequest -func (p *TraceServiceGetTrajectoryConfigArgs) GetReq() (v *GetTrajectoryConfigRequest) { +func (p *TraceServiceListTrajectoryArgs) GetReq() (v *ListTrajectoryRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceGetTrajectoryConfigArgs_Req_DEFAULT + return TraceServiceListTrajectoryArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceGetTrajectoryConfigArgs) SetReq(val *GetTrajectoryConfigRequest) { +func (p *TraceServiceListTrajectoryArgs) SetReq(val *ListTrajectoryRequest) { p.Req = val } -var fieldIDToName_TraceServiceGetTrajectoryConfigArgs = map[int16]string{ +var fieldIDToName_TraceServiceListTrajectoryArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceGetTrajectoryConfigArgs) IsSetReq() bool { +func (p *TraceServiceListTrajectoryArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceGetTrajectoryConfigArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListTrajectoryArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -29646,7 +32246,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTrajectoryConfigArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListTrajectoryArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -29656,8 +32256,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceGetTrajectoryConfigArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewGetTrajectoryConfigRequest() +func (p *TraceServiceListTrajectoryArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewListTrajectoryRequest() if err := _field.Read(iprot); err != nil { return err } @@ -29665,9 +32265,9 @@ func (p *TraceServiceGetTrajectoryConfigArgs) ReadField1(iprot thrift.TProtocol) return nil } -func (p *TraceServiceGetTrajectoryConfigArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListTrajectoryArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("GetTrajectoryConfig_args"); err != nil { + if err = oprot.WriteStructBegin("ListTrajectory_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -29693,7 +32293,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceGetTrajectoryConfigArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListTrajectoryArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -29710,15 +32310,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceGetTrajectoryConfigArgs) String() string { +func (p *TraceServiceListTrajectoryArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceGetTrajectoryConfigArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceListTrajectoryArgs(%+v)", *p) } -func (p *TraceServiceGetTrajectoryConfigArgs) DeepEqual(ano *TraceServiceGetTrajectoryConfigArgs) bool { +func (p *TraceServiceListTrajectoryArgs) DeepEqual(ano *TraceServiceListTrajectoryArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -29730,7 +32330,7 @@ func (p *TraceServiceGetTrajectoryConfigArgs) DeepEqual(ano *TraceServiceGetTraj return true } -func (p *TraceServiceGetTrajectoryConfigArgs) Field1DeepEqual(src *GetTrajectoryConfigRequest) bool { +func (p *TraceServiceListTrajectoryArgs) Field1DeepEqual(src *ListTrajectoryRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -29738,41 +32338,41 @@ func (p *TraceServiceGetTrajectoryConfigArgs) Field1DeepEqual(src *GetTrajectory return true } -type TraceServiceGetTrajectoryConfigResult struct { - Success *GetTrajectoryConfigResponse `thrift:"success,0,optional" frugal:"0,optional,GetTrajectoryConfigResponse"` +type TraceServiceListTrajectoryResult struct { + Success *ListTrajectoryResponse `thrift:"success,0,optional" frugal:"0,optional,ListTrajectoryResponse"` } -func NewTraceServiceGetTrajectoryConfigResult() *TraceServiceGetTrajectoryConfigResult { - return &TraceServiceGetTrajectoryConfigResult{} +func NewTraceServiceListTrajectoryResult() *TraceServiceListTrajectoryResult { + return &TraceServiceListTrajectoryResult{} } -func (p *TraceServiceGetTrajectoryConfigResult) InitDefault() { +func (p *TraceServiceListTrajectoryResult) InitDefault() { } -var TraceServiceGetTrajectoryConfigResult_Success_DEFAULT *GetTrajectoryConfigResponse +var TraceServiceListTrajectoryResult_Success_DEFAULT *ListTrajectoryResponse -func (p *TraceServiceGetTrajectoryConfigResult) GetSuccess() (v *GetTrajectoryConfigResponse) { +func (p *TraceServiceListTrajectoryResult) GetSuccess() (v *ListTrajectoryResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceGetTrajectoryConfigResult_Success_DEFAULT + return TraceServiceListTrajectoryResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceGetTrajectoryConfigResult) SetSuccess(x interface{}) { - p.Success = x.(*GetTrajectoryConfigResponse) +func (p *TraceServiceListTrajectoryResult) SetSuccess(x interface{}) { + p.Success = x.(*ListTrajectoryResponse) } -var fieldIDToName_TraceServiceGetTrajectoryConfigResult = map[int16]string{ +var fieldIDToName_TraceServiceListTrajectoryResult = map[int16]string{ 0: "success", } -func (p *TraceServiceGetTrajectoryConfigResult) IsSetSuccess() bool { +func (p *TraceServiceListTrajectoryResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceGetTrajectoryConfigResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListTrajectoryResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -29817,7 +32417,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceGetTrajectoryConfigResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListTrajectoryResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -29827,8 +32427,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceGetTrajectoryConfigResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewGetTrajectoryConfigResponse() +func (p *TraceServiceListTrajectoryResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewListTrajectoryResponse() if err := _field.Read(iprot); err != nil { return err } @@ -29836,9 +32436,9 @@ func (p *TraceServiceGetTrajectoryConfigResult) ReadField0(iprot thrift.TProtoco return nil } -func (p *TraceServiceGetTrajectoryConfigResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListTrajectoryResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("GetTrajectoryConfig_result"); err != nil { + if err = oprot.WriteStructBegin("ListTrajectory_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -29864,7 +32464,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceGetTrajectoryConfigResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListTrajectoryResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -29883,15 +32483,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceGetTrajectoryConfigResult) String() string { +func (p *TraceServiceListTrajectoryResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceGetTrajectoryConfigResult(%+v)", *p) + return fmt.Sprintf("TraceServiceListTrajectoryResult(%+v)", *p) } -func (p *TraceServiceGetTrajectoryConfigResult) DeepEqual(ano *TraceServiceGetTrajectoryConfigResult) bool { +func (p *TraceServiceListTrajectoryResult) DeepEqual(ano *TraceServiceListTrajectoryResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -29903,7 +32503,7 @@ func (p *TraceServiceGetTrajectoryConfigResult) DeepEqual(ano *TraceServiceGetTr return true } -func (p *TraceServiceGetTrajectoryConfigResult) Field0DeepEqual(src *GetTrajectoryConfigResponse) bool { +func (p *TraceServiceListTrajectoryResult) Field0DeepEqual(src *ListTrajectoryResponse) bool { if !p.Success.DeepEqual(src) { return false @@ -29911,41 +32511,41 @@ func (p *TraceServiceGetTrajectoryConfigResult) Field0DeepEqual(src *GetTrajecto return true } -type TraceServiceListTrajectoryArgs struct { - Req *ListTrajectoryRequest `thrift:"req,1" frugal:"1,default,ListTrajectoryRequest"` +type TraceServiceListMetadataArgs struct { + Req *ListMetadataRequest `thrift:"req,1" frugal:"1,default,ListMetadataRequest"` } -func NewTraceServiceListTrajectoryArgs() *TraceServiceListTrajectoryArgs { - return &TraceServiceListTrajectoryArgs{} +func NewTraceServiceListMetadataArgs() *TraceServiceListMetadataArgs { + return &TraceServiceListMetadataArgs{} } -func (p *TraceServiceListTrajectoryArgs) InitDefault() { +func (p *TraceServiceListMetadataArgs) InitDefault() { } -var TraceServiceListTrajectoryArgs_Req_DEFAULT *ListTrajectoryRequest +var TraceServiceListMetadataArgs_Req_DEFAULT *ListMetadataRequest -func (p *TraceServiceListTrajectoryArgs) GetReq() (v *ListTrajectoryRequest) { +func (p *TraceServiceListMetadataArgs) GetReq() (v *ListMetadataRequest) { if p == nil { return } if !p.IsSetReq() { - return TraceServiceListTrajectoryArgs_Req_DEFAULT + return TraceServiceListMetadataArgs_Req_DEFAULT } return p.Req } -func (p *TraceServiceListTrajectoryArgs) SetReq(val *ListTrajectoryRequest) { +func (p *TraceServiceListMetadataArgs) SetReq(val *ListMetadataRequest) { p.Req = val } -var fieldIDToName_TraceServiceListTrajectoryArgs = map[int16]string{ +var fieldIDToName_TraceServiceListMetadataArgs = map[int16]string{ 1: "req", } -func (p *TraceServiceListTrajectoryArgs) IsSetReq() bool { +func (p *TraceServiceListMetadataArgs) IsSetReq() bool { return p.Req != nil } -func (p *TraceServiceListTrajectoryArgs) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListMetadataArgs) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -29990,7 +32590,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListTrajectoryArgs[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListMetadataArgs[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -30000,8 +32600,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListTrajectoryArgs) ReadField1(iprot thrift.TProtocol) error { - _field := NewListTrajectoryRequest() +func (p *TraceServiceListMetadataArgs) ReadField1(iprot thrift.TProtocol) error { + _field := NewListMetadataRequest() if err := _field.Read(iprot); err != nil { return err } @@ -30009,9 +32609,9 @@ func (p *TraceServiceListTrajectoryArgs) ReadField1(iprot thrift.TProtocol) erro return nil } -func (p *TraceServiceListTrajectoryArgs) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListMetadataArgs) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListTrajectory_args"); err != nil { + if err = oprot.WriteStructBegin("ListMetadata_args"); err != nil { goto WriteStructBeginError } if p != nil { @@ -30037,7 +32637,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListTrajectoryArgs) writeField1(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListMetadataArgs) writeField1(oprot thrift.TProtocol) (err error) { if err = oprot.WriteFieldBegin("req", thrift.STRUCT, 1); err != nil { goto WriteFieldBeginError } @@ -30054,15 +32654,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 1 end error: ", p), err) } -func (p *TraceServiceListTrajectoryArgs) String() string { +func (p *TraceServiceListMetadataArgs) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListTrajectoryArgs(%+v)", *p) + return fmt.Sprintf("TraceServiceListMetadataArgs(%+v)", *p) } -func (p *TraceServiceListTrajectoryArgs) DeepEqual(ano *TraceServiceListTrajectoryArgs) bool { +func (p *TraceServiceListMetadataArgs) DeepEqual(ano *TraceServiceListMetadataArgs) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -30074,7 +32674,7 @@ func (p *TraceServiceListTrajectoryArgs) DeepEqual(ano *TraceServiceListTrajecto return true } -func (p *TraceServiceListTrajectoryArgs) Field1DeepEqual(src *ListTrajectoryRequest) bool { +func (p *TraceServiceListMetadataArgs) Field1DeepEqual(src *ListMetadataRequest) bool { if !p.Req.DeepEqual(src) { return false @@ -30082,41 +32682,41 @@ func (p *TraceServiceListTrajectoryArgs) Field1DeepEqual(src *ListTrajectoryRequ return true } -type TraceServiceListTrajectoryResult struct { - Success *ListTrajectoryResponse `thrift:"success,0,optional" frugal:"0,optional,ListTrajectoryResponse"` +type TraceServiceListMetadataResult struct { + Success *ListMetadataResponse `thrift:"success,0,optional" frugal:"0,optional,ListMetadataResponse"` } -func NewTraceServiceListTrajectoryResult() *TraceServiceListTrajectoryResult { - return &TraceServiceListTrajectoryResult{} +func NewTraceServiceListMetadataResult() *TraceServiceListMetadataResult { + return &TraceServiceListMetadataResult{} } -func (p *TraceServiceListTrajectoryResult) InitDefault() { +func (p *TraceServiceListMetadataResult) InitDefault() { } -var TraceServiceListTrajectoryResult_Success_DEFAULT *ListTrajectoryResponse +var TraceServiceListMetadataResult_Success_DEFAULT *ListMetadataResponse -func (p *TraceServiceListTrajectoryResult) GetSuccess() (v *ListTrajectoryResponse) { +func (p *TraceServiceListMetadataResult) GetSuccess() (v *ListMetadataResponse) { if p == nil { return } if !p.IsSetSuccess() { - return TraceServiceListTrajectoryResult_Success_DEFAULT + return TraceServiceListMetadataResult_Success_DEFAULT } return p.Success } -func (p *TraceServiceListTrajectoryResult) SetSuccess(x interface{}) { - p.Success = x.(*ListTrajectoryResponse) +func (p *TraceServiceListMetadataResult) SetSuccess(x interface{}) { + p.Success = x.(*ListMetadataResponse) } -var fieldIDToName_TraceServiceListTrajectoryResult = map[int16]string{ +var fieldIDToName_TraceServiceListMetadataResult = map[int16]string{ 0: "success", } -func (p *TraceServiceListTrajectoryResult) IsSetSuccess() bool { +func (p *TraceServiceListMetadataResult) IsSetSuccess() bool { return p.Success != nil } -func (p *TraceServiceListTrajectoryResult) Read(iprot thrift.TProtocol) (err error) { +func (p *TraceServiceListMetadataResult) Read(iprot thrift.TProtocol) (err error) { var fieldTypeId thrift.TType var fieldId int16 @@ -30161,7 +32761,7 @@ ReadStructBeginError: ReadFieldBeginError: return thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListTrajectoryResult[fieldId]), err) + return thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListMetadataResult[fieldId]), err) SkipFieldError: return thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) @@ -30171,8 +32771,8 @@ ReadStructEndError: return thrift.PrependError(fmt.Sprintf("%T read struct end error: ", p), err) } -func (p *TraceServiceListTrajectoryResult) ReadField0(iprot thrift.TProtocol) error { - _field := NewListTrajectoryResponse() +func (p *TraceServiceListMetadataResult) ReadField0(iprot thrift.TProtocol) error { + _field := NewListMetadataResponse() if err := _field.Read(iprot); err != nil { return err } @@ -30180,9 +32780,9 @@ func (p *TraceServiceListTrajectoryResult) ReadField0(iprot thrift.TProtocol) er return nil } -func (p *TraceServiceListTrajectoryResult) Write(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListMetadataResult) Write(oprot thrift.TProtocol) (err error) { var fieldId int16 - if err = oprot.WriteStructBegin("ListTrajectory_result"); err != nil { + if err = oprot.WriteStructBegin("ListMetadata_result"); err != nil { goto WriteStructBeginError } if p != nil { @@ -30208,7 +32808,7 @@ WriteStructEndError: return thrift.PrependError(fmt.Sprintf("%T write struct end error: ", p), err) } -func (p *TraceServiceListTrajectoryResult) writeField0(oprot thrift.TProtocol) (err error) { +func (p *TraceServiceListMetadataResult) writeField0(oprot thrift.TProtocol) (err error) { if p.IsSetSuccess() { if err = oprot.WriteFieldBegin("success", thrift.STRUCT, 0); err != nil { goto WriteFieldBeginError @@ -30227,15 +32827,15 @@ WriteFieldEndError: return thrift.PrependError(fmt.Sprintf("%T write field 0 end error: ", p), err) } -func (p *TraceServiceListTrajectoryResult) String() string { +func (p *TraceServiceListMetadataResult) String() string { if p == nil { return "" } - return fmt.Sprintf("TraceServiceListTrajectoryResult(%+v)", *p) + return fmt.Sprintf("TraceServiceListMetadataResult(%+v)", *p) } -func (p *TraceServiceListTrajectoryResult) DeepEqual(ano *TraceServiceListTrajectoryResult) bool { +func (p *TraceServiceListMetadataResult) DeepEqual(ano *TraceServiceListMetadataResult) bool { if p == ano { return true } else if p == nil || ano == nil { @@ -30247,7 +32847,7 @@ func (p *TraceServiceListTrajectoryResult) DeepEqual(ano *TraceServiceListTrajec return true } -func (p *TraceServiceListTrajectoryResult) Field0DeepEqual(src *ListTrajectoryResponse) bool { +func (p *TraceServiceListMetadataResult) Field0DeepEqual(src *ListMetadataResponse) bool { if !p.Success.DeepEqual(src) { return false diff --git a/backend/kitex_gen/coze/loop/observability/trace/coze.loop.observability.trace_validator.go b/backend/kitex_gen/coze/loop/observability/trace/coze.loop.observability.trace_validator.go index 096a4eaa9..308a95be1 100644 --- a/backend/kitex_gen/coze/loop/observability/trace/coze.loop.observability.trace_validator.go +++ b/backend/kitex_gen/coze/loop/observability/trace/coze.loop.observability.trace_validator.go @@ -340,6 +340,25 @@ func (p *ListAnnotationsResponse) IsValid() error { } return nil } +func (p *ListWorkspaceAnnotationsRequest) IsValid() error { + if p.WorkspaceID <= int64(0) { + return fmt.Errorf("field WorkspaceID gt rule failed, current value: %v", p.WorkspaceID) + } + if p.Base != nil { + if err := p.Base.IsValid(); err != nil { + return fmt.Errorf("field Base not valid, %w", err) + } + } + return nil +} +func (p *ListWorkspaceAnnotationsResponse) IsValid() error { + if p.BaseResp != nil { + if err := p.BaseResp.IsValid(); err != nil { + return fmt.Errorf("field BaseResp not valid, %w", err) + } + } + return nil +} func (p *ExportTracesToDatasetRequest) IsValid() error { if p.WorkspaceID <= int64(0) { return fmt.Errorf("field WorkspaceID gt rule failed, current value: %v", p.WorkspaceID) @@ -392,9 +411,6 @@ func (p *PreviewExportTracesToDatasetRequest) IsValid() error { if p.WorkspaceID <= int64(0) { return fmt.Errorf("field WorkspaceID gt rule failed, current value: %v", p.WorkspaceID) } - if len(p.SpanIds) < int(1) { - return fmt.Errorf("field SpanIds MinLen rule failed, current value: %v", p.SpanIds) - } if len(p.SpanIds) > int(500) { return fmt.Errorf("field SpanIds MaxLen rule failed, current value: %v", p.SpanIds) } @@ -409,6 +425,16 @@ func (p *PreviewExportTracesToDatasetRequest) IsValid() error { if len(p.FieldMappings) > int(100) { return fmt.Errorf("field FieldMappings MaxLen rule failed, current value: %v", p.FieldMappings) } + if p.SpanFilters != nil { + if err := p.SpanFilters.IsValid(); err != nil { + return fmt.Errorf("field SpanFilters not valid, %w", err) + } + } + if p.Limit != nil { + if *p.Limit <= int64(0) { + return fmt.Errorf("field Limit gt rule failed, current value: %v", *p.Limit) + } + } if p.Base != nil { if err := p.Base.IsValid(); err != nil { return fmt.Errorf("field Base not valid, %w", err) @@ -592,3 +618,22 @@ func (p *ListTrajectoryResponse) IsValid() error { } return nil } +func (p *ListMetadataRequest) IsValid() error { + if p.Base != nil { + if err := p.Base.IsValid(); err != nil { + return fmt.Errorf("field Base not valid, %w", err) + } + } + return nil +} +func (p *MetadataItemInfo) IsValid() error { + return nil +} +func (p *ListMetadataResponse) IsValid() error { + if p.BaseResp != nil { + if err := p.BaseResp.IsValid(); err != nil { + return fmt.Errorf("field BaseResp not valid, %w", err) + } + } + return nil +} diff --git a/backend/kitex_gen/coze/loop/observability/trace/k-coze.loop.observability.trace.go b/backend/kitex_gen/coze/loop/observability/trace/k-coze.loop.observability.trace.go index c6cfda974..2a70aecf2 100644 --- a/backend/kitex_gen/coze/loop/observability/trace/k-coze.loop.observability.trace.go +++ b/backend/kitex_gen/coze/loop/observability/trace/k-coze.loop.observability.trace.go @@ -9596,7 +9596,7 @@ func (p *ListAnnotationsResponse) DeepCopy(s interface{}) error { return nil } -func (p *ExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsRequest) FastRead(buf []byte) (int, error) { var err error var offset int @@ -9604,12 +9604,6 @@ func (p *ExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { var fieldTypeId thrift.TType var fieldId int16 var issetWorkspaceID bool = false - var issetSpanIds bool = false - var issetCategory bool = false - var issetConfig bool = false - var issetStartTime bool = false - var issetEndTime bool = false - var issetExportType bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -9636,13 +9630,12 @@ func (p *ExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { } } case 2: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.STRING { l, err = p.FastReadField2(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetSpanIds = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -9651,13 +9644,12 @@ func (p *ExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { } } case 3: - if fieldTypeId == thrift.I32 { + if fieldTypeId == thrift.STRING { l, err = p.FastReadField3(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetCategory = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -9666,82 +9658,8 @@ func (p *ExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { } } case 4: - if fieldTypeId == thrift.STRUCT { - l, err = p.FastReadField4(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetConfig = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 5: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField5(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetStartTime = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 6: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField6(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetEndTime = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 7: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField7(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 8: if fieldTypeId == thrift.STRING { - l, err = p.FastReadField8(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetExportType = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 9: - if fieldTypeId == thrift.LIST { - l, err = p.FastReadField9(buf[offset:]) + l, err = p.FastReadField4(buf[offset:]) offset += l if err != nil { goto ReadFieldError @@ -9780,48 +9698,18 @@ func (p *ExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { fieldId = 1 goto RequiredFieldNotSetError } - - if !issetSpanIds { - fieldId = 2 - goto RequiredFieldNotSetError - } - - if !issetCategory { - fieldId = 3 - goto RequiredFieldNotSetError - } - - if !issetConfig { - fieldId = 4 - goto RequiredFieldNotSetError - } - - if !issetStartTime { - fieldId = 5 - goto RequiredFieldNotSetError - } - - if !issetEndTime { - fieldId = 6 - goto RequiredFieldNotSetError - } - - if !issetExportType { - fieldId = 8 - goto RequiredFieldNotSetError - } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExportTracesToDatasetRequest[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListWorkspaceAnnotationsRequest[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ExportTracesToDatasetRequest[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListWorkspaceAnnotationsRequest[fieldId])) } -func (p *ExportTracesToDatasetRequest) FastReadField1(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsRequest) FastReadField1(buf []byte) (int, error) { offset := 0 var _field int64 @@ -9835,87 +9723,7 @@ func (p *ExportTracesToDatasetRequest) FastReadField1(buf []byte) (int, error) { return offset, nil } -func (p *ExportTracesToDatasetRequest) FastReadField2(buf []byte) (int, error) { - offset := 0 - - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { - return offset, err - } - _field := make([]*SpanID, 0, size) - values := make([]SpanID, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - - _field = append(_field, _elem) - } - p.SpanIds = _field - return offset, nil -} - -func (p *ExportTracesToDatasetRequest) FastReadField3(buf []byte) (int, error) { - offset := 0 - - var _field dataset.DatasetCategory - if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - - _field = dataset.DatasetCategory(v) - } - p.Category = _field - return offset, nil -} - -func (p *ExportTracesToDatasetRequest) FastReadField4(buf []byte) (int, error) { - offset := 0 - _field := NewDatasetConfig() - if l, err := _field.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - p.Config = _field - return offset, nil -} - -func (p *ExportTracesToDatasetRequest) FastReadField5(buf []byte) (int, error) { - offset := 0 - - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = v - } - p.StartTime = _field - return offset, nil -} - -func (p *ExportTracesToDatasetRequest) FastReadField6(buf []byte) (int, error) { - offset := 0 - - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = v - } - p.EndTime = _field - return offset, nil -} - -func (p *ExportTracesToDatasetRequest) FastReadField7(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsRequest) FastReadField2(buf []byte) (int, error) { offset := 0 var _field *common.PlatformType @@ -9929,46 +9737,35 @@ func (p *ExportTracesToDatasetRequest) FastReadField7(buf []byte) (int, error) { return offset, nil } -func (p *ExportTracesToDatasetRequest) FastReadField8(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsRequest) FastReadField3(buf []byte) (int, error) { offset := 0 - var _field dataset0.ExportType + var _field *annotation.AnnotationType if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = v + _field = &v } - p.ExportType = _field + p.AnnotationType = _field return offset, nil } -func (p *ExportTracesToDatasetRequest) FastReadField9(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsRequest) FastReadField4(buf []byte) (int, error) { offset := 0 - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { + var _field *common.SpanListType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { return offset, err + } else { + offset += l + _field = &v } - _field := make([]*dataset0.FieldMapping, 0, size) - values := make([]dataset0.FieldMapping, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - - _field = append(_field, _elem) - } - p.FieldMappings = _field + p.SpanListType = _field return offset, nil } -func (p *ExportTracesToDatasetRequest) FastReadField255(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsRequest) FastReadField255(buf []byte) (int, error) { offset := 0 _field := base.NewBase() if l, err := _field.FastRead(buf[offset:]); err != nil { @@ -9980,128 +9777,71 @@ func (p *ExportTracesToDatasetRequest) FastReadField255(buf []byte) (int, error) return offset, nil } -func (p *ExportTracesToDatasetRequest) FastWrite(buf []byte) int { +func (p *ListWorkspaceAnnotationsRequest) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ExportTracesToDatasetRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField5(buf[offset:], w) - offset += p.fastWriteField6(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) offset += p.fastWriteField3(buf[offset:], w) offset += p.fastWriteField4(buf[offset:], w) - offset += p.fastWriteField7(buf[offset:], w) - offset += p.fastWriteField8(buf[offset:], w) - offset += p.fastWriteField9(buf[offset:], w) offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *ExportTracesToDatasetRequest) BLength() int { +func (p *ListWorkspaceAnnotationsRequest) BLength() int { l := 0 if p != nil { l += p.field1Length() l += p.field2Length() l += p.field3Length() l += p.field4Length() - l += p.field5Length() - l += p.field6Length() - l += p.field7Length() - l += p.field8Length() - l += p.field9Length() l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *ExportTracesToDatasetRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) return offset } -func (p *ExportTracesToDatasetRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.SpanIds { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) - return offset -} - -func (p *ExportTracesToDatasetRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 3) - offset += thrift.Binary.WriteI32(buf[offset:], int32(p.Category)) - return offset -} - -func (p *ExportTracesToDatasetRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 4) - offset += p.Config.FastWriteNocopy(buf[offset:], w) - return offset -} - -func (p *ExportTracesToDatasetRequest) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 5) - offset += thrift.Binary.WriteI64(buf[offset:], p.StartTime) - return offset -} - -func (p *ExportTracesToDatasetRequest) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 6) - offset += thrift.Binary.WriteI64(buf[offset:], p.EndTime) - return offset -} - -func (p *ExportTracesToDatasetRequest) fastWriteField7(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetPlatformType() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 7) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.PlatformType) } return offset } -func (p *ExportTracesToDatasetRequest) fastWriteField8(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 8) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.ExportType) + if p.IsSetAnnotationType() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 3) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.AnnotationType) + } return offset } -func (p *ExportTracesToDatasetRequest) fastWriteField9(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetFieldMappings() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 9) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.FieldMappings { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + if p.IsSetSpanListType() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 4) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.SpanListType) } return offset } -func (p *ExportTracesToDatasetRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetBase() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) @@ -10110,53 +9850,14 @@ func (p *ExportTracesToDatasetRequest) fastWriteField255(buf []byte, w thrift.No return offset } -func (p *ExportTracesToDatasetRequest) field1Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - return l -} - -func (p *ExportTracesToDatasetRequest) field2Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.SpanIds { - _ = v - l += v.BLength() - } - return l -} - -func (p *ExportTracesToDatasetRequest) field3Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I32Length() - return l -} - -func (p *ExportTracesToDatasetRequest) field4Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += p.Config.BLength() - return l -} - -func (p *ExportTracesToDatasetRequest) field5Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - return l -} - -func (p *ExportTracesToDatasetRequest) field6Length() int { +func (p *ListWorkspaceAnnotationsRequest) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() l += thrift.Binary.I64Length() return l } -func (p *ExportTracesToDatasetRequest) field7Length() int { +func (p *ListWorkspaceAnnotationsRequest) field2Length() int { l := 0 if p.IsSetPlatformType() { l += thrift.Binary.FieldBeginLength() @@ -10165,27 +9866,25 @@ func (p *ExportTracesToDatasetRequest) field7Length() int { return l } -func (p *ExportTracesToDatasetRequest) field8Length() int { +func (p *ListWorkspaceAnnotationsRequest) field3Length() int { l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.ExportType) + if p.IsSetAnnotationType() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.AnnotationType) + } return l } -func (p *ExportTracesToDatasetRequest) field9Length() int { +func (p *ListWorkspaceAnnotationsRequest) field4Length() int { l := 0 - if p.IsSetFieldMappings() { + if p.IsSetSpanListType() { l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.FieldMappings { - _ = v - l += v.BLength() - } + l += thrift.Binary.StringLengthNocopy(*p.SpanListType) } return l } -func (p *ExportTracesToDatasetRequest) field255Length() int { +func (p *ListWorkspaceAnnotationsRequest) field255Length() int { l := 0 if p.IsSetBase() { l += thrift.Binary.FieldBeginLength() @@ -10194,64 +9893,27 @@ func (p *ExportTracesToDatasetRequest) field255Length() int { return l } -func (p *ExportTracesToDatasetRequest) DeepCopy(s interface{}) error { - src, ok := s.(*ExportTracesToDatasetRequest) +func (p *ListWorkspaceAnnotationsRequest) DeepCopy(s interface{}) error { + src, ok := s.(*ListWorkspaceAnnotationsRequest) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } p.WorkspaceID = src.WorkspaceID - if src.SpanIds != nil { - p.SpanIds = make([]*SpanID, 0, len(src.SpanIds)) - for _, elem := range src.SpanIds { - var _elem *SpanID - if elem != nil { - _elem = &SpanID{} - if err := _elem.DeepCopy(elem); err != nil { - return err - } - } - - p.SpanIds = append(p.SpanIds, _elem) - } - } - - p.Category = src.Category - - var _config *DatasetConfig - if src.Config != nil { - _config = &DatasetConfig{} - if err := _config.DeepCopy(src.Config); err != nil { - return err - } - } - p.Config = _config - - p.StartTime = src.StartTime - - p.EndTime = src.EndTime - if src.PlatformType != nil { tmp := *src.PlatformType p.PlatformType = &tmp } - p.ExportType = src.ExportType - - if src.FieldMappings != nil { - p.FieldMappings = make([]*dataset0.FieldMapping, 0, len(src.FieldMappings)) - for _, elem := range src.FieldMappings { - var _elem *dataset0.FieldMapping - if elem != nil { - _elem = &dataset0.FieldMapping{} - if err := _elem.DeepCopy(elem); err != nil { - return err - } - } + if src.AnnotationType != nil { + tmp := *src.AnnotationType + p.AnnotationType = &tmp + } - p.FieldMappings = append(p.FieldMappings, _elem) - } + if src.SpanListType != nil { + tmp := *src.SpanListType + p.SpanListType = &tmp } var _base *base.Base @@ -10266,15 +9928,14 @@ func (p *ExportTracesToDatasetRequest) DeepCopy(s interface{}) error { return nil } -func (p *SpanID) FastRead(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsResponse) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetTraceID bool = false - var issetSpanID bool = false + var issetSimpleAnnotationList bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -10286,13 +9947,13 @@ func (p *SpanID) FastRead(buf []byte) (int, error) { } switch fieldId { case 1: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.LIST { l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetTraceID = true + issetSimpleAnnotationList = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -10300,14 +9961,13 @@ func (p *SpanID) FastRead(buf []byte) (int, error) { goto SkipFieldError } } - case 2: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField2(buf[offset:]) + case 255: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField255(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetSpanID = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -10324,131 +9984,172 @@ func (p *SpanID) FastRead(buf []byte) (int, error) { } } - if !issetTraceID { + if !issetSimpleAnnotationList { fieldId = 1 goto RequiredFieldNotSetError } - - if !issetSpanID { - fieldId = 2 - goto RequiredFieldNotSetError - } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SpanID[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListWorkspaceAnnotationsResponse[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_SpanID[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListWorkspaceAnnotationsResponse[fieldId])) } -func (p *SpanID) FastReadField1(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsResponse) FastReadField1(buf []byte) (int, error) { offset := 0 - var _field string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { return offset, err - } else { - offset += l - _field = v } - p.TraceID = _field + _field := make([]*annotation.SimpleAnnotationInfo, 0, size) + values := make([]annotation.SimpleAnnotationInfo, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.SimpleAnnotationList = _field return offset, nil } -func (p *SpanID) FastReadField2(buf []byte) (int, error) { +func (p *ListWorkspaceAnnotationsResponse) FastReadField255(buf []byte) (int, error) { offset := 0 - - var _field string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + _field := base.NewBaseResp() + if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = v } - p.SpanID = _field + p.BaseResp = _field return offset, nil } -func (p *SpanID) FastWrite(buf []byte) int { +func (p *ListWorkspaceAnnotationsResponse) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *SpanID) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *SpanID) BLength() int { +func (p *ListWorkspaceAnnotationsResponse) BLength() int { l := 0 if p != nil { l += p.field1Length() - l += p.field2Length() + l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *SpanID) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 1) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.TraceID) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.SimpleAnnotationList { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) return offset } -func (p *SpanID) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { +func (p *ListWorkspaceAnnotationsResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.SpanID) + if p.IsSetBaseResp() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + } return offset } -func (p *SpanID) field1Length() int { +func (p *ListWorkspaceAnnotationsResponse) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.TraceID) + l += thrift.Binary.ListBeginLength() + for _, v := range p.SimpleAnnotationList { + _ = v + l += v.BLength() + } return l } -func (p *SpanID) field2Length() int { +func (p *ListWorkspaceAnnotationsResponse) field255Length() int { l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.SpanID) + if p.IsSetBaseResp() { + l += thrift.Binary.FieldBeginLength() + l += p.BaseResp.BLength() + } return l } -func (p *SpanID) DeepCopy(s interface{}) error { - src, ok := s.(*SpanID) +func (p *ListWorkspaceAnnotationsResponse) DeepCopy(s interface{}) error { + src, ok := s.(*ListWorkspaceAnnotationsResponse) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - if src.TraceID != "" { - p.TraceID = kutils.StringDeepCopy(src.TraceID) - } + if src.SimpleAnnotationList != nil { + p.SimpleAnnotationList = make([]*annotation.SimpleAnnotationInfo, 0, len(src.SimpleAnnotationList)) + for _, elem := range src.SimpleAnnotationList { + var _elem *annotation.SimpleAnnotationInfo + if elem != nil { + _elem = &annotation.SimpleAnnotationInfo{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } - if src.SpanID != "" { - p.SpanID = kutils.StringDeepCopy(src.SpanID) + p.SimpleAnnotationList = append(p.SimpleAnnotationList, _elem) + } + } + + var _baseResp *base.BaseResp + if src.BaseResp != nil { + _baseResp = &base.BaseResp{} + if err := _baseResp.DeepCopy(src.BaseResp); err != nil { + return err + } } + p.BaseResp = _baseResp return nil } -func (p *DatasetConfig) FastRead(buf []byte) (int, error) { +func (p *ExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetIsNewDataset bool = false + var issetWorkspaceID bool = false + var issetSpanIds bool = false + var issetCategory bool = false + var issetConfig bool = false + var issetStartTime bool = false + var issetEndTime bool = false + var issetExportType bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -10460,13 +10161,13 @@ func (p *DatasetConfig) FastRead(buf []byte) (int, error) { } switch fieldId { case 1: - if fieldTypeId == thrift.BOOL { + if fieldTypeId == thrift.I64 { l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetIsNewDataset = true + issetWorkspaceID = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -10475,12 +10176,13 @@ func (p *DatasetConfig) FastRead(buf []byte) (int, error) { } } case 2: - if fieldTypeId == thrift.I64 { + if fieldTypeId == thrift.LIST { l, err = p.FastReadField2(buf[offset:]) offset += l if err != nil { goto ReadFieldError } + issetSpanIds = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -10489,12 +10191,13 @@ func (p *DatasetConfig) FastRead(buf []byte) (int, error) { } } case 3: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.I32 { l, err = p.FastReadField3(buf[offset:]) offset += l if err != nil { goto ReadFieldError } + issetCategory = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -10509,6 +10212,94 @@ func (p *DatasetConfig) FastRead(buf []byte) (int, error) { if err != nil { goto ReadFieldError } + issetConfig = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 5: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField5(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetStartTime = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 6: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField6(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetEndTime = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 7: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField7(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 8: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField8(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetExportType = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 9: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField9(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 255: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField255(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -10525,352 +10316,174 @@ func (p *DatasetConfig) FastRead(buf []byte) (int, error) { } } - if !issetIsNewDataset { + if !issetWorkspaceID { fieldId = 1 goto RequiredFieldNotSetError } + + if !issetSpanIds { + fieldId = 2 + goto RequiredFieldNotSetError + } + + if !issetCategory { + fieldId = 3 + goto RequiredFieldNotSetError + } + + if !issetConfig { + fieldId = 4 + goto RequiredFieldNotSetError + } + + if !issetStartTime { + fieldId = 5 + goto RequiredFieldNotSetError + } + + if !issetEndTime { + fieldId = 6 + goto RequiredFieldNotSetError + } + + if !issetExportType { + fieldId = 8 + goto RequiredFieldNotSetError + } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_DatasetConfig[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExportTracesToDatasetRequest[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_DatasetConfig[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ExportTracesToDatasetRequest[fieldId])) } -func (p *DatasetConfig) FastReadField1(buf []byte) (int, error) { +func (p *ExportTracesToDatasetRequest) FastReadField1(buf []byte) (int, error) { offset := 0 - var _field bool - if v, l, err := thrift.Binary.ReadBool(buf[offset:]); err != nil { + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { return offset, err } else { offset += l _field = v } - p.IsNewDataset = _field + p.WorkspaceID = _field return offset, nil } -func (p *DatasetConfig) FastReadField2(buf []byte) (int, error) { +func (p *ExportTracesToDatasetRequest) FastReadField2(buf []byte) (int, error) { offset := 0 - var _field *int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { return offset, err - } else { - offset += l - _field = &v } - p.DatasetID = _field + _field := make([]*SpanID, 0, size) + values := make([]SpanID, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.SpanIds = _field return offset, nil } -func (p *DatasetConfig) FastReadField3(buf []byte) (int, error) { +func (p *ExportTracesToDatasetRequest) FastReadField3(buf []byte) (int, error) { offset := 0 - var _field *string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + var _field dataset.DatasetCategory + if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = &v + + _field = dataset.DatasetCategory(v) } - p.DatasetName = _field + p.Category = _field return offset, nil } -func (p *DatasetConfig) FastReadField4(buf []byte) (int, error) { +func (p *ExportTracesToDatasetRequest) FastReadField4(buf []byte) (int, error) { offset := 0 - _field := dataset0.NewDatasetSchema() + _field := NewDatasetConfig() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l } - p.DatasetSchema = _field + p.Config = _field return offset, nil } -func (p *DatasetConfig) FastWrite(buf []byte) int { - return p.FastWriteNocopy(buf, nil) -} - -func (p *DatasetConfig) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) FastReadField5(buf []byte) (int, error) { offset := 0 - if p != nil { - offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField2(buf[offset:], w) - offset += p.fastWriteField3(buf[offset:], w) - offset += p.fastWriteField4(buf[offset:], w) + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v } - offset += thrift.Binary.WriteFieldStop(buf[offset:]) - return offset + p.StartTime = _field + return offset, nil } -func (p *DatasetConfig) BLength() int { - l := 0 - if p != nil { - l += p.field1Length() - l += p.field2Length() - l += p.field3Length() - l += p.field4Length() +func (p *ExportTracesToDatasetRequest) FastReadField6(buf []byte) (int, error) { + offset := 0 + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v } - l += thrift.Binary.FieldStopLength() - return l + p.EndTime = _field + return offset, nil } -func (p *DatasetConfig) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) FastReadField7(buf []byte) (int, error) { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.BOOL, 1) - offset += thrift.Binary.WriteBool(buf[offset:], p.IsNewDataset) - return offset -} - -func (p *DatasetConfig) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetDatasetID() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 2) - offset += thrift.Binary.WriteI64(buf[offset:], *p.DatasetID) - } - return offset -} - -func (p *DatasetConfig) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetDatasetName() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 3) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.DatasetName) - } - return offset -} - -func (p *DatasetConfig) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetDatasetSchema() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 4) - offset += p.DatasetSchema.FastWriteNocopy(buf[offset:], w) - } - return offset -} - -func (p *DatasetConfig) field1Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.BoolLength() - return l -} - -func (p *DatasetConfig) field2Length() int { - l := 0 - if p.IsSetDatasetID() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - } - return l -} - -func (p *DatasetConfig) field3Length() int { - l := 0 - if p.IsSetDatasetName() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(*p.DatasetName) - } - return l -} - -func (p *DatasetConfig) field4Length() int { - l := 0 - if p.IsSetDatasetSchema() { - l += thrift.Binary.FieldBeginLength() - l += p.DatasetSchema.BLength() - } - return l -} -func (p *DatasetConfig) DeepCopy(s interface{}) error { - src, ok := s.(*DatasetConfig) - if !ok { - return fmt.Errorf("%T's type not matched %T", s, p) - } - - p.IsNewDataset = src.IsNewDataset - - if src.DatasetID != nil { - tmp := *src.DatasetID - p.DatasetID = &tmp - } - - if src.DatasetName != nil { - var tmp string - if *src.DatasetName != "" { - tmp = kutils.StringDeepCopy(*src.DatasetName) - } - p.DatasetName = &tmp - } - - var _datasetSchema *dataset0.DatasetSchema - if src.DatasetSchema != nil { - _datasetSchema = &dataset0.DatasetSchema{} - if err := _datasetSchema.DeepCopy(src.DatasetSchema); err != nil { - return err - } - } - p.DatasetSchema = _datasetSchema - - return nil -} - -func (p *ExportTracesToDatasetResponse) FastRead(buf []byte) (int, error) { - - var err error - var offset int - var l int - var fieldTypeId thrift.TType - var fieldId int16 - for { - fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + var _field *common.PlatformType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { offset += l - if err != nil { - goto ReadFieldBeginError - } - if fieldTypeId == thrift.STOP { - break - } - switch fieldId { - case 1: - if fieldTypeId == thrift.I32 { - l, err = p.FastReadField1(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 2: - if fieldTypeId == thrift.LIST { - l, err = p.FastReadField2(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 3: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField3(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 4: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField4(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 255: - if fieldTypeId == thrift.STRUCT { - l, err = p.FastReadField255(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 256: - if fieldTypeId == thrift.I32 { - l, err = p.FastReadField256(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 257: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField257(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - default: - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } + _field = &v } - + p.PlatformType = _field return offset, nil -ReadFieldBeginError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) -ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExportTracesToDatasetResponse[fieldId]), err) -SkipFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) } -func (p *ExportTracesToDatasetResponse) FastReadField1(buf []byte) (int, error) { +func (p *ExportTracesToDatasetRequest) FastReadField8(buf []byte) (int, error) { offset := 0 - var _field *int32 - if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { + var _field dataset0.ExportType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = &v + _field = v } - p.SuccessCount = _field + p.ExportType = _field return offset, nil } -func (p *ExportTracesToDatasetResponse) FastReadField2(buf []byte) (int, error) { +func (p *ExportTracesToDatasetRequest) FastReadField9(buf []byte) (int, error) { offset := 0 _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) @@ -10878,8 +10491,8 @@ func (p *ExportTracesToDatasetResponse) FastReadField2(buf []byte) (int, error) if err != nil { return offset, err } - _field := make([]*dataset.ItemErrorGroup, 0, size) - values := make([]dataset.ItemErrorGroup, size) + _field := make([]*dataset0.FieldMapping, 0, size) + values := make([]dataset0.FieldMapping, size) for i := 0; i < size; i++ { _elem := &values[i] _elem.InitDefault() @@ -10891,327 +10504,317 @@ func (p *ExportTracesToDatasetResponse) FastReadField2(buf []byte) (int, error) _field = append(_field, _elem) } - p.Errors = _field + p.FieldMappings = _field return offset, nil } -func (p *ExportTracesToDatasetResponse) FastReadField3(buf []byte) (int, error) { +func (p *ExportTracesToDatasetRequest) FastReadField255(buf []byte) (int, error) { offset := 0 - - var _field *int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + _field := base.NewBase() + if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = &v - } - p.DatasetID = _field - return offset, nil -} - -func (p *ExportTracesToDatasetResponse) FastReadField4(buf []byte) (int, error) { - offset := 0 - - var _field *string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = &v - } - p.DatasetName = _field - return offset, nil -} - -func (p *ExportTracesToDatasetResponse) FastReadField255(buf []byte) (int, error) { - offset := 0 - _field := base.NewBaseResp() - if l, err := _field.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - p.BaseResp = _field - return offset, nil -} - -func (p *ExportTracesToDatasetResponse) FastReadField256(buf []byte) (int, error) { - offset := 0 - - var _field *int32 - if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = &v - } - p.Code = _field - return offset, nil -} - -func (p *ExportTracesToDatasetResponse) FastReadField257(buf []byte) (int, error) { - offset := 0 - - var _field *string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = &v } - p.Msg = _field + p.Base = _field return offset, nil } -func (p *ExportTracesToDatasetResponse) FastWrite(buf []byte) int { +func (p *ExportTracesToDatasetRequest) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ExportTracesToDatasetResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField3(buf[offset:], w) - offset += p.fastWriteField256(buf[offset:], w) + offset += p.fastWriteField5(buf[offset:], w) + offset += p.fastWriteField6(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField7(buf[offset:], w) + offset += p.fastWriteField8(buf[offset:], w) + offset += p.fastWriteField9(buf[offset:], w) offset += p.fastWriteField255(buf[offset:], w) - offset += p.fastWriteField257(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *ExportTracesToDatasetResponse) BLength() int { +func (p *ExportTracesToDatasetRequest) BLength() int { l := 0 if p != nil { l += p.field1Length() l += p.field2Length() l += p.field3Length() l += p.field4Length() + l += p.field5Length() + l += p.field6Length() + l += p.field7Length() + l += p.field8Length() + l += p.field9Length() l += p.field255Length() - l += p.field256Length() - l += p.field257Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *ExportTracesToDatasetResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetSuccessCount() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 1) - offset += thrift.Binary.WriteI32(buf[offset:], *p.SuccessCount) - } + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) + offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) return offset } -func (p *ExportTracesToDatasetResponse) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetErrors() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.Errors { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.SpanIds { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) return offset } -func (p *ExportTracesToDatasetResponse) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetDatasetID() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 3) - offset += thrift.Binary.WriteI64(buf[offset:], *p.DatasetID) - } + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 3) + offset += thrift.Binary.WriteI32(buf[offset:], int32(p.Category)) return offset } -func (p *ExportTracesToDatasetResponse) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetDatasetName() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 4) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.DatasetName) - } + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 4) + offset += p.Config.FastWriteNocopy(buf[offset:], w) return offset } -func (p *ExportTracesToDatasetResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetBaseResp() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 5) + offset += thrift.Binary.WriteI64(buf[offset:], p.StartTime) + return offset +} + +func (p *ExportTracesToDatasetRequest) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 6) + offset += thrift.Binary.WriteI64(buf[offset:], p.EndTime) + return offset +} + +func (p *ExportTracesToDatasetRequest) fastWriteField7(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetPlatformType() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 7) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.PlatformType) } return offset } -func (p *ExportTracesToDatasetResponse) fastWriteField256(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) fastWriteField8(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetCode() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 256) - offset += thrift.Binary.WriteI32(buf[offset:], *p.Code) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 8) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.ExportType) + return offset +} + +func (p *ExportTracesToDatasetRequest) fastWriteField9(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetFieldMappings() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 9) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.FieldMappings { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) } return offset } -func (p *ExportTracesToDatasetResponse) fastWriteField257(buf []byte, w thrift.NocopyWriter) int { +func (p *ExportTracesToDatasetRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetMsg() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 257) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.Msg) + if p.IsSetBase() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.Base.FastWriteNocopy(buf[offset:], w) } return offset } -func (p *ExportTracesToDatasetResponse) field1Length() int { +func (p *ExportTracesToDatasetRequest) field1Length() int { l := 0 - if p.IsSetSuccessCount() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I32Length() - } + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() return l } -func (p *ExportTracesToDatasetResponse) field2Length() int { +func (p *ExportTracesToDatasetRequest) field2Length() int { l := 0 - if p.IsSetErrors() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.Errors { - _ = v - l += v.BLength() - } + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.SpanIds { + _ = v + l += v.BLength() } return l } -func (p *ExportTracesToDatasetResponse) field3Length() int { +func (p *ExportTracesToDatasetRequest) field3Length() int { l := 0 - if p.IsSetDatasetID() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - } + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I32Length() return l } -func (p *ExportTracesToDatasetResponse) field4Length() int { +func (p *ExportTracesToDatasetRequest) field4Length() int { l := 0 - if p.IsSetDatasetName() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(*p.DatasetName) - } + l += thrift.Binary.FieldBeginLength() + l += p.Config.BLength() return l } -func (p *ExportTracesToDatasetResponse) field255Length() int { +func (p *ExportTracesToDatasetRequest) field5Length() int { l := 0 - if p.IsSetBaseResp() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *ExportTracesToDatasetRequest) field6Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *ExportTracesToDatasetRequest) field7Length() int { + l := 0 + if p.IsSetPlatformType() { l += thrift.Binary.FieldBeginLength() - l += p.BaseResp.BLength() + l += thrift.Binary.StringLengthNocopy(*p.PlatformType) } return l } -func (p *ExportTracesToDatasetResponse) field256Length() int { +func (p *ExportTracesToDatasetRequest) field8Length() int { l := 0 - if p.IsSetCode() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(p.ExportType) + return l +} + +func (p *ExportTracesToDatasetRequest) field9Length() int { + l := 0 + if p.IsSetFieldMappings() { l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I32Length() + l += thrift.Binary.ListBeginLength() + for _, v := range p.FieldMappings { + _ = v + l += v.BLength() + } } return l } -func (p *ExportTracesToDatasetResponse) field257Length() int { +func (p *ExportTracesToDatasetRequest) field255Length() int { l := 0 - if p.IsSetMsg() { + if p.IsSetBase() { l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(*p.Msg) + l += p.Base.BLength() } return l } -func (p *ExportTracesToDatasetResponse) DeepCopy(s interface{}) error { - src, ok := s.(*ExportTracesToDatasetResponse) +func (p *ExportTracesToDatasetRequest) DeepCopy(s interface{}) error { + src, ok := s.(*ExportTracesToDatasetRequest) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - if src.SuccessCount != nil { - tmp := *src.SuccessCount - p.SuccessCount = &tmp - } + p.WorkspaceID = src.WorkspaceID - if src.Errors != nil { - p.Errors = make([]*dataset.ItemErrorGroup, 0, len(src.Errors)) - for _, elem := range src.Errors { - var _elem *dataset.ItemErrorGroup + if src.SpanIds != nil { + p.SpanIds = make([]*SpanID, 0, len(src.SpanIds)) + for _, elem := range src.SpanIds { + var _elem *SpanID if elem != nil { - _elem = &dataset.ItemErrorGroup{} + _elem = &SpanID{} if err := _elem.DeepCopy(elem); err != nil { return err } } - p.Errors = append(p.Errors, _elem) + p.SpanIds = append(p.SpanIds, _elem) } } - if src.DatasetID != nil { - tmp := *src.DatasetID - p.DatasetID = &tmp - } + p.Category = src.Category - if src.DatasetName != nil { - var tmp string - if *src.DatasetName != "" { - tmp = kutils.StringDeepCopy(*src.DatasetName) + var _config *DatasetConfig + if src.Config != nil { + _config = &DatasetConfig{} + if err := _config.DeepCopy(src.Config); err != nil { + return err } - p.DatasetName = &tmp } + p.Config = _config - var _baseResp *base.BaseResp - if src.BaseResp != nil { - _baseResp = &base.BaseResp{} - if err := _baseResp.DeepCopy(src.BaseResp); err != nil { - return err - } + p.StartTime = src.StartTime + + p.EndTime = src.EndTime + + if src.PlatformType != nil { + tmp := *src.PlatformType + p.PlatformType = &tmp } - p.BaseResp = _baseResp - if src.Code != nil { - tmp := *src.Code - p.Code = &tmp + p.ExportType = src.ExportType + + if src.FieldMappings != nil { + p.FieldMappings = make([]*dataset0.FieldMapping, 0, len(src.FieldMappings)) + for _, elem := range src.FieldMappings { + var _elem *dataset0.FieldMapping + if elem != nil { + _elem = &dataset0.FieldMapping{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.FieldMappings = append(p.FieldMappings, _elem) + } } - if src.Msg != nil { - var tmp string - if *src.Msg != "" { - tmp = kutils.StringDeepCopy(*src.Msg) + var _base *base.Base + if src.Base != nil { + _base = &base.Base{} + if err := _base.DeepCopy(src.Base); err != nil { + return err } - p.Msg = &tmp } + p.Base = _base return nil } -func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { +func (p *SpanID) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetWorkspaceID bool = false - var issetSpanIds bool = false - var issetCategory bool = false - var issetConfig bool = false - var issetStartTime bool = false - var issetEndTime bool = false - var issetExportType bool = false + var issetTraceID bool = false + var issetSpanID bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -11223,13 +10826,13 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) } switch fieldId { case 1: - if fieldTypeId == thrift.I64 { + if fieldTypeId == thrift.STRING { l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetWorkspaceID = true + issetTraceID = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11238,13 +10841,13 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) } } case 2: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.STRING { l, err = p.FastReadField2(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetSpanIds = true + issetSpanID = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11252,14 +10855,158 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) goto SkipFieldError } } - case 3: - if fieldTypeId == thrift.I32 { - l, err = p.FastReadField3(buf[offset:]) + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + if !issetTraceID { + fieldId = 1 + goto RequiredFieldNotSetError + } + + if !issetSpanID { + fieldId = 2 + goto RequiredFieldNotSetError + } + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SpanID[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_SpanID[fieldId])) +} + +func (p *SpanID) FastReadField1(buf []byte) (int, error) { + offset := 0 + + var _field string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.TraceID = _field + return offset, nil +} + +func (p *SpanID) FastReadField2(buf []byte) (int, error) { + offset := 0 + + var _field string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.SpanID = _field + return offset, nil +} + +func (p *SpanID) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *SpanID) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *SpanID) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *SpanID) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 1) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.TraceID) + return offset +} + +func (p *SpanID) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.SpanID) + return offset +} + +func (p *SpanID) field1Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(p.TraceID) + return l +} + +func (p *SpanID) field2Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(p.SpanID) + return l +} + +func (p *SpanID) DeepCopy(s interface{}) error { + src, ok := s.(*SpanID) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + if src.TraceID != "" { + p.TraceID = kutils.StringDeepCopy(src.TraceID) + } + + if src.SpanID != "" { + p.SpanID = kutils.StringDeepCopy(src.SpanID) + } + + return nil +} + +func (p *DatasetConfig) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + var issetIsNewDataset bool = false + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.BOOL { + l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetCategory = true + issetIsNewDataset = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11267,14 +11014,13 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) goto SkipFieldError } } - case 4: - if fieldTypeId == thrift.STRUCT { - l, err = p.FastReadField4(buf[offset:]) - offset += l - if err != nil { + case 2: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { goto ReadFieldError } - issetConfig = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11282,14 +11028,13 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) goto SkipFieldError } } - case 5: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField5(buf[offset:]) + case 3: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField3(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetStartTime = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11297,14 +11042,13 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) goto SkipFieldError } } - case 6: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField6(buf[offset:]) + case 4: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField4(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetEndTime = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11312,9 +11056,233 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) goto SkipFieldError } } - case 7: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField7(buf[offset:]) + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + if !issetIsNewDataset { + fieldId = 1 + goto RequiredFieldNotSetError + } + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_DatasetConfig[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_DatasetConfig[fieldId])) +} + +func (p *DatasetConfig) FastReadField1(buf []byte) (int, error) { + offset := 0 + + var _field bool + if v, l, err := thrift.Binary.ReadBool(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.IsNewDataset = _field + return offset, nil +} + +func (p *DatasetConfig) FastReadField2(buf []byte) (int, error) { + offset := 0 + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.DatasetID = _field + return offset, nil +} + +func (p *DatasetConfig) FastReadField3(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.DatasetName = _field + return offset, nil +} + +func (p *DatasetConfig) FastReadField4(buf []byte) (int, error) { + offset := 0 + _field := dataset0.NewDatasetSchema() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.DatasetSchema = _field + return offset, nil +} + +func (p *DatasetConfig) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *DatasetConfig) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) + offset += p.fastWriteField4(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *DatasetConfig) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + l += p.field3Length() + l += p.field4Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *DatasetConfig) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.BOOL, 1) + offset += thrift.Binary.WriteBool(buf[offset:], p.IsNewDataset) + return offset +} + +func (p *DatasetConfig) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetDatasetID() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 2) + offset += thrift.Binary.WriteI64(buf[offset:], *p.DatasetID) + } + return offset +} + +func (p *DatasetConfig) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetDatasetName() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 3) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.DatasetName) + } + return offset +} + +func (p *DatasetConfig) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetDatasetSchema() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 4) + offset += p.DatasetSchema.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *DatasetConfig) field1Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.BoolLength() + return l +} + +func (p *DatasetConfig) field2Length() int { + l := 0 + if p.IsSetDatasetID() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + } + return l +} + +func (p *DatasetConfig) field3Length() int { + l := 0 + if p.IsSetDatasetName() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.DatasetName) + } + return l +} + +func (p *DatasetConfig) field4Length() int { + l := 0 + if p.IsSetDatasetSchema() { + l += thrift.Binary.FieldBeginLength() + l += p.DatasetSchema.BLength() + } + return l +} + +func (p *DatasetConfig) DeepCopy(s interface{}) error { + src, ok := s.(*DatasetConfig) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + p.IsNewDataset = src.IsNewDataset + + if src.DatasetID != nil { + tmp := *src.DatasetID + p.DatasetID = &tmp + } + + if src.DatasetName != nil { + var tmp string + if *src.DatasetName != "" { + tmp = kutils.StringDeepCopy(*src.DatasetName) + } + p.DatasetName = &tmp + } + + var _datasetSchema *dataset0.DatasetSchema + if src.DatasetSchema != nil { + _datasetSchema = &dataset0.DatasetSchema{} + if err := _datasetSchema.DeepCopy(src.DatasetSchema); err != nil { + return err + } + } + p.DatasetSchema = _datasetSchema + + return nil +} + +func (p *ExportTracesToDatasetResponse) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.I32 { + l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError @@ -11326,14 +11294,13 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) goto SkipFieldError } } - case 8: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField8(buf[offset:]) + case 2: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField2(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetExportType = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11341,13 +11308,2085 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) goto SkipFieldError } } - case 9: - if fieldTypeId == thrift.LIST { - l, err = p.FastReadField9(buf[offset:]) + case 3: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField3(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 4: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField4(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 255: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField255(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 256: + if fieldTypeId == thrift.I32 { + l, err = p.FastReadField256(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 257: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField257(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExportTracesToDatasetResponse[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +} + +func (p *ExportTracesToDatasetResponse) FastReadField1(buf []byte) (int, error) { + offset := 0 + + var _field *int32 + if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.SuccessCount = _field + return offset, nil +} + +func (p *ExportTracesToDatasetResponse) FastReadField2(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*dataset.ItemErrorGroup, 0, size) + values := make([]dataset.ItemErrorGroup, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.Errors = _field + return offset, nil +} + +func (p *ExportTracesToDatasetResponse) FastReadField3(buf []byte) (int, error) { + offset := 0 + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.DatasetID = _field + return offset, nil +} + +func (p *ExportTracesToDatasetResponse) FastReadField4(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.DatasetName = _field + return offset, nil +} + +func (p *ExportTracesToDatasetResponse) FastReadField255(buf []byte) (int, error) { + offset := 0 + _field := base.NewBaseResp() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.BaseResp = _field + return offset, nil +} + +func (p *ExportTracesToDatasetResponse) FastReadField256(buf []byte) (int, error) { + offset := 0 + + var _field *int32 + if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.Code = _field + return offset, nil +} + +func (p *ExportTracesToDatasetResponse) FastReadField257(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.Msg = _field + return offset, nil +} + +func (p *ExportTracesToDatasetResponse) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *ExportTracesToDatasetResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) + offset += p.fastWriteField256(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField255(buf[offset:], w) + offset += p.fastWriteField257(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *ExportTracesToDatasetResponse) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + l += p.field3Length() + l += p.field4Length() + l += p.field255Length() + l += p.field256Length() + l += p.field257Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *ExportTracesToDatasetResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetSuccessCount() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 1) + offset += thrift.Binary.WriteI32(buf[offset:], *p.SuccessCount) + } + return offset +} + +func (p *ExportTracesToDatasetResponse) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetErrors() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.Errors { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + +func (p *ExportTracesToDatasetResponse) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetDatasetID() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 3) + offset += thrift.Binary.WriteI64(buf[offset:], *p.DatasetID) + } + return offset +} + +func (p *ExportTracesToDatasetResponse) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetDatasetName() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 4) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.DatasetName) + } + return offset +} + +func (p *ExportTracesToDatasetResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetBaseResp() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *ExportTracesToDatasetResponse) fastWriteField256(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetCode() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 256) + offset += thrift.Binary.WriteI32(buf[offset:], *p.Code) + } + return offset +} + +func (p *ExportTracesToDatasetResponse) fastWriteField257(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetMsg() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 257) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.Msg) + } + return offset +} + +func (p *ExportTracesToDatasetResponse) field1Length() int { + l := 0 + if p.IsSetSuccessCount() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I32Length() + } + return l +} + +func (p *ExportTracesToDatasetResponse) field2Length() int { + l := 0 + if p.IsSetErrors() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.Errors { + _ = v + l += v.BLength() + } + } + return l +} + +func (p *ExportTracesToDatasetResponse) field3Length() int { + l := 0 + if p.IsSetDatasetID() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + } + return l +} + +func (p *ExportTracesToDatasetResponse) field4Length() int { + l := 0 + if p.IsSetDatasetName() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.DatasetName) + } + return l +} + +func (p *ExportTracesToDatasetResponse) field255Length() int { + l := 0 + if p.IsSetBaseResp() { + l += thrift.Binary.FieldBeginLength() + l += p.BaseResp.BLength() + } + return l +} + +func (p *ExportTracesToDatasetResponse) field256Length() int { + l := 0 + if p.IsSetCode() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I32Length() + } + return l +} + +func (p *ExportTracesToDatasetResponse) field257Length() int { + l := 0 + if p.IsSetMsg() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.Msg) + } + return l +} + +func (p *ExportTracesToDatasetResponse) DeepCopy(s interface{}) error { + src, ok := s.(*ExportTracesToDatasetResponse) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + if src.SuccessCount != nil { + tmp := *src.SuccessCount + p.SuccessCount = &tmp + } + + if src.Errors != nil { + p.Errors = make([]*dataset.ItemErrorGroup, 0, len(src.Errors)) + for _, elem := range src.Errors { + var _elem *dataset.ItemErrorGroup + if elem != nil { + _elem = &dataset.ItemErrorGroup{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.Errors = append(p.Errors, _elem) + } + } + + if src.DatasetID != nil { + tmp := *src.DatasetID + p.DatasetID = &tmp + } + + if src.DatasetName != nil { + var tmp string + if *src.DatasetName != "" { + tmp = kutils.StringDeepCopy(*src.DatasetName) + } + p.DatasetName = &tmp + } + + var _baseResp *base.BaseResp + if src.BaseResp != nil { + _baseResp = &base.BaseResp{} + if err := _baseResp.DeepCopy(src.BaseResp); err != nil { + return err + } + } + p.BaseResp = _baseResp + + if src.Code != nil { + tmp := *src.Code + p.Code = &tmp + } + + if src.Msg != nil { + var tmp string + if *src.Msg != "" { + tmp = kutils.StringDeepCopy(*src.Msg) + } + p.Msg = &tmp + } + + return nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + var issetWorkspaceID bool = false + var issetSpanIds bool = false + var issetCategory bool = false + var issetConfig bool = false + var issetStartTime bool = false + var issetEndTime bool = false + var issetExportType bool = false + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetWorkspaceID = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetSpanIds = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 3: + if fieldTypeId == thrift.I32 { + l, err = p.FastReadField3(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetCategory = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 4: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField4(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetConfig = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 5: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField5(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetStartTime = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 6: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField6(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetEndTime = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 7: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField7(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 8: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField8(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetExportType = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 9: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField9(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 10: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField10(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 11: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField11(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 255: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField255(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + if !issetWorkspaceID { + fieldId = 1 + goto RequiredFieldNotSetError + } + + if !issetSpanIds { + fieldId = 2 + goto RequiredFieldNotSetError + } + + if !issetCategory { + fieldId = 3 + goto RequiredFieldNotSetError + } + + if !issetConfig { + fieldId = 4 + goto RequiredFieldNotSetError + } + + if !issetStartTime { + fieldId = 5 + goto RequiredFieldNotSetError + } + + if !issetEndTime { + fieldId = 6 + goto RequiredFieldNotSetError + } + + if !issetExportType { + fieldId = 8 + goto RequiredFieldNotSetError + } + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_PreviewExportTracesToDatasetRequest[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_PreviewExportTracesToDatasetRequest[fieldId])) +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField1(buf []byte) (int, error) { + offset := 0 + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.WorkspaceID = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField2(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*SpanID, 0, size) + values := make([]SpanID, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.SpanIds = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField3(buf []byte) (int, error) { + offset := 0 + + var _field dataset.DatasetCategory + if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + + _field = dataset.DatasetCategory(v) + } + p.Category = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField4(buf []byte) (int, error) { + offset := 0 + _field := NewDatasetConfig() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.Config = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField5(buf []byte) (int, error) { + offset := 0 + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.StartTime = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField6(buf []byte) (int, error) { + offset := 0 + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.EndTime = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField7(buf []byte) (int, error) { + offset := 0 + + var _field *common.PlatformType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.PlatformType = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField8(buf []byte) (int, error) { + offset := 0 + + var _field dataset0.ExportType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.ExportType = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField9(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*dataset0.FieldMapping, 0, size) + values := make([]dataset0.FieldMapping, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.FieldMappings = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField10(buf []byte) (int, error) { + offset := 0 + _field := filter.NewSpanFilterFields() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.SpanFilters = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField11(buf []byte) (int, error) { + offset := 0 + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.Limit = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastReadField255(buf []byte) (int, error) { + offset := 0 + _field := base.NewBase() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.Base = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetRequest) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *PreviewExportTracesToDatasetRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField5(buf[offset:], w) + offset += p.fastWriteField6(buf[offset:], w) + offset += p.fastWriteField11(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) + offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField7(buf[offset:], w) + offset += p.fastWriteField8(buf[offset:], w) + offset += p.fastWriteField9(buf[offset:], w) + offset += p.fastWriteField10(buf[offset:], w) + offset += p.fastWriteField255(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + l += p.field3Length() + l += p.field4Length() + l += p.field5Length() + l += p.field6Length() + l += p.field7Length() + l += p.field8Length() + l += p.field9Length() + l += p.field10Length() + l += p.field11Length() + l += p.field255Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) + offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.SpanIds { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 3) + offset += thrift.Binary.WriteI32(buf[offset:], int32(p.Category)) + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 4) + offset += p.Config.FastWriteNocopy(buf[offset:], w) + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 5) + offset += thrift.Binary.WriteI64(buf[offset:], p.StartTime) + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 6) + offset += thrift.Binary.WriteI64(buf[offset:], p.EndTime) + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField7(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetPlatformType() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 7) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.PlatformType) + } + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField8(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 8) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.ExportType) + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField9(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetFieldMappings() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 9) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.FieldMappings { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField10(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetSpanFilters() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 10) + offset += p.SpanFilters.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField11(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetLimit() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 11) + offset += thrift.Binary.WriteI64(buf[offset:], *p.Limit) + } + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetBase() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.Base.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *PreviewExportTracesToDatasetRequest) field1Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field2Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.SpanIds { + _ = v + l += v.BLength() + } + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field3Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I32Length() + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field4Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += p.Config.BLength() + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field5Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field6Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field7Length() int { + l := 0 + if p.IsSetPlatformType() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.PlatformType) + } + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field8Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(p.ExportType) + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field9Length() int { + l := 0 + if p.IsSetFieldMappings() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.FieldMappings { + _ = v + l += v.BLength() + } + } + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field10Length() int { + l := 0 + if p.IsSetSpanFilters() { + l += thrift.Binary.FieldBeginLength() + l += p.SpanFilters.BLength() + } + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field11Length() int { + l := 0 + if p.IsSetLimit() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + } + return l +} + +func (p *PreviewExportTracesToDatasetRequest) field255Length() int { + l := 0 + if p.IsSetBase() { + l += thrift.Binary.FieldBeginLength() + l += p.Base.BLength() + } + return l +} + +func (p *PreviewExportTracesToDatasetRequest) DeepCopy(s interface{}) error { + src, ok := s.(*PreviewExportTracesToDatasetRequest) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + p.WorkspaceID = src.WorkspaceID + + if src.SpanIds != nil { + p.SpanIds = make([]*SpanID, 0, len(src.SpanIds)) + for _, elem := range src.SpanIds { + var _elem *SpanID + if elem != nil { + _elem = &SpanID{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.SpanIds = append(p.SpanIds, _elem) + } + } + + p.Category = src.Category + + var _config *DatasetConfig + if src.Config != nil { + _config = &DatasetConfig{} + if err := _config.DeepCopy(src.Config); err != nil { + return err + } + } + p.Config = _config + + p.StartTime = src.StartTime + + p.EndTime = src.EndTime + + if src.PlatformType != nil { + tmp := *src.PlatformType + p.PlatformType = &tmp + } + + p.ExportType = src.ExportType + + if src.FieldMappings != nil { + p.FieldMappings = make([]*dataset0.FieldMapping, 0, len(src.FieldMappings)) + for _, elem := range src.FieldMappings { + var _elem *dataset0.FieldMapping + if elem != nil { + _elem = &dataset0.FieldMapping{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.FieldMappings = append(p.FieldMappings, _elem) + } + } + + var _spanFilters *filter.SpanFilterFields + if src.SpanFilters != nil { + _spanFilters = &filter.SpanFilterFields{} + if err := _spanFilters.DeepCopy(src.SpanFilters); err != nil { + return err + } + } + p.SpanFilters = _spanFilters + + if src.Limit != nil { + tmp := *src.Limit + p.Limit = &tmp + } + + var _base *base.Base + if src.Base != nil { + _base = &base.Base{} + if err := _base.DeepCopy(src.Base); err != nil { + return err + } + } + p.Base = _base + + return nil +} + +func (p *PreviewExportTracesToDatasetResponse) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 255: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField255(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 256: + if fieldTypeId == thrift.I32 { + l, err = p.FastReadField256(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 257: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField257(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_PreviewExportTracesToDatasetResponse[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +} + +func (p *PreviewExportTracesToDatasetResponse) FastReadField1(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*dataset0.Item, 0, size) + values := make([]dataset0.Item, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.Items = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetResponse) FastReadField2(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*dataset.ItemErrorGroup, 0, size) + values := make([]dataset.ItemErrorGroup, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.Errors = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetResponse) FastReadField255(buf []byte) (int, error) { + offset := 0 + _field := base.NewBaseResp() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.BaseResp = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetResponse) FastReadField256(buf []byte) (int, error) { + offset := 0 + + var _field *int32 + if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.Code = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetResponse) FastReadField257(buf []byte) (int, error) { + offset := 0 + + var _field *string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.Msg = _field + return offset, nil +} + +func (p *PreviewExportTracesToDatasetResponse) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *PreviewExportTracesToDatasetResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField256(buf[offset:], w) + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField255(buf[offset:], w) + offset += p.fastWriteField257(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *PreviewExportTracesToDatasetResponse) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + l += p.field255Length() + l += p.field256Length() + l += p.field257Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *PreviewExportTracesToDatasetResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetItems() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.Items { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + +func (p *PreviewExportTracesToDatasetResponse) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetErrors() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.Errors { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + +func (p *PreviewExportTracesToDatasetResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetBaseResp() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *PreviewExportTracesToDatasetResponse) fastWriteField256(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetCode() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 256) + offset += thrift.Binary.WriteI32(buf[offset:], *p.Code) + } + return offset +} + +func (p *PreviewExportTracesToDatasetResponse) fastWriteField257(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetMsg() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 257) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.Msg) + } + return offset +} + +func (p *PreviewExportTracesToDatasetResponse) field1Length() int { + l := 0 + if p.IsSetItems() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.Items { + _ = v + l += v.BLength() + } + } + return l +} + +func (p *PreviewExportTracesToDatasetResponse) field2Length() int { + l := 0 + if p.IsSetErrors() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.Errors { + _ = v + l += v.BLength() + } + } + return l +} + +func (p *PreviewExportTracesToDatasetResponse) field255Length() int { + l := 0 + if p.IsSetBaseResp() { + l += thrift.Binary.FieldBeginLength() + l += p.BaseResp.BLength() + } + return l +} + +func (p *PreviewExportTracesToDatasetResponse) field256Length() int { + l := 0 + if p.IsSetCode() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I32Length() + } + return l +} + +func (p *PreviewExportTracesToDatasetResponse) field257Length() int { + l := 0 + if p.IsSetMsg() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.Msg) + } + return l +} + +func (p *PreviewExportTracesToDatasetResponse) DeepCopy(s interface{}) error { + src, ok := s.(*PreviewExportTracesToDatasetResponse) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + if src.Items != nil { + p.Items = make([]*dataset0.Item, 0, len(src.Items)) + for _, elem := range src.Items { + var _elem *dataset0.Item + if elem != nil { + _elem = &dataset0.Item{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.Items = append(p.Items, _elem) + } + } + + if src.Errors != nil { + p.Errors = make([]*dataset.ItemErrorGroup, 0, len(src.Errors)) + for _, elem := range src.Errors { + var _elem *dataset.ItemErrorGroup + if elem != nil { + _elem = &dataset.ItemErrorGroup{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.Errors = append(p.Errors, _elem) + } + } + + var _baseResp *base.BaseResp + if src.BaseResp != nil { + _baseResp = &base.BaseResp{} + if err := _baseResp.DeepCopy(src.BaseResp); err != nil { + return err + } + } + p.BaseResp = _baseResp + + if src.Code != nil { + tmp := *src.Code + p.Code = &tmp + } + + if src.Msg != nil { + var tmp string + if *src.Msg != "" { + tmp = kutils.StringDeepCopy(*src.Msg) + } + p.Msg = &tmp + } + + return nil +} + +func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + var issetWorkspaceID bool = false + var issetAnnotationID bool = false + var issetSpanID bool = false + var issetStartTime bool = false + var issetCorrection bool = false + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetWorkspaceID = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetAnnotationID = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 3: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField3(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetSpanID = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 4: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField4(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetStartTime = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 5: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField5(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetCorrection = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 6: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField6(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 255: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField255(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + if !issetWorkspaceID { + fieldId = 1 + goto RequiredFieldNotSetError + } + + if !issetAnnotationID { + fieldId = 2 + goto RequiredFieldNotSetError + } + + if !issetSpanID { + fieldId = 3 + goto RequiredFieldNotSetError + } + + if !issetStartTime { + fieldId = 4 + goto RequiredFieldNotSetError + } + + if !issetCorrection { + fieldId = 5 + goto RequiredFieldNotSetError + } + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ChangeEvaluatorScoreRequest[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ChangeEvaluatorScoreRequest[fieldId])) +} + +func (p *ChangeEvaluatorScoreRequest) FastReadField1(buf []byte) (int, error) { + offset := 0 + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.WorkspaceID = _field + return offset, nil +} + +func (p *ChangeEvaluatorScoreRequest) FastReadField2(buf []byte) (int, error) { + offset := 0 + + var _field string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.AnnotationID = _field + return offset, nil +} + +func (p *ChangeEvaluatorScoreRequest) FastReadField3(buf []byte) (int, error) { + offset := 0 + + var _field string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.SpanID = _field + return offset, nil +} + +func (p *ChangeEvaluatorScoreRequest) FastReadField4(buf []byte) (int, error) { + offset := 0 + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.StartTime = _field + return offset, nil +} + +func (p *ChangeEvaluatorScoreRequest) FastReadField5(buf []byte) (int, error) { + offset := 0 + _field := annotation.NewCorrection() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.Correction = _field + return offset, nil +} + +func (p *ChangeEvaluatorScoreRequest) FastReadField6(buf []byte) (int, error) { + offset := 0 + + var _field *common.PlatformType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.PlatformType = _field + return offset, nil +} + +func (p *ChangeEvaluatorScoreRequest) FastReadField255(buf []byte) (int, error) { + offset := 0 + _field := base.NewBase() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.Base = _field + return offset, nil +} + +func (p *ChangeEvaluatorScoreRequest) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *ChangeEvaluatorScoreRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) + offset += p.fastWriteField5(buf[offset:], w) + offset += p.fastWriteField6(buf[offset:], w) + offset += p.fastWriteField255(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *ChangeEvaluatorScoreRequest) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field2Length() + l += p.field3Length() + l += p.field4Length() + l += p.field5Length() + l += p.field6Length() + l += p.field255Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *ChangeEvaluatorScoreRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) + offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) + return offset +} + +func (p *ChangeEvaluatorScoreRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.AnnotationID) + return offset +} + +func (p *ChangeEvaluatorScoreRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 3) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.SpanID) + return offset +} + +func (p *ChangeEvaluatorScoreRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 4) + offset += thrift.Binary.WriteI64(buf[offset:], p.StartTime) + return offset +} + +func (p *ChangeEvaluatorScoreRequest) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 5) + offset += p.Correction.FastWriteNocopy(buf[offset:], w) + return offset +} + +func (p *ChangeEvaluatorScoreRequest) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetPlatformType() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 6) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.PlatformType) + } + return offset +} + +func (p *ChangeEvaluatorScoreRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetBase() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.Base.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *ChangeEvaluatorScoreRequest) field1Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *ChangeEvaluatorScoreRequest) field2Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(p.AnnotationID) + return l +} + +func (p *ChangeEvaluatorScoreRequest) field3Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(p.SpanID) + return l +} + +func (p *ChangeEvaluatorScoreRequest) field4Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *ChangeEvaluatorScoreRequest) field5Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += p.Correction.BLength() + return l +} + +func (p *ChangeEvaluatorScoreRequest) field6Length() int { + l := 0 + if p.IsSetPlatformType() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.PlatformType) + } + return l +} + +func (p *ChangeEvaluatorScoreRequest) field255Length() int { + l := 0 + if p.IsSetBase() { + l += thrift.Binary.FieldBeginLength() + l += p.Base.BLength() + } + return l +} + +func (p *ChangeEvaluatorScoreRequest) DeepCopy(s interface{}) error { + src, ok := s.(*ChangeEvaluatorScoreRequest) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + p.WorkspaceID = src.WorkspaceID + + if src.AnnotationID != "" { + p.AnnotationID = kutils.StringDeepCopy(src.AnnotationID) + } + + if src.SpanID != "" { + p.SpanID = kutils.StringDeepCopy(src.SpanID) + } + + p.StartTime = src.StartTime + + var _correction *annotation.Correction + if src.Correction != nil { + _correction = &annotation.Correction{} + if err := _correction.DeepCopy(src.Correction); err != nil { + return err + } + } + p.Correction = _correction + + if src.PlatformType != nil { + tmp := *src.PlatformType + p.PlatformType = &tmp + } + + var _base *base.Base + if src.Base != nil { + _base = &base.Base{} + if err := _base.DeepCopy(src.Base); err != nil { + return err + } + } + p.Base = _base + + return nil +} + +func (p *ChangeEvaluatorScoreResponse) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + var issetAnnotation bool = false + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } + issetAnnotation = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11378,132 +13417,214 @@ func (p *PreviewExportTracesToDatasetRequest) FastRead(buf []byte) (int, error) } } - if !issetWorkspaceID { + if !issetAnnotation { fieldId = 1 goto RequiredFieldNotSetError } - - if !issetSpanIds { - fieldId = 2 - goto RequiredFieldNotSetError - } - - if !issetCategory { - fieldId = 3 - goto RequiredFieldNotSetError - } - - if !issetConfig { - fieldId = 4 - goto RequiredFieldNotSetError - } - - if !issetStartTime { - fieldId = 5 - goto RequiredFieldNotSetError - } - - if !issetEndTime { - fieldId = 6 - goto RequiredFieldNotSetError - } - - if !issetExportType { - fieldId = 8 - goto RequiredFieldNotSetError - } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_PreviewExportTracesToDatasetRequest[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ChangeEvaluatorScoreResponse[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_PreviewExportTracesToDatasetRequest[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ChangeEvaluatorScoreResponse[fieldId])) } -func (p *PreviewExportTracesToDatasetRequest) FastReadField1(buf []byte) (int, error) { +func (p *ChangeEvaluatorScoreResponse) FastReadField1(buf []byte) (int, error) { offset := 0 - - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + _field := annotation.NewAnnotation() + if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = v } - p.WorkspaceID = _field + p.Annotation = _field return offset, nil } -func (p *PreviewExportTracesToDatasetRequest) FastReadField2(buf []byte) (int, error) { +func (p *ChangeEvaluatorScoreResponse) FastReadField255(buf []byte) (int, error) { offset := 0 - - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { + _field := base.NewBaseResp() + if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err + } else { + offset += l } - _field := make([]*SpanID, 0, size) - values := make([]SpanID, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } + p.BaseResp = _field + return offset, nil +} - _field = append(_field, _elem) +func (p *ChangeEvaluatorScoreResponse) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *ChangeEvaluatorScoreResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField255(buf[offset:], w) } - p.SpanIds = _field - return offset, nil + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset } -func (p *PreviewExportTracesToDatasetRequest) FastReadField3(buf []byte) (int, error) { +func (p *ChangeEvaluatorScoreResponse) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + l += p.field255Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *ChangeEvaluatorScoreResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 1) + offset += p.Annotation.FastWriteNocopy(buf[offset:], w) + return offset +} - var _field dataset.DatasetCategory - if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { - return offset, err - } else { - offset += l +func (p *ChangeEvaluatorScoreResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetBaseResp() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + } + return offset +} - _field = dataset.DatasetCategory(v) +func (p *ChangeEvaluatorScoreResponse) field1Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += p.Annotation.BLength() + return l +} + +func (p *ChangeEvaluatorScoreResponse) field255Length() int { + l := 0 + if p.IsSetBaseResp() { + l += thrift.Binary.FieldBeginLength() + l += p.BaseResp.BLength() } - p.Category = _field - return offset, nil + return l } -func (p *PreviewExportTracesToDatasetRequest) FastReadField4(buf []byte) (int, error) { - offset := 0 - _field := NewDatasetConfig() - if l, err := _field.FastRead(buf[offset:]); err != nil { - return offset, err - } else { +func (p *ChangeEvaluatorScoreResponse) DeepCopy(s interface{}) error { + src, ok := s.(*ChangeEvaluatorScoreResponse) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + var _annotation *annotation.Annotation + if src.Annotation != nil { + _annotation = &annotation.Annotation{} + if err := _annotation.DeepCopy(src.Annotation); err != nil { + return err + } + } + p.Annotation = _annotation + + var _baseResp *base.BaseResp + if src.BaseResp != nil { + _baseResp = &base.BaseResp{} + if err := _baseResp.DeepCopy(src.BaseResp); err != nil { + return err + } + } + p.BaseResp = _baseResp + + return nil +} + +func (p *ListAnnotationEvaluatorsRequest) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + var issetWorkspaceID bool = false + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetWorkspaceID = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 255: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField255(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } } - p.Config = _field - return offset, nil -} - -func (p *PreviewExportTracesToDatasetRequest) FastReadField5(buf []byte) (int, error) { - offset := 0 - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = v + if !issetWorkspaceID { + fieldId = 1 + goto RequiredFieldNotSetError } - p.StartTime = _field return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListAnnotationEvaluatorsRequest[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListAnnotationEvaluatorsRequest[fieldId])) } -func (p *PreviewExportTracesToDatasetRequest) FastReadField6(buf []byte) (int, error) { +func (p *ListAnnotationEvaluatorsRequest) FastReadField1(buf []byte) (int, error) { offset := 0 var _field int64 @@ -11513,64 +13634,25 @@ func (p *PreviewExportTracesToDatasetRequest) FastReadField6(buf []byte) (int, e offset += l _field = v } - p.EndTime = _field + p.WorkspaceID = _field return offset, nil } -func (p *PreviewExportTracesToDatasetRequest) FastReadField7(buf []byte) (int, error) { +func (p *ListAnnotationEvaluatorsRequest) FastReadField2(buf []byte) (int, error) { offset := 0 - var _field *common.PlatformType + var _field *string if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { return offset, err } else { offset += l _field = &v } - p.PlatformType = _field - return offset, nil -} - -func (p *PreviewExportTracesToDatasetRequest) FastReadField8(buf []byte) (int, error) { - offset := 0 - - var _field dataset0.ExportType - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = v - } - p.ExportType = _field - return offset, nil -} - -func (p *PreviewExportTracesToDatasetRequest) FastReadField9(buf []byte) (int, error) { - offset := 0 - - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { - return offset, err - } - _field := make([]*dataset0.FieldMapping, 0, size) - values := make([]dataset0.FieldMapping, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - - _field = append(_field, _elem) - } - p.FieldMappings = _field + p.Name = _field return offset, nil } -func (p *PreviewExportTracesToDatasetRequest) FastReadField255(buf []byte) (int, error) { +func (p *ListAnnotationEvaluatorsRequest) FastReadField255(buf []byte) (int, error) { offset := 0 _field := base.NewBase() if l, err := _field.FastRead(buf[offset:]); err != nil { @@ -11582,128 +13664,49 @@ func (p *PreviewExportTracesToDatasetRequest) FastReadField255(buf []byte) (int, return offset, nil } -func (p *PreviewExportTracesToDatasetRequest) FastWrite(buf []byte) int { +func (p *ListAnnotationEvaluatorsRequest) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *PreviewExportTracesToDatasetRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ListAnnotationEvaluatorsRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField5(buf[offset:], w) - offset += p.fastWriteField6(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) - offset += p.fastWriteField3(buf[offset:], w) - offset += p.fastWriteField4(buf[offset:], w) - offset += p.fastWriteField7(buf[offset:], w) - offset += p.fastWriteField8(buf[offset:], w) - offset += p.fastWriteField9(buf[offset:], w) offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *PreviewExportTracesToDatasetRequest) BLength() int { +func (p *ListAnnotationEvaluatorsRequest) BLength() int { l := 0 if p != nil { l += p.field1Length() l += p.field2Length() - l += p.field3Length() - l += p.field4Length() - l += p.field5Length() - l += p.field6Length() - l += p.field7Length() - l += p.field8Length() - l += p.field9Length() l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *PreviewExportTracesToDatasetRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ListAnnotationEvaluatorsRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) return offset } -func (p *PreviewExportTracesToDatasetRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.SpanIds { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) - return offset -} - -func (p *PreviewExportTracesToDatasetRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 3) - offset += thrift.Binary.WriteI32(buf[offset:], int32(p.Category)) - return offset -} - -func (p *PreviewExportTracesToDatasetRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 4) - offset += p.Config.FastWriteNocopy(buf[offset:], w) - return offset -} - -func (p *PreviewExportTracesToDatasetRequest) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 5) - offset += thrift.Binary.WriteI64(buf[offset:], p.StartTime) - return offset -} - -func (p *PreviewExportTracesToDatasetRequest) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 6) - offset += thrift.Binary.WriteI64(buf[offset:], p.EndTime) - return offset -} - -func (p *PreviewExportTracesToDatasetRequest) fastWriteField7(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetPlatformType() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 7) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.PlatformType) - } - return offset -} - -func (p *PreviewExportTracesToDatasetRequest) fastWriteField8(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 8) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.ExportType) - return offset -} - -func (p *PreviewExportTracesToDatasetRequest) fastWriteField9(buf []byte, w thrift.NocopyWriter) int { +func (p *ListAnnotationEvaluatorsRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetFieldMappings() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 9) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.FieldMappings { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + if p.IsSetName() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.Name) } return offset } -func (p *PreviewExportTracesToDatasetRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ListAnnotationEvaluatorsRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetBase() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) @@ -11712,82 +13715,23 @@ func (p *PreviewExportTracesToDatasetRequest) fastWriteField255(buf []byte, w th return offset } -func (p *PreviewExportTracesToDatasetRequest) field1Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - return l -} - -func (p *PreviewExportTracesToDatasetRequest) field2Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.SpanIds { - _ = v - l += v.BLength() - } - return l -} - -func (p *PreviewExportTracesToDatasetRequest) field3Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I32Length() - return l -} - -func (p *PreviewExportTracesToDatasetRequest) field4Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += p.Config.BLength() - return l -} - -func (p *PreviewExportTracesToDatasetRequest) field5Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - return l -} - -func (p *PreviewExportTracesToDatasetRequest) field6Length() int { +func (p *ListAnnotationEvaluatorsRequest) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() l += thrift.Binary.I64Length() return l } -func (p *PreviewExportTracesToDatasetRequest) field7Length() int { - l := 0 - if p.IsSetPlatformType() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(*p.PlatformType) - } - return l -} - -func (p *PreviewExportTracesToDatasetRequest) field8Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.ExportType) - return l -} - -func (p *PreviewExportTracesToDatasetRequest) field9Length() int { - l := 0 - if p.IsSetFieldMappings() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.FieldMappings { - _ = v - l += v.BLength() - } +func (p *ListAnnotationEvaluatorsRequest) field2Length() int { + l := 0 + if p.IsSetName() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.Name) } return l } -func (p *PreviewExportTracesToDatasetRequest) field255Length() int { +func (p *ListAnnotationEvaluatorsRequest) field255Length() int { l := 0 if p.IsSetBase() { l += thrift.Binary.FieldBeginLength() @@ -11796,64 +13740,20 @@ func (p *PreviewExportTracesToDatasetRequest) field255Length() int { return l } -func (p *PreviewExportTracesToDatasetRequest) DeepCopy(s interface{}) error { - src, ok := s.(*PreviewExportTracesToDatasetRequest) +func (p *ListAnnotationEvaluatorsRequest) DeepCopy(s interface{}) error { + src, ok := s.(*ListAnnotationEvaluatorsRequest) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } p.WorkspaceID = src.WorkspaceID - if src.SpanIds != nil { - p.SpanIds = make([]*SpanID, 0, len(src.SpanIds)) - for _, elem := range src.SpanIds { - var _elem *SpanID - if elem != nil { - _elem = &SpanID{} - if err := _elem.DeepCopy(elem); err != nil { - return err - } - } - - p.SpanIds = append(p.SpanIds, _elem) - } - } - - p.Category = src.Category - - var _config *DatasetConfig - if src.Config != nil { - _config = &DatasetConfig{} - if err := _config.DeepCopy(src.Config); err != nil { - return err - } - } - p.Config = _config - - p.StartTime = src.StartTime - - p.EndTime = src.EndTime - - if src.PlatformType != nil { - tmp := *src.PlatformType - p.PlatformType = &tmp - } - - p.ExportType = src.ExportType - - if src.FieldMappings != nil { - p.FieldMappings = make([]*dataset0.FieldMapping, 0, len(src.FieldMappings)) - for _, elem := range src.FieldMappings { - var _elem *dataset0.FieldMapping - if elem != nil { - _elem = &dataset0.FieldMapping{} - if err := _elem.DeepCopy(elem); err != nil { - return err - } - } - - p.FieldMappings = append(p.FieldMappings, _elem) + if src.Name != nil { + var tmp string + if *src.Name != "" { + tmp = kutils.StringDeepCopy(*src.Name) } + p.Name = &tmp } var _base *base.Base @@ -11868,13 +13768,14 @@ func (p *PreviewExportTracesToDatasetRequest) DeepCopy(s interface{}) error { return nil } -func (p *PreviewExportTracesToDatasetResponse) FastRead(buf []byte) (int, error) { +func (p *ListAnnotationEvaluatorsResponse) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 + var issetEvaluators bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -11892,20 +13793,7 @@ func (p *PreviewExportTracesToDatasetResponse) FastRead(buf []byte) (int, error) if err != nil { goto ReadFieldError } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 2: - if fieldTypeId == thrift.LIST { - l, err = p.FastReadField2(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } + issetEvaluators = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11927,34 +13815,6 @@ func (p *PreviewExportTracesToDatasetResponse) FastRead(buf []byte) (int, error) goto SkipFieldError } } - case 256: - if fieldTypeId == thrift.I32 { - l, err = p.FastReadField256(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 257: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField257(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } default: l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -11964,41 +13824,22 @@ func (p *PreviewExportTracesToDatasetResponse) FastRead(buf []byte) (int, error) } } + if !issetEvaluators { + fieldId = 1 + goto RequiredFieldNotSetError + } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_PreviewExportTracesToDatasetResponse[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListAnnotationEvaluatorsResponse[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListAnnotationEvaluatorsResponse[fieldId])) } -func (p *PreviewExportTracesToDatasetResponse) FastReadField1(buf []byte) (int, error) { - offset := 0 - - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { - return offset, err - } - _field := make([]*dataset0.Item, 0, size) - values := make([]dataset0.Item, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - - _field = append(_field, _elem) - } - p.Items = _field - return offset, nil -} - -func (p *PreviewExportTracesToDatasetResponse) FastReadField2(buf []byte) (int, error) { +func (p *ListAnnotationEvaluatorsResponse) FastReadField1(buf []byte) (int, error) { offset := 0 _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) @@ -12006,8 +13847,8 @@ func (p *PreviewExportTracesToDatasetResponse) FastReadField2(buf []byte) (int, if err != nil { return offset, err } - _field := make([]*dataset.ItemErrorGroup, 0, size) - values := make([]dataset.ItemErrorGroup, size) + _field := make([]*annotation.AnnotationEvaluator, 0, size) + values := make([]annotation.AnnotationEvaluator, size) for i := 0; i < size; i++ { _elem := &values[i] _elem.InitDefault() @@ -12019,225 +13860,107 @@ func (p *PreviewExportTracesToDatasetResponse) FastReadField2(buf []byte) (int, _field = append(_field, _elem) } - p.Errors = _field + p.Evaluators = _field return offset, nil } -func (p *PreviewExportTracesToDatasetResponse) FastReadField255(buf []byte) (int, error) { +func (p *ListAnnotationEvaluatorsResponse) FastReadField255(buf []byte) (int, error) { offset := 0 _field := base.NewBaseResp() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err - } else { - offset += l - } - p.BaseResp = _field - return offset, nil -} - -func (p *PreviewExportTracesToDatasetResponse) FastReadField256(buf []byte) (int, error) { - offset := 0 - - var _field *int32 - if v, l, err := thrift.Binary.ReadI32(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = &v - } - p.Code = _field - return offset, nil -} - -func (p *PreviewExportTracesToDatasetResponse) FastReadField257(buf []byte) (int, error) { - offset := 0 - - var _field *string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = &v - } - p.Msg = _field - return offset, nil -} - -func (p *PreviewExportTracesToDatasetResponse) FastWrite(buf []byte) int { - return p.FastWriteNocopy(buf, nil) -} - -func (p *PreviewExportTracesToDatasetResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p != nil { - offset += p.fastWriteField256(buf[offset:], w) - offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField2(buf[offset:], w) - offset += p.fastWriteField255(buf[offset:], w) - offset += p.fastWriteField257(buf[offset:], w) - } - offset += thrift.Binary.WriteFieldStop(buf[offset:]) - return offset -} - -func (p *PreviewExportTracesToDatasetResponse) BLength() int { - l := 0 - if p != nil { - l += p.field1Length() - l += p.field2Length() - l += p.field255Length() - l += p.field256Length() - l += p.field257Length() - } - l += thrift.Binary.FieldStopLength() - return l -} - -func (p *PreviewExportTracesToDatasetResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetItems() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.Items { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) - } - return offset -} - -func (p *PreviewExportTracesToDatasetResponse) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetErrors() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.Errors { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) - } - return offset -} - -func (p *PreviewExportTracesToDatasetResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetBaseResp() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + } else { + offset += l } - return offset + p.BaseResp = _field + return offset, nil } -func (p *PreviewExportTracesToDatasetResponse) fastWriteField256(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetCode() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I32, 256) - offset += thrift.Binary.WriteI32(buf[offset:], *p.Code) - } - return offset +func (p *ListAnnotationEvaluatorsResponse) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) } -func (p *PreviewExportTracesToDatasetResponse) fastWriteField257(buf []byte, w thrift.NocopyWriter) int { +func (p *ListAnnotationEvaluatorsResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetMsg() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 257) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.Msg) + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField255(buf[offset:], w) } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *PreviewExportTracesToDatasetResponse) field1Length() int { +func (p *ListAnnotationEvaluatorsResponse) BLength() int { l := 0 - if p.IsSetItems() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.Items { - _ = v - l += v.BLength() - } + if p != nil { + l += p.field1Length() + l += p.field255Length() } + l += thrift.Binary.FieldStopLength() return l } -func (p *PreviewExportTracesToDatasetResponse) field2Length() int { - l := 0 - if p.IsSetErrors() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.Errors { - _ = v - l += v.BLength() - } +func (p *ListAnnotationEvaluatorsResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.Evaluators { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) } - return l + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + return offset } -func (p *PreviewExportTracesToDatasetResponse) field255Length() int { - l := 0 +func (p *ListAnnotationEvaluatorsResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { + offset := 0 if p.IsSetBaseResp() { - l += thrift.Binary.FieldBeginLength() - l += p.BaseResp.BLength() + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) } - return l + return offset } -func (p *PreviewExportTracesToDatasetResponse) field256Length() int { +func (p *ListAnnotationEvaluatorsResponse) field1Length() int { l := 0 - if p.IsSetCode() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I32Length() + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.Evaluators { + _ = v + l += v.BLength() } return l } -func (p *PreviewExportTracesToDatasetResponse) field257Length() int { +func (p *ListAnnotationEvaluatorsResponse) field255Length() int { l := 0 - if p.IsSetMsg() { + if p.IsSetBaseResp() { l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(*p.Msg) + l += p.BaseResp.BLength() } return l } -func (p *PreviewExportTracesToDatasetResponse) DeepCopy(s interface{}) error { - src, ok := s.(*PreviewExportTracesToDatasetResponse) +func (p *ListAnnotationEvaluatorsResponse) DeepCopy(s interface{}) error { + src, ok := s.(*ListAnnotationEvaluatorsResponse) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - if src.Items != nil { - p.Items = make([]*dataset0.Item, 0, len(src.Items)) - for _, elem := range src.Items { - var _elem *dataset0.Item - if elem != nil { - _elem = &dataset0.Item{} - if err := _elem.DeepCopy(elem); err != nil { - return err - } - } - - p.Items = append(p.Items, _elem) - } - } - - if src.Errors != nil { - p.Errors = make([]*dataset.ItemErrorGroup, 0, len(src.Errors)) - for _, elem := range src.Errors { - var _elem *dataset.ItemErrorGroup + if src.Evaluators != nil { + p.Evaluators = make([]*annotation.AnnotationEvaluator, 0, len(src.Evaluators)) + for _, elem := range src.Evaluators { + var _elem *annotation.AnnotationEvaluator if elem != nil { - _elem = &dataset.ItemErrorGroup{} + _elem = &annotation.AnnotationEvaluator{} if err := _elem.DeepCopy(elem); err != nil { return err } } - p.Errors = append(p.Errors, _elem) + p.Evaluators = append(p.Evaluators, _elem) } } @@ -12250,23 +13973,10 @@ func (p *PreviewExportTracesToDatasetResponse) DeepCopy(s interface{}) error { } p.BaseResp = _baseResp - if src.Code != nil { - tmp := *src.Code - p.Code = &tmp - } - - if src.Msg != nil { - var tmp string - if *src.Msg != "" { - tmp = kutils.StringDeepCopy(*src.Msg) - } - p.Msg = &tmp - } - return nil } -func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { +func (p *ExtractSpanInfoRequest) FastRead(buf []byte) (int, error) { var err error var offset int @@ -12274,10 +13984,8 @@ func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { var fieldTypeId thrift.TType var fieldId int16 var issetWorkspaceID bool = false - var issetAnnotationID bool = false - var issetSpanID bool = false - var issetStartTime bool = false - var issetCorrection bool = false + var issetTraceID bool = false + var issetSpanIds bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -12310,7 +14018,7 @@ func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { if err != nil { goto ReadFieldError } - issetAnnotationID = true + issetTraceID = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -12319,13 +14027,13 @@ func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { } } case 3: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.LIST { l, err = p.FastReadField3(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetSpanID = true + issetSpanIds = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -12340,7 +14048,6 @@ func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { if err != nil { goto ReadFieldError } - issetStartTime = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -12349,13 +14056,12 @@ func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { } } case 5: - if fieldTypeId == thrift.STRUCT { + if fieldTypeId == thrift.I64 { l, err = p.FastReadField5(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetCorrection = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -12377,6 +14083,20 @@ func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { goto SkipFieldError } } + case 7: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField7(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } case 255: if fieldTypeId == thrift.STRUCT { l, err = p.FastReadField255(buf[offset:]) @@ -12405,37 +14125,27 @@ func (p *ChangeEvaluatorScoreRequest) FastRead(buf []byte) (int, error) { goto RequiredFieldNotSetError } - if !issetAnnotationID { + if !issetTraceID { fieldId = 2 goto RequiredFieldNotSetError } - if !issetSpanID { + if !issetSpanIds { fieldId = 3 goto RequiredFieldNotSetError } - - if !issetStartTime { - fieldId = 4 - goto RequiredFieldNotSetError - } - - if !issetCorrection { - fieldId = 5 - goto RequiredFieldNotSetError - } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ChangeEvaluatorScoreRequest[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExtractSpanInfoRequest[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ChangeEvaluatorScoreRequest[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ExtractSpanInfoRequest[fieldId])) } -func (p *ChangeEvaluatorScoreRequest) FastReadField1(buf []byte) (int, error) { +func (p *ExtractSpanInfoRequest) FastReadField1(buf []byte) (int, error) { offset := 0 var _field int64 @@ -12449,7 +14159,7 @@ func (p *ChangeEvaluatorScoreRequest) FastReadField1(buf []byte) (int, error) { return offset, nil } -func (p *ChangeEvaluatorScoreRequest) FastReadField2(buf []byte) (int, error) { +func (p *ExtractSpanInfoRequest) FastReadField2(buf []byte) (int, error) { offset := 0 var _field string @@ -12459,51 +14169,63 @@ func (p *ChangeEvaluatorScoreRequest) FastReadField2(buf []byte) (int, error) { offset += l _field = v } - p.AnnotationID = _field + p.TraceID = _field return offset, nil } -func (p *ChangeEvaluatorScoreRequest) FastReadField3(buf []byte) (int, error) { +func (p *ExtractSpanInfoRequest) FastReadField3(buf []byte) (int, error) { offset := 0 - var _field string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { return offset, err - } else { - offset += l - _field = v } - p.SpanID = _field + _field := make([]string, 0, size) + for i := 0; i < size; i++ { + var _elem string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _elem = v + } + + _field = append(_field, _elem) + } + p.SpanIds = _field return offset, nil } -func (p *ChangeEvaluatorScoreRequest) FastReadField4(buf []byte) (int, error) { +func (p *ExtractSpanInfoRequest) FastReadField4(buf []byte) (int, error) { offset := 0 - var _field int64 + var _field *int64 if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = v + _field = &v } p.StartTime = _field return offset, nil } -func (p *ChangeEvaluatorScoreRequest) FastReadField5(buf []byte) (int, error) { +func (p *ExtractSpanInfoRequest) FastReadField5(buf []byte) (int, error) { offset := 0 - _field := annotation.NewCorrection() - if l, err := _field.FastRead(buf[offset:]); err != nil { + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { return offset, err } else { offset += l + _field = &v } - p.Correction = _field + p.EndTime = _field return offset, nil } -func (p *ChangeEvaluatorScoreRequest) FastReadField6(buf []byte) (int, error) { +func (p *ExtractSpanInfoRequest) FastReadField6(buf []byte) (int, error) { offset := 0 var _field *common.PlatformType @@ -12517,7 +14239,32 @@ func (p *ChangeEvaluatorScoreRequest) FastReadField6(buf []byte) (int, error) { return offset, nil } -func (p *ChangeEvaluatorScoreRequest) FastReadField255(buf []byte) (int, error) { +func (p *ExtractSpanInfoRequest) FastReadField7(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]*dataset0.FieldMapping, 0, size) + values := make([]dataset0.FieldMapping, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.FieldMappings = _field + return offset, nil +} + +func (p *ExtractSpanInfoRequest) FastReadField255(buf []byte) (int, error) { offset := 0 _field := base.NewBase() if l, err := _field.FastRead(buf[offset:]); err != nil { @@ -12529,26 +14276,27 @@ func (p *ChangeEvaluatorScoreRequest) FastReadField255(buf []byte) (int, error) return offset, nil } -func (p *ChangeEvaluatorScoreRequest) FastWrite(buf []byte) int { +func (p *ExtractSpanInfoRequest) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ChangeEvaluatorScoreRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField5(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) offset += p.fastWriteField3(buf[offset:], w) - offset += p.fastWriteField5(buf[offset:], w) offset += p.fastWriteField6(buf[offset:], w) + offset += p.fastWriteField7(buf[offset:], w) offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *ChangeEvaluatorScoreRequest) BLength() int { +func (p *ExtractSpanInfoRequest) BLength() int { l := 0 if p != nil { l += p.field1Length() @@ -12557,48 +14305,60 @@ func (p *ChangeEvaluatorScoreRequest) BLength() int { l += p.field4Length() l += p.field5Length() l += p.field6Length() + l += p.field7Length() l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *ChangeEvaluatorScoreRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) return offset } -func (p *ChangeEvaluatorScoreRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { offset := 0 offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.AnnotationID) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.TraceID) return offset } -func (p *ChangeEvaluatorScoreRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 3) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.SpanID) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 3) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.SpanIds { + length++ + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, v) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRING, length) return offset } -func (p *ChangeEvaluatorScoreRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 4) - offset += thrift.Binary.WriteI64(buf[offset:], p.StartTime) + if p.IsSetStartTime() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 4) + offset += thrift.Binary.WriteI64(buf[offset:], *p.StartTime) + } return offset } -func (p *ChangeEvaluatorScoreRequest) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoRequest) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 5) - offset += p.Correction.FastWriteNocopy(buf[offset:], w) + if p.IsSetEndTime() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 5) + offset += thrift.Binary.WriteI64(buf[offset:], *p.EndTime) + } return offset } -func (p *ChangeEvaluatorScoreRequest) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoRequest) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetPlatformType() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 6) @@ -12607,7 +14367,23 @@ func (p *ChangeEvaluatorScoreRequest) fastWriteField6(buf []byte, w thrift.Nocop return offset } -func (p *ChangeEvaluatorScoreRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoRequest) fastWriteField7(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetFieldMappings() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 7) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.FieldMappings { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } + return offset +} + +func (p *ExtractSpanInfoRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetBase() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) @@ -12616,289 +14392,154 @@ func (p *ChangeEvaluatorScoreRequest) fastWriteField255(buf []byte, w thrift.Noc return offset } -func (p *ChangeEvaluatorScoreRequest) field1Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - return l -} - -func (p *ChangeEvaluatorScoreRequest) field2Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.AnnotationID) - return l -} - -func (p *ChangeEvaluatorScoreRequest) field3Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.SpanID) - return l -} - -func (p *ChangeEvaluatorScoreRequest) field4Length() int { +func (p *ExtractSpanInfoRequest) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() l += thrift.Binary.I64Length() return l } -func (p *ChangeEvaluatorScoreRequest) field5Length() int { +func (p *ExtractSpanInfoRequest) field2Length() int { l := 0 l += thrift.Binary.FieldBeginLength() - l += p.Correction.BLength() - return l -} - -func (p *ChangeEvaluatorScoreRequest) field6Length() int { - l := 0 - if p.IsSetPlatformType() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(*p.PlatformType) - } - return l -} - -func (p *ChangeEvaluatorScoreRequest) field255Length() int { - l := 0 - if p.IsSetBase() { - l += thrift.Binary.FieldBeginLength() - l += p.Base.BLength() - } + l += thrift.Binary.StringLengthNocopy(p.TraceID) return l -} - -func (p *ChangeEvaluatorScoreRequest) DeepCopy(s interface{}) error { - src, ok := s.(*ChangeEvaluatorScoreRequest) - if !ok { - return fmt.Errorf("%T's type not matched %T", s, p) - } - - p.WorkspaceID = src.WorkspaceID - - if src.AnnotationID != "" { - p.AnnotationID = kutils.StringDeepCopy(src.AnnotationID) - } - - if src.SpanID != "" { - p.SpanID = kutils.StringDeepCopy(src.SpanID) - } - - p.StartTime = src.StartTime - - var _correction *annotation.Correction - if src.Correction != nil { - _correction = &annotation.Correction{} - if err := _correction.DeepCopy(src.Correction); err != nil { - return err - } - } - p.Correction = _correction - - if src.PlatformType != nil { - tmp := *src.PlatformType - p.PlatformType = &tmp - } - - var _base *base.Base - if src.Base != nil { - _base = &base.Base{} - if err := _base.DeepCopy(src.Base); err != nil { - return err - } - } - p.Base = _base - - return nil -} - -func (p *ChangeEvaluatorScoreResponse) FastRead(buf []byte) (int, error) { - - var err error - var offset int - var l int - var fieldTypeId thrift.TType - var fieldId int16 - var issetAnnotation bool = false - for { - fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldBeginError - } - if fieldTypeId == thrift.STOP { - break - } - switch fieldId { - case 1: - if fieldTypeId == thrift.STRUCT { - l, err = p.FastReadField1(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetAnnotation = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 255: - if fieldTypeId == thrift.STRUCT { - l, err = p.FastReadField255(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - default: - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - } - - if !issetAnnotation { - fieldId = 1 - goto RequiredFieldNotSetError - } - return offset, nil -ReadFieldBeginError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) -ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ChangeEvaluatorScoreResponse[fieldId]), err) -SkipFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) -RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ChangeEvaluatorScoreResponse[fieldId])) -} - -func (p *ChangeEvaluatorScoreResponse) FastReadField1(buf []byte) (int, error) { - offset := 0 - _field := annotation.NewAnnotation() - if l, err := _field.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - p.Annotation = _field - return offset, nil -} - -func (p *ChangeEvaluatorScoreResponse) FastReadField255(buf []byte) (int, error) { - offset := 0 - _field := base.NewBaseResp() - if l, err := _field.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - p.BaseResp = _field - return offset, nil -} - -func (p *ChangeEvaluatorScoreResponse) FastWrite(buf []byte) int { - return p.FastWriteNocopy(buf, nil) -} - -func (p *ChangeEvaluatorScoreResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p != nil { - offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField255(buf[offset:], w) +} + +func (p *ExtractSpanInfoRequest) field3Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.SpanIds { + _ = v + l += thrift.Binary.StringLengthNocopy(v) } - offset += thrift.Binary.WriteFieldStop(buf[offset:]) - return offset + return l } -func (p *ChangeEvaluatorScoreResponse) BLength() int { +func (p *ExtractSpanInfoRequest) field4Length() int { l := 0 - if p != nil { - l += p.field1Length() - l += p.field255Length() + if p.IsSetStartTime() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() } - l += thrift.Binary.FieldStopLength() return l } -func (p *ChangeEvaluatorScoreResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 1) - offset += p.Annotation.FastWriteNocopy(buf[offset:], w) - return offset +func (p *ExtractSpanInfoRequest) field5Length() int { + l := 0 + if p.IsSetEndTime() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + } + return l } -func (p *ChangeEvaluatorScoreResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetBaseResp() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) +func (p *ExtractSpanInfoRequest) field6Length() int { + l := 0 + if p.IsSetPlatformType() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(*p.PlatformType) } - return offset + return l } -func (p *ChangeEvaluatorScoreResponse) field1Length() int { +func (p *ExtractSpanInfoRequest) field7Length() int { l := 0 - l += thrift.Binary.FieldBeginLength() - l += p.Annotation.BLength() + if p.IsSetFieldMappings() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.FieldMappings { + _ = v + l += v.BLength() + } + } return l } -func (p *ChangeEvaluatorScoreResponse) field255Length() int { +func (p *ExtractSpanInfoRequest) field255Length() int { l := 0 - if p.IsSetBaseResp() { + if p.IsSetBase() { l += thrift.Binary.FieldBeginLength() - l += p.BaseResp.BLength() + l += p.Base.BLength() } return l } -func (p *ChangeEvaluatorScoreResponse) DeepCopy(s interface{}) error { - src, ok := s.(*ChangeEvaluatorScoreResponse) +func (p *ExtractSpanInfoRequest) DeepCopy(s interface{}) error { + src, ok := s.(*ExtractSpanInfoRequest) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - var _annotation *annotation.Annotation - if src.Annotation != nil { - _annotation = &annotation.Annotation{} - if err := _annotation.DeepCopy(src.Annotation); err != nil { - return err + p.WorkspaceID = src.WorkspaceID + + if src.TraceID != "" { + p.TraceID = kutils.StringDeepCopy(src.TraceID) + } + + if src.SpanIds != nil { + p.SpanIds = make([]string, 0, len(src.SpanIds)) + for _, elem := range src.SpanIds { + var _elem string + if elem != "" { + _elem = kutils.StringDeepCopy(elem) + } + p.SpanIds = append(p.SpanIds, _elem) } } - p.Annotation = _annotation - var _baseResp *base.BaseResp - if src.BaseResp != nil { - _baseResp = &base.BaseResp{} - if err := _baseResp.DeepCopy(src.BaseResp); err != nil { + if src.StartTime != nil { + tmp := *src.StartTime + p.StartTime = &tmp + } + + if src.EndTime != nil { + tmp := *src.EndTime + p.EndTime = &tmp + } + + if src.PlatformType != nil { + tmp := *src.PlatformType + p.PlatformType = &tmp + } + + if src.FieldMappings != nil { + p.FieldMappings = make([]*dataset0.FieldMapping, 0, len(src.FieldMappings)) + for _, elem := range src.FieldMappings { + var _elem *dataset0.FieldMapping + if elem != nil { + _elem = &dataset0.FieldMapping{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.FieldMappings = append(p.FieldMappings, _elem) + } + } + + var _base *base.Base + if src.Base != nil { + _base = &base.Base{} + if err := _base.DeepCopy(src.Base); err != nil { return err } } - p.BaseResp = _baseResp + p.Base = _base return nil } -func (p *ListAnnotationEvaluatorsRequest) FastRead(buf []byte) (int, error) { +func (p *SpanInfo) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetWorkspaceID bool = false + var issetSpanID bool = false + var issetFieldList bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -12910,13 +14551,13 @@ func (p *ListAnnotationEvaluatorsRequest) FastRead(buf []byte) (int, error) { } switch fieldId { case 1: - if fieldTypeId == thrift.I64 { + if fieldTypeId == thrift.STRING { l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetWorkspaceID = true + issetSpanID = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -12925,26 +14566,13 @@ func (p *ListAnnotationEvaluatorsRequest) FastRead(buf []byte) (int, error) { } } case 2: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.LIST { l, err = p.FastReadField2(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 255: - if fieldTypeId == thrift.STRUCT { - l, err = p.FastReadField255(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } + issetFieldList = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -12961,173 +14589,164 @@ func (p *ListAnnotationEvaluatorsRequest) FastRead(buf []byte) (int, error) { } } - if !issetWorkspaceID { + if !issetSpanID { fieldId = 1 goto RequiredFieldNotSetError } + + if !issetFieldList { + fieldId = 2 + goto RequiredFieldNotSetError + } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListAnnotationEvaluatorsRequest[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SpanInfo[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListAnnotationEvaluatorsRequest[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_SpanInfo[fieldId])) } -func (p *ListAnnotationEvaluatorsRequest) FastReadField1(buf []byte) (int, error) { +func (p *SpanInfo) FastReadField1(buf []byte) (int, error) { offset := 0 - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + var _field string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { return offset, err } else { offset += l _field = v } - p.WorkspaceID = _field + p.SpanID = _field return offset, nil } -func (p *ListAnnotationEvaluatorsRequest) FastReadField2(buf []byte) (int, error) { +func (p *SpanInfo) FastReadField2(buf []byte) (int, error) { offset := 0 - var _field *string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { return offset, err - } else { - offset += l - _field = &v } - p.Name = _field - return offset, nil -} + _field := make([]*dataset0.FieldData, 0, size) + values := make([]dataset0.FieldData, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } -func (p *ListAnnotationEvaluatorsRequest) FastReadField255(buf []byte) (int, error) { - offset := 0 - _field := base.NewBase() - if l, err := _field.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l + _field = append(_field, _elem) } - p.Base = _field + p.FieldList = _field return offset, nil } -func (p *ListAnnotationEvaluatorsRequest) FastWrite(buf []byte) int { +func (p *SpanInfo) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ListAnnotationEvaluatorsRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *SpanInfo) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) - offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *ListAnnotationEvaluatorsRequest) BLength() int { +func (p *SpanInfo) BLength() int { l := 0 if p != nil { l += p.field1Length() l += p.field2Length() - l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *ListAnnotationEvaluatorsRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) - offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) - return offset -} - -func (p *ListAnnotationEvaluatorsRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { +func (p *SpanInfo) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetName() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.Name) - } + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 1) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.SpanID) return offset } -func (p *ListAnnotationEvaluatorsRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *SpanInfo) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetBase() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.Base.FastWriteNocopy(buf[offset:], w) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.FieldList { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) return offset } -func (p *ListAnnotationEvaluatorsRequest) field1Length() int { +func (p *SpanInfo) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - return l -} - -func (p *ListAnnotationEvaluatorsRequest) field2Length() int { - l := 0 - if p.IsSetName() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(*p.Name) - } + l += thrift.Binary.StringLengthNocopy(p.SpanID) return l } -func (p *ListAnnotationEvaluatorsRequest) field255Length() int { +func (p *SpanInfo) field2Length() int { l := 0 - if p.IsSetBase() { - l += thrift.Binary.FieldBeginLength() - l += p.Base.BLength() + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.FieldList { + _ = v + l += v.BLength() } return l } -func (p *ListAnnotationEvaluatorsRequest) DeepCopy(s interface{}) error { - src, ok := s.(*ListAnnotationEvaluatorsRequest) +func (p *SpanInfo) DeepCopy(s interface{}) error { + src, ok := s.(*SpanInfo) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - p.WorkspaceID = src.WorkspaceID - - if src.Name != nil { - var tmp string - if *src.Name != "" { - tmp = kutils.StringDeepCopy(*src.Name) - } - p.Name = &tmp + if src.SpanID != "" { + p.SpanID = kutils.StringDeepCopy(src.SpanID) } - var _base *base.Base - if src.Base != nil { - _base = &base.Base{} - if err := _base.DeepCopy(src.Base); err != nil { - return err + if src.FieldList != nil { + p.FieldList = make([]*dataset0.FieldData, 0, len(src.FieldList)) + for _, elem := range src.FieldList { + var _elem *dataset0.FieldData + if elem != nil { + _elem = &dataset0.FieldData{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } + + p.FieldList = append(p.FieldList, _elem) } } - p.Base = _base return nil } -func (p *ListAnnotationEvaluatorsResponse) FastRead(buf []byte) (int, error) { +func (p *ExtractSpanInfoResponse) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetEvaluators bool = false + var issetSpanInfos bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -13145,7 +14764,7 @@ func (p *ListAnnotationEvaluatorsResponse) FastRead(buf []byte) (int, error) { if err != nil { goto ReadFieldError } - issetEvaluators = true + issetSpanInfos = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -13176,7 +14795,7 @@ func (p *ListAnnotationEvaluatorsResponse) FastRead(buf []byte) (int, error) { } } - if !issetEvaluators { + if !issetSpanInfos { fieldId = 1 goto RequiredFieldNotSetError } @@ -13184,14 +14803,14 @@ func (p *ListAnnotationEvaluatorsResponse) FastRead(buf []byte) (int, error) { ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListAnnotationEvaluatorsResponse[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExtractSpanInfoResponse[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListAnnotationEvaluatorsResponse[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ExtractSpanInfoResponse[fieldId])) } -func (p *ListAnnotationEvaluatorsResponse) FastReadField1(buf []byte) (int, error) { +func (p *ExtractSpanInfoResponse) FastReadField1(buf []byte) (int, error) { offset := 0 _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) @@ -13199,8 +14818,8 @@ func (p *ListAnnotationEvaluatorsResponse) FastReadField1(buf []byte) (int, erro if err != nil { return offset, err } - _field := make([]*annotation.AnnotationEvaluator, 0, size) - values := make([]annotation.AnnotationEvaluator, size) + _field := make([]*SpanInfo, 0, size) + values := make([]SpanInfo, size) for i := 0; i < size; i++ { _elem := &values[i] _elem.InitDefault() @@ -13212,11 +14831,11 @@ func (p *ListAnnotationEvaluatorsResponse) FastReadField1(buf []byte) (int, erro _field = append(_field, _elem) } - p.Evaluators = _field + p.SpanInfos = _field return offset, nil } -func (p *ListAnnotationEvaluatorsResponse) FastReadField255(buf []byte) (int, error) { +func (p *ExtractSpanInfoResponse) FastReadField255(buf []byte) (int, error) { offset := 0 _field := base.NewBaseResp() if l, err := _field.FastRead(buf[offset:]); err != nil { @@ -13228,11 +14847,11 @@ func (p *ListAnnotationEvaluatorsResponse) FastReadField255(buf []byte) (int, er return offset, nil } -func (p *ListAnnotationEvaluatorsResponse) FastWrite(buf []byte) int { +func (p *ExtractSpanInfoResponse) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ListAnnotationEvaluatorsResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) @@ -13242,7 +14861,7 @@ func (p *ListAnnotationEvaluatorsResponse) FastWriteNocopy(buf []byte, w thrift. return offset } -func (p *ListAnnotationEvaluatorsResponse) BLength() int { +func (p *ExtractSpanInfoResponse) BLength() int { l := 0 if p != nil { l += p.field1Length() @@ -13252,13 +14871,13 @@ func (p *ListAnnotationEvaluatorsResponse) BLength() int { return l } -func (p *ListAnnotationEvaluatorsResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) listBeginOffset := offset offset += thrift.Binary.ListBeginLength() var length int - for _, v := range p.Evaluators { + for _, v := range p.SpanInfos { length++ offset += v.FastWriteNocopy(buf[offset:], w) } @@ -13266,7 +14885,7 @@ func (p *ListAnnotationEvaluatorsResponse) fastWriteField1(buf []byte, w thrift. return offset } -func (p *ListAnnotationEvaluatorsResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ExtractSpanInfoResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetBaseResp() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) @@ -13275,18 +14894,18 @@ func (p *ListAnnotationEvaluatorsResponse) fastWriteField255(buf []byte, w thrif return offset } -func (p *ListAnnotationEvaluatorsResponse) field1Length() int { +func (p *ExtractSpanInfoResponse) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() l += thrift.Binary.ListBeginLength() - for _, v := range p.Evaluators { + for _, v := range p.SpanInfos { _ = v l += v.BLength() } return l } -func (p *ListAnnotationEvaluatorsResponse) field255Length() int { +func (p *ExtractSpanInfoResponse) field255Length() int { l := 0 if p.IsSetBaseResp() { l += thrift.Binary.FieldBeginLength() @@ -13295,24 +14914,24 @@ func (p *ListAnnotationEvaluatorsResponse) field255Length() int { return l } -func (p *ListAnnotationEvaluatorsResponse) DeepCopy(s interface{}) error { - src, ok := s.(*ListAnnotationEvaluatorsResponse) +func (p *ExtractSpanInfoResponse) DeepCopy(s interface{}) error { + src, ok := s.(*ExtractSpanInfoResponse) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - if src.Evaluators != nil { - p.Evaluators = make([]*annotation.AnnotationEvaluator, 0, len(src.Evaluators)) - for _, elem := range src.Evaluators { - var _elem *annotation.AnnotationEvaluator + if src.SpanInfos != nil { + p.SpanInfos = make([]*SpanInfo, 0, len(src.SpanInfos)) + for _, elem := range src.SpanInfos { + var _elem *SpanInfo if elem != nil { - _elem = &annotation.AnnotationEvaluator{} + _elem = &SpanInfo{} if err := _elem.DeepCopy(elem); err != nil { return err } } - p.Evaluators = append(p.Evaluators, _elem) + p.SpanInfos = append(p.SpanInfos, _elem) } } @@ -13328,7 +14947,7 @@ func (p *ListAnnotationEvaluatorsResponse) DeepCopy(s interface{}) error { return nil } -func (p *ExtractSpanInfoRequest) FastRead(buf []byte) (int, error) { +func (p *UpsertTrajectoryConfigRequest) FastRead(buf []byte) (int, error) { var err error var offset int @@ -13336,8 +14955,6 @@ func (p *ExtractSpanInfoRequest) FastRead(buf []byte) (int, error) { var fieldTypeId thrift.TType var fieldId int16 var issetWorkspaceID bool = false - var issetTraceID bool = false - var issetSpanIds bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -13364,84 +14981,12 @@ func (p *ExtractSpanInfoRequest) FastRead(buf []byte) (int, error) { } } case 2: - if fieldTypeId == thrift.STRING { + if fieldTypeId == thrift.STRUCT { l, err = p.FastReadField2(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetTraceID = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 3: - if fieldTypeId == thrift.LIST { - l, err = p.FastReadField3(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetSpanIds = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 4: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField4(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 5: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField5(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 6: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField6(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 7: - if fieldTypeId == thrift.LIST { - l, err = p.FastReadField7(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -13469,154 +15014,51 @@ func (p *ExtractSpanInfoRequest) FastRead(buf []byte) (int, error) { if err != nil { goto SkipFieldError } - } - } - - if !issetWorkspaceID { - fieldId = 1 - goto RequiredFieldNotSetError - } - - if !issetTraceID { - fieldId = 2 - goto RequiredFieldNotSetError - } - - if !issetSpanIds { - fieldId = 3 - goto RequiredFieldNotSetError - } - return offset, nil -ReadFieldBeginError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) -ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExtractSpanInfoRequest[fieldId]), err) -SkipFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) -RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ExtractSpanInfoRequest[fieldId])) -} - -func (p *ExtractSpanInfoRequest) FastReadField1(buf []byte) (int, error) { - offset := 0 - - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = v - } - p.WorkspaceID = _field - return offset, nil -} - -func (p *ExtractSpanInfoRequest) FastReadField2(buf []byte) (int, error) { - offset := 0 - - var _field string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = v - } - p.TraceID = _field - return offset, nil -} - -func (p *ExtractSpanInfoRequest) FastReadField3(buf []byte) (int, error) { - offset := 0 - - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { - return offset, err - } - _field := make([]string, 0, size) - for i := 0; i < size; i++ { - var _elem string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _elem = v - } - - _field = append(_field, _elem) - } - p.SpanIds = _field - return offset, nil -} - -func (p *ExtractSpanInfoRequest) FastReadField4(buf []byte) (int, error) { - offset := 0 + } + } - var _field *int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = &v + if !issetWorkspaceID { + fieldId = 1 + goto RequiredFieldNotSetError } - p.StartTime = _field return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_UpsertTrajectoryConfigRequest[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_UpsertTrajectoryConfigRequest[fieldId])) } -func (p *ExtractSpanInfoRequest) FastReadField5(buf []byte) (int, error) { +func (p *UpsertTrajectoryConfigRequest) FastReadField1(buf []byte) (int, error) { offset := 0 - var _field *int64 + var _field int64 if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = &v + _field = v } - p.EndTime = _field + p.WorkspaceID = _field return offset, nil } -func (p *ExtractSpanInfoRequest) FastReadField6(buf []byte) (int, error) { +func (p *UpsertTrajectoryConfigRequest) FastReadField2(buf []byte) (int, error) { offset := 0 - - var _field *common.PlatformType - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + _field := filter.NewFilterFields() + if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = &v - } - p.PlatformType = _field - return offset, nil -} - -func (p *ExtractSpanInfoRequest) FastReadField7(buf []byte) (int, error) { - offset := 0 - - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { - return offset, err } - _field := make([]*dataset0.FieldMapping, 0, size) - values := make([]dataset0.FieldMapping, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - - _field = append(_field, _elem) - } - p.FieldMappings = _field + p.Filters = _field return offset, nil } -func (p *ExtractSpanInfoRequest) FastReadField255(buf []byte) (int, error) { +func (p *UpsertTrajectoryConfigRequest) FastReadField255(buf []byte) (int, error) { offset := 0 _field := base.NewBase() if l, err := _field.FastRead(buf[offset:]); err != nil { @@ -13628,114 +15070,49 @@ func (p *ExtractSpanInfoRequest) FastReadField255(buf []byte) (int, error) { return offset, nil } -func (p *ExtractSpanInfoRequest) FastWrite(buf []byte) int { +func (p *UpsertTrajectoryConfigRequest) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ExtractSpanInfoRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *UpsertTrajectoryConfigRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField4(buf[offset:], w) - offset += p.fastWriteField5(buf[offset:], w) offset += p.fastWriteField2(buf[offset:], w) - offset += p.fastWriteField3(buf[offset:], w) - offset += p.fastWriteField6(buf[offset:], w) - offset += p.fastWriteField7(buf[offset:], w) offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *ExtractSpanInfoRequest) BLength() int { +func (p *UpsertTrajectoryConfigRequest) BLength() int { l := 0 if p != nil { l += p.field1Length() l += p.field2Length() - l += p.field3Length() - l += p.field4Length() - l += p.field5Length() - l += p.field6Length() - l += p.field7Length() l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *ExtractSpanInfoRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *UpsertTrajectoryConfigRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) return offset } -func (p *ExtractSpanInfoRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.TraceID) - return offset -} - -func (p *ExtractSpanInfoRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 3) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.SpanIds { - length++ - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, v) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRING, length) - return offset -} - -func (p *ExtractSpanInfoRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetStartTime() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 4) - offset += thrift.Binary.WriteI64(buf[offset:], *p.StartTime) - } - return offset -} - -func (p *ExtractSpanInfoRequest) fastWriteField5(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetEndTime() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 5) - offset += thrift.Binary.WriteI64(buf[offset:], *p.EndTime) - } - return offset -} - -func (p *ExtractSpanInfoRequest) fastWriteField6(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetPlatformType() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 6) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.PlatformType) - } - return offset -} - -func (p *ExtractSpanInfoRequest) fastWriteField7(buf []byte, w thrift.NocopyWriter) int { +func (p *UpsertTrajectoryConfigRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetFieldMappings() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 7) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.FieldMappings { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + if p.IsSetFilters() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 2) + offset += p.Filters.FastWriteNocopy(buf[offset:], w) } return offset } -func (p *ExtractSpanInfoRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *UpsertTrajectoryConfigRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetBase() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) @@ -13744,72 +15121,23 @@ func (p *ExtractSpanInfoRequest) fastWriteField255(buf []byte, w thrift.NocopyWr return offset } -func (p *ExtractSpanInfoRequest) field1Length() int { +func (p *UpsertTrajectoryConfigRequest) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() l += thrift.Binary.I64Length() return l } -func (p *ExtractSpanInfoRequest) field2Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.TraceID) - return l -} - -func (p *ExtractSpanInfoRequest) field3Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.SpanIds { - _ = v - l += thrift.Binary.StringLengthNocopy(v) - } - return l -} - -func (p *ExtractSpanInfoRequest) field4Length() int { - l := 0 - if p.IsSetStartTime() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - } - return l -} - -func (p *ExtractSpanInfoRequest) field5Length() int { - l := 0 - if p.IsSetEndTime() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - } - return l -} - -func (p *ExtractSpanInfoRequest) field6Length() int { - l := 0 - if p.IsSetPlatformType() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(*p.PlatformType) - } - return l -} - -func (p *ExtractSpanInfoRequest) field7Length() int { +func (p *UpsertTrajectoryConfigRequest) field2Length() int { l := 0 - if p.IsSetFieldMappings() { + if p.IsSetFilters() { l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.FieldMappings { - _ = v - l += v.BLength() - } + l += p.Filters.BLength() } return l } -func (p *ExtractSpanInfoRequest) field255Length() int { +func (p *UpsertTrajectoryConfigRequest) field255Length() int { l := 0 if p.IsSetBase() { l += thrift.Binary.FieldBeginLength() @@ -13818,58 +15146,22 @@ func (p *ExtractSpanInfoRequest) field255Length() int { return l } -func (p *ExtractSpanInfoRequest) DeepCopy(s interface{}) error { - src, ok := s.(*ExtractSpanInfoRequest) +func (p *UpsertTrajectoryConfigRequest) DeepCopy(s interface{}) error { + src, ok := s.(*UpsertTrajectoryConfigRequest) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } p.WorkspaceID = src.WorkspaceID - if src.TraceID != "" { - p.TraceID = kutils.StringDeepCopy(src.TraceID) - } - - if src.SpanIds != nil { - p.SpanIds = make([]string, 0, len(src.SpanIds)) - for _, elem := range src.SpanIds { - var _elem string - if elem != "" { - _elem = kutils.StringDeepCopy(elem) - } - p.SpanIds = append(p.SpanIds, _elem) - } - } - - if src.StartTime != nil { - tmp := *src.StartTime - p.StartTime = &tmp - } - - if src.EndTime != nil { - tmp := *src.EndTime - p.EndTime = &tmp - } - - if src.PlatformType != nil { - tmp := *src.PlatformType - p.PlatformType = &tmp - } - - if src.FieldMappings != nil { - p.FieldMappings = make([]*dataset0.FieldMapping, 0, len(src.FieldMappings)) - for _, elem := range src.FieldMappings { - var _elem *dataset0.FieldMapping - if elem != nil { - _elem = &dataset0.FieldMapping{} - if err := _elem.DeepCopy(elem); err != nil { - return err - } - } - - p.FieldMappings = append(p.FieldMappings, _elem) + var _filters *filter.FilterFields + if src.Filters != nil { + _filters = &filter.FilterFields{} + if err := _filters.DeepCopy(src.Filters); err != nil { + return err } } + p.Filters = _filters var _base *base.Base if src.Base != nil { @@ -13883,15 +15175,13 @@ func (p *ExtractSpanInfoRequest) DeepCopy(s interface{}) error { return nil } -func (p *SpanInfo) FastRead(buf []byte) (int, error) { +func (p *UpsertTrajectoryConfigResponse) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetSpanID bool = false - var issetFieldList bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -13902,29 +15192,13 @@ func (p *SpanInfo) FastRead(buf []byte) (int, error) { break } switch fieldId { - case 1: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField1(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetSpanID = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 2: - if fieldTypeId == thrift.LIST { - l, err = p.FastReadField2(buf[offset:]) + case 255: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField255(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetFieldList = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -13941,164 +15215,93 @@ func (p *SpanInfo) FastRead(buf []byte) (int, error) { } } - if !issetSpanID { - fieldId = 1 - goto RequiredFieldNotSetError - } - - if !issetFieldList { - fieldId = 2 - goto RequiredFieldNotSetError - } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_SpanInfo[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_UpsertTrajectoryConfigResponse[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) -RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_SpanInfo[fieldId])) } -func (p *SpanInfo) FastReadField1(buf []byte) (int, error) { +func (p *UpsertTrajectoryConfigResponse) FastReadField255(buf []byte) (int, error) { offset := 0 - - var _field string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + _field := base.NewBaseResp() + if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l - _field = v - } - p.SpanID = _field - return offset, nil -} - -func (p *SpanInfo) FastReadField2(buf []byte) (int, error) { - offset := 0 - - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { - return offset, err - } - _field := make([]*dataset0.FieldData, 0, size) - values := make([]dataset0.FieldData, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - - _field = append(_field, _elem) } - p.FieldList = _field + p.BaseResp = _field return offset, nil } -func (p *SpanInfo) FastWrite(buf []byte) int { +func (p *UpsertTrajectoryConfigResponse) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *SpanInfo) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *UpsertTrajectoryConfigResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { - offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *SpanInfo) BLength() int { +func (p *UpsertTrajectoryConfigResponse) BLength() int { l := 0 if p != nil { - l += p.field1Length() - l += p.field2Length() + l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *SpanInfo) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 1) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.SpanID) - return offset -} - -func (p *SpanInfo) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { +func (p *UpsertTrajectoryConfigResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 2) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.FieldList { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) + if p.IsSetBaseResp() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) + offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) return offset } -func (p *SpanInfo) field1Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.SpanID) - return l -} - -func (p *SpanInfo) field2Length() int { +func (p *UpsertTrajectoryConfigResponse) field255Length() int { l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.FieldList { - _ = v - l += v.BLength() + if p.IsSetBaseResp() { + l += thrift.Binary.FieldBeginLength() + l += p.BaseResp.BLength() } return l } -func (p *SpanInfo) DeepCopy(s interface{}) error { - src, ok := s.(*SpanInfo) +func (p *UpsertTrajectoryConfigResponse) DeepCopy(s interface{}) error { + src, ok := s.(*UpsertTrajectoryConfigResponse) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - if src.SpanID != "" { - p.SpanID = kutils.StringDeepCopy(src.SpanID) - } - - if src.FieldList != nil { - p.FieldList = make([]*dataset0.FieldData, 0, len(src.FieldList)) - for _, elem := range src.FieldList { - var _elem *dataset0.FieldData - if elem != nil { - _elem = &dataset0.FieldData{} - if err := _elem.DeepCopy(elem); err != nil { - return err - } - } - - p.FieldList = append(p.FieldList, _elem) + var _baseResp *base.BaseResp + if src.BaseResp != nil { + _baseResp = &base.BaseResp{} + if err := _baseResp.DeepCopy(src.BaseResp); err != nil { + return err } } + p.BaseResp = _baseResp return nil } -func (p *ExtractSpanInfoResponse) FastRead(buf []byte) (int, error) { +func (p *GetTrajectoryConfigRequest) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetSpanInfos bool = false + var issetWorkspaceID bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -14110,13 +15313,13 @@ func (p *ExtractSpanInfoResponse) FastRead(buf []byte) (int, error) { } switch fieldId { case 1: - if fieldTypeId == thrift.LIST { + if fieldTypeId == thrift.I64 { l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetSpanInfos = true + issetWorkspaceID = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -14147,7 +15350,7 @@ func (p *ExtractSpanInfoResponse) FastRead(buf []byte) (int, error) { } } - if !issetSpanInfos { + if !issetWorkspaceID { fieldId = 1 goto RequiredFieldNotSetError } @@ -14155,55 +15358,44 @@ func (p *ExtractSpanInfoResponse) FastRead(buf []byte) (int, error) { ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ExtractSpanInfoResponse[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_GetTrajectoryConfigRequest[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ExtractSpanInfoResponse[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_GetTrajectoryConfigRequest[fieldId])) } -func (p *ExtractSpanInfoResponse) FastReadField1(buf []byte) (int, error) { +func (p *GetTrajectoryConfigRequest) FastReadField1(buf []byte) (int, error) { offset := 0 - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { return offset, err + } else { + offset += l + _field = v } - _field := make([]*SpanInfo, 0, size) - values := make([]SpanInfo, size) - for i := 0; i < size; i++ { - _elem := &values[i] - _elem.InitDefault() - if l, err := _elem.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - - _field = append(_field, _elem) - } - p.SpanInfos = _field + p.WorkspaceID = _field return offset, nil } -func (p *ExtractSpanInfoResponse) FastReadField255(buf []byte) (int, error) { +func (p *GetTrajectoryConfigRequest) FastReadField255(buf []byte) (int, error) { offset := 0 - _field := base.NewBaseResp() + _field := base.NewBase() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l } - p.BaseResp = _field + p.Base = _field return offset, nil } -func (p *ExtractSpanInfoResponse) FastWrite(buf []byte) int { +func (p *GetTrajectoryConfigRequest) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ExtractSpanInfoResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *GetTrajectoryConfigRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) @@ -14213,7 +15405,7 @@ func (p *ExtractSpanInfoResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWri return offset } -func (p *ExtractSpanInfoResponse) BLength() int { +func (p *GetTrajectoryConfigRequest) BLength() int { l := 0 if p != nil { l += p.field1Length() @@ -14223,90 +15415,65 @@ func (p *ExtractSpanInfoResponse) BLength() int { return l } -func (p *ExtractSpanInfoResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *GetTrajectoryConfigRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.SpanInfos { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) + offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) return offset } -func (p *ExtractSpanInfoResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *GetTrajectoryConfigRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetBaseResp() { + if p.IsSetBase() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + offset += p.Base.FastWriteNocopy(buf[offset:], w) } return offset } -func (p *ExtractSpanInfoResponse) field1Length() int { +func (p *GetTrajectoryConfigRequest) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.SpanInfos { - _ = v - l += v.BLength() - } + l += thrift.Binary.I64Length() return l } -func (p *ExtractSpanInfoResponse) field255Length() int { +func (p *GetTrajectoryConfigRequest) field255Length() int { l := 0 - if p.IsSetBaseResp() { + if p.IsSetBase() { l += thrift.Binary.FieldBeginLength() - l += p.BaseResp.BLength() + l += p.Base.BLength() } return l } - -func (p *ExtractSpanInfoResponse) DeepCopy(s interface{}) error { - src, ok := s.(*ExtractSpanInfoResponse) - if !ok { - return fmt.Errorf("%T's type not matched %T", s, p) - } - - if src.SpanInfos != nil { - p.SpanInfos = make([]*SpanInfo, 0, len(src.SpanInfos)) - for _, elem := range src.SpanInfos { - var _elem *SpanInfo - if elem != nil { - _elem = &SpanInfo{} - if err := _elem.DeepCopy(elem); err != nil { - return err - } - } - - p.SpanInfos = append(p.SpanInfos, _elem) - } + +func (p *GetTrajectoryConfigRequest) DeepCopy(s interface{}) error { + src, ok := s.(*GetTrajectoryConfigRequest) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) } - var _baseResp *base.BaseResp - if src.BaseResp != nil { - _baseResp = &base.BaseResp{} - if err := _baseResp.DeepCopy(src.BaseResp); err != nil { + p.WorkspaceID = src.WorkspaceID + + var _base *base.Base + if src.Base != nil { + _base = &base.Base{} + if err := _base.DeepCopy(src.Base); err != nil { return err } } - p.BaseResp = _baseResp + p.Base = _base return nil } -func (p *UpsertTrajectoryConfigRequest) FastRead(buf []byte) (int, error) { +func (p *GetTrajectoryConfigResponse) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetWorkspaceID bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -14318,23 +15485,8 @@ func (p *UpsertTrajectoryConfigRequest) FastRead(buf []byte) (int, error) { } switch fieldId { case 1: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField1(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetWorkspaceID = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 2: if fieldTypeId == thrift.STRUCT { - l, err = p.FastReadField2(buf[offset:]) + l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError @@ -14369,36 +15521,16 @@ func (p *UpsertTrajectoryConfigRequest) FastRead(buf []byte) (int, error) { } } - if !issetWorkspaceID { - fieldId = 1 - goto RequiredFieldNotSetError - } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_UpsertTrajectoryConfigRequest[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_GetTrajectoryConfigResponse[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) -RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_UpsertTrajectoryConfigRequest[fieldId])) -} - -func (p *UpsertTrajectoryConfigRequest) FastReadField1(buf []byte) (int, error) { - offset := 0 - - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = v - } - p.WorkspaceID = _field - return offset, nil } -func (p *UpsertTrajectoryConfigRequest) FastReadField2(buf []byte) (int, error) { +func (p *GetTrajectoryConfigResponse) FastReadField1(buf []byte) (int, error) { offset := 0 _field := filter.NewFilterFields() if l, err := _field.FastRead(buf[offset:]); err != nil { @@ -14410,77 +15542,61 @@ func (p *UpsertTrajectoryConfigRequest) FastReadField2(buf []byte) (int, error) return offset, nil } -func (p *UpsertTrajectoryConfigRequest) FastReadField255(buf []byte) (int, error) { +func (p *GetTrajectoryConfigResponse) FastReadField255(buf []byte) (int, error) { offset := 0 - _field := base.NewBase() + _field := base.NewBaseResp() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l } - p.Base = _field + p.BaseResp = _field return offset, nil } -func (p *UpsertTrajectoryConfigRequest) FastWrite(buf []byte) int { +func (p *GetTrajectoryConfigResponse) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *UpsertTrajectoryConfigRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *GetTrajectoryConfigResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField2(buf[offset:], w) offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *UpsertTrajectoryConfigRequest) BLength() int { +func (p *GetTrajectoryConfigResponse) BLength() int { l := 0 if p != nil { l += p.field1Length() - l += p.field2Length() l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *UpsertTrajectoryConfigRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) - offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) - return offset -} - -func (p *UpsertTrajectoryConfigRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { +func (p *GetTrajectoryConfigResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetFilters() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 2) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 1) offset += p.Filters.FastWriteNocopy(buf[offset:], w) } return offset } -func (p *UpsertTrajectoryConfigRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *GetTrajectoryConfigResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetBase() { + if p.IsSetBaseResp() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.Base.FastWriteNocopy(buf[offset:], w) + offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) } return offset } -func (p *UpsertTrajectoryConfigRequest) field1Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - return l -} - -func (p *UpsertTrajectoryConfigRequest) field2Length() int { +func (p *GetTrajectoryConfigResponse) field1Length() int { l := 0 if p.IsSetFilters() { l += thrift.Binary.FieldBeginLength() @@ -14489,23 +15605,21 @@ func (p *UpsertTrajectoryConfigRequest) field2Length() int { return l } -func (p *UpsertTrajectoryConfigRequest) field255Length() int { +func (p *GetTrajectoryConfigResponse) field255Length() int { l := 0 - if p.IsSetBase() { + if p.IsSetBaseResp() { l += thrift.Binary.FieldBeginLength() - l += p.Base.BLength() + l += p.BaseResp.BLength() } return l } -func (p *UpsertTrajectoryConfigRequest) DeepCopy(s interface{}) error { - src, ok := s.(*UpsertTrajectoryConfigRequest) +func (p *GetTrajectoryConfigResponse) DeepCopy(s interface{}) error { + src, ok := s.(*GetTrajectoryConfigResponse) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - p.WorkspaceID = src.WorkspaceID - var _filters *filter.FilterFields if src.Filters != nil { _filters = &filter.FilterFields{} @@ -14515,25 +15629,28 @@ func (p *UpsertTrajectoryConfigRequest) DeepCopy(s interface{}) error { } p.Filters = _filters - var _base *base.Base - if src.Base != nil { - _base = &base.Base{} - if err := _base.DeepCopy(src.Base); err != nil { + var _baseResp *base.BaseResp + if src.BaseResp != nil { + _baseResp = &base.BaseResp{} + if err := _baseResp.DeepCopy(src.BaseResp); err != nil { return err } } - p.Base = _base + p.BaseResp = _baseResp return nil } -func (p *UpsertTrajectoryConfigResponse) FastRead(buf []byte) (int, error) { +func (p *ListTrajectoryRequest) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 + var issetPlatformType bool = false + var issetWorkspaceID bool = false + var issetTraceIds bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -14544,6 +15661,65 @@ func (p *UpsertTrajectoryConfigResponse) FastRead(buf []byte) (int, error) { break } switch fieldId { + case 1: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetPlatformType = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetWorkspaceID = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 3: + if fieldTypeId == thrift.LIST { + l, err = p.FastReadField3(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + issetTraceIds = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 4: + if fieldTypeId == thrift.I64 { + l, err = p.FastReadField4(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } case 255: if fieldTypeId == thrift.STRUCT { l, err = p.FastReadField255(buf[offset:]) @@ -14567,93 +15743,273 @@ func (p *UpsertTrajectoryConfigResponse) FastRead(buf []byte) (int, error) { } } + if !issetPlatformType { + fieldId = 1 + goto RequiredFieldNotSetError + } + + if !issetWorkspaceID { + fieldId = 2 + goto RequiredFieldNotSetError + } + + if !issetTraceIds { + fieldId = 3 + goto RequiredFieldNotSetError + } + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListTrajectoryRequest[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListTrajectoryRequest[fieldId])) +} + +func (p *ListTrajectoryRequest) FastReadField1(buf []byte) (int, error) { + offset := 0 + + var _field common.PlatformType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.PlatformType = _field + return offset, nil +} + +func (p *ListTrajectoryRequest) FastReadField2(buf []byte) (int, error) { + offset := 0 + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = v + } + p.WorkspaceID = _field + return offset, nil +} + +func (p *ListTrajectoryRequest) FastReadField3(buf []byte) (int, error) { + offset := 0 + + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { + return offset, err + } + _field := make([]string, 0, size) + for i := 0; i < size; i++ { + var _elem string + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _elem = v + } + + _field = append(_field, _elem) + } + p.TraceIds = _field + return offset, nil +} + +func (p *ListTrajectoryRequest) FastReadField4(buf []byte) (int, error) { + offset := 0 + + var _field *int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.StartTime = _field return offset, nil -ReadFieldBeginError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) -ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_UpsertTrajectoryConfigResponse[fieldId]), err) -SkipFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) } -func (p *UpsertTrajectoryConfigResponse) FastReadField255(buf []byte) (int, error) { +func (p *ListTrajectoryRequest) FastReadField255(buf []byte) (int, error) { offset := 0 - _field := base.NewBaseResp() + _field := base.NewBase() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l } - p.BaseResp = _field + p.Base = _field return offset, nil } -func (p *UpsertTrajectoryConfigResponse) FastWrite(buf []byte) int { +func (p *ListTrajectoryRequest) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *UpsertTrajectoryConfigResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ListTrajectoryRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField4(buf[offset:], w) + offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *UpsertTrajectoryConfigResponse) BLength() int { +func (p *ListTrajectoryRequest) BLength() int { l := 0 if p != nil { + l += p.field1Length() + l += p.field2Length() + l += p.field3Length() + l += p.field4Length() l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *UpsertTrajectoryConfigResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ListTrajectoryRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetBaseResp() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 1) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.PlatformType) + return offset +} + +func (p *ListTrajectoryRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 2) + offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) + return offset +} + +func (p *ListTrajectoryRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 3) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.TraceIds { + length++ + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, v) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRING, length) + return offset +} + +func (p *ListTrajectoryRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetStartTime() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 4) + offset += thrift.Binary.WriteI64(buf[offset:], *p.StartTime) + } + return offset +} + +func (p *ListTrajectoryRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetBase() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + offset += p.Base.FastWriteNocopy(buf[offset:], w) } return offset } -func (p *UpsertTrajectoryConfigResponse) field255Length() int { +func (p *ListTrajectoryRequest) field1Length() int { l := 0 - if p.IsSetBaseResp() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.StringLengthNocopy(p.PlatformType) + return l +} + +func (p *ListTrajectoryRequest) field2Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *ListTrajectoryRequest) field3Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.TraceIds { + _ = v + l += thrift.Binary.StringLengthNocopy(v) + } + return l +} + +func (p *ListTrajectoryRequest) field4Length() int { + l := 0 + if p.IsSetStartTime() { l += thrift.Binary.FieldBeginLength() - l += p.BaseResp.BLength() + l += thrift.Binary.I64Length() } return l } -func (p *UpsertTrajectoryConfigResponse) DeepCopy(s interface{}) error { - src, ok := s.(*UpsertTrajectoryConfigResponse) +func (p *ListTrajectoryRequest) field255Length() int { + l := 0 + if p.IsSetBase() { + l += thrift.Binary.FieldBeginLength() + l += p.Base.BLength() + } + return l +} + +func (p *ListTrajectoryRequest) DeepCopy(s interface{}) error { + src, ok := s.(*ListTrajectoryRequest) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - var _baseResp *base.BaseResp - if src.BaseResp != nil { - _baseResp = &base.BaseResp{} - if err := _baseResp.DeepCopy(src.BaseResp); err != nil { + p.PlatformType = src.PlatformType + + p.WorkspaceID = src.WorkspaceID + + if src.TraceIds != nil { + p.TraceIds = make([]string, 0, len(src.TraceIds)) + for _, elem := range src.TraceIds { + var _elem string + if elem != "" { + _elem = kutils.StringDeepCopy(elem) + } + p.TraceIds = append(p.TraceIds, _elem) + } + } + + if src.StartTime != nil { + tmp := *src.StartTime + p.StartTime = &tmp + } + + var _base *base.Base + if src.Base != nil { + _base = &base.Base{} + if err := _base.DeepCopy(src.Base); err != nil { return err } } - p.BaseResp = _baseResp + p.Base = _base return nil } -func (p *GetTrajectoryConfigRequest) FastRead(buf []byte) (int, error) { +func (p *ListTrajectoryResponse) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 - var issetWorkspaceID bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -14665,13 +16021,12 @@ func (p *GetTrajectoryConfigRequest) FastRead(buf []byte) (int, error) { } switch fieldId { case 1: - if fieldTypeId == thrift.I64 { + if fieldTypeId == thrift.LIST { l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } - issetWorkspaceID = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -14702,52 +16057,57 @@ func (p *GetTrajectoryConfigRequest) FastRead(buf []byte) (int, error) { } } - if !issetWorkspaceID { - fieldId = 1 - goto RequiredFieldNotSetError - } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_GetTrajectoryConfigRequest[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListTrajectoryResponse[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) -RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_GetTrajectoryConfigRequest[fieldId])) } -func (p *GetTrajectoryConfigRequest) FastReadField1(buf []byte) (int, error) { +func (p *ListTrajectoryResponse) FastReadField1(buf []byte) (int, error) { offset := 0 - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) + offset += l + if err != nil { return offset, err - } else { - offset += l - _field = v } - p.WorkspaceID = _field + _field := make([]*trajectory.Trajectory, 0, size) + values := make([]trajectory.Trajectory, size) + for i := 0; i < size; i++ { + _elem := &values[i] + _elem.InitDefault() + if l, err := _elem.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + + _field = append(_field, _elem) + } + p.Trajectories = _field return offset, nil } -func (p *GetTrajectoryConfigRequest) FastReadField255(buf []byte) (int, error) { +func (p *ListTrajectoryResponse) FastReadField255(buf []byte) (int, error) { offset := 0 - _field := base.NewBase() + _field := base.NewBaseResp() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l } - p.Base = _field + p.BaseResp = _field return offset, nil } -func (p *GetTrajectoryConfigRequest) FastWrite(buf []byte) int { +func (p *ListTrajectoryResponse) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *GetTrajectoryConfigRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ListTrajectoryResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) @@ -14757,7 +16117,7 @@ func (p *GetTrajectoryConfigRequest) FastWriteNocopy(buf []byte, w thrift.Nocopy return offset } -func (p *GetTrajectoryConfigRequest) BLength() int { +func (p *ListTrajectoryResponse) BLength() int { l := 0 if p != nil { l += p.field1Length() @@ -14767,65 +16127,94 @@ func (p *GetTrajectoryConfigRequest) BLength() int { return l } -func (p *GetTrajectoryConfigRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ListTrajectoryResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) - offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) + if p.IsSetTrajectories() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.Trajectories { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) + } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + } return offset } -func (p *GetTrajectoryConfigRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ListTrajectoryResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetBase() { + if p.IsSetBaseResp() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.Base.FastWriteNocopy(buf[offset:], w) + offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) } return offset } -func (p *GetTrajectoryConfigRequest) field1Length() int { +func (p *ListTrajectoryResponse) field1Length() int { l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() + if p.IsSetTrajectories() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.Trajectories { + _ = v + l += v.BLength() + } + } return l } -func (p *GetTrajectoryConfigRequest) field255Length() int { +func (p *ListTrajectoryResponse) field255Length() int { l := 0 - if p.IsSetBase() { + if p.IsSetBaseResp() { l += thrift.Binary.FieldBeginLength() - l += p.Base.BLength() + l += p.BaseResp.BLength() } return l } -func (p *GetTrajectoryConfigRequest) DeepCopy(s interface{}) error { - src, ok := s.(*GetTrajectoryConfigRequest) +func (p *ListTrajectoryResponse) DeepCopy(s interface{}) error { + src, ok := s.(*ListTrajectoryResponse) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - p.WorkspaceID = src.WorkspaceID + if src.Trajectories != nil { + p.Trajectories = make([]*trajectory.Trajectory, 0, len(src.Trajectories)) + for _, elem := range src.Trajectories { + var _elem *trajectory.Trajectory + if elem != nil { + _elem = &trajectory.Trajectory{} + if err := _elem.DeepCopy(elem); err != nil { + return err + } + } - var _base *base.Base - if src.Base != nil { - _base = &base.Base{} - if err := _base.DeepCopy(src.Base); err != nil { + p.Trajectories = append(p.Trajectories, _elem) + } + } + + var _baseResp *base.BaseResp + if src.BaseResp != nil { + _baseResp = &base.BaseResp{} + if err := _baseResp.DeepCopy(src.BaseResp); err != nil { return err } } - p.Base = _base + p.BaseResp = _baseResp return nil } -func (p *GetTrajectoryConfigResponse) FastRead(buf []byte) (int, error) { +func (p *ListMetadataRequest) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 + var issetWorkspaceID bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -14837,12 +16226,41 @@ func (p *GetTrajectoryConfigResponse) FastRead(buf []byte) (int, error) { } switch fieldId { case 1: - if fieldTypeId == thrift.STRUCT { + if fieldTypeId == thrift.I64 { l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } + issetWorkspaceID = true + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 2: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField2(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + case 3: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField3(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -14873,198 +16291,228 @@ func (p *GetTrajectoryConfigResponse) FastRead(buf []byte) (int, error) { } } + if !issetWorkspaceID { + fieldId = 1 + goto RequiredFieldNotSetError + } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_GetTrajectoryConfigResponse[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListMetadataRequest[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListMetadataRequest[fieldId])) } -func (p *GetTrajectoryConfigResponse) FastReadField1(buf []byte) (int, error) { +func (p *ListMetadataRequest) FastReadField1(buf []byte) (int, error) { offset := 0 - _field := filter.NewFilterFields() - if l, err := _field.FastRead(buf[offset:]); err != nil { + + var _field int64 + if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { return offset, err } else { offset += l + _field = v } - p.Filters = _field + p.WorkspaceID = _field return offset, nil } -func (p *GetTrajectoryConfigResponse) FastReadField255(buf []byte) (int, error) { +func (p *ListMetadataRequest) FastReadField2(buf []byte) (int, error) { offset := 0 - _field := base.NewBaseResp() + + var _field *common.PlatformType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.PlatformType = _field + return offset, nil +} + +func (p *ListMetadataRequest) FastReadField3(buf []byte) (int, error) { + offset := 0 + + var _field *common.SpanListType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + _field = &v + } + p.SpanListType = _field + return offset, nil +} + +func (p *ListMetadataRequest) FastReadField255(buf []byte) (int, error) { + offset := 0 + _field := base.NewBase() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { offset += l } - p.BaseResp = _field + p.Base = _field return offset, nil } -func (p *GetTrajectoryConfigResponse) FastWrite(buf []byte) int { +func (p *ListMetadataRequest) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *GetTrajectoryConfigResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ListMetadataRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) + offset += p.fastWriteField3(buf[offset:], w) offset += p.fastWriteField255(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *GetTrajectoryConfigResponse) BLength() int { +func (p *ListMetadataRequest) BLength() int { l := 0 if p != nil { l += p.field1Length() + l += p.field2Length() + l += p.field3Length() l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *GetTrajectoryConfigResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ListMetadataRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetFilters() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 1) - offset += p.Filters.FastWriteNocopy(buf[offset:], w) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 1) + offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) + return offset +} + +func (p *ListMetadataRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetPlatformType() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.PlatformType) } return offset } -func (p *GetTrajectoryConfigResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ListMetadataRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetBaseResp() { + if p.IsSetSpanListType() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 3) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, *p.SpanListType) + } + return offset +} + +func (p *ListMetadataRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetBase() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.BaseResp.FastWriteNocopy(buf[offset:], w) + offset += p.Base.FastWriteNocopy(buf[offset:], w) } return offset } -func (p *GetTrajectoryConfigResponse) field1Length() int { +func (p *ListMetadataRequest) field1Length() int { l := 0 - if p.IsSetFilters() { + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.I64Length() + return l +} + +func (p *ListMetadataRequest) field2Length() int { + l := 0 + if p.IsSetPlatformType() { l += thrift.Binary.FieldBeginLength() - l += p.Filters.BLength() + l += thrift.Binary.StringLengthNocopy(*p.PlatformType) } return l } -func (p *GetTrajectoryConfigResponse) field255Length() int { +func (p *ListMetadataRequest) field3Length() int { l := 0 - if p.IsSetBaseResp() { + if p.IsSetSpanListType() { l += thrift.Binary.FieldBeginLength() - l += p.BaseResp.BLength() + l += thrift.Binary.StringLengthNocopy(*p.SpanListType) } return l } -func (p *GetTrajectoryConfigResponse) DeepCopy(s interface{}) error { - src, ok := s.(*GetTrajectoryConfigResponse) +func (p *ListMetadataRequest) field255Length() int { + l := 0 + if p.IsSetBase() { + l += thrift.Binary.FieldBeginLength() + l += p.Base.BLength() + } + return l +} + +func (p *ListMetadataRequest) DeepCopy(s interface{}) error { + src, ok := s.(*ListMetadataRequest) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - var _filters *filter.FilterFields - if src.Filters != nil { - _filters = &filter.FilterFields{} - if err := _filters.DeepCopy(src.Filters); err != nil { - return err - } - } - p.Filters = _filters + p.WorkspaceID = src.WorkspaceID - var _baseResp *base.BaseResp - if src.BaseResp != nil { - _baseResp = &base.BaseResp{} - if err := _baseResp.DeepCopy(src.BaseResp); err != nil { - return err - } + if src.PlatformType != nil { + tmp := *src.PlatformType + p.PlatformType = &tmp } - p.BaseResp = _baseResp - - return nil -} - -func (p *ListTrajectoryRequest) FastRead(buf []byte) (int, error) { - var err error - var offset int - var l int - var fieldTypeId thrift.TType - var fieldId int16 - var issetPlatformType bool = false - var issetWorkspaceID bool = false - var issetTraceIds bool = false - for { - fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldBeginError - } - if fieldTypeId == thrift.STOP { - break - } - switch fieldId { - case 1: - if fieldTypeId == thrift.STRING { - l, err = p.FastReadField1(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetPlatformType = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 2: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField2(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetWorkspaceID = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 3: - if fieldTypeId == thrift.LIST { - l, err = p.FastReadField3(buf[offset:]) - offset += l - if err != nil { - goto ReadFieldError - } - issetTraceIds = true - } else { - l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) - offset += l - if err != nil { - goto SkipFieldError - } - } - case 4: - if fieldTypeId == thrift.I64 { - l, err = p.FastReadField4(buf[offset:]) + if src.SpanListType != nil { + tmp := *src.SpanListType + p.SpanListType = &tmp + } + + var _base *base.Base + if src.Base != nil { + _base = &base.Base{} + if err := _base.DeepCopy(src.Base); err != nil { + return err + } + } + p.Base = _base + + return nil +} + +func (p *MetadataItemInfo) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + var issetKey bool = false + var issetValueType bool = false + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField1(buf[offset:]) offset += l if err != nil { goto ReadFieldError } + issetKey = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -15072,13 +16520,14 @@ func (p *ListTrajectoryRequest) FastRead(buf []byte) (int, error) { goto SkipFieldError } } - case 255: - if fieldTypeId == thrift.STRUCT { - l, err = p.FastReadField255(buf[offset:]) + case 2: + if fieldTypeId == thrift.STRING { + l, err = p.FastReadField2(buf[offset:]) offset += l if err != nil { goto ReadFieldError } + issetValueType = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -15095,273 +16544,129 @@ func (p *ListTrajectoryRequest) FastRead(buf []byte) (int, error) { } } - if !issetPlatformType { + if !issetKey { fieldId = 1 goto RequiredFieldNotSetError } - if !issetWorkspaceID { + if !issetValueType { fieldId = 2 goto RequiredFieldNotSetError } - - if !issetTraceIds { - fieldId = 3 - goto RequiredFieldNotSetError - } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListTrajectoryRequest[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_MetadataItemInfo[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) RequiredFieldNotSetError: - return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListTrajectoryRequest[fieldId])) + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_MetadataItemInfo[fieldId])) } -func (p *ListTrajectoryRequest) FastReadField1(buf []byte) (int, error) { +func (p *MetadataItemInfo) FastReadField1(buf []byte) (int, error) { offset := 0 - var _field common.PlatformType + var _field string if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { return offset, err } else { offset += l _field = v } - p.PlatformType = _field + p.Key = _field return offset, nil } -func (p *ListTrajectoryRequest) FastReadField2(buf []byte) (int, error) { +func (p *MetadataItemInfo) FastReadField2(buf []byte) (int, error) { offset := 0 - var _field int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { + var _field span.MetadataValueType + if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { return offset, err } else { offset += l _field = v } - p.WorkspaceID = _field - return offset, nil -} - -func (p *ListTrajectoryRequest) FastReadField3(buf []byte) (int, error) { - offset := 0 - - _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) - offset += l - if err != nil { - return offset, err - } - _field := make([]string, 0, size) - for i := 0; i < size; i++ { - var _elem string - if v, l, err := thrift.Binary.ReadString(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _elem = v - } - - _field = append(_field, _elem) - } - p.TraceIds = _field - return offset, nil -} - -func (p *ListTrajectoryRequest) FastReadField4(buf []byte) (int, error) { - offset := 0 - - var _field *int64 - if v, l, err := thrift.Binary.ReadI64(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - _field = &v - } - p.StartTime = _field - return offset, nil -} - -func (p *ListTrajectoryRequest) FastReadField255(buf []byte) (int, error) { - offset := 0 - _field := base.NewBase() - if l, err := _field.FastRead(buf[offset:]); err != nil { - return offset, err - } else { - offset += l - } - p.Base = _field + p.ValueType = _field return offset, nil } -func (p *ListTrajectoryRequest) FastWrite(buf []byte) int { +func (p *MetadataItemInfo) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ListTrajectoryRequest) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *MetadataItemInfo) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { - offset += p.fastWriteField2(buf[offset:], w) - offset += p.fastWriteField4(buf[offset:], w) offset += p.fastWriteField1(buf[offset:], w) - offset += p.fastWriteField3(buf[offset:], w) - offset += p.fastWriteField255(buf[offset:], w) + offset += p.fastWriteField2(buf[offset:], w) } offset += thrift.Binary.WriteFieldStop(buf[offset:]) return offset } -func (p *ListTrajectoryRequest) BLength() int { +func (p *MetadataItemInfo) BLength() int { l := 0 if p != nil { l += p.field1Length() l += p.field2Length() - l += p.field3Length() - l += p.field4Length() - l += p.field255Length() } l += thrift.Binary.FieldStopLength() return l } -func (p *ListTrajectoryRequest) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *MetadataItemInfo) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 1) - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.PlatformType) - return offset -} - -func (p *ListTrajectoryRequest) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 2) - offset += thrift.Binary.WriteI64(buf[offset:], p.WorkspaceID) - return offset -} - -func (p *ListTrajectoryRequest) fastWriteField3(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 3) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.TraceIds { - length++ - offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, v) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRING, length) - return offset -} - -func (p *ListTrajectoryRequest) fastWriteField4(buf []byte, w thrift.NocopyWriter) int { - offset := 0 - if p.IsSetStartTime() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.I64, 4) - offset += thrift.Binary.WriteI64(buf[offset:], *p.StartTime) - } + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.Key) return offset } -func (p *ListTrajectoryRequest) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *MetadataItemInfo) fastWriteField2(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetBase() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) - offset += p.Base.FastWriteNocopy(buf[offset:], w) - } + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRING, 2) + offset += thrift.Binary.WriteStringNocopy(buf[offset:], w, p.ValueType) return offset } -func (p *ListTrajectoryRequest) field1Length() int { - l := 0 - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.StringLengthNocopy(p.PlatformType) - return l -} - -func (p *ListTrajectoryRequest) field2Length() int { +func (p *MetadataItemInfo) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() + l += thrift.Binary.StringLengthNocopy(p.Key) return l } -func (p *ListTrajectoryRequest) field3Length() int { +func (p *MetadataItemInfo) field2Length() int { l := 0 l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.TraceIds { - _ = v - l += thrift.Binary.StringLengthNocopy(v) - } - return l -} - -func (p *ListTrajectoryRequest) field4Length() int { - l := 0 - if p.IsSetStartTime() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.I64Length() - } - return l -} - -func (p *ListTrajectoryRequest) field255Length() int { - l := 0 - if p.IsSetBase() { - l += thrift.Binary.FieldBeginLength() - l += p.Base.BLength() - } + l += thrift.Binary.StringLengthNocopy(p.ValueType) return l } -func (p *ListTrajectoryRequest) DeepCopy(s interface{}) error { - src, ok := s.(*ListTrajectoryRequest) +func (p *MetadataItemInfo) DeepCopy(s interface{}) error { + src, ok := s.(*MetadataItemInfo) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - p.PlatformType = src.PlatformType - - p.WorkspaceID = src.WorkspaceID - - if src.TraceIds != nil { - p.TraceIds = make([]string, 0, len(src.TraceIds)) - for _, elem := range src.TraceIds { - var _elem string - if elem != "" { - _elem = kutils.StringDeepCopy(elem) - } - p.TraceIds = append(p.TraceIds, _elem) - } - } - - if src.StartTime != nil { - tmp := *src.StartTime - p.StartTime = &tmp + if src.Key != "" { + p.Key = kutils.StringDeepCopy(src.Key) } - var _base *base.Base - if src.Base != nil { - _base = &base.Base{} - if err := _base.DeepCopy(src.Base); err != nil { - return err - } - } - p.Base = _base + p.ValueType = src.ValueType return nil } -func (p *ListTrajectoryResponse) FastRead(buf []byte) (int, error) { +func (p *ListMetadataResponse) FastRead(buf []byte) (int, error) { var err error var offset int var l int var fieldTypeId thrift.TType var fieldId int16 + var issetMetadataItemList bool = false for { fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) offset += l @@ -15379,6 +16684,7 @@ func (p *ListTrajectoryResponse) FastRead(buf []byte) (int, error) { if err != nil { goto ReadFieldError } + issetMetadataItemList = true } else { l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) offset += l @@ -15409,16 +16715,22 @@ func (p *ListTrajectoryResponse) FastRead(buf []byte) (int, error) { } } + if !issetMetadataItemList { + fieldId = 1 + goto RequiredFieldNotSetError + } return offset, nil ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListTrajectoryResponse[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_ListMetadataResponse[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +RequiredFieldNotSetError: + return offset, thrift.NewProtocolException(thrift.INVALID_DATA, fmt.Sprintf("required field %s is not set", fieldIDToName_ListMetadataResponse[fieldId])) } -func (p *ListTrajectoryResponse) FastReadField1(buf []byte) (int, error) { +func (p *ListMetadataResponse) FastReadField1(buf []byte) (int, error) { offset := 0 _, size, l, err := thrift.Binary.ReadListBegin(buf[offset:]) @@ -15426,8 +16738,8 @@ func (p *ListTrajectoryResponse) FastReadField1(buf []byte) (int, error) { if err != nil { return offset, err } - _field := make([]*trajectory.Trajectory, 0, size) - values := make([]trajectory.Trajectory, size) + _field := make([]*MetadataItemInfo, 0, size) + values := make([]MetadataItemInfo, size) for i := 0; i < size; i++ { _elem := &values[i] _elem.InitDefault() @@ -15439,11 +16751,11 @@ func (p *ListTrajectoryResponse) FastReadField1(buf []byte) (int, error) { _field = append(_field, _elem) } - p.Trajectories = _field + p.MetadataItemList = _field return offset, nil } -func (p *ListTrajectoryResponse) FastReadField255(buf []byte) (int, error) { +func (p *ListMetadataResponse) FastReadField255(buf []byte) (int, error) { offset := 0 _field := base.NewBaseResp() if l, err := _field.FastRead(buf[offset:]); err != nil { @@ -15455,11 +16767,11 @@ func (p *ListTrajectoryResponse) FastReadField255(buf []byte) (int, error) { return offset, nil } -func (p *ListTrajectoryResponse) FastWrite(buf []byte) int { +func (p *ListMetadataResponse) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *ListTrajectoryResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *ListMetadataResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) @@ -15469,7 +16781,7 @@ func (p *ListTrajectoryResponse) FastWriteNocopy(buf []byte, w thrift.NocopyWrit return offset } -func (p *ListTrajectoryResponse) BLength() int { +func (p *ListMetadataResponse) BLength() int { l := 0 if p != nil { l += p.field1Length() @@ -15479,23 +16791,21 @@ func (p *ListTrajectoryResponse) BLength() int { return l } -func (p *ListTrajectoryResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *ListMetadataResponse) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 - if p.IsSetTrajectories() { - offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) - listBeginOffset := offset - offset += thrift.Binary.ListBeginLength() - var length int - for _, v := range p.Trajectories { - length++ - offset += v.FastWriteNocopy(buf[offset:], w) - } - thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.LIST, 1) + listBeginOffset := offset + offset += thrift.Binary.ListBeginLength() + var length int + for _, v := range p.MetadataItemList { + length++ + offset += v.FastWriteNocopy(buf[offset:], w) } + thrift.Binary.WriteListBegin(buf[listBeginOffset:], thrift.STRUCT, length) return offset } -func (p *ListTrajectoryResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { +func (p *ListMetadataResponse) fastWriteField255(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetBaseResp() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 255) @@ -15504,20 +16814,18 @@ func (p *ListTrajectoryResponse) fastWriteField255(buf []byte, w thrift.NocopyWr return offset } -func (p *ListTrajectoryResponse) field1Length() int { +func (p *ListMetadataResponse) field1Length() int { l := 0 - if p.IsSetTrajectories() { - l += thrift.Binary.FieldBeginLength() - l += thrift.Binary.ListBeginLength() - for _, v := range p.Trajectories { - _ = v - l += v.BLength() - } + l += thrift.Binary.FieldBeginLength() + l += thrift.Binary.ListBeginLength() + for _, v := range p.MetadataItemList { + _ = v + l += v.BLength() } return l } -func (p *ListTrajectoryResponse) field255Length() int { +func (p *ListMetadataResponse) field255Length() int { l := 0 if p.IsSetBaseResp() { l += thrift.Binary.FieldBeginLength() @@ -15526,24 +16834,24 @@ func (p *ListTrajectoryResponse) field255Length() int { return l } -func (p *ListTrajectoryResponse) DeepCopy(s interface{}) error { - src, ok := s.(*ListTrajectoryResponse) +func (p *ListMetadataResponse) DeepCopy(s interface{}) error { + src, ok := s.(*ListMetadataResponse) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - if src.Trajectories != nil { - p.Trajectories = make([]*trajectory.Trajectory, 0, len(src.Trajectories)) - for _, elem := range src.Trajectories { - var _elem *trajectory.Trajectory + if src.MetadataItemList != nil { + p.MetadataItemList = make([]*MetadataItemInfo, 0, len(src.MetadataItemList)) + for _, elem := range src.MetadataItemList { + var _elem *MetadataItemInfo if elem != nil { - _elem = &trajectory.Trajectory{} + _elem = &MetadataItemInfo{} if err := _elem.DeepCopy(elem); err != nil { return err } } - p.Trajectories = append(p.Trajectories, _elem) + p.MetadataItemList = append(p.MetadataItemList, _elem) } } @@ -18808,7 +20116,241 @@ func (p *TraceServiceDeleteManualAnnotationResult) fastWriteField0(buf []byte, w return offset } -func (p *TraceServiceDeleteManualAnnotationResult) field0Length() int { +func (p *TraceServiceDeleteManualAnnotationResult) field0Length() int { + l := 0 + if p.IsSetSuccess() { + l += thrift.Binary.FieldBeginLength() + l += p.Success.BLength() + } + return l +} + +func (p *TraceServiceDeleteManualAnnotationResult) DeepCopy(s interface{}) error { + src, ok := s.(*TraceServiceDeleteManualAnnotationResult) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + var _success *DeleteManualAnnotationResponse + if src.Success != nil { + _success = &DeleteManualAnnotationResponse{} + if err := _success.DeepCopy(src.Success); err != nil { + return err + } + } + p.Success = _success + + return nil +} + +func (p *TraceServiceListAnnotationsArgs) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationsArgs[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +} + +func (p *TraceServiceListAnnotationsArgs) FastReadField1(buf []byte) (int, error) { + offset := 0 + _field := NewListAnnotationsRequest() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.Req = _field + return offset, nil +} + +func (p *TraceServiceListAnnotationsArgs) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *TraceServiceListAnnotationsArgs) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *TraceServiceListAnnotationsArgs) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *TraceServiceListAnnotationsArgs) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 1) + offset += p.Req.FastWriteNocopy(buf[offset:], w) + return offset +} + +func (p *TraceServiceListAnnotationsArgs) field1Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += p.Req.BLength() + return l +} + +func (p *TraceServiceListAnnotationsArgs) DeepCopy(s interface{}) error { + src, ok := s.(*TraceServiceListAnnotationsArgs) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + var _req *ListAnnotationsRequest + if src.Req != nil { + _req = &ListAnnotationsRequest{} + if err := _req.DeepCopy(src.Req); err != nil { + return err + } + } + p.Req = _req + + return nil +} + +func (p *TraceServiceListAnnotationsResult) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 0: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField0(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationsResult[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +} + +func (p *TraceServiceListAnnotationsResult) FastReadField0(buf []byte) (int, error) { + offset := 0 + _field := NewListAnnotationsResponse() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.Success = _field + return offset, nil +} + +func (p *TraceServiceListAnnotationsResult) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *TraceServiceListAnnotationsResult) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField0(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *TraceServiceListAnnotationsResult) BLength() int { + l := 0 + if p != nil { + l += p.field0Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *TraceServiceListAnnotationsResult) fastWriteField0(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetSuccess() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 0) + offset += p.Success.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *TraceServiceListAnnotationsResult) field0Length() int { l := 0 if p.IsSetSuccess() { l += thrift.Binary.FieldBeginLength() @@ -18817,15 +20359,15 @@ func (p *TraceServiceDeleteManualAnnotationResult) field0Length() int { return l } -func (p *TraceServiceDeleteManualAnnotationResult) DeepCopy(s interface{}) error { - src, ok := s.(*TraceServiceDeleteManualAnnotationResult) +func (p *TraceServiceListAnnotationsResult) DeepCopy(s interface{}) error { + src, ok := s.(*TraceServiceListAnnotationsResult) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - var _success *DeleteManualAnnotationResponse + var _success *ListAnnotationsResponse if src.Success != nil { - _success = &DeleteManualAnnotationResponse{} + _success = &ListAnnotationsResponse{} if err := _success.DeepCopy(src.Success); err != nil { return err } @@ -18835,7 +20377,7 @@ func (p *TraceServiceDeleteManualAnnotationResult) DeepCopy(s interface{}) error return nil } -func (p *TraceServiceListAnnotationsArgs) FastRead(buf []byte) (int, error) { +func (p *TraceServiceListWorkspaceAnnotationsArgs) FastRead(buf []byte) (int, error) { var err error var offset int @@ -18879,14 +20421,14 @@ func (p *TraceServiceListAnnotationsArgs) FastRead(buf []byte) (int, error) { ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationsArgs[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListWorkspaceAnnotationsArgs[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) } -func (p *TraceServiceListAnnotationsArgs) FastReadField1(buf []byte) (int, error) { +func (p *TraceServiceListWorkspaceAnnotationsArgs) FastReadField1(buf []byte) (int, error) { offset := 0 - _field := NewListAnnotationsRequest() + _field := NewListWorkspaceAnnotationsRequest() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { @@ -18896,11 +20438,11 @@ func (p *TraceServiceListAnnotationsArgs) FastReadField1(buf []byte) (int, error return offset, nil } -func (p *TraceServiceListAnnotationsArgs) FastWrite(buf []byte) int { +func (p *TraceServiceListWorkspaceAnnotationsArgs) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *TraceServiceListAnnotationsArgs) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *TraceServiceListWorkspaceAnnotationsArgs) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField1(buf[offset:], w) @@ -18909,7 +20451,7 @@ func (p *TraceServiceListAnnotationsArgs) FastWriteNocopy(buf []byte, w thrift.N return offset } -func (p *TraceServiceListAnnotationsArgs) BLength() int { +func (p *TraceServiceListWorkspaceAnnotationsArgs) BLength() int { l := 0 if p != nil { l += p.field1Length() @@ -18918,29 +20460,29 @@ func (p *TraceServiceListAnnotationsArgs) BLength() int { return l } -func (p *TraceServiceListAnnotationsArgs) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { +func (p *TraceServiceListWorkspaceAnnotationsArgs) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { offset := 0 offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 1) offset += p.Req.FastWriteNocopy(buf[offset:], w) return offset } -func (p *TraceServiceListAnnotationsArgs) field1Length() int { +func (p *TraceServiceListWorkspaceAnnotationsArgs) field1Length() int { l := 0 l += thrift.Binary.FieldBeginLength() l += p.Req.BLength() return l } -func (p *TraceServiceListAnnotationsArgs) DeepCopy(s interface{}) error { - src, ok := s.(*TraceServiceListAnnotationsArgs) +func (p *TraceServiceListWorkspaceAnnotationsArgs) DeepCopy(s interface{}) error { + src, ok := s.(*TraceServiceListWorkspaceAnnotationsArgs) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - var _req *ListAnnotationsRequest + var _req *ListWorkspaceAnnotationsRequest if src.Req != nil { - _req = &ListAnnotationsRequest{} + _req = &ListWorkspaceAnnotationsRequest{} if err := _req.DeepCopy(src.Req); err != nil { return err } @@ -18950,7 +20492,7 @@ func (p *TraceServiceListAnnotationsArgs) DeepCopy(s interface{}) error { return nil } -func (p *TraceServiceListAnnotationsResult) FastRead(buf []byte) (int, error) { +func (p *TraceServiceListWorkspaceAnnotationsResult) FastRead(buf []byte) (int, error) { var err error var offset int @@ -18994,14 +20536,14 @@ func (p *TraceServiceListAnnotationsResult) FastRead(buf []byte) (int, error) { ReadFieldBeginError: return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) ReadFieldError: - return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListAnnotationsResult[fieldId]), err) + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListWorkspaceAnnotationsResult[fieldId]), err) SkipFieldError: return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) } -func (p *TraceServiceListAnnotationsResult) FastReadField0(buf []byte) (int, error) { +func (p *TraceServiceListWorkspaceAnnotationsResult) FastReadField0(buf []byte) (int, error) { offset := 0 - _field := NewListAnnotationsResponse() + _field := NewListWorkspaceAnnotationsResponse() if l, err := _field.FastRead(buf[offset:]); err != nil { return offset, err } else { @@ -19011,11 +20553,11 @@ func (p *TraceServiceListAnnotationsResult) FastReadField0(buf []byte) (int, err return offset, nil } -func (p *TraceServiceListAnnotationsResult) FastWrite(buf []byte) int { +func (p *TraceServiceListWorkspaceAnnotationsResult) FastWrite(buf []byte) int { return p.FastWriteNocopy(buf, nil) } -func (p *TraceServiceListAnnotationsResult) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { +func (p *TraceServiceListWorkspaceAnnotationsResult) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p != nil { offset += p.fastWriteField0(buf[offset:], w) @@ -19024,7 +20566,7 @@ func (p *TraceServiceListAnnotationsResult) FastWriteNocopy(buf []byte, w thrift return offset } -func (p *TraceServiceListAnnotationsResult) BLength() int { +func (p *TraceServiceListWorkspaceAnnotationsResult) BLength() int { l := 0 if p != nil { l += p.field0Length() @@ -19033,7 +20575,7 @@ func (p *TraceServiceListAnnotationsResult) BLength() int { return l } -func (p *TraceServiceListAnnotationsResult) fastWriteField0(buf []byte, w thrift.NocopyWriter) int { +func (p *TraceServiceListWorkspaceAnnotationsResult) fastWriteField0(buf []byte, w thrift.NocopyWriter) int { offset := 0 if p.IsSetSuccess() { offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 0) @@ -19042,7 +20584,7 @@ func (p *TraceServiceListAnnotationsResult) fastWriteField0(buf []byte, w thrift return offset } -func (p *TraceServiceListAnnotationsResult) field0Length() int { +func (p *TraceServiceListWorkspaceAnnotationsResult) field0Length() int { l := 0 if p.IsSetSuccess() { l += thrift.Binary.FieldBeginLength() @@ -19051,15 +20593,15 @@ func (p *TraceServiceListAnnotationsResult) field0Length() int { return l } -func (p *TraceServiceListAnnotationsResult) DeepCopy(s interface{}) error { - src, ok := s.(*TraceServiceListAnnotationsResult) +func (p *TraceServiceListWorkspaceAnnotationsResult) DeepCopy(s interface{}) error { + src, ok := s.(*TraceServiceListWorkspaceAnnotationsResult) if !ok { return fmt.Errorf("%T's type not matched %T", s, p) } - var _success *ListAnnotationsResponse + var _success *ListWorkspaceAnnotationsResponse if src.Success != nil { - _success = &ListAnnotationsResponse{} + _success = &ListWorkspaceAnnotationsResponse{} if err := _success.DeepCopy(src.Success); err != nil { return err } @@ -20941,6 +22483,240 @@ func (p *TraceServiceListTrajectoryResult) DeepCopy(s interface{}) error { return nil } +func (p *TraceServiceListMetadataArgs) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 1: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField1(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListMetadataArgs[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +} + +func (p *TraceServiceListMetadataArgs) FastReadField1(buf []byte) (int, error) { + offset := 0 + _field := NewListMetadataRequest() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.Req = _field + return offset, nil +} + +func (p *TraceServiceListMetadataArgs) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *TraceServiceListMetadataArgs) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField1(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *TraceServiceListMetadataArgs) BLength() int { + l := 0 + if p != nil { + l += p.field1Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *TraceServiceListMetadataArgs) fastWriteField1(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 1) + offset += p.Req.FastWriteNocopy(buf[offset:], w) + return offset +} + +func (p *TraceServiceListMetadataArgs) field1Length() int { + l := 0 + l += thrift.Binary.FieldBeginLength() + l += p.Req.BLength() + return l +} + +func (p *TraceServiceListMetadataArgs) DeepCopy(s interface{}) error { + src, ok := s.(*TraceServiceListMetadataArgs) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + var _req *ListMetadataRequest + if src.Req != nil { + _req = &ListMetadataRequest{} + if err := _req.DeepCopy(src.Req); err != nil { + return err + } + } + p.Req = _req + + return nil +} + +func (p *TraceServiceListMetadataResult) FastRead(buf []byte) (int, error) { + + var err error + var offset int + var l int + var fieldTypeId thrift.TType + var fieldId int16 + for { + fieldTypeId, fieldId, l, err = thrift.Binary.ReadFieldBegin(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldBeginError + } + if fieldTypeId == thrift.STOP { + break + } + switch fieldId { + case 0: + if fieldTypeId == thrift.STRUCT { + l, err = p.FastReadField0(buf[offset:]) + offset += l + if err != nil { + goto ReadFieldError + } + } else { + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + default: + l, err = thrift.Binary.Skip(buf[offset:], fieldTypeId) + offset += l + if err != nil { + goto SkipFieldError + } + } + } + + return offset, nil +ReadFieldBeginError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d begin error: ", p, fieldId), err) +ReadFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T read field %d '%s' error: ", p, fieldId, fieldIDToName_TraceServiceListMetadataResult[fieldId]), err) +SkipFieldError: + return offset, thrift.PrependError(fmt.Sprintf("%T field %d skip type %d error: ", p, fieldId, fieldTypeId), err) +} + +func (p *TraceServiceListMetadataResult) FastReadField0(buf []byte) (int, error) { + offset := 0 + _field := NewListMetadataResponse() + if l, err := _field.FastRead(buf[offset:]); err != nil { + return offset, err + } else { + offset += l + } + p.Success = _field + return offset, nil +} + +func (p *TraceServiceListMetadataResult) FastWrite(buf []byte) int { + return p.FastWriteNocopy(buf, nil) +} + +func (p *TraceServiceListMetadataResult) FastWriteNocopy(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p != nil { + offset += p.fastWriteField0(buf[offset:], w) + } + offset += thrift.Binary.WriteFieldStop(buf[offset:]) + return offset +} + +func (p *TraceServiceListMetadataResult) BLength() int { + l := 0 + if p != nil { + l += p.field0Length() + } + l += thrift.Binary.FieldStopLength() + return l +} + +func (p *TraceServiceListMetadataResult) fastWriteField0(buf []byte, w thrift.NocopyWriter) int { + offset := 0 + if p.IsSetSuccess() { + offset += thrift.Binary.WriteFieldBegin(buf[offset:], thrift.STRUCT, 0) + offset += p.Success.FastWriteNocopy(buf[offset:], w) + } + return offset +} + +func (p *TraceServiceListMetadataResult) field0Length() int { + l := 0 + if p.IsSetSuccess() { + l += thrift.Binary.FieldBeginLength() + l += p.Success.BLength() + } + return l +} + +func (p *TraceServiceListMetadataResult) DeepCopy(s interface{}) error { + src, ok := s.(*TraceServiceListMetadataResult) + if !ok { + return fmt.Errorf("%T's type not matched %T", s, p) + } + + var _success *ListMetadataResponse + if src.Success != nil { + _success = &ListMetadataResponse{} + if err := _success.DeepCopy(src.Success); err != nil { + return err + } + } + p.Success = _success + + return nil +} + func (p *TraceServiceListSpansArgs) GetFirstArgument() interface{} { return p.Req } @@ -21061,6 +22837,14 @@ func (p *TraceServiceListAnnotationsResult) GetResult() interface{} { return p.Success } +func (p *TraceServiceListWorkspaceAnnotationsArgs) GetFirstArgument() interface{} { + return p.Req +} + +func (p *TraceServiceListWorkspaceAnnotationsResult) GetResult() interface{} { + return p.Success +} + func (p *TraceServiceExportTracesToDatasetArgs) GetFirstArgument() interface{} { return p.Req } @@ -21124,3 +22908,11 @@ func (p *TraceServiceListTrajectoryArgs) GetFirstArgument() interface{} { func (p *TraceServiceListTrajectoryResult) GetResult() interface{} { return p.Success } + +func (p *TraceServiceListMetadataArgs) GetFirstArgument() interface{} { + return p.Req +} + +func (p *TraceServiceListMetadataResult) GetResult() interface{} { + return p.Success +} diff --git a/backend/kitex_gen/coze/loop/observability/trace/traceservice/client.go b/backend/kitex_gen/coze/loop/observability/trace/traceservice/client.go index c069a0593..60262c39c 100644 --- a/backend/kitex_gen/coze/loop/observability/trace/traceservice/client.go +++ b/backend/kitex_gen/coze/loop/observability/trace/traceservice/client.go @@ -26,6 +26,7 @@ type Client interface { UpdateManualAnnotation(ctx context.Context, req *trace.UpdateManualAnnotationRequest, callOptions ...callopt.Option) (r *trace.UpdateManualAnnotationResponse, err error) DeleteManualAnnotation(ctx context.Context, req *trace.DeleteManualAnnotationRequest, callOptions ...callopt.Option) (r *trace.DeleteManualAnnotationResponse, err error) ListAnnotations(ctx context.Context, req *trace.ListAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListAnnotationsResponse, err error) + ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListWorkspaceAnnotationsResponse, err error) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.ExportTracesToDatasetResponse, err error) PreviewExportTracesToDataset(ctx context.Context, req *trace.PreviewExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.PreviewExportTracesToDatasetResponse, err error) ChangeEvaluatorScore(ctx context.Context, req *trace.ChangeEvaluatorScoreRequest, callOptions ...callopt.Option) (r *trace.ChangeEvaluatorScoreResponse, err error) @@ -34,6 +35,7 @@ type Client interface { UpsertTrajectoryConfig(ctx context.Context, req *trace.UpsertTrajectoryConfigRequest, callOptions ...callopt.Option) (r *trace.UpsertTrajectoryConfigResponse, err error) GetTrajectoryConfig(ctx context.Context, req *trace.GetTrajectoryConfigRequest, callOptions ...callopt.Option) (r *trace.GetTrajectoryConfigResponse, err error) ListTrajectory(ctx context.Context, req *trace.ListTrajectoryRequest, callOptions ...callopt.Option) (r *trace.ListTrajectoryResponse, err error) + ListMetadata(ctx context.Context, req *trace.ListMetadataRequest, callOptions ...callopt.Option) (r *trace.ListMetadataResponse, err error) } // NewClient creates a client for the service defined in IDL. @@ -140,6 +142,11 @@ func (p *kTraceServiceClient) ListAnnotations(ctx context.Context, req *trace.Li return p.kClient.ListAnnotations(ctx, req) } +func (p *kTraceServiceClient) ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest, callOptions ...callopt.Option) (r *trace.ListWorkspaceAnnotationsResponse, err error) { + ctx = client.NewCtxWithCallOptions(ctx, callOptions) + return p.kClient.ListWorkspaceAnnotations(ctx, req) +} + func (p *kTraceServiceClient) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest, callOptions ...callopt.Option) (r *trace.ExportTracesToDatasetResponse, err error) { ctx = client.NewCtxWithCallOptions(ctx, callOptions) return p.kClient.ExportTracesToDataset(ctx, req) @@ -179,3 +186,8 @@ func (p *kTraceServiceClient) ListTrajectory(ctx context.Context, req *trace.Lis ctx = client.NewCtxWithCallOptions(ctx, callOptions) return p.kClient.ListTrajectory(ctx, req) } + +func (p *kTraceServiceClient) ListMetadata(ctx context.Context, req *trace.ListMetadataRequest, callOptions ...callopt.Option) (r *trace.ListMetadataResponse, err error) { + ctx = client.NewCtxWithCallOptions(ctx, callOptions) + return p.kClient.ListMetadata(ctx, req) +} diff --git a/backend/kitex_gen/coze/loop/observability/trace/traceservice/traceservice.go b/backend/kitex_gen/coze/loop/observability/trace/traceservice/traceservice.go index 5b9655007..183301476 100644 --- a/backend/kitex_gen/coze/loop/observability/trace/traceservice/traceservice.go +++ b/backend/kitex_gen/coze/loop/observability/trace/traceservice/traceservice.go @@ -118,6 +118,13 @@ var serviceMethods = map[string]kitex.MethodInfo{ false, kitex.WithStreamingMode(kitex.StreamingNone), ), + "ListWorkspaceAnnotations": kitex.NewMethodInfo( + listWorkspaceAnnotationsHandler, + newTraceServiceListWorkspaceAnnotationsArgs, + newTraceServiceListWorkspaceAnnotationsResult, + false, + kitex.WithStreamingMode(kitex.StreamingNone), + ), "ExportTracesToDataset": kitex.NewMethodInfo( exportTracesToDatasetHandler, newTraceServiceExportTracesToDatasetArgs, @@ -174,6 +181,13 @@ var serviceMethods = map[string]kitex.MethodInfo{ false, kitex.WithStreamingMode(kitex.StreamingNone), ), + "ListMetadata": kitex.NewMethodInfo( + listMetadataHandler, + newTraceServiceListMetadataArgs, + newTraceServiceListMetadataResult, + false, + kitex.WithStreamingMode(kitex.StreamingNone), + ), } var ( @@ -492,6 +506,25 @@ func newTraceServiceListAnnotationsResult() interface{} { return trace.NewTraceServiceListAnnotationsResult() } +func listWorkspaceAnnotationsHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { + realArg := arg.(*trace.TraceServiceListWorkspaceAnnotationsArgs) + realResult := result.(*trace.TraceServiceListWorkspaceAnnotationsResult) + success, err := handler.(trace.TraceService).ListWorkspaceAnnotations(ctx, realArg.Req) + if err != nil { + return err + } + realResult.Success = success + return nil +} + +func newTraceServiceListWorkspaceAnnotationsArgs() interface{} { + return trace.NewTraceServiceListWorkspaceAnnotationsArgs() +} + +func newTraceServiceListWorkspaceAnnotationsResult() interface{} { + return trace.NewTraceServiceListWorkspaceAnnotationsResult() +} + func exportTracesToDatasetHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { realArg := arg.(*trace.TraceServiceExportTracesToDatasetArgs) realResult := result.(*trace.TraceServiceExportTracesToDatasetResult) @@ -644,6 +677,25 @@ func newTraceServiceListTrajectoryResult() interface{} { return trace.NewTraceServiceListTrajectoryResult() } +func listMetadataHandler(ctx context.Context, handler interface{}, arg, result interface{}) error { + realArg := arg.(*trace.TraceServiceListMetadataArgs) + realResult := result.(*trace.TraceServiceListMetadataResult) + success, err := handler.(trace.TraceService).ListMetadata(ctx, realArg.Req) + if err != nil { + return err + } + realResult.Success = success + return nil +} + +func newTraceServiceListMetadataArgs() interface{} { + return trace.NewTraceServiceListMetadataArgs() +} + +func newTraceServiceListMetadataResult() interface{} { + return trace.NewTraceServiceListMetadataResult() +} + type kClient struct { c client.Client sc client.Streaming @@ -806,6 +858,16 @@ func (p *kClient) ListAnnotations(ctx context.Context, req *trace.ListAnnotation return _result.GetSuccess(), nil } +func (p *kClient) ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest) (r *trace.ListWorkspaceAnnotationsResponse, err error) { + var _args trace.TraceServiceListWorkspaceAnnotationsArgs + _args.Req = req + var _result trace.TraceServiceListWorkspaceAnnotationsResult + if err = p.c.Call(ctx, "ListWorkspaceAnnotations", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} + func (p *kClient) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest) (r *trace.ExportTracesToDatasetResponse, err error) { var _args trace.TraceServiceExportTracesToDatasetArgs _args.Req = req @@ -885,3 +947,13 @@ func (p *kClient) ListTrajectory(ctx context.Context, req *trace.ListTrajectoryR } return _result.GetSuccess(), nil } + +func (p *kClient) ListMetadata(ctx context.Context, req *trace.ListMetadataRequest) (r *trace.ListMetadataResponse, err error) { + var _args trace.TraceServiceListMetadataArgs + _args.Req = req + var _result trace.TraceServiceListMetadataResult + if err = p.c.Call(ctx, "ListMetadata", &_args, &_result); err != nil { + return + } + return _result.GetSuccess(), nil +} diff --git a/backend/loop_gen/coze/loop/observability/lotrace/local_traceservice.go b/backend/loop_gen/coze/loop/observability/lotrace/local_traceservice.go index 296a86ba4..3d44c8dac 100644 --- a/backend/loop_gen/coze/loop/observability/lotrace/local_traceservice.go +++ b/backend/loop_gen/coze/loop/observability/lotrace/local_traceservice.go @@ -337,6 +337,27 @@ func (l *LocalTraceService) ListAnnotations(ctx context.Context, req *trace.List return result.GetSuccess(), nil } +func (l *LocalTraceService) ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest, callOptions ...callopt.Option) (*trace.ListWorkspaceAnnotationsResponse, error) { + chain := l.mds(func(ctx context.Context, in, out interface{}) error { + arg := in.(*trace.TraceServiceListWorkspaceAnnotationsArgs) + result := out.(*trace.TraceServiceListWorkspaceAnnotationsResult) + resp, err := l.impl.ListWorkspaceAnnotations(ctx, arg.Req) + if err != nil { + return err + } + result.SetSuccess(resp) + return nil + }) + + arg := &trace.TraceServiceListWorkspaceAnnotationsArgs{Req: req} + result := &trace.TraceServiceListWorkspaceAnnotationsResult{} + ctx = l.injectRPCInfo(ctx, "ListWorkspaceAnnotations") + if err := chain(ctx, arg, result); err != nil { + return nil, err + } + return result.GetSuccess(), nil +} + func (l *LocalTraceService) ExportTracesToDataset(ctx context.Context, Req *trace.ExportTracesToDatasetRequest, callOptions ...callopt.Option) (*trace.ExportTracesToDatasetResponse, error) { chain := l.mds(func(ctx context.Context, in, out interface{}) error { arg := in.(*trace.TraceServiceExportTracesToDatasetArgs) @@ -505,6 +526,27 @@ func (l *LocalTraceService) ListTrajectory(ctx context.Context, req *trace.ListT return result.GetSuccess(), nil } +func (l *LocalTraceService) ListMetadata(ctx context.Context, req *trace.ListMetadataRequest, callOptions ...callopt.Option) (*trace.ListMetadataResponse, error) { + chain := l.mds(func(ctx context.Context, in, out interface{}) error { + arg := in.(*trace.TraceServiceListMetadataArgs) + result := out.(*trace.TraceServiceListMetadataResult) + resp, err := l.impl.ListMetadata(ctx, arg.Req) + if err != nil { + return err + } + result.SetSuccess(resp) + return nil + }) + + arg := &trace.TraceServiceListMetadataArgs{Req: req} + result := &trace.TraceServiceListMetadataResult{} + ctx = l.injectRPCInfo(ctx, "ListMetadata") + if err := chain(ctx, arg, result); err != nil { + return nil, err + } + return result.GetSuccess(), nil +} + func (l *LocalTraceService) injectRPCInfo(ctx context.Context, method string) context.Context { rpcStats := rpcinfo.AsMutableRPCStats(rpcinfo.NewRPCStats()) ri := rpcinfo.NewRPCInfo( diff --git a/backend/modules/observability/application/convertor/task/task.go b/backend/modules/observability/application/convertor/task/task.go index 0501039ee..f8ae51ddd 100644 --- a/backend/modules/observability/application/convertor/task/task.go +++ b/backend/modules/observability/application/convertor/task/task.go @@ -78,6 +78,18 @@ func TaskDO2DTO(ctx context.Context, v *entity.ObservabilityTask, userMap map[st taskInfo.TaskSource = gptr.Of(*v.TaskSource) } + if v.WorkflowID != 0 { + taskInfo.WorkflowID = ptr.Of(v.WorkflowID) + } + + if len(v.TaskRuns) > 0 { + taskRuns := make([]*task.TaskRun, 0, len(v.TaskRuns)) + for _, tr := range v.TaskRuns { + taskRuns = append(taskRuns, TaskRunDO2DTO(ctx, tr, userMap)) + } + taskInfo.TaskRuns = taskRuns + } + return taskInfo } @@ -117,9 +129,16 @@ func TaskConfigDO2DTO(v *entity.TaskConfig) *task.TaskConfig { dataReflowConfigs = append(dataReflowConfigs, DataReflowConfigDO2DTO(config)) } } + var evaluationExperimentConfig *task.EvaluationExperimentConfig + if v.EvaluationExperimentConfig != nil { + evaluationExperimentConfig = EvaluationExperimentConfigDO2DTO(v.EvaluationExperimentConfig) + } return &task.TaskConfig{ - AutoEvaluateConfigs: autoEvaluateConfigs, - DataReflowConfig: dataReflowConfigs, + AutoEvaluateConfigs: autoEvaluateConfigs, + DataReflowConfig: dataReflowConfigs, + EvaluationExperimentConfig: evaluationExperimentConfig, + SourceInfo: SourceInfoListDO2DTO(v.SourceInfo), + IsWorkflowScheduled: v.IsWorkflowScheduled, } } @@ -142,7 +161,66 @@ func AutoEvaluateConfigDO2DTO(v *entity.AutoEvaluateConfig) *task.AutoEvaluateCo EvaluatorVersionID: v.EvaluatorVersionID, EvaluatorID: v.EvaluatorID, FieldMappings: fieldMappings, + ScoreWeight: v.ScoreWeight, + EvaluatorVersion: v.EvaluatorVersion, + } +} + +func EvaluationExperimentConfigDO2DTO(v *entity.EvaluationExperimentConfig) *task.EvaluationExperimentConfig { + if v == nil { + return nil + } + var fullEvalSetFieldMappings []*task.EvaluateFieldMapping + for _, fm := range v.FullEvalSetFieldMappings { + fullEvalSetFieldMappings = append(fullEvalSetFieldMappings, &task.EvaluateFieldMapping{ + FieldSchema: fm.FieldSchema, + TraceFieldKey: fm.TraceFieldKey, + TraceFieldJsonpath: fm.TraceFieldJsonpath, + EvalSetName: fm.EvalSetName, + }) + } + return &task.EvaluationExperimentConfig{ + ItemConcurrencyCount: v.ItemConcurrencyCount, + ItemMaxRetryCount: v.ItemMaxRetryCount, + SourceTargetID: v.SourceTargetID, + ExptTemplateID: v.ExptTemplateID, + SourceTargetVersion: v.SourceTargetVersion, + FullEvalSetFieldMappings: fullEvalSetFieldMappings, + } +} + +func SourceInfoListDO2DTO(vs []*entity.SourceInfo) []*task.SourceInfo { + if len(vs) == 0 { + return nil + } + result := make([]*task.SourceInfo, 0, len(vs)) + for _, v := range vs { + if v == nil { + continue + } + result = append(result, &task.SourceInfo{ + Name: v.Name, + Version: v.Version, + }) + } + return result +} + +func SourceInfoListDTO2DO(vs []*task.SourceInfo) []*entity.SourceInfo { + if len(vs) == 0 { + return nil + } + result := make([]*entity.SourceInfo, 0, len(vs)) + for _, v := range vs { + if v == nil { + continue + } + result = append(result, &entity.SourceInfo{ + Name: v.Name, + Version: v.Version, + }) } + return result } func DataReflowConfigDO2DTO(v *entity.DataReflowConfig) *task.DataReflowConfig { @@ -156,10 +234,11 @@ func DataReflowConfigDO2DTO(v *entity.DataReflowConfig) *task.DataReflowConfig { } } return &task.DataReflowConfig{ - DatasetID: v.DatasetID, - DatasetName: v.DatasetName, - DatasetSchema: ptr.Of(v.DatasetSchema), - FieldMappings: fieldMappings, + DatasetID: v.DatasetID, + DatasetName: v.DatasetName, + DatasetSchema: ptr.Of(v.DatasetSchema), + FieldMappings: fieldMappings, + DatasetCategory: v.DatasetCategory, } } @@ -341,6 +420,7 @@ func TaskDTO2DO(taskDTO *task.Task) *entity.ObservabilityTask { CreatedBy: createdBy, UpdatedBy: updatedBy, BackfillEffectiveTime: EffectiveTimeDTO2DO(taskDTO.GetRule().GetBackfillEffectiveTime()), + WorkflowID: gptr.Indirect(taskDTO.WorkflowID), } if taskDTO.TaskSource != nil { @@ -351,8 +431,9 @@ func TaskDTO2DO(taskDTO *task.Task) *entity.ObservabilityTask { } func SpanFilterDTO2DO(spanFilterFields *filter.SpanFilterFields) *entity.SpanFilterFields { + // avoid npe if spanFilterFields == nil { - return nil + return new(entity.SpanFilterFields) } return &entity.SpanFilterFields{ PlatformType: loop_span.PlatformType(*spanFilterFields.PlatformType), @@ -405,26 +486,33 @@ func TaskConfigDTO2DO(taskConfig *task.TaskConfig) *entity.TaskConfig { var fieldMappings []*entity.EvaluateFieldMapping if len(autoEvaluateConfig.FieldMappings) > 0 { // todo tyf 这段逻辑挪到service层 - var evalSetNames []string + var datasetKeys []string jspnPathMapping := make(map[string]string) for _, config := range autoEvaluateConfig.FieldMappings { var evalSetName string + var datasetKey string jspnPath := fmt.Sprintf("%s.%s", config.TraceFieldKey, config.TraceFieldJsonpath) if _, exits := jspnPathMapping[jspnPath]; exits { - evalSetName = jspnPathMapping[jspnPath] + datasetKey = jspnPathMapping[jspnPath] } else { - evalSetName = getLastPartAfterDot(jspnPath) - for exists := slices.Contains(evalSetNames, evalSetName); exists; exists = slices.Contains(evalSetNames, evalSetName) { - evalSetName += "_" + datasetKey = getLastPartAfterDot(jspnPath) + for exists := slices.Contains(datasetKeys, datasetKey); exists; exists = slices.Contains(datasetKeys, datasetKey) { + datasetKey += "_" } } - evalSetNames = append(evalSetNames, evalSetName) - jspnPathMapping[jspnPath] = evalSetName + if config.IsSetEvalSetName() { + evalSetName = config.GetEvalSetName() + } else { + evalSetName = datasetKey + } + datasetKeys = append(datasetKeys, datasetKey) + jspnPathMapping[jspnPath] = datasetKey fieldMappings = append(fieldMappings, &entity.EvaluateFieldMapping{ FieldSchema: config.FieldSchema, TraceFieldKey: config.TraceFieldKey, TraceFieldJsonpath: config.TraceFieldJsonpath, EvalSetName: ptr.Of(evalSetName), + DatasetKey: ptr.Of(datasetKey), }) } @@ -433,6 +521,8 @@ func TaskConfigDTO2DO(taskConfig *task.TaskConfig) *entity.TaskConfig { EvaluatorVersionID: autoEvaluateConfig.EvaluatorVersionID, EvaluatorID: autoEvaluateConfig.EvaluatorID, FieldMappings: fieldMappings, + ScoreWeight: autoEvaluateConfig.ScoreWeight, + EvaluatorVersion: autoEvaluateConfig.EvaluatorVersion, }) } dataReflowConfigs := make([]*entity.DataReflowConfig, 0, len(taskConfig.DataReflowConfig)) @@ -448,15 +538,60 @@ func TaskConfigDTO2DO(taskConfig *task.TaskConfig) *entity.TaskConfig { } } dataReflowConfigs = append(dataReflowConfigs, &entity.DataReflowConfig{ - DatasetID: dataReflowConfig.DatasetID, - DatasetName: dataReflowConfig.DatasetName, - DatasetSchema: *dataReflowConfig.DatasetSchema, - FieldMappings: fieldMappings, + DatasetID: dataReflowConfig.DatasetID, + DatasetName: dataReflowConfig.DatasetName, + DatasetSchema: *dataReflowConfig.DatasetSchema, + FieldMappings: fieldMappings, + DatasetCategory: dataReflowConfig.DatasetCategory, }) } + var evaluationExperimentConfig *entity.EvaluationExperimentConfig + if taskConfig.EvaluationExperimentConfig != nil { + var fullEvalSetFieldMappings []*entity.EvaluateFieldMapping + if len(taskConfig.EvaluationExperimentConfig.FullEvalSetFieldMappings) > 0 { + var datasetKeys []string + jspnPathMapping := make(map[string]string) + for _, config := range taskConfig.EvaluationExperimentConfig.FullEvalSetFieldMappings { + var evalSetName string + var datasetKey string + jspnPath := fmt.Sprintf("%s.%s", config.TraceFieldKey, config.TraceFieldJsonpath) + if config.IsSetEvalSetName() { + evalSetName = config.GetEvalSetName() + } + if _, exits := jspnPathMapping[jspnPath]; exits { + datasetKey = jspnPathMapping[jspnPath] + } else { + datasetKey = getLastPartAfterDot(jspnPath) + for exists := slices.Contains(datasetKeys, datasetKey); exists; exists = slices.Contains(datasetKeys, datasetKey) { + datasetKey += "_" + } + } + datasetKeys = append(datasetKeys, datasetKey) + jspnPathMapping[jspnPath] = datasetKey + fullEvalSetFieldMappings = append(fullEvalSetFieldMappings, &entity.EvaluateFieldMapping{ + FieldSchema: config.FieldSchema, + TraceFieldKey: config.TraceFieldKey, + TraceFieldJsonpath: config.TraceFieldJsonpath, + EvalSetName: ptr.Of(evalSetName), + DatasetKey: ptr.Of(datasetKey), + }) + } + } + evaluationExperimentConfig = &entity.EvaluationExperimentConfig{ + ItemConcurrencyCount: taskConfig.EvaluationExperimentConfig.ItemConcurrencyCount, + ItemMaxRetryCount: taskConfig.EvaluationExperimentConfig.ItemMaxRetryCount, + SourceTargetID: taskConfig.EvaluationExperimentConfig.SourceTargetID, + ExptTemplateID: taskConfig.EvaluationExperimentConfig.ExptTemplateID, + SourceTargetVersion: taskConfig.EvaluationExperimentConfig.SourceTargetVersion, + FullEvalSetFieldMappings: fullEvalSetFieldMappings, + } + } return &entity.TaskConfig{ - AutoEvaluateConfigs: autoEvaluateConfigs, - DataReflowConfig: dataReflowConfigs, + AutoEvaluateConfigs: autoEvaluateConfigs, + DataReflowConfig: dataReflowConfigs, + EvaluationExperimentConfig: evaluationExperimentConfig, + SourceInfo: SourceInfoListDTO2DO(taskConfig.SourceInfo), + IsWorkflowScheduled: taskConfig.IsWorkflowScheduled, } } diff --git a/backend/modules/observability/application/convertor/task/task_test.go b/backend/modules/observability/application/convertor/task/task_test.go index 78605fcbd..5bafe81f2 100755 --- a/backend/modules/observability/application/convertor/task/task_test.go +++ b/backend/modules/observability/application/convertor/task/task_test.go @@ -11,6 +11,7 @@ import ( "github.com/bytedance/gg/gptr" "github.com/bytedance/sonic" "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" kitCommon "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/common" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/dataset" @@ -165,6 +166,10 @@ func TestTaskConfigDTO2DO(t *testing.T) { }, }, }, + EvaluationExperimentConfig: &kitTask.EvaluationExperimentConfig{ + ItemConcurrencyCount: gptr.Of(int32(3)), + ItemMaxRetryCount: gptr.Of(int32(4)), + }, DataReflowConfig: []*kitTask.DataReflowConfig{ { DatasetID: gptr.Of(int64(10)), @@ -185,11 +190,18 @@ func TestTaskConfigDTO2DO(t *testing.T) { cfg := TaskConfigDTO2DO(dto) if assert.NotNil(t, cfg) && assert.Len(t, cfg.AutoEvaluateConfigs, 1) { + if assert.NotNil(t, cfg.EvaluationExperimentConfig) { + assert.Equal(t, int32(3), ptr.From(cfg.EvaluationExperimentConfig.ItemConcurrencyCount)) + assert.Equal(t, int32(4), ptr.From(cfg.EvaluationExperimentConfig.ItemMaxRetryCount)) + } mappings := cfg.AutoEvaluateConfigs[0].FieldMappings if assert.Len(t, mappings, 3) { assert.Equal(t, "result", ptr.From(mappings[0].EvalSetName)) assert.Equal(t, "result_", ptr.From(mappings[1].EvalSetName)) assert.Equal(t, "result_0", ptr.From(mappings[2].EvalSetName)) + assert.Equal(t, "result", ptr.From(mappings[0].DatasetKey)) + assert.Equal(t, "result_", ptr.From(mappings[1].DatasetKey)) + assert.Equal(t, "result_0", ptr.From(mappings[2].DatasetKey)) } } @@ -201,6 +213,50 @@ func TestTaskConfigDTO2DO(t *testing.T) { } } +func TestTaskConfigDO2DTO_EvaluationExperimentConfig(t *testing.T) { + t.Parallel() + + do := &entity.TaskConfig{ + EvaluationExperimentConfig: &entity.EvaluationExperimentConfig{ + ItemConcurrencyCount: gptr.Of(int32(5)), + ItemMaxRetryCount: gptr.Of(int32(6)), + SourceTargetID: gptr.Of("target-123"), + ExptTemplateID: gptr.Of(int64(999)), + SourceTargetVersion: gptr.Of("v1.0"), + FullEvalSetFieldMappings: []*entity.EvaluateFieldMapping{ + { + FieldSchema: &dataset.FieldSchema{ + Key: gptr.Of("key1"), + Name: gptr.Of("field1"), + }, + TraceFieldKey: "Input", + TraceFieldJsonpath: "$.input", + EvalSetName: gptr.Of("field1"), + }, + }, + }, + } + dto := TaskConfigDO2DTO(do) + if assert.NotNil(t, dto) { + ec := dto.EvaluationExperimentConfig + if assert.NotNil(t, ec) { + assert.Equal(t, int32(5), ptr.From(ec.ItemConcurrencyCount)) + assert.Equal(t, int32(6), ptr.From(ec.ItemMaxRetryCount)) + assert.Equal(t, "target-123", ptr.From(ec.SourceTargetID)) + assert.Equal(t, int64(999), ptr.From(ec.ExptTemplateID)) + assert.Equal(t, "v1.0", ptr.From(ec.SourceTargetVersion)) + if assert.Len(t, ec.FullEvalSetFieldMappings, 1) { + fm := ec.FullEvalSetFieldMappings[0] + assert.Equal(t, "Input", fm.TraceFieldKey) + assert.Equal(t, "$.input", fm.TraceFieldJsonpath) + assert.Equal(t, "field1", ptr.From(fm.EvalSetName)) + } + } + } + + assert.Nil(t, EvaluationExperimentConfigDO2DTO(nil)) +} + func TestTaskDTO2DO(t *testing.T) { t.Parallel() @@ -329,3 +385,198 @@ func TestBuildTaskRunBaseInfo(t *testing.T) { assert.Equal(t, "", ptr.From(base.UpdatedBy.UserID)) } } + +func TestSourceInfoListDO2DTO(t *testing.T) { + t.Parallel() + + t.Run("nil input returns nil", func(t *testing.T) { + t.Parallel() + assert.Nil(t, SourceInfoListDO2DTO(nil)) + }) + + t.Run("empty input returns nil", func(t *testing.T) { + t.Parallel() + assert.Nil(t, SourceInfoListDO2DTO([]*entity.SourceInfo{})) + }) + + t.Run("list with nil elements skips them", func(t *testing.T) { + t.Parallel() + input := []*entity.SourceInfo{ + nil, + {Name: gptr.Of("sdk-go"), Version: gptr.Of("1.0.0")}, + nil, + } + result := SourceInfoListDO2DTO(input) + if assert.Len(t, result, 1) { + assert.Equal(t, gptr.Of("sdk-go"), result[0].Name) + assert.Equal(t, gptr.Of("1.0.0"), result[0].Version) + } + }) + + t.Run("normal conversion", func(t *testing.T) { + t.Parallel() + input := []*entity.SourceInfo{ + {Name: gptr.Of("sdk-go"), Version: gptr.Of("1.0.0")}, + {Name: gptr.Of("sdk-python"), Version: gptr.Of("2.0.0")}, + } + result := SourceInfoListDO2DTO(input) + if assert.Len(t, result, 2) { + assert.Equal(t, gptr.Of("sdk-go"), result[0].Name) + assert.Equal(t, gptr.Of("1.0.0"), result[0].Version) + assert.Equal(t, gptr.Of("sdk-python"), result[1].Name) + assert.Equal(t, gptr.Of("2.0.0"), result[1].Version) + } + }) +} + +func TestSpanFilterDTO2DO(t *testing.T) { + t.Parallel() + + t.Run("nil input returns empty struct", func(t *testing.T) { + t.Parallel() + result := SpanFilterDTO2DO(nil) + assert.NotNil(t, result) + assert.Equal(t, new(entity.SpanFilterFields), result) + }) + + t.Run("normal conversion", func(t *testing.T) { + t.Parallel() + dto := &filter.SpanFilterFields{ + PlatformType: gptr.Of(kitCommon.PlatformTypeCozeloop), + SpanListType: gptr.Of(kitCommon.SpanListTypeRootSpan), + Filters: &filter.FilterFields{ + QueryAndOr: ptr.Of(filter.QueryRelationAnd), + FilterFields: []*filter.FilterField{}, + }, + } + result := SpanFilterDTO2DO(dto) + if assert.NotNil(t, result) { + assert.Equal(t, kitCommon.PlatformTypeCozeloop, string(result.PlatformType)) + assert.Equal(t, kitCommon.SpanListTypeRootSpan, string(result.SpanListType)) + } + }) +} + +func TestSourceInfoListDTO2DO(t *testing.T) { + t.Parallel() + + t.Run("nil input returns nil", func(t *testing.T) { + t.Parallel() + assert.Nil(t, SourceInfoListDTO2DO(nil)) + }) + + t.Run("empty input returns nil", func(t *testing.T) { + t.Parallel() + assert.Nil(t, SourceInfoListDTO2DO([]*kitTask.SourceInfo{})) + }) + + t.Run("skips nil elements", func(t *testing.T) { + t.Parallel() + result := SourceInfoListDTO2DO([]*kitTask.SourceInfo{ + nil, + {Name: gptr.Of("svc"), Version: gptr.Of("v1")}, + }) + if assert.Len(t, result, 1) { + assert.Equal(t, "svc", ptr.From(result[0].Name)) + assert.Equal(t, "v1", ptr.From(result[0].Version)) + } + }) + + t.Run("normal conversion", func(t *testing.T) { + t.Parallel() + result := SourceInfoListDTO2DO([]*kitTask.SourceInfo{ + {Name: gptr.Of("a"), Version: gptr.Of("1.0")}, + {Name: gptr.Of("b"), Version: gptr.Of("2.0")}, + }) + if assert.Len(t, result, 2) { + assert.Equal(t, "a", ptr.From(result[0].Name)) + assert.Equal(t, "1.0", ptr.From(result[0].Version)) + assert.Equal(t, "b", ptr.From(result[1].Name)) + assert.Equal(t, "2.0", ptr.From(result[1].Version)) + } + }) +} + +func TestTaskConfigDTO2DO_FullEvalSetAndSourceInfo(t *testing.T) { + t.Parallel() + + schema := &dataset.FieldSchema{ + Key: gptr.Of("output_key"), + Name: gptr.Of("Output"), + ContentType: gptr.Of("Text"), + } + + dto := &kitTask.TaskConfig{ + AutoEvaluateConfigs: []*kitTask.AutoEvaluateConfig{ + { + EvaluatorVersionID: 1, + EvaluatorID: 2, + FieldMappings: []*kitTask.EvaluateFieldMapping{ + { + FieldSchema: schema, + TraceFieldKey: "Input", + TraceFieldJsonpath: "$.input", + EvalSetName: gptr.Of("input_field"), + }, + }, + }, + }, + EvaluationExperimentConfig: &kitTask.EvaluationExperimentConfig{ + ItemConcurrencyCount: gptr.Of(int32(5)), + ItemMaxRetryCount: gptr.Of(int32(3)), + ExptTemplateID: gptr.Of(int64(888)), + SourceTargetID: gptr.Of("target-1"), + SourceTargetVersion: gptr.Of("v2"), + FullEvalSetFieldMappings: []*kitTask.EvaluateFieldMapping{ + { + FieldSchema: schema, + TraceFieldKey: "Output", + TraceFieldJsonpath: "$.output", + EvalSetName: gptr.Of("output_field"), + }, + { + FieldSchema: schema, + TraceFieldKey: "Output", + TraceFieldJsonpath: "$.output", + }, + }, + }, + SourceInfo: []*kitTask.SourceInfo{ + {Name: gptr.Of("my-service"), Version: gptr.Of("1.0.0")}, + }, + IsWorkflowScheduled: gptr.Of(true), + } + + cfg := TaskConfigDTO2DO(dto) + require.NotNil(t, cfg) + + ec := cfg.EvaluationExperimentConfig + if assert.NotNil(t, ec) { + assert.Equal(t, int32(5), ptr.From(ec.ItemConcurrencyCount)) + assert.Equal(t, int32(3), ptr.From(ec.ItemMaxRetryCount)) + assert.Equal(t, int64(888), ptr.From(ec.ExptTemplateID)) + assert.Equal(t, "target-1", ptr.From(ec.SourceTargetID)) + assert.Equal(t, "v2", ptr.From(ec.SourceTargetVersion)) + if assert.Len(t, ec.FullEvalSetFieldMappings, 2) { + assert.Equal(t, "output_field", ptr.From(ec.FullEvalSetFieldMappings[0].EvalSetName)) + assert.Equal(t, "output", ptr.From(ec.FullEvalSetFieldMappings[0].DatasetKey)) + assert.Equal(t, "output", ptr.From(ec.FullEvalSetFieldMappings[1].DatasetKey)) + } + } + + if assert.Len(t, cfg.SourceInfo, 1) { + assert.Equal(t, "my-service", ptr.From(cfg.SourceInfo[0].Name)) + assert.Equal(t, "1.0.0", ptr.From(cfg.SourceInfo[0].Version)) + } + + assert.Equal(t, true, ptr.From(cfg.IsWorkflowScheduled)) + + if assert.Len(t, cfg.AutoEvaluateConfigs, 1) { + fm := cfg.AutoEvaluateConfigs[0].FieldMappings + if assert.Len(t, fm, 1) { + assert.Equal(t, "input_field", ptr.From(fm[0].EvalSetName)) + } + } + + assert.Nil(t, TaskConfigDTO2DO(nil)) +} diff --git a/backend/modules/observability/application/convertor/trace/annotation.go b/backend/modules/observability/application/convertor/trace/annotation.go index c4ef608d3..6157f30a3 100644 --- a/backend/modules/observability/application/convertor/trace/annotation.go +++ b/backend/modules/observability/application/convertor/trace/annotation.go @@ -8,6 +8,8 @@ import ( "strconv" "time" + "github.com/bytedance/gg/gptr" + annodto "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/annotation" commdto "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/common" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/rpc" @@ -111,6 +113,8 @@ func AnnotationDO2DTO( ret.AutoEvaluate.EvaluatorVersionID = meta.EvaluatorVersionID ret.AutoEvaluate.TaskID = strconv.FormatInt(meta.TaskID, 10) ret.AutoEvaluate.RecordID = meta.EvaluatorRecordID + ret.AutoEvaluate.ExptID = gptr.Of(meta.ExptID) + ret.AutoEvaluate.ExptTemplateID = gptr.Of(meta.ExptTemplateID) ret.AutoEvaluate.EvaluatorResult_ = annodto.NewEvaluatorResult_() ret.AutoEvaluate.EvaluatorResult_.Score = ptr.Of(a.Value.FloatValue) ret.AutoEvaluate.EvaluatorResult_.Reasoning = ptr.Of(a.Reasoning) @@ -188,3 +192,27 @@ func AnnotationListDO2DTO( } return ret } + +func AnnotationListKeyConv(annos []*annodto.Annotation) { + for _, anno := range annos { + if anno == nil || anno.Type == nil { + continue + } + switch *anno.Type { + case annodto.AnnotationTypeAutoEvaluate: + if anno.AutoEvaluate != nil { + anno.Key = gptr.Of(anno.AutoEvaluate.EvaluatorName) + } + case annodto.AnnotationTypeManualFeedback: + if anno.ManualFeedback != nil { + anno.Key = gptr.Of(anno.ManualFeedback.TagKeyName) + } + case annodto.AnnotationTypeCozeFeedback: + continue + case annodto.AnnotationTypeOpenAPIFeedback: + continue + default: + continue + } + } +} diff --git a/backend/modules/observability/application/convertor/trace/trace_export.go b/backend/modules/observability/application/convertor/trace/trace_export.go index e8e9e2aeb..ca1337eff 100644 --- a/backend/modules/observability/application/convertor/trace/trace_export.go +++ b/backend/modules/observability/application/convertor/trace/trace_export.go @@ -84,6 +84,7 @@ func PreviewRequestDTO2DO(req *trace.PreviewExportTracesToDatasetRequest) *servi SpanIds: convertSpanIdsDTO2DO(req.GetSpanIds()), Category: convertDatasetCategoryDTO2DO(req.GetCategory()), Config: convertDatasetConfigDTO2DO(req.GetConfig()), + Limit: req.Limit, } result.StartTime = req.GetStartTime() result.EndTime = req.GetEndTime() @@ -109,6 +110,10 @@ func PreviewRequestDTO2DO(req *trace.PreviewExportTracesToDatasetRequest) *servi result.FieldMappings = ConvertFieldMappingsDTO2DO(req.GetFieldMappings()) } + if req.IsSetSpanFilters() { + result.SpanFilters = req.SpanFilters + } + return result } diff --git a/backend/modules/observability/application/task.go b/backend/modules/observability/application/task.go index bbd8179f1..3fa2317fb 100644 --- a/backend/modules/observability/application/task.go +++ b/backend/modules/observability/application/task.go @@ -364,7 +364,6 @@ func (t *TaskApplication) BackFill(ctx context.Context, event *entity.BackFillEv // 结构校验失败，不处理 return nil } - return t.tracehubSvc.BackFill(ctx, event) } diff --git a/backend/modules/observability/application/trace.go b/backend/modules/observability/application/trace.go index 22c5d9be7..a55dc03d6 100644 --- a/backend/modules/observability/application/trace.go +++ b/backend/modules/observability/application/trace.go @@ -5,9 +5,12 @@ package application import ( "context" + "sort" "strconv" "time" + annodto "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/annotation" + "github.com/coze-dev/coze-loop/backend/infra/external/benefit" "github.com/coze-dev/coze-loop/backend/infra/middleware/session" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/common" @@ -37,11 +40,12 @@ import ( ) const ( - MaxSpanLength = 500 - MaxListSpansLimit = 1000 - MaxTraceTreeLength = 10000 - MaxOApiListSpansLimit = 200 - QueryLimitDefault = 100 + MaxSpanLength = 500 + MaxListSpansLimit = 1000 + MaxTraceTreeLength = 10000 + MaxOApiListSpansLimit = 200 + QueryLimitDefault = 100 + MaxListMetadataSpansList int64 = 3000 ) //go:generate mockgen -destination=mocks/trace_application.go -package=mocks . ITraceApplication @@ -608,6 +612,63 @@ func (t *TraceApplication) buildGetTracesMetaInfoReq(req *trace.GetTracesMetaInf return ret } +func (t *TraceApplication) ListMetadata(ctx context.Context, req *trace.ListMetadataRequest) (*trace.ListMetadataResponse, error) { + if err := t.validateListMetadataReq(ctx, req); err != nil { + return nil, err + } + if err := t.authSvc.CheckWorkspacePermission(ctx, + rpc.AuthActionTraceRead, + strconv.FormatInt(req.GetWorkspaceID(), 10), false); err != nil { + return nil, err + } + sReq, err := t.buildListMetadataSvcReq(req) + if err != nil { + return nil, errorx.WrapByCode(err, obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("list metadata req is invalid")) + } + sResp, err := t.traceService.ListMetadata(ctx, sReq) + if err != nil { + return nil, err + } + logs.CtxInfo(ctx, "List metadata successfully, items count: %d", len(sResp.MetadataItemList)) + return &trace.ListMetadataResponse{ + MetadataItemList: sResp.MetadataItemList, + }, nil +} + +func (t *TraceApplication) validateListMetadataReq(ctx context.Context, req *trace.ListMetadataRequest) error { + if req == nil { + return errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("no request provided")) + } else if req.GetWorkspaceID() <= 0 { + return errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("invalid workspace_id")) + } + return nil +} + +func (t *TraceApplication) buildListMetadataSvcReq(req *trace.ListMetadataRequest) (*service.ListMetadataReq, error) { + // default 3 days + ret := &service.ListMetadataReq{ + WorkspaceID: req.GetWorkspaceID(), + StartTime: time.Now().Add(-3 * 24 * time.Hour).UnixMilli(), + EndTime: time.Now().UnixMilli(), + } + platformType := loop_span.PlatformType(req.GetPlatformType()) + if req.PlatformType == nil { + platformType = loop_span.PlatformCozeLoop + } + ret.PlatformType = platformType + switch req.GetSpanListType() { + case common.SpanListTypeRootSpan: + ret.SpanListType = loop_span.SpanListTypeRootSpan + case common.SpanListTypeAllSpan: + ret.SpanListType = loop_span.SpanListTypeAllSpan + case common.SpanListTypeLlmSpan: + ret.SpanListType = loop_span.SpanListTypeLLMSpan + default: + ret.SpanListType = loop_span.SpanListTypeRootSpan + } + return ret, nil +} + func (t *TraceApplication) CreateView(ctx context.Context, req *trace.CreateViewRequest) (*trace.CreateViewResponse, error) { if req == nil { return nil, errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("no request provided")) @@ -878,6 +939,105 @@ func (t *TraceApplication) ListAnnotations(ctx context.Context, req *trace.ListA }, nil } +func (t *TraceApplication) ListWorkspaceAnnotations(ctx context.Context, req *trace.ListWorkspaceAnnotationsRequest) (*trace.ListWorkspaceAnnotationsResponse, error) { + if req == nil { + return nil, errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("request is nil")) + } + if req.GetWorkspaceID() <= 0 { + return nil, errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("invalid workspace_id")) + } + if err := t.authSvc.CheckWorkspacePermission(ctx, + rpc.AuthActionTraceRead, + strconv.FormatInt(req.GetWorkspaceID(), 10), false); err != nil { + return nil, err + } + + platformType := loop_span.PlatformCozeLoop + if req.PlatformType != nil { + platformType = loop_span.PlatformType(*req.PlatformType) + } + + // 3 days forward by default + svcReq := &service.ListWorkspaceAnnotationsReq{ + WorkspaceID: req.WorkspaceID, + StartTime: time.Now().Add(-3 * 24 * time.Hour).UnixMilli(), + PlatformType: platformType, + } + + if req.AnnotationType != nil { + svcReq.AnnotationType = *req.AnnotationType + } + switch req.GetSpanListType() { + case common.SpanListTypeRootSpan: + svcReq.SpanListType = loop_span.SpanListTypeRootSpan + case common.SpanListTypeAllSpan: + svcReq.SpanListType = loop_span.SpanListTypeAllSpan + case common.SpanListTypeLlmSpan: + svcReq.SpanListType = loop_span.SpanListTypeLLMSpan + default: + svcReq.SpanListType = loop_span.SpanListTypeRootSpan + } + + resp, err := t.traceService.ListWorkspaceAnnotations(ctx, svcReq) + if err != nil { + return nil, err + } + + dResp := t.GetDisplayInfo(ctx, &GetDisplayInfoRequest{ + WorkspaceID: req.GetWorkspaceID(), + UserIDs: resp.Annotations.GetUserIDs(), + EvaluatorIDs: resp.Annotations.GetEvaluatorVersionIDs(), + TagKeyIDs: resp.Annotations.GetAnnotationTagIDs(), + }) + _ = dResp + annoWithInfo := tconv.AnnotationListDO2DTO(resp.Annotations, dResp.UserMap, dResp.EvalMap, dResp.TagMap) + originalKeys := make([]string, len(annoWithInfo)) + for i, anno := range annoWithInfo { + if anno != nil { + originalKeys[i] = anno.GetKey() + } + } + tconv.AnnotationListKeyConv(annoWithInfo) + type annoKey struct { + Key string + OriginalKey string + AnnotationType loop_span.AnnotationType + } + keyCount := make(map[annoKey]int) + for i, anno := range annoWithInfo { + if anno == nil { + continue + } + k := annoKey{ + Key: anno.GetKey(), + OriginalKey: originalKeys[i], + AnnotationType: loop_span.AnnotationType(anno.GetType()), + } + keyCount[k]++ + } + + keys := lo.Keys(keyCount) + sort.Slice(keys, func(i, j int) bool { + if keyCount[keys[i]] != keyCount[keys[j]] { + return keyCount[keys[i]] > keyCount[keys[j]] + } + return keys[i].Key < keys[j].Key + }) + + simpleList := make([]*annodto.SimpleAnnotationInfo, 0, len(keys)) + for _, k := range keys { + simpleList = append(simpleList, &annodto.SimpleAnnotationInfo{ + Key: k.Key, + AnnotationType: ptr.Of(annodto.AnnotationType(k.AnnotationType)), + OriginalKey: ptr.Of(k.OriginalKey), + }) + } + + return &trace.ListWorkspaceAnnotationsResponse{ + SimpleAnnotationList: simpleList, + }, nil +} + func (t *TraceApplication) ExportTracesToDataset(ctx context.Context, req *trace.ExportTracesToDatasetRequest) ( r *trace.ExportTracesToDatasetResponse, err error, ) { @@ -920,6 +1080,9 @@ func (t *TraceApplication) PreviewExportTracesToDataset(ctx context.Context, req if err := req.IsValid(); err != nil { return nil, errorx.WrapByCode(err, obErrorx.CommercialCommonInvalidParamCodeCode) } + if len(req.SpanIds) == 0 && req.SpanFilters == nil { + return nil, errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("span_ids and span_filters cannot be both nil")) + } v := utils.DateValidator{ Start: req.GetStartTime(), End: req.GetEndTime(), diff --git a/backend/modules/observability/application/trace_test.go b/backend/modules/observability/application/trace_test.go index 8de82d417..95e9a4fd0 100644 --- a/backend/modules/observability/application/trace_test.go +++ b/backend/modules/observability/application/trace_test.go @@ -507,6 +507,8 @@ func TestTraceApplication_ListSpans(t *testing.T) { TaskID: "123", RecordID: 123, EvaluatorVersionID: 123, + ExptID: ptr.Of(int64(0)), + ExptTemplateID: ptr.Of(int64(0)), EvaluatorResult_: &annodto.EvaluatorResult_{ Score: ptr.Of(1.0), Reasoning: ptr.Of(""), @@ -2346,6 +2348,319 @@ func TestTraceApplication_ListAnnotationEvaluators(t *testing.T) { } } +func TestTraceApplication_ListMetadata(t *testing.T) { + t.Run("nil request returns error", func(t *testing.T) { + tr := &TraceApplication{} + _, err := tr.ListMetadata(context.Background(), nil) + assert.Error(t, err) + }) + + t.Run("invalid workspace_id returns error", func(t *testing.T) { + tr := &TraceApplication{} + _, err := tr.ListMetadata(context.Background(), &trace.ListMetadataRequest{ + WorkspaceID: 0, + }) + assert.Error(t, err) + }) + + t.Run("permission error", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(fmt.Errorf("denied")) + tr := &TraceApplication{authSvc: mockAuth} + _, err := tr.ListMetadata(context.Background(), &trace.ListMetadataRequest{ + WorkspaceID: 123, + }) + assert.Error(t, err) + }) + + t.Run("service error", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListMetadata(gomock.Any(), gomock.Any()).Return(nil, assert.AnError) + tr := &TraceApplication{authSvc: mockAuth, traceService: mockSvc} + _, err := tr.ListMetadata(context.Background(), &trace.ListMetadataRequest{ + WorkspaceID: 123, + }) + assert.Error(t, err) + }) + + t.Run("success with default platform and span list type", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListMetadata(gomock.Any(), gomock.Any()).DoAndReturn( + func(ctx context.Context, req *service.ListMetadataReq) (*service.ListMetadataResp, error) { + assert.Equal(t, int64(123), req.WorkspaceID) + assert.Equal(t, loop_span.PlatformCozeLoop, req.PlatformType) + assert.Equal(t, loop_span.SpanListTypeRootSpan, req.SpanListType) + return &service.ListMetadataResp{ + MetadataItemList: []*trace.MetadataItemInfo{ + {Key: "k1", ValueType: "string"}, + }, + }, nil + }, + ) + tr := &TraceApplication{authSvc: mockAuth, traceService: mockSvc} + resp, err := tr.ListMetadata(context.Background(), &trace.ListMetadataRequest{ + WorkspaceID: 123, + }) + assert.NoError(t, err) + assert.NotNil(t, resp) + assert.Len(t, resp.MetadataItemList, 1) + assert.Equal(t, "k1", resp.MetadataItemList[0].Key) + }) + + t.Run("explicit platform type and all_span list type", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListMetadata(gomock.Any(), gomock.Any()).DoAndReturn( + func(ctx context.Context, req *service.ListMetadataReq) (*service.ListMetadataResp, error) { + assert.Equal(t, loop_span.PlatformType(commondto.PlatformTypeCozeBot), req.PlatformType) + assert.Equal(t, loop_span.SpanListTypeAllSpan, req.SpanListType) + return &service.ListMetadataResp{}, nil + }, + ) + pt := commondto.PlatformTypeCozeBot + slt := commondto.SpanListTypeAllSpan + tr := &TraceApplication{authSvc: mockAuth, traceService: mockSvc} + resp, err := tr.ListMetadata(context.Background(), &trace.ListMetadataRequest{ + WorkspaceID: 123, + PlatformType: &pt, + SpanListType: &slt, + }) + assert.NoError(t, err) + assert.NotNil(t, resp) + }) + + t.Run("llm_span list type", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListMetadata(gomock.Any(), gomock.Any()).DoAndReturn( + func(ctx context.Context, req *service.ListMetadataReq) (*service.ListMetadataResp, error) { + assert.Equal(t, loop_span.SpanListTypeLLMSpan, req.SpanListType) + return &service.ListMetadataResp{}, nil + }, + ) + slt := commondto.SpanListTypeLlmSpan + tr := &TraceApplication{authSvc: mockAuth, traceService: mockSvc} + resp, err := tr.ListMetadata(context.Background(), &trace.ListMetadataRequest{ + WorkspaceID: 123, + SpanListType: &slt, + }) + assert.NoError(t, err) + assert.NotNil(t, resp) + }) + + t.Run("root_span list type explicit", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListMetadata(gomock.Any(), gomock.Any()).DoAndReturn( + func(ctx context.Context, req *service.ListMetadataReq) (*service.ListMetadataResp, error) { + assert.Equal(t, loop_span.SpanListTypeRootSpan, req.SpanListType) + return &service.ListMetadataResp{}, nil + }, + ) + slt := commondto.SpanListTypeRootSpan + tr := &TraceApplication{authSvc: mockAuth, traceService: mockSvc} + resp, err := tr.ListMetadata(context.Background(), &trace.ListMetadataRequest{ + WorkspaceID: 123, + SpanListType: &slt, + }) + assert.NoError(t, err) + assert.NotNil(t, resp) + }) +} + +func TestTraceApplication_ListWorkspaceAnnotations(t *testing.T) { + t.Run("nil request returns error", func(t *testing.T) { + tr := &TraceApplication{} + _, err := tr.ListWorkspaceAnnotations(context.Background(), nil) + assert.Error(t, err) + }) + + t.Run("invalid workspace_id returns error", func(t *testing.T) { + tr := &TraceApplication{} + _, err := tr.ListWorkspaceAnnotations(context.Background(), &trace.ListWorkspaceAnnotationsRequest{ + WorkspaceID: 0, + }) + assert.Error(t, err) + }) + + t.Run("permission error", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(fmt.Errorf("denied")) + tr := &TraceApplication{authSvc: mockAuth} + _, err := tr.ListWorkspaceAnnotations(context.Background(), &trace.ListWorkspaceAnnotationsRequest{ + WorkspaceID: 123, + }) + assert.Error(t, err) + }) + + t.Run("service error", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListWorkspaceAnnotations(gomock.Any(), gomock.Any()).Return(nil, assert.AnError) + tr := &TraceApplication{authSvc: mockAuth, traceService: mockSvc} + _, err := tr.ListWorkspaceAnnotations(context.Background(), &trace.ListWorkspaceAnnotationsRequest{ + WorkspaceID: 123, + }) + assert.Error(t, err) + }) + + t.Run("empty annotations returns empty list", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListWorkspaceAnnotations(gomock.Any(), gomock.Any()).Return(&service.ListWorkspaceAnnotationsResp{ + Annotations: loop_span.AnnotationList{}, + }, nil) + tr := &TraceApplication{authSvc: mockAuth, traceService: mockSvc} + resp, err := tr.ListWorkspaceAnnotations(context.Background(), &trace.ListWorkspaceAnnotationsRequest{ + WorkspaceID: 123, + }) + assert.NoError(t, err) + assert.NotNil(t, resp) + assert.Len(t, resp.SimpleAnnotationList, 0) + }) + + t.Run("auto_evaluate annotation key is converted with original_key preserved", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockUser := rpcmock.NewMockIUserProvider(ctrl) + mockEval := rpcmock.NewMockIEvaluatorRPCAdapter(ctrl) + mockTag := rpcmock.NewMockITagRPCAdapter(ctrl) + + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListWorkspaceAnnotations(gomock.Any(), gomock.Any()).Return(&service.ListWorkspaceAnnotationsResp{ + Annotations: loop_span.AnnotationList{ + { + Key: "100:200", + AnnotationType: loop_span.AnnotationTypeAutoEvaluate, + Metadata: loop_span.AutoEvaluateMetadata{ + TaskID: 100, + EvaluatorVersionID: 200, + }, + CreatedBy: "user1", + UpdatedBy: "user1", + }, + }, + }, nil) + mockUser.EXPECT().GetUserInfo(gomock.Any(), gomock.Any()).Return(nil, nil, nil).AnyTimes() + mockEval.EXPECT().BatchGetEvaluatorVersions(gomock.Any(), gomock.Any()).Return(nil, map[int64]*rpc.Evaluator{ + 200: {EvaluatorName: "MyEvaluator"}, + }, nil).AnyTimes() + mockTag.EXPECT().BatchGetTagInfo(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes() + + tr := &TraceApplication{ + authSvc: mockAuth, + traceService: mockSvc, + userSvc: mockUser, + evalSvc: mockEval, + tagSvc: mockTag, + } + resp, err := tr.ListWorkspaceAnnotations(context.Background(), &trace.ListWorkspaceAnnotationsRequest{ + WorkspaceID: 123, + }) + assert.NoError(t, err) + assert.NotNil(t, resp) + assert.Len(t, resp.SimpleAnnotationList, 1) + assert.Equal(t, "MyEvaluator", resp.SimpleAnnotationList[0].Key) + assert.Equal(t, "100:200", resp.SimpleAnnotationList[0].GetOriginalKey()) + }) + + t.Run("dedup and sort by frequency with original_key", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + mockUser := rpcmock.NewMockIUserProvider(ctrl) + mockEval := rpcmock.NewMockIEvaluatorRPCAdapter(ctrl) + mockTag := rpcmock.NewMockITagRPCAdapter(ctrl) + + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListWorkspaceAnnotations(gomock.Any(), gomock.Any()).Return(&service.ListWorkspaceAnnotationsResp{ + Annotations: loop_span.AnnotationList{ + {Key: "100:200", AnnotationType: loop_span.AnnotationTypeAutoEvaluate, Metadata: loop_span.AutoEvaluateMetadata{TaskID: 100, EvaluatorVersionID: 200}, CreatedBy: "u1", UpdatedBy: "u1"}, + {Key: "100:200", AnnotationType: loop_span.AnnotationTypeAutoEvaluate, Metadata: loop_span.AutoEvaluateMetadata{TaskID: 100, EvaluatorVersionID: 200}, CreatedBy: "u1", UpdatedBy: "u1"}, + {Key: "100:300", AnnotationType: loop_span.AnnotationTypeAutoEvaluate, Metadata: loop_span.AutoEvaluateMetadata{TaskID: 100, EvaluatorVersionID: 300}, CreatedBy: "u1", UpdatedBy: "u1"}, + }, + }, nil) + mockUser.EXPECT().GetUserInfo(gomock.Any(), gomock.Any()).Return(nil, nil, nil).AnyTimes() + mockEval.EXPECT().BatchGetEvaluatorVersions(gomock.Any(), gomock.Any()).Return(nil, map[int64]*rpc.Evaluator{ + 200: {EvaluatorName: "EvalA"}, + 300: {EvaluatorName: "EvalB"}, + }, nil).AnyTimes() + mockTag.EXPECT().BatchGetTagInfo(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes() + + tr := &TraceApplication{ + authSvc: mockAuth, + traceService: mockSvc, + userSvc: mockUser, + evalSvc: mockEval, + tagSvc: mockTag, + } + resp, err := tr.ListWorkspaceAnnotations(context.Background(), &trace.ListWorkspaceAnnotationsRequest{ + WorkspaceID: 123, + }) + assert.NoError(t, err) + assert.NotNil(t, resp) + assert.Len(t, resp.SimpleAnnotationList, 2) + assert.Equal(t, "EvalA", resp.SimpleAnnotationList[0].Key) + assert.Equal(t, "100:200", resp.SimpleAnnotationList[0].GetOriginalKey()) + assert.Equal(t, "EvalB", resp.SimpleAnnotationList[1].Key) + assert.Equal(t, "100:300", resp.SimpleAnnotationList[1].GetOriginalKey()) + }) + + t.Run("with PlatformType and SpanListType params", func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + mockAuth := rpcmock.NewMockIAuthProvider(ctrl) + mockSvc := svcmock.NewMockITraceService(ctrl) + + mockAuth.EXPECT().CheckWorkspacePermission(gomock.Any(), rpc.AuthActionTraceRead, "123", false).Return(nil) + mockSvc.EXPECT().ListWorkspaceAnnotations(gomock.Any(), gomock.Any()).Return(&service.ListWorkspaceAnnotationsResp{ + Annotations: loop_span.AnnotationList{}, + }, nil) + + slt := commondto.SpanListTypeAllSpan + pt := commondto.PlatformTypeCozeBot + tr := &TraceApplication{authSvc: mockAuth, traceService: mockSvc} + resp, err := tr.ListWorkspaceAnnotations(context.Background(), &trace.ListWorkspaceAnnotationsRequest{ + WorkspaceID: 123, + PlatformType: &pt, + SpanListType: &slt, + }) + assert.NoError(t, err) + assert.NotNil(t, resp) + assert.Len(t, resp.SimpleAnnotationList, 0) + }) +} + func TestTraceApplication_ExtractSpanInfo(t *testing.T) { type fields struct { traceSvc service.ITraceService diff --git a/backend/modules/observability/application/wire.go b/backend/modules/observability/application/wire.go index 6460039f1..e670dc293 100644 --- a/backend/modules/observability/application/wire.go +++ b/backend/modules/observability/application/wire.go @@ -29,6 +29,7 @@ import ( "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/rpc" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/scheduledtask" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/storage" + taskhook "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/task" metrics_entity "github.com/coze-dev/coze-loop/backend/modules/observability/domain/metric/entity" metric_repo "github.com/coze-dev/coze-loop/backend/modules/observability/domain/metric/repo" metric_service "github.com/coze-dev/coze-loop/backend/modules/observability/domain/metric/service" @@ -370,11 +371,11 @@ func NewDatasetServiceAdapter(evalSetService evaluationsetservice.Client, datase } func NewInitTaskProcessor(datasetServiceProvider *service.DatasetServiceAdaptor, evalService rpc.IEvaluatorRPCAdapter, - evaluationService rpc.IEvaluationRPCAdapter, taskRepo trepo.ITaskRepo, + evaluationService rpc.IEvaluationRPCAdapter, taskRepo trepo.ITaskRepo, taskHookProvider taskhook.IWorkflowProvider, ) *task_processor.TaskProcessor { taskProcessor := task_processor.NewTaskProcessor() taskProcessor.Register(task_entity.TaskTypeAutoEval, task_processor.NewAutoEvaluateProcessor( - 0, datasetServiceProvider, evalService, evaluationService, taskRepo, &task_processor.EvalTargetBuilderImpl{})) + 0, datasetServiceProvider, evalService, evaluationService, taskRepo, &task_processor.EvalTargetBuilderImpl{}, taskHookProvider)) return taskProcessor } @@ -478,6 +479,7 @@ func InitTaskApplication( taskProcessor task_processor.TaskProcessor, aid int32, persistentCmdable redis.PersistentCmdable, + taskHookProvider taskhook.IWorkflowProvider, ) (ITaskApplication, error) { wire.Build(taskSet) return nil, nil diff --git a/backend/modules/observability/application/wire_gen.go b/backend/modules/observability/application/wire_gen.go index 0a06e208e..4dd23574d 100644 --- a/backend/modules/observability/application/wire_gen.go +++ b/backend/modules/observability/application/wire_gen.go @@ -29,6 +29,8 @@ import ( "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/rpc" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/scheduledtask" storage2 "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/storage" + taskhook "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/task" + entity2 "github.com/coze-dev/coze-loop/backend/modules/observability/domain/metric/entity" repo3 "github.com/coze-dev/coze-loop/backend/modules/observability/domain/metric/repo" service2 "github.com/coze-dev/coze-loop/backend/modules/observability/domain/metric/service" @@ -259,7 +261,7 @@ func InitTraceIngestionApplication(configFactory conf.IConfigLoaderFactory, stor return iTraceIngestionApplication, nil } -func InitTaskApplication(db2 db.Provider, idgen2 idgen.IIDGenerator, configFactory conf.IConfigLoaderFactory, benefit2 benefit.IBenefitService, ckDb ck.Provider, meter metrics.Meter, redis3 redis.Cmdable, mqFactory mq.IFactory, userClient userservice.Client, authClient authservice.Client, evalService evaluatorservice.Client, evalSetService evaluationsetservice.Client, exptService experimentservice.Client, datasetService datasetservice.Client, fileClient fileservice.Client, taskProcessor processor.TaskProcessor, aid int32, persistentCmdable redis.PersistentCmdable) (ITaskApplication, error) { +func InitTaskApplication(db2 db.Provider, idgen2 idgen.IIDGenerator, configFactory conf.IConfigLoaderFactory, benefit2 benefit.IBenefitService, ckDb ck.Provider, meter metrics.Meter, redis3 redis.Cmdable, mqFactory mq.IFactory, userClient userservice.Client, authClient authservice.Client, evalService evaluatorservice.Client, evalSetService evaluationsetservice.Client, exptService experimentservice.Client, datasetService datasetservice.Client, fileClient fileservice.Client, taskProcessor processor.TaskProcessor, aid int32, persistentCmdable redis.PersistentCmdable, taskHookProvider taskhook.IWorkflowProvider) (ITaskApplication, error) { iTaskDao := mysql.NewTaskDaoImpl(db2) iTaskDAO := redis2.NewTaskDAO(redis3) iTaskRunDao := mysql.NewTaskRunDaoImpl(db2) @@ -277,7 +279,10 @@ func InitTaskApplication(db2 db.Provider, idgen2 idgen.IIDGenerator, configFacto datasetServiceAdaptor := NewDatasetServiceAdapter(evalSetService, datasetService) iEvaluatorRPCAdapter := evaluator.NewEvaluatorRPCProvider(evalService) iEvaluationRPCAdapter := evaluation.NewEvaluationRPCProvider(exptService) - processorTaskProcessor := NewInitTaskProcessor(datasetServiceAdaptor, iEvaluatorRPCAdapter, iEvaluationRPCAdapter, iTaskRepo) + if taskHookProvider == nil { + taskHookProvider = taskhook.NewNoopTaskHookProvider() + } + processorTaskProcessor := NewInitTaskProcessor(datasetServiceAdaptor, iEvaluatorRPCAdapter, iEvaluationRPCAdapter, iTaskRepo, taskHookProvider) iStorageProvider := storage.NewTraceStorageProvider() iTenantProvider := tenant.NewTenantProvider(iTraceConfig) iFileProvider := file.NewFileRPCProvider(fileClient) @@ -455,11 +460,11 @@ func NewDatasetServiceAdapter(evalSetService evaluationsetservice.Client, datase } func NewInitTaskProcessor(datasetServiceProvider *service.DatasetServiceAdaptor, evalService rpc.IEvaluatorRPCAdapter, - evaluationService rpc.IEvaluationRPCAdapter, taskRepo repo4.ITaskRepo, + evaluationService rpc.IEvaluationRPCAdapter, taskRepo repo4.ITaskRepo, taskHookProvider taskhook.IWorkflowProvider, ) *processor.TaskProcessor { taskProcessor := processor.NewTaskProcessor() taskProcessor.Register(entity3.TaskTypeAutoEval, processor.NewAutoEvaluateProcessor( - 0, datasetServiceProvider, evalService, evaluationService, taskRepo, &processor.EvalTargetBuilderImpl{})) + 0, datasetServiceProvider, evalService, evaluationService, taskRepo, &processor.EvalTargetBuilderImpl{}, taskHookProvider)) return taskProcessor } diff --git a/backend/modules/observability/domain/component/rpc/evaluation.go b/backend/modules/observability/domain/component/rpc/evaluation.go index 80c0070c8..28771ae10 100644 --- a/backend/modules/observability/domain/component/rpc/evaluation.go +++ b/backend/modules/observability/domain/component/rpc/evaluation.go @@ -33,6 +33,7 @@ type SubmitExperimentReq struct { TargetFieldMapping *expt.TargetFieldMapping EvaluatorFieldMapping []*expt.EvaluatorFieldMapping ItemConcurNum *int32 + ItemRetryNum *int32 EvaluatorsConcurNum *int32 CreateEvalTargetParam *eval_target.CreateEvalTargetParam ExptType *expt.ExptType @@ -40,6 +41,9 @@ type SubmitExperimentReq struct { SourceType *expt.SourceType SourceID *string Session *common.Session + ExptTemplateID *int64 + IsWorkflowScheduled *bool + TimeRange *expt.TaskTimeRange } type InvokeExperimentReq struct { WorkspaceID int64 diff --git a/backend/modules/observability/domain/component/task/mocks/workflow_provider.go b/backend/modules/observability/domain/component/task/mocks/workflow_provider.go new file mode 100644 index 000000000..181b94635 --- /dev/null +++ b/backend/modules/observability/domain/component/task/mocks/workflow_provider.go @@ -0,0 +1,56 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/task (interfaces: IWorkflowProvider) +// +// Generated by this command: +// +// mockgen -destination=mocks/task_hook.go -package=mocks . IWorkflowProvider +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + task "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/task" + gomock "go.uber.org/mock/gomock" +) + +// MockIWorkflowProvider is a mock of IWorkflowProvider interface. +type MockIWorkflowProvider struct { + ctrl *gomock.Controller + recorder *MockIWorkflowProviderMockRecorder + isgomock struct{} +} + +// MockIWorkflowProviderMockRecorder is the mock recorder for MockIWorkflowProvider. +type MockIWorkflowProviderMockRecorder struct { + mock *MockIWorkflowProvider +} + +// NewMockIWorkflowProvider creates a new mock instance. +func NewMockIWorkflowProvider(ctrl *gomock.Controller) *MockIWorkflowProvider { + mock := &MockIWorkflowProvider{ctrl: ctrl} + mock.recorder = &MockIWorkflowProviderMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockIWorkflowProvider) EXPECT() *MockIWorkflowProviderMockRecorder { + return m.recorder +} + +// WorkflowCallback mocks base method. +func (m *MockIWorkflowProvider) WorkflowCallback(ctx context.Context, event *task.WorkflowCallbackParam) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WorkflowCallback", ctx, event) + ret0, _ := ret[0].(error) + return ret0 +} + +// WorkflowCallback indicates an expected call of WorkflowCallback. +func (mr *MockIWorkflowProviderMockRecorder) WorkflowCallback(ctx, event any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WorkflowCallback", reflect.TypeOf((*MockIWorkflowProvider)(nil).WorkflowCallback), ctx, event) +} diff --git a/backend/modules/observability/domain/component/task/workflow_provider.go b/backend/modules/observability/domain/component/task/workflow_provider.go new file mode 100644 index 000000000..88486cf67 --- /dev/null +++ b/backend/modules/observability/domain/component/task/workflow_provider.go @@ -0,0 +1,34 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package task + +import ( + "context" + + "github.com/coze-dev/coze-loop/backend/pkg/logs" + + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/entity" +) + +//go:generate mockgen -destination=mocks/task_hook.go -package=mocks . IWorkflowProvider + +type WorkflowCallbackParam struct { + Task *entity.ObservabilityTask + TaskRun *entity.TaskRun +} + +type IWorkflowProvider interface { + WorkflowCallback(ctx context.Context, event *WorkflowCallbackParam) error +} + +type NoopTaskHookProvider struct{} + +func NewNoopTaskHookProvider() IWorkflowProvider { + return &NoopTaskHookProvider{} +} + +func (n *NoopTaskHookProvider) WorkflowCallback(ctx context.Context, event *WorkflowCallbackParam) error { + logs.CtxInfo(ctx, "workflow callback called noop!") + return nil +} diff --git a/backend/modules/observability/domain/task/entity/event.go b/backend/modules/observability/domain/task/entity/event.go index 529314298..4fd4f1c17 100644 --- a/backend/modules/observability/domain/task/entity/event.go +++ b/backend/modules/observability/domain/task/entity/event.go @@ -291,6 +291,18 @@ func (s *OnlineExptTurnEvalResult) GetTaskIDFromExt() int64 { return taskID } +func (s *OnlineExptTurnEvalResult) GetExptTemplateIDFromExt() int64 { + if s == nil { + return 0 + } + exptTemplateIDStr := s.Ext["expt_template_id"] + exptTemplateID, err := strconv.ParseInt(exptTemplateIDStr, 10, 64) + if err != nil { + return 0 + } + return exptTemplateID +} + func (s *OnlineExptTurnEvalResult) GetWorkspaceIDFromExt() (string, int64) { if s == nil { return "", 0 diff --git a/backend/modules/observability/domain/task/entity/filter.go b/backend/modules/observability/domain/task/entity/filter.go index 8f5d7306d..f981f2694 100755 --- a/backend/modules/observability/domain/task/entity/filter.go +++ b/backend/modules/observability/domain/task/entity/filter.go @@ -39,14 +39,14 @@ const ( FieldTypeDouble FieldType = "double" FieldTypeBool FieldType = "bool" - TaskFieldNameTaskStatus TaskFieldName = "task_status" - TaskFieldNameTaskName TaskFieldName = "task_name" - TaskFieldNameTaskType TaskFieldName = "task_type" - TaskFieldNameSampleRate TaskFieldName = "sample_rate" - TaskFieldNameCreatedBy TaskFieldName = "created_by" - TaskFieldNameTaskSource TaskFieldName = "task_source" - - TaskSourceUser TaskSourceValue = "user" + TaskFieldNameTaskStatus TaskFieldName = "task_status" + TaskFieldNameTaskName TaskFieldName = "task_name" + TaskFieldNameTaskType TaskFieldName = "task_type" + TaskFieldNameSampleRate TaskFieldName = "sample_rate" + TaskFieldNameCreatedBy TaskFieldName = "created_by" + TaskFieldNameTaskSource TaskFieldName = "task_source" + TaskFieldNameUpdatedBy TaskFieldName = "updated_by" + TaskSourceUser TaskSourceValue = "user" ) // TaskFilterFields aggregates multiple TaskFilterField expressions. diff --git a/backend/modules/observability/domain/task/entity/task.go b/backend/modules/observability/domain/task/entity/task.go index 6a03ade1e..9b8432126 100644 --- a/backend/modules/observability/domain/task/entity/task.go +++ b/backend/modules/observability/domain/task/entity/task.go @@ -7,6 +7,8 @@ import ( "context" "time" + datadataset "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/dataset" taskdto "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/task" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/loop_span" @@ -79,8 +81,8 @@ type ObservabilityTask struct { CreatedBy string // 创建人 UpdatedBy string // 更新人 TaskSource *string // 创建来源 - - TaskRuns []*TaskRun + WorkflowID int64 // 关联 workflow ID + TaskRuns []*TaskRun } type RunDetail struct { @@ -108,13 +110,31 @@ type Sampler struct { CycleTimeUnit TimeUnit `json:"cycle_time_unit"` } type TaskConfig struct { - AutoEvaluateConfigs []*AutoEvaluateConfig `json:"auto_evaluate_configs"` - DataReflowConfig []*DataReflowConfig + AutoEvaluateConfigs []*AutoEvaluateConfig `json:"auto_evaluate_configs"` + DataReflowConfig []*DataReflowConfig + EvaluationExperimentConfig *EvaluationExperimentConfig `json:"evaluation_experiment_config,omitempty"` + SourceInfo []*SourceInfo `json:"source_info,omitempty"` + IsWorkflowScheduled *bool `json:"is_workflow_scheduled,omitempty"` } type AutoEvaluateConfig struct { EvaluatorVersionID int64 `json:"evaluator_version_id"` EvaluatorID int64 `json:"evaluator_id"` FieldMappings []*EvaluateFieldMapping `json:"field_mappings"` + ScoreWeight *float64 `json:"score_weight,omitempty"` + EvaluatorVersion *string `json:"evaluator_version,omitempty"` +} + +type EvaluationExperimentConfig struct { + ItemConcurrencyCount *int32 `json:"item_concurrency_count,omitempty"` + ItemMaxRetryCount *int32 `json:"item_max_retry_count,omitempty"` + SourceTargetID *string `json:"source_target_id,omitempty"` + ExptTemplateID *int64 `json:"expt_template_id,omitempty"` + SourceTargetVersion *string `json:"source_target_version,omitempty"` + FullEvalSetFieldMappings []*EvaluateFieldMapping `json:"full_eval_set_field_mappings,omitempty"` +} +type SourceInfo struct { + Name *string `json:"name,omitempty"` + Version *string `json:"version,omitempty"` } type EvaluateFieldMapping struct { // 数据集字段约束 @@ -122,12 +142,14 @@ type EvaluateFieldMapping struct { TraceFieldKey string `json:"trace_field_key"` TraceFieldJsonpath string `json:"trace_field_jsonpath"` EvalSetName *string `json:"eval_set_name"` + DatasetKey *string `json:"dataset_key"` } type DataReflowConfig struct { - DatasetID *int64 `json:"dataset_id"` - DatasetName *string `json:"dataset_name"` - DatasetSchema dataset.DatasetSchema `json:"dataset_schema"` - FieldMappings []dataset.FieldMapping `json:"field_mappings"` + DatasetID *int64 `json:"dataset_id"` + DatasetName *string `json:"dataset_name"` + DatasetSchema dataset.DatasetSchema `json:"dataset_schema"` + FieldMappings []dataset.FieldMapping `json:"field_mappings"` + DatasetCategory *datadataset.DatasetCategory `json:"dataset_category"` } type TaskRun struct { @@ -327,3 +349,7 @@ func (t *ObservabilityTask) GetPlatformType() loop_span.PlatformType { } return loop_span.PlatformDefault } + +func (t *ObservabilityTask) IsNewWorkflowTask() bool { + return t.WorkflowID != 0 +} diff --git a/backend/modules/observability/domain/task/repo/task.go b/backend/modules/observability/domain/task/repo/task.go index 056115b97..c638746c2 100644 --- a/backend/modules/observability/domain/task/repo/task.go +++ b/backend/modules/observability/domain/task/repo/task.go @@ -16,6 +16,7 @@ type ListTaskParam struct { ReqLimit int32 ReqOffset int32 OrderBy *common.OrderBy + NeedOnlyOld bool } //go:generate mockgen -destination=mocks/Task.go -package=mocks . ITaskRepo diff --git a/backend/modules/observability/domain/task/service/task_callback.go b/backend/modules/observability/domain/task/service/task_callback.go index f1e54dcd4..c94d47d33 100644 --- a/backend/modules/observability/domain/task/service/task_callback.go +++ b/backend/modules/observability/domain/task/service/task_callback.go @@ -119,6 +119,8 @@ func (t *TaskCallbackServiceImpl) AutoEvalCallback(ctx context.Context, event *e turn.Score, turn.Reasoning, turn.GetUserID(), + event.ExptID, + turn.GetExptTemplateIDFromExt(), ) if err != nil { return err diff --git a/backend/modules/observability/domain/task/service/taskexe/processor.go b/backend/modules/observability/domain/task/service/taskexe/processor.go index d41555b21..625ca29c1 100644 --- a/backend/modules/observability/domain/task/service/taskexe/processor.go +++ b/backend/modules/observability/domain/task/service/taskexe/processor.go @@ -11,9 +11,10 @@ import ( ) type Trigger struct { - Task *entity.ObservabilityTask - Span *loop_span.Span - TaskRun *entity.TaskRun + Task *entity.ObservabilityTask + Span *loop_span.Span + TaskRun *entity.TaskRun + Trajectory *loop_span.Trajectory } type OnTaskRunCreatedReq struct { diff --git a/backend/modules/observability/domain/task/service/taskexe/processor/auto_evaluate.go b/backend/modules/observability/domain/task/service/taskexe/processor/auto_evaluate.go index dd7d082a1..a47b80482 100644 --- a/backend/modules/observability/domain/task/service/taskexe/processor/auto_evaluate.go +++ b/backend/modules/observability/domain/task/service/taskexe/processor/auto_evaluate.go @@ -9,17 +9,23 @@ import ( "strconv" "time" + "github.com/bytedance/gg/gslice" + "github.com/samber/lo" + "github.com/coze-dev/coze-loop/backend/modules/evaluation/pkg/errno" "github.com/bytedance/gg/gptr" + "github.com/coze-dev/coze-loop/backend/infra/backoff" "github.com/coze-dev/coze-loop/backend/infra/middleware/session" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/common" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/eval_set" + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/evaluator" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/expt" dataset0 "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/dataset" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/task" tconv "github.com/coze-dev/coze-loop/backend/modules/observability/application/convertor/task" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/rpc" + taskhook "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/task" task_entity "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/entity" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/repo" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/service/taskexe" @@ -43,6 +49,7 @@ type AutoEvaluateProcessor struct { taskRepo repo.ITaskRepo aid int32 evalTargetBuilder EvalTargetBuilder + workflowProvider taskhook.IWorkflowProvider } func NewAutoEvaluateProcessor( @@ -52,7 +59,12 @@ func NewAutoEvaluateProcessor( evaluationService rpc.IEvaluationRPCAdapter, taskRepo repo.ITaskRepo, evalTargetBuilder EvalTargetBuilder, + workflowProvider taskhook.IWorkflowProvider, ) *AutoEvaluateProcessor { + if workflowProvider == nil { + logs.Info("Auto Evaluate workflowProvider is nil") + workflowProvider = taskhook.NewNoopTaskHookProvider() + } return &AutoEvaluateProcessor{ datasetServiceAdaptor: datasetServiceProvider, evalSvc: evalService, @@ -60,6 +72,7 @@ func NewAutoEvaluateProcessor( taskRepo: taskRepo, aid: aid, evalTargetBuilder: evalTargetBuilder, + workflowProvider: workflowProvider, } } @@ -86,6 +99,7 @@ func (p *AutoEvaluateProcessor) ValidateConfig(ctx context.Context, config any) return errorx.NewByCode(obErrorx.CommonInvalidParamCode) } // Verify evaluator version validity + evaluatorVersionIDs = lo.Uniq(evaluatorVersionIDs) evaluators, _, err := p.evalSvc.BatchGetEvaluatorVersions(ctx, &rpc.BatchGetEvaluatorVersionsParam{ WorkspaceID: cfg.WorkspaceID, EvaluatorVersionIds: evaluatorVersionIDs, @@ -107,8 +121,26 @@ func (p *AutoEvaluateProcessor) Invoke(ctx context.Context, trigger *taskexe.Tri workspaceID := trigger.Task.WorkspaceID sessionInfo := p.getSession(ctx, trigger.Task) var mapping []*task_entity.EvaluateFieldMapping + evalsetName := make(map[string]bool) + for _, autoEvaluateConfig := range trigger.Task.TaskConfig.AutoEvaluateConfigs { mapping = append(mapping, autoEvaluateConfig.FieldMappings...) + for _, fieldMapping := range autoEvaluateConfig.FieldMappings { + if fieldMapping.EvalSetName != nil { + evalsetName[*fieldMapping.EvalSetName] = true + } + } + } + if trigger.Task.TaskConfig.EvaluationExperimentConfig != nil && trigger.Task.TaskConfig.EvaluationExperimentConfig.FullEvalSetFieldMappings != nil { + for _, fieldMapping := range trigger.Task.TaskConfig.EvaluationExperimentConfig.FullEvalSetFieldMappings { + if fieldMapping.EvalSetName == nil { + continue + } + if ok := evalsetName[*fieldMapping.EvalSetName]; !ok { + mapping = append(mapping, fieldMapping) + evalsetName[*fieldMapping.EvalSetName] = true + } + } } turns := buildItems(ctx, []*loop_span.Span{trigger.Span}, mapping, taskRun.GetTaskRunConfig().GetAutoEvaluateRunConfig().GetSchema(), strconv.FormatInt(taskRun.ID, 10)) if len(turns) == 0 { @@ -127,6 +159,18 @@ func (p *AutoEvaluateProcessor) Invoke(ctx context.Context, trigger *taskexe.Tri _ = p.taskRepo.DecrTaskRunCount(ctx, trigger.Task.ID, taskRun.ID, taskTTL) return nil } + extMap := map[string]string{ + "workspace_id": strconv.FormatInt(trigger.Task.WorkspaceID, 10), + "span_id": trigger.Span.SpanID, + "task_id": cast.ToString(trigger.Task.ID), + "task_run_id": cast.ToString(taskRun.ID), + "span_start_time": cast.ToString(trigger.Span.StartTime/1000 - time.Hour.Milliseconds()), + "span_end_time": cast.ToString(trigger.Span.StartTime/1000 + time.Hour.Milliseconds()), + "platform_type": string(trigger.Task.GetPlatformType()), + } + if trigger.Task.TaskConfig.EvaluationExperimentConfig != nil && trigger.Task.TaskConfig.EvaluationExperimentConfig.ExptTemplateID != nil { + extMap["expt_template_id"] = cast.ToString(trigger.Task.TaskConfig.EvaluationExperimentConfig.ExptTemplateID) + } addedItems, err := p.evaluationSvc.InvokeExperiment(ctx, &rpc.InvokeExperimentReq{ WorkspaceID: workspaceID, EvaluationSetID: taskRun.GetTaskRunConfig().GetAutoEvaluateRunConfig().GetEvalID(), @@ -144,15 +188,7 @@ func (p *AutoEvaluateProcessor) Invoke(ctx context.Context, trigger *taskexe.Tri ExperimentID: gptr.Of(taskRun.GetTaskRunConfig().GetAutoEvaluateRunConfig().GetExptID()), ExperimentRunID: gptr.Of(taskRun.GetTaskRunConfig().GetAutoEvaluateRunConfig().GetExptRunID()), Session: sessionInfo, - Ext: map[string]string{ - "workspace_id": strconv.FormatInt(trigger.Task.WorkspaceID, 10), - "span_id": trigger.Span.SpanID, - "task_id": cast.ToString(trigger.Task.ID), - "task_run_id": cast.ToString(taskRun.ID), - "span_start_time": cast.ToString(trigger.Span.StartTime/1000 - time.Hour.Milliseconds()), - "span_end_time": cast.ToString(trigger.Span.StartTime/1000 + time.Hour.Milliseconds()), - "platform_type": string(trigger.Task.GetPlatformType()), - }, + Ext: extMap, }) if err != nil { _ = p.taskRepo.DecrTaskCount(ctx, trigger.Task.ID, taskTTL) @@ -268,6 +304,15 @@ func (p *AutoEvaluateProcessor) OnTaskFinished(ctx context.Context, param taskex logs.CtxError(ctx, "OnUpdateChangeProcessor failed, taskID:%d, err:%v", param.Task.ID, err) return err } + + //if err := p.workflowProvider.WorkflowCallback(ctx, &taskhook.WorkflowCallbackParam{ + // Task: param.Task, + // TaskRun: param.TaskRun, + //}); err != nil { + // logs.CtxError(ctx, "workflowProvider.WorkflowCallback failed, taskID:%d, err:%v", param.Task.ID, err) + // return err + //} + if err := p.taskRepo.RemoveNonFinalTask(ctx, strconv.FormatInt(param.Task.WorkspaceID, 10), param.Task.ID); err != nil { logs.CtxError(ctx, "RemoveNonFinalTask failed, taskID:%d, err:%v", param.Task.ID, err) return err @@ -290,7 +335,7 @@ func (p *AutoEvaluateProcessor) OnTaskRunCreated(ctx context.Context, param task var evaluationSetColumns []string var evaluatorVersionIds []int64 var evaluatorFieldMappings []*expt.EvaluatorFieldMapping - evaluationSetColumns = append(evaluationSetColumns, "span_id", "trace_id", "run_id") + evaluationSetColumns = append(evaluationSetColumns, "span_id", "trace_id") autoEvaluateConfigs := currentTask.TaskConfig.AutoEvaluateConfigs evaluationSetSchema, fromEvalSet := getBasicEvaluationSetSchema(evaluationSetColumns) for _, autoEvaluateConfig := range autoEvaluateConfigs { @@ -300,31 +345,69 @@ func (p *AutoEvaluateProcessor) OnTaskRunCreated(ctx context.Context, param task if fieldMapping.FieldSchema == nil { continue } + logs.CtxInfo(ctx, "Data key fall back: %v", fieldMapping) + // 兼容老版 无 evalsetname + var evalsetName *string + if fieldMapping.EvalSetName == nil || *fieldMapping.EvalSetName == "" { + evalsetName = fieldMapping.DatasetKey + } else { + evalsetName = fieldMapping.EvalSetName + } fromEvalSet = append(fromEvalSet, &expt.FieldMapping{ FieldName: fieldMapping.FieldSchema.Name, - FromFieldName: fieldMapping.EvalSetName, + FromFieldName: evalsetName, }) - if slices.Contains(evaluationSetColumns, *fieldMapping.EvalSetName) { + if slices.Contains(evaluationSetColumns, *evalsetName) { continue } - // historical data compatibility, convert plain_text to text, data needs to be refreshed - evaluationSetSchema.FieldSchemas = append(evaluationSetSchema.FieldSchemas, &dataset0.FieldSchema{ - Key: gptr.Of(*fieldMapping.EvalSetName), - Name: gptr.Of(*fieldMapping.EvalSetName), + evalsetSchema := &dataset0.FieldSchema{ + // todo fby test + Key: fieldMapping.DatasetKey, + Name: evalsetName, Description: gptr.Of(fieldMapping.TraceFieldJsonpath), ContentType: fieldMapping.FieldSchema.ContentType, // DefaultDisplayFormat: gptr.Of(dataset.FieldDisplayFormat_PlainText), TextSchema: fieldMapping.FieldSchema.TextSchema, // Hidden: gptr.Of(false), - }) - evaluationSetColumns = append(evaluationSetColumns, *fieldMapping.EvalSetName) + } + // historical data compatibility, convert plain_text to text, data needs to be refreshed + evaluationSetSchema.FieldSchemas = append(evaluationSetSchema.FieldSchemas, evalsetSchema) + evaluationSetColumns = append(evaluationSetColumns, *evalsetName) } evaluatorFieldMappings = append(evaluatorFieldMappings, &expt.EvaluatorFieldMapping{ EvaluatorVersionID: autoEvaluateConfig.EvaluatorVersionID, FromEvalSet: fromEvalSet, + EvaluatorIDVersionItem: &evaluator.EvaluatorIDVersionItem{ + EvaluatorID: gptr.Of(autoEvaluateConfig.EvaluatorID), + EvaluatorVersionID: gptr.Of(autoEvaluateConfig.EvaluatorVersionID), + ScoreWeight: autoEvaluateConfig.ScoreWeight, + Version: autoEvaluateConfig.EvaluatorVersion, + }, }) } + if currentTask.TaskConfig.EvaluationExperimentConfig != nil && currentTask.TaskConfig.EvaluationExperimentConfig.FullEvalSetFieldMappings != nil { + addEvalSetNameList := gslice.ToMap(evaluationSetSchema.FieldSchemas, func(schema *dataset0.FieldSchema) (string, bool) { + return schema.GetName(), true + }) + for _, mapping := range currentTask.TaskConfig.EvaluationExperimentConfig.FullEvalSetFieldMappings { + if mapping.EvalSetName == nil { + continue + } + if ok := addEvalSetNameList[*mapping.EvalSetName]; ok { + continue + } + evalsetSchema := &dataset0.FieldSchema{ + Key: mapping.DatasetKey, + Name: mapping.EvalSetName, + Description: gptr.Of(mapping.TraceFieldJsonpath), + ContentType: mapping.FieldSchema.ContentType, + TextSchema: mapping.FieldSchema.TextSchema, + } + addEvalSetNameList[*mapping.EvalSetName] = true + evaluationSetSchema.FieldSchemas = append(evaluationSetSchema.FieldSchemas, evalsetSchema) + } + } category := getCategory(task.TaskType(currentTask.TaskType)) schema := convertDatasetSchemaDTO2DO(evaluationSetSchema) logs.CtxInfo(ctx, "[auto_task] CreateDataset,category:%s", category) @@ -344,7 +427,7 @@ func (p *AutoEvaluateProcessor) OnTaskRunCreated(ctx context.Context, param task category, schema, sessionInfo, - ptr.Of(entity.BizCategoryFromOnlineTrace), + ptr.Of(entity.BizCategoryFromOnlineTrace), false, 0, )) if err != nil { logs.CtxError(ctx, "CreateDataset failed, workspace_id=%d, err=%#v", currentTask.WorkspaceID, err) @@ -358,7 +441,7 @@ func (p *AutoEvaluateProcessor) OnTaskRunCreated(ctx context.Context, param task EvalSetVersionID: gptr.Of(datasetID), EvaluatorVersionIds: evaluatorVersionIds, Name: ptr.Of(exptName), - Desc: gptr.Of("Auto Task Experiment"), + Desc: currentTask.Description, EvalSetID: gptr.Of(datasetID), EvaluatorFieldMapping: evaluatorFieldMappings, TargetFieldMapping: &expt.TargetFieldMapping{ @@ -370,6 +453,21 @@ func (p *AutoEvaluateProcessor) OnTaskRunCreated(ctx context.Context, param task SourceType: gptr.Of(expt.SourceType_AutoTask), SourceID: gptr.Of(cast.ToString(currentTask.ID)), Session: sessionInfo, + IsWorkflowScheduled: currentTask.TaskConfig.IsWorkflowScheduled, + TimeRange: &expt.TaskTimeRange{ + StartTime: gptr.Of(param.RunStartAt), + EndTime: gptr.Of(param.RunEndAt), + }, + } + if currentTask.TaskConfig.EvaluationExperimentConfig != nil { + submitExperimentReq.ItemRetryNum = currentTask.TaskConfig.EvaluationExperimentConfig.ItemMaxRetryCount + submitExperimentReq.ItemConcurNum = currentTask.TaskConfig.EvaluationExperimentConfig.ItemConcurrencyCount + submitExperimentReq.ExptTemplateID = currentTask.TaskConfig.EvaluationExperimentConfig.ExptTemplateID + } + if currentTask.WorkflowID != 0 { + // triggered by workflow + submitExperimentReq.SourceType = gptr.Of(expt.SourceType(3)) + submitExperimentReq.SourceID = gptr.Of(cast.ToString(currentTask.WorkflowID)) } logs.CtxInfo(ctx, "[auto_task] SubmitExperiment:%+v", submitExperimentReq) exptID, exptRunID, err := p.evaluationSvc.SubmitExperiment(ctx, &submitExperimentReq) @@ -440,6 +538,16 @@ func (p *AutoEvaluateProcessor) OnTaskRunFinished(ctx context.Context, param tas logs.CtxError(ctx, "[auto_task] OnFinishTaskRunProcessor, UpdateTaskRun err, taskRunID:%d, err:%v", taskRun.ID, err) return err } + if param.Task.TaskSource != nil && *param.Task.TaskSource == task.TaskSourceWorkflow { + if err := backoff.RetryWithMaxTimes(ctx, 3, func() error { + return p.workflowProvider.WorkflowCallback(ctx, &taskhook.WorkflowCallbackParam{ + Task: param.Task, + TaskRun: param.TaskRun, + }) + }); err != nil { + logs.CtxError(ctx, "workflowProvider.WorkflowCallback failed, taskID:%d, err:%v", param.Task.ID, err) + } + } return nil } diff --git a/backend/modules/observability/domain/task/service/taskexe/processor/auto_evaluate_test.go b/backend/modules/observability/domain/task/service/taskexe/processor/auto_evaluate_test.go index b2260f19b..39fa0ef2f 100755 --- a/backend/modules/observability/domain/task/service/taskexe/processor/auto_evaluate_test.go +++ b/backend/modules/observability/domain/task/service/taskexe/processor/auto_evaluate_test.go @@ -22,12 +22,15 @@ import ( "github.com/coze-dev/coze-loop/backend/infra/middleware/session" datadataset "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/common" + evDomain "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/expt" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/expt" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/dataset" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/task" "github.com/coze-dev/coze-loop/backend/modules/evaluation/pkg/errno" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/rpc" rpcmock "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/rpc/mocks" + taskhook "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/task" + hookmocks "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/task/mocks" taskentity "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/entity" repomocks "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/repo/mocks" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/service/taskexe" @@ -1301,7 +1304,7 @@ func TestAutoEvaluateProcessor_NewAutoEvaluateProcessor(t *testing.T) { taskRepo := repomocks.NewMockITaskRepo(ctrl) // Test constructor - proc := NewAutoEvaluateProcessor(123, datasetServiceAdaptor, evalMock, evaluationMock, taskRepo, &EvalTargetBuilderImpl{}) + proc := NewAutoEvaluateProcessor(123, datasetServiceAdaptor, evalMock, evaluationMock, taskRepo, &EvalTargetBuilderImpl{}, nil) assert.NotNil(t, proc) assert.Equal(t, int32(123), proc.aid) @@ -1309,6 +1312,248 @@ func TestAutoEvaluateProcessor_NewAutoEvaluateProcessor(t *testing.T) { assert.Equal(t, evalMock, proc.evalSvc) assert.Equal(t, evaluationMock, proc.evaluationSvc) assert.Equal(t, taskRepo, proc.taskRepo) + assert.NotNil(t, proc.workflowProvider) +} + +func TestAutoEvaluateProcessor_Invoke_ExptTemplateID_InExtMap(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + repoMock := repomocks.NewMockITaskRepo(ctrl) + evalAdapter := &fakeEvaluationAdapter{} + + workspaceID := int64(12345) + taskID := int64(1002) + runID := int64(2003) + + schema := []*traceentity.FieldSchema{{ + Key: gptr.Of("input_text"), + Name: "input_text", + ContentType: traceentity.ContentType_Text, + }} + schemaBytes, _ := json.Marshal(schema) + schemaStr := string(schemaBytes) + + exptTemplateID := int64(9999) + taskObj := &taskentity.ObservabilityTask{ + ID: taskID, + WorkspaceID: workspaceID, + TaskType: taskentity.TaskTypeAutoEval, + TaskStatus: taskentity.TaskStatusRunning, + Sampler: &taskentity.Sampler{ + SampleRate: 1, + SampleSize: 1000, + }, + EffectiveTime: &taskentity.EffectiveTime{ + StartAt: time.Now().Add(-time.Hour).UnixMilli(), + EndAt: time.Now().Add(time.Hour).UnixMilli(), + }, + CreatedBy: "1001", + TaskConfig: &taskentity.TaskConfig{ + AutoEvaluateConfigs: []*taskentity.AutoEvaluateConfig{{ + FieldMappings: []*taskentity.EvaluateFieldMapping{{ + EvalSetName: gptr.Of("input_text"), + TraceFieldKey: "Input", + TraceFieldJsonpath: "", + }}, + }}, + EvaluationExperimentConfig: &taskentity.EvaluationExperimentConfig{ + ExptTemplateID: &exptTemplateID, + }, + }, + } + + taskRun := &taskentity.TaskRun{ + ID: runID, TaskID: taskID, WorkspaceID: workspaceID, + TaskType: taskentity.TaskRunTypeNewData, RunStatus: taskentity.TaskRunStatusRunning, + RunStartAt: time.Now().Add(-30 * time.Minute), RunEndAt: time.Now().Add(30 * time.Minute), + TaskRunConfig: &taskentity.TaskRunConfig{AutoEvaluateRunConfig: &taskentity.AutoEvaluateRunConfig{ + ExptID: 3003, ExptRunID: 3004, EvalID: 4004, SchemaID: 5005, + Schema: gptr.Of(schemaStr), + EndAt: time.Now().Add(30 * time.Minute).UnixMilli(), + Status: string(taskentity.TaskRunStatusRunning), + }}, + } + + span := &loop_span.Span{TraceID: "trace-abc", SpanID: "span-xyz", StartTime: time.Now().UnixMilli(), Input: "hello"} + + repoMock.EXPECT().IncrTaskCount(gomock.Any(), taskID, gomock.Any()).Return(nil).AnyTimes() + repoMock.EXPECT().IncrTaskRunCount(gomock.Any(), taskID, runID, gomock.Any()).Return(nil).AnyTimes() + repoMock.EXPECT().GetTaskCount(gomock.Any(), taskID).Return(int64(0), nil).AnyTimes() + repoMock.EXPECT().GetTaskRunCount(gomock.Any(), taskID, runID).Return(int64(0), nil).AnyTimes() + repoMock.EXPECT().DecrTaskCount(gomock.Any(), taskID, gomock.Any()).Return(nil).AnyTimes() + repoMock.EXPECT().DecrTaskRunCount(gomock.Any(), taskID, runID, gomock.Any()).Return(nil).AnyTimes() + + proc := &AutoEvaluateProcessor{taskRepo: repoMock, evaluationSvc: evalAdapter} + _ = proc.Invoke(context.Background(), &taskexe.Trigger{Task: taskObj, Span: span, TaskRun: taskRun}) + + require.NotNil(t, evalAdapter.lastInvoke) + assert.Equal(t, "9999", evalAdapter.lastInvoke.Ext["expt_template_id"]) +} + +func TestAutoEvaluateProcessor_OnCreateTaskRunChange_WithEvalExperimentConfig(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + datasetProvider := rpcmock.NewMockIDatasetProvider(ctrl) + repoMock := repomocks.NewMockITaskRepo(ctrl) + repoAdapter := &taskRepoMockAdapter{MockITaskRepo: repoMock} + + taskObj := buildTestTask(t) + itemRetry := int32(3) + itemConcur := int32(5) + exptTplID := int64(7777) + taskObj.TaskConfig.EvaluationExperimentConfig = &taskentity.EvaluationExperimentConfig{ + ItemMaxRetryCount: &itemRetry, + ItemConcurrencyCount: &itemConcur, + ExptTemplateID: &exptTplID, + FullEvalSetFieldMappings: []*taskentity.EvaluateFieldMapping{ + { + EvalSetName: gptr.Of("extra_field"), + DatasetKey: gptr.Of("extra_field"), + TraceFieldJsonpath: "$.output", + FieldSchema: &dataset.FieldSchema{ + ContentType: gptr.Of(common.ContentTypeText), + TextSchema: gptr.Of("{}"), + }, + }, + {EvalSetName: nil}, + {EvalSetName: gptr.Of("field_1")}, + }, + } + + param := taskexe.OnTaskRunCreatedReq{ + CurrentTask: taskObj, + RunType: taskentity.TaskRunTypeNewData, + RunStartAt: time.Now().Add(-time.Minute).UnixMilli(), + RunEndAt: time.Now().Add(time.Hour).UnixMilli(), + } + + datasetProvider.EXPECT().CreateDataset(gomock.Any(), gomock.Any()).Return(int64(9001), nil) + datasetProvider.EXPECT().GetDataset(gomock.Any(), taskObj.WorkspaceID, int64(9001), traceentity.DatasetCategory_Evaluation). + Return(&traceentity.Dataset{DatasetVersion: traceentity.DatasetVersion{DatasetSchema: traceentity.DatasetSchema{ID: 7001}}}, nil) + repoMock.EXPECT().CreateTaskRun(gomock.Any(), gomock.AssignableToTypeOf(&taskentity.TaskRun{})).Return(int64(1), nil) + + adaptor := service.NewDatasetServiceAdaptor() + adaptor.Register(traceentity.DatasetCategory_Evaluation, datasetProvider) + + evalAdapter := &fakeEvaluationAdapter{} + evalAdapter.submitResp.exptID = 1111 + evalAdapter.submitResp.exptRunID = 2222 + + proc := &AutoEvaluateProcessor{ + datasetServiceAdaptor: adaptor, + evaluationSvc: evalAdapter, + taskRepo: repoAdapter, + aid: 321, + evalTargetBuilder: &EvalTargetBuilderImpl{}, + } + + ctx := session.WithCtxUser(context.Background(), &session.User{ID: taskObj.CreatedBy}) + err := proc.OnTaskRunCreated(ctx, param) + assert.NoError(t, err) + + require.NotNil(t, evalAdapter.lastSubmit) + assert.Equal(t, &itemRetry, evalAdapter.lastSubmit.ItemRetryNum) + assert.Equal(t, &itemConcur, evalAdapter.lastSubmit.ItemConcurNum) + assert.Equal(t, &exptTplID, evalAdapter.lastSubmit.ExptTemplateID) +} + +func TestAutoEvaluateProcessor_OnCreateTaskRunChange_WithWorkflowID(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + datasetProvider := rpcmock.NewMockIDatasetProvider(ctrl) + repoMock := repomocks.NewMockITaskRepo(ctrl) + repoAdapter := &taskRepoMockAdapter{MockITaskRepo: repoMock} + + taskObj := buildTestTask(t) + taskObj.WorkflowID = 888 + + param := taskexe.OnTaskRunCreatedReq{ + CurrentTask: taskObj, + RunType: taskentity.TaskRunTypeNewData, + RunStartAt: time.Now().Add(-time.Minute).UnixMilli(), + RunEndAt: time.Now().Add(time.Hour).UnixMilli(), + } + + datasetProvider.EXPECT().CreateDataset(gomock.Any(), gomock.Any()).Return(int64(9001), nil) + datasetProvider.EXPECT().GetDataset(gomock.Any(), taskObj.WorkspaceID, int64(9001), traceentity.DatasetCategory_Evaluation). + Return(&traceentity.Dataset{DatasetVersion: traceentity.DatasetVersion{DatasetSchema: traceentity.DatasetSchema{ID: 7001}}}, nil) + repoMock.EXPECT().CreateTaskRun(gomock.Any(), gomock.AssignableToTypeOf(&taskentity.TaskRun{})).Return(int64(1), nil) + + adaptor := service.NewDatasetServiceAdaptor() + adaptor.Register(traceentity.DatasetCategory_Evaluation, datasetProvider) + + evalAdapter := &fakeEvaluationAdapter{} + evalAdapter.submitResp.exptID = 1111 + evalAdapter.submitResp.exptRunID = 2222 + + proc := &AutoEvaluateProcessor{ + datasetServiceAdaptor: adaptor, + evaluationSvc: evalAdapter, + taskRepo: repoAdapter, + aid: 321, + evalTargetBuilder: &EvalTargetBuilderImpl{}, + } + + ctx := session.WithCtxUser(context.Background(), &session.User{ID: taskObj.CreatedBy}) + err := proc.OnTaskRunCreated(ctx, param) + assert.NoError(t, err) + + require.NotNil(t, evalAdapter.lastSubmit) + assert.Equal(t, gptr.Of(evDomain.SourceType(3)), evalAdapter.lastSubmit.SourceType) + assert.Equal(t, gptr.Of("888"), evalAdapter.lastSubmit.SourceID) +} + +func TestAutoEvaluateProcessor_OnFinishTaskRunChange_WorkflowCallback(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + repoMock := repomocks.NewMockITaskRepo(ctrl) + repoAdapter := &taskRepoMockAdapter{MockITaskRepo: repoMock} + evalMock := &fakeEvaluationAdapter{} + workflowMock := hookmocks.NewMockIWorkflowProvider(ctrl) + + taskRun := &taskentity.TaskRun{ + ID: 8001, + TaskRunConfig: &taskentity.TaskRunConfig{ + AutoEvaluateRunConfig: &taskentity.AutoEvaluateRunConfig{ + ExptID: 9001, ExptRunID: 9002, + }, + }, + } + taskObj := &taskentity.ObservabilityTask{ + ID: 101, + WorkspaceID: 1234, + CreatedBy: "1001", + TaskSource: gptr.Of(task.TaskSourceWorkflow), + } + + repoMock.EXPECT().UpdateTaskRun(gomock.Any(), taskRun).Return(nil) + workflowMock.EXPECT().WorkflowCallback(gomock.Any(), &taskhook.WorkflowCallbackParam{ + Task: taskObj, TaskRun: taskRun, + }).Return(errors.New("temp error")).Times(2) + workflowMock.EXPECT().WorkflowCallback(gomock.Any(), &taskhook.WorkflowCallbackParam{ + Task: taskObj, TaskRun: taskRun, + }).Return(nil).Times(1) + + proc := &AutoEvaluateProcessor{ + taskRepo: repoAdapter, + evaluationSvc: evalMock, + workflowProvider: workflowMock, + } + + err := proc.OnTaskRunFinished(context.Background(), taskexe.OnTaskRunFinishedReq{ + Task: taskObj, + TaskRun: taskRun, + }) + assert.NoError(t, err) + assert.Equal(t, taskentity.TaskRunStatusDone, taskRun.RunStatus) } // fakeEvaluatorAdapter 是 IEvaluatorRPCAdapter 的fake实现 @@ -1350,11 +1595,12 @@ type fakeEvaluationAdapter struct { finishResp struct { err error } - // capture last invoke param for assertions lastInvoke *rpc.InvokeExperimentReq + lastSubmit *rpc.SubmitExperimentReq } func (f *fakeEvaluationAdapter) SubmitExperiment(ctx context.Context, param *rpc.SubmitExperimentReq) (exptID, exptRunID int64, err error) { + f.lastSubmit = param return f.submitResp.exptID, f.submitResp.exptRunID, f.submitResp.err } diff --git a/backend/modules/observability/domain/task/service/taskexe/processor/eval_target_builder.go b/backend/modules/observability/domain/task/service/taskexe/processor/eval_target_builder.go index 6a38d574c..b50b5d58c 100644 --- a/backend/modules/observability/domain/task/service/taskexe/processor/eval_target_builder.go +++ b/backend/modules/observability/domain/task/service/taskexe/processor/eval_target_builder.go @@ -5,11 +5,12 @@ package processor import ( "context" - "strconv" eval_target_d "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/eval_target" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/eval_target" + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/common" task_entity "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/entity" + "github.com/coze-dev/coze-loop/backend/pkg/logs" "github.com/samber/lo" ) @@ -22,8 +23,36 @@ type EvalTargetBuilderImpl struct { } func (b *EvalTargetBuilderImpl) Build(ctx context.Context, task *task_entity.ObservabilityTask) *eval_target.CreateEvalTargetParam { - return &eval_target.CreateEvalTargetParam{ + var sourceTargetID *string = nil + if task.TaskConfig.EvaluationExperimentConfig != nil { + sourceTargetID = task.TaskConfig.EvaluationExperimentConfig.SourceTargetID + } + ret := &eval_target.CreateEvalTargetParam{ EvalTargetType: lo.ToPtr(eval_target_d.EvalTargetType_Trace), - SourceTargetID: lo.ToPtr(strconv.FormatInt(task.ID, 10)), + SourceTargetID: sourceTargetID, + } + if task.SpanFilter == nil { + return ret + } + evalTargetType := eval_target_d.EvalTargetType_Trace + switch string(task.SpanFilter.PlatformType) { + case common.PlatformTypeInnerCozeBot, common.PlatformTypeCozeBot: + evalTargetType = eval_target_d.EvalTargetType_CozeBotOnline + case common.PlatformTypeInnerPrompt, common.PlatformTypePrompt: + evalTargetType = eval_target_d.EvalTargetType_CozeLoopPromptOnline + + case common.PlatformTypeInnerCozeloop, common.PlatformTypeCozeloop: + evalTargetType = eval_target_d.EvalTargetType_CustomRPCServerOnline + + case common.PlatformTypeWorkflow: + evalTargetType = eval_target_d.EvalTargetType_CozeWorkflowOnline + case common.PlatformTypeVeAgentkit: + evalTargetType = eval_target_d.EvalTargetType_VolcengineAgentAgentkitOnline + case common.PlatformTypeVeADK: + evalTargetType = eval_target_d.EvalTargetType_VolcengineAgentOnline + } + logs.CtxInfo(ctx, "Evaluation target type: %s", evalTargetType) + ret.EvalTargetType = lo.ToPtr(evalTargetType) + return ret } diff --git a/backend/modules/observability/domain/task/service/taskexe/processor/eval_target_builder_test.go b/backend/modules/observability/domain/task/service/taskexe/processor/eval_target_builder_test.go new file mode 100644 index 000000000..46173c69d --- /dev/null +++ b/backend/modules/observability/domain/task/service/taskexe/processor/eval_target_builder_test.go @@ -0,0 +1,223 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package processor + +import ( + "context" + "testing" + + "github.com/samber/lo" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" + + eval_target_d "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/eval_target" + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/common" + task_entity "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/entity" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/loop_span" +) + +func TestEvalTargetBuilderImpl_Build(t *testing.T) { + builder := &EvalTargetBuilderImpl{} + ctx := context.Background() + + t.Run("nil SpanFilter returns Trace type", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: nil, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_Trace, *result.EvalTargetType) + assert.Nil(t, result.SourceTargetID) + }) + + t.Run("nil SpanFilter with EvaluationExperimentConfig returns sourceTargetID", func(t *testing.T) { + srcID := "src-123" + task := &task_entity.ObservabilityTask{ + SpanFilter: nil, + TaskConfig: &task_entity.TaskConfig{ + EvaluationExperimentConfig: &task_entity.EvaluationExperimentConfig{ + SourceTargetID: &srcID, + }, + }, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_Trace, *result.EvalTargetType) + require.NotNil(t, result.SourceTargetID) + assert.Equal(t, "src-123", *result.SourceTargetID) + }) + + t.Run("PlatformTypeCozeBot returns CozeBotOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeCozeBot), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_CozeBotOnline, *result.EvalTargetType) + }) + + t.Run("PlatformTypeInnerCozeBot returns CozeBotOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeInnerCozeBot), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_CozeBotOnline, *result.EvalTargetType) + }) + + t.Run("PlatformTypePrompt returns CozeLoopPromptOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypePrompt), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_CozeLoopPromptOnline, *result.EvalTargetType) + }) + + t.Run("PlatformTypeInnerPrompt returns CozeLoopPromptOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeInnerPrompt), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_CozeLoopPromptOnline, *result.EvalTargetType) + }) + + t.Run("PlatformTypeCozeloop returns CustomRPCServerOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeCozeloop), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_CustomRPCServerOnline, *result.EvalTargetType) + }) + + t.Run("PlatformTypeInnerCozeloop returns CustomRPCServerOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeInnerCozeloop), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_CustomRPCServerOnline, *result.EvalTargetType) + }) + + t.Run("PlatformTypeWorkflow returns CozeWorkflowOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeWorkflow), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_CozeWorkflowOnline, *result.EvalTargetType) + }) + + t.Run("PlatformTypeVeAgentkit returns VolcengineAgentAgentkitOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeVeAgentkit), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_VolcengineAgentAgentkitOnline, *result.EvalTargetType) + }) + + t.Run("PlatformTypeVeADK returns VolcengineAgentOnline", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeVeADK), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_VolcengineAgentOnline, *result.EvalTargetType) + }) + + t.Run("unknown PlatformType returns Trace type", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType("unknown_platform"), + }, + TaskConfig: &task_entity.TaskConfig{}, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_Trace, *result.EvalTargetType) + }) + + t.Run("with sourceTargetID and valid PlatformType", func(t *testing.T) { + srcID := "target-456" + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeCozeBot), + }, + TaskConfig: &task_entity.TaskConfig{ + EvaluationExperimentConfig: &task_entity.EvaluationExperimentConfig{ + SourceTargetID: &srcID, + }, + }, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Equal(t, eval_target_d.EvalTargetType_CozeBotOnline, *result.EvalTargetType) + require.NotNil(t, result.SourceTargetID) + assert.Equal(t, "target-456", *result.SourceTargetID) + }) + + t.Run("nil EvaluationExperimentConfig sourceTargetID is nil", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypeWorkflow), + }, + TaskConfig: &task_entity.TaskConfig{ + EvaluationExperimentConfig: nil, + }, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Nil(t, result.SourceTargetID) + assert.Equal(t, eval_target_d.EvalTargetType_CozeWorkflowOnline, *result.EvalTargetType) + }) + + t.Run("EvaluationExperimentConfig with nil SourceTargetID", func(t *testing.T) { + task := &task_entity.ObservabilityTask{ + SpanFilter: &task_entity.SpanFilterFields{ + PlatformType: loop_span.PlatformType(common.PlatformTypePrompt), + }, + TaskConfig: &task_entity.TaskConfig{ + EvaluationExperimentConfig: &task_entity.EvaluationExperimentConfig{ + SourceTargetID: nil, + ExptTemplateID: lo.ToPtr(int64(999)), + }, + }, + } + result := builder.Build(ctx, task) + require.NotNil(t, result) + assert.Nil(t, result.SourceTargetID) + assert.Equal(t, eval_target_d.EvalTargetType_CozeLoopPromptOnline, *result.EvalTargetType) + }) +} diff --git a/backend/modules/observability/domain/task/service/taskexe/tracehub/backfill.go b/backend/modules/observability/domain/task/service/taskexe/tracehub/backfill.go index 79f477bb4..56c36653e 100644 --- a/backend/modules/observability/domain/task/service/taskexe/tracehub/backfill.go +++ b/backend/modules/observability/domain/task/service/taskexe/tracehub/backfill.go @@ -113,12 +113,14 @@ func (h *TraceHubServiceImpl) buildSubscriber(ctx context.Context, event *entity proc := h.taskProcessor.GetTaskProcessor(taskDO.TaskType) sub := &spanSubscriber{ - taskID: taskDO.ID, - t: taskDO, - tr: taskRun, - processor: proc, - taskRepo: h.taskRepo, - runType: entity.TaskRunTypeBackFill, + taskID: taskDO.ID, + t: taskDO, + tr: taskRun, + processor: proc, + taskRepo: h.taskRepo, + runType: entity.TaskRunTypeBackFill, + buildHelper: h.buildHelper, + traceService: h.traceService, } return sub, nil diff --git a/backend/modules/observability/domain/task/service/taskexe/tracehub/subscriber.go b/backend/modules/observability/domain/task/service/taskexe/tracehub/subscriber.go index a311e4760..c399a471b 100644 --- a/backend/modules/observability/domain/task/service/taskexe/tracehub/subscriber.go +++ b/backend/modules/observability/domain/task/service/taskexe/tracehub/subscriber.go @@ -8,6 +8,7 @@ import ( "math/rand" "time" + datadataset "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/entity" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/repo" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/service/taskexe" @@ -32,6 +33,20 @@ type spanSubscriber struct { traceService service.ITraceService } +func (s *spanSubscriber) hasTrajectory() bool { + if s.t == nil || s.t.TaskConfig == nil { + return false + } + for _, config := range s.t.TaskConfig.DataReflowConfig { + for _, mapping := range config.FieldMappings { + if mapping.FieldSchema.SchemaKey != nil && *mapping.FieldSchema.SchemaKey == datadataset.SchemaKey_Trajectory { + return true + } + } + } + return false +} + // Sampled determines whether a span is sampled based on the sampling rate; the sample size will be validated during flush. func (s *spanSubscriber) Sampled() bool { if s.t == nil || s.t.Sampler == nil { @@ -210,6 +225,7 @@ func (s *spanSubscriber) AddSpan(ctx context.Context, span *loop_span.Span) erro } trigger := &taskexe.Trigger{Task: s.t, Span: span, TaskRun: taskRunConfig} logs.CtxDebug(ctx, "invoke processor, trigger: %v", trigger) + // For response API // New Data 在这里处理 // Back fill 在前置批量处理 if s.runType == entity.TaskRunTypeNewData { @@ -219,6 +235,22 @@ func (s *spanSubscriber) AddSpan(ctx context.Context, span *loop_span.Span) erro return err } } + + // 轨迹计算 + if s.runType == entity.TaskRunTypeBackFill && s.hasTrajectory() { + // 可能有延迟 + startAt := time.Now().Add(-90 * 24 * time.Hour).UnixMilli() + endAt := time.Now().UnixMilli() + trajectoryMap, err := s.traceService.GetTrajectories(ctx, s.t.WorkspaceID, []string{span.TraceID}, startAt, endAt, s.t.GetPlatformType()) + if err != nil { + logs.CtxError(ctx, "get trajectories failed, task_id=%d, trace_id=%s err: %v", s.t.ID, span.TraceID, err) + return err + } + if trajectory, ok := trajectoryMap[span.TraceID]; ok { + trigger.Trajectory = trajectory + } + } + err = s.processor.Invoke(ctx, trigger) if err != nil { logs.CtxWarn(ctx, "invoke processor failed, trace_id=%s, span_id=%s, err: %v", span.TraceID, span.SpanID, err) diff --git a/backend/modules/observability/domain/task/service/taskexe/tracehub/subscriber_trajectory_test.go b/backend/modules/observability/domain/task/service/taskexe/tracehub/subscriber_trajectory_test.go new file mode 100644 index 000000000..ecfbd7d27 --- /dev/null +++ b/backend/modules/observability/domain/task/service/taskexe/tracehub/subscriber_trajectory_test.go @@ -0,0 +1,208 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package tracehub + +import ( + "context" + "testing" + "time" + + "github.com/stretchr/testify/assert" + "go.uber.org/mock/gomock" + + datadataset "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/dataset" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/entity" + repo_mocks "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/repo/mocks" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/service/taskexe" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/loop_span" + service_mocks "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service/mocks" + "github.com/coze-dev/coze-loop/backend/pkg/lang/ptr" +) + +type capturingProcessor struct { + trigger *taskexe.Trigger +} + +func (c *capturingProcessor) ValidateConfig(ctx context.Context, config any) error { return nil } +func (c *capturingProcessor) Invoke(ctx context.Context, trigger *taskexe.Trigger) error { + c.trigger = trigger + return nil +} + +func (c *capturingProcessor) OnTaskRunCreated(ctx context.Context, param taskexe.OnTaskRunCreatedReq) error { + return nil +} + +func (c *capturingProcessor) OnTaskRunFinished(ctx context.Context, param taskexe.OnTaskRunFinishedReq) error { + return nil +} + +func (c *capturingProcessor) OnTaskFinished(ctx context.Context, param taskexe.OnTaskFinishedReq) error { + return nil +} + +func (c *capturingProcessor) OnTaskUpdated(ctx context.Context, currentTask *entity.ObservabilityTask, taskOp entity.TaskStatus) error { + return nil +} + +func (c *capturingProcessor) OnTaskCreated(ctx context.Context, currentTask *entity.ObservabilityTask) error { + return nil +} + +func TestSpanSubscriber_AddSpan_WithTrajectory(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + mockRepo := repo_mocks.NewMockITaskRepo(ctrl) + mockTraceService := service_mocks.NewMockITraceService(ctrl) + processor := &capturingProcessor{} + + // Setup Task with Trajectory configuration + task := &entity.ObservabilityTask{ + ID: 1, + WorkspaceID: 7, + TaskStatus: entity.TaskStatusRunning, + SpanFilter: &entity.SpanFilterFields{ + PlatformType: loop_span.PlatformCozeLoop, + SpanListType: loop_span.SpanListTypeRootSpan, + }, + TaskConfig: &entity.TaskConfig{ + DataReflowConfig: []*entity.DataReflowConfig{ + { + FieldMappings: []dataset.FieldMapping{ + { + FieldSchema: &dataset.FieldSchema{ + SchemaKey: ptr.Of(datadataset.SchemaKey_Trajectory), + }, + }, + }, + }, + }, + }, + } + + sub := &spanSubscriber{ + taskID: task.ID, + t: task, + processor: processor, + taskRepo: mockRepo, + runType: entity.TaskRunTypeBackFill, + traceService: mockTraceService, + // buildHelper can be nil for this test as we don't test Match() logic deeply + } + + // Mock TaskRun + run := &entity.TaskRun{ + ID: 1001, + TaskID: task.ID, + WorkspaceID: task.WorkspaceID, + TaskType: entity.TaskRunTypeNewData, + RunStatus: entity.TaskRunStatusRunning, + RunStartAt: time.Now().Add(-time.Hour), + RunEndAt: time.Now().Add(time.Hour), + } + mockRepo.EXPECT().GetBackfillTaskRun(gomock.Any(), gomock.Nil(), task.ID).Return(run, nil) + + // Span + span := &loop_span.Span{ + TraceID: "trace123", + SpanID: "span123", + StartTime: time.Now().UnixMilli(), + } + + // Expect GetTrajectories + expectedTrajectory := &loop_span.Trajectory{ + ID: ptr.Of("trace123"), + } + trajectoryMap := map[string]*loop_span.Trajectory{ + "trace123": expectedTrajectory, + } + mockTraceService.EXPECT().GetTrajectories(gomock.Any(), task.WorkspaceID, []string{"trace123"}, gomock.Any(), gomock.Any(), gomock.Any()).Return(trajectoryMap, nil) + + // Execute + err := sub.AddSpan(context.Background(), span) + assert.NoError(t, err) + + // Verify Trigger + assert.NotNil(t, processor.trigger) + assert.NotNil(t, processor.trigger.Trajectory) + assert.Equal(t, expectedTrajectory, processor.trigger.Trajectory) +} + +func TestSpanSubscriber_AddSpan_WithoutTrajectory(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + mockRepo := repo_mocks.NewMockITaskRepo(ctrl) + mockTraceService := service_mocks.NewMockITraceService(ctrl) + processor := &capturingProcessor{} + + // Setup Task WITHOUT Trajectory configuration + task := &entity.ObservabilityTask{ + ID: 2, + WorkspaceID: 7, + TaskStatus: entity.TaskStatusRunning, + SpanFilter: &entity.SpanFilterFields{ + PlatformType: loop_span.PlatformCozeLoop, + SpanListType: loop_span.SpanListTypeRootSpan, + }, + TaskConfig: &entity.TaskConfig{ + DataReflowConfig: []*entity.DataReflowConfig{ + { + FieldMappings: []dataset.FieldMapping{ + { + FieldSchema: &dataset.FieldSchema{ + SchemaKey: ptr.Of(datadataset.SchemaKey_String), // Not Trajectory + }, + }, + }, + }, + }, + }, + } + + sub := &spanSubscriber{ + taskID: task.ID, + t: task, + processor: processor, + taskRepo: mockRepo, + runType: entity.TaskRunTypeNewData, + traceService: mockTraceService, + } + + // Mock TaskRun + run := &entity.TaskRun{ + ID: 1002, + TaskID: task.ID, + WorkspaceID: task.WorkspaceID, + TaskType: entity.TaskRunTypeNewData, + RunStatus: entity.TaskRunStatusRunning, + RunStartAt: time.Now().Add(-time.Hour), + RunEndAt: time.Now().Add(time.Hour), + } + mockRepo.EXPECT().GetLatestNewDataTaskRun(gomock.Any(), gomock.Nil(), task.ID).Return(run, nil) + + // Span + span := &loop_span.Span{ + TraceID: "trace456", + SpanID: "span456", + StartTime: time.Now().UnixMilli(), + } + + // Expect MergeHistoryMessagesByRespIDBatch + mockTraceService.EXPECT().MergeHistoryMessagesByRespIDBatch(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil) + + // DO NOT Expect GetTrajectories (should not be called) + + // Execute + err := sub.AddSpan(context.Background(), span) + assert.NoError(t, err) + + // Verify Trigger + assert.NotNil(t, processor.trigger) + assert.Nil(t, processor.trigger.Trajectory) +} diff --git a/backend/modules/observability/domain/trace/entity/dataset.go b/backend/modules/observability/domain/trace/entity/dataset.go index 63d51fbab..1d26f9da8 100644 --- a/backend/modules/observability/domain/trace/entity/dataset.go +++ b/backend/modules/observability/domain/trace/entity/dataset.go @@ -7,6 +7,8 @@ import ( "context" "strconv" + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" + "github.com/bytedance/gg/gptr" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/common" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/loop_span" @@ -80,6 +82,9 @@ type Dataset struct { EvaluationBizCategory *EvaluationBizCategory Seesion *common.Session UserID *string + // 数据集属性 + Visibility dataset.DatasetVisibility + WorkflowID int64 } type DatasetVersion struct { @@ -119,12 +124,18 @@ type FieldSchema struct { DisplayFormat FieldDisplayFormat } -func NewDataset(id, spaceID int64, name string, category DatasetCategory, schema DatasetSchema, session *common.Session, evaluationBizCategory *EvaluationBizCategory) *Dataset { +func NewDataset(id, spaceID int64, name string, category DatasetCategory, + schema DatasetSchema, + session *common.Session, + evaluationBizCategory *EvaluationBizCategory, + isNewWorkflowTask bool, + workflowID int64, +) *Dataset { var userID *string if session != nil { userID = ptr.Of(strconv.FormatInt(*session.UserID, 10)) } - dataset := &Dataset{ + ds := &Dataset{ ID: id, WorkspaceID: spaceID, Name: name, @@ -136,7 +147,14 @@ func NewDataset(id, spaceID int64, name string, category DatasetCategory, schema Seesion: session, UserID: userID, } - return dataset + if isNewWorkflowTask { + ds.Visibility = dataset.DatasetVisibility_System + } else { + ds.Visibility = dataset.DatasetVisibility_Space + } + ds.WorkflowID = workflowID + + return ds } func (d *Dataset) GetFieldSchemaKeyByName(fieldSchemaName string) string { @@ -467,6 +485,7 @@ func GetContentInfo(ctx context.Context, contentType ContentType, value string) Text: value, } } + logs.CtxInfo(ctx, "Get content info success, content:%v", content) return content, 0 } diff --git a/backend/modules/observability/domain/trace/entity/dataset_test.go b/backend/modules/observability/domain/trace/entity/dataset_test.go index 4194c9791..0decc4f62 100644 --- a/backend/modules/observability/domain/trace/entity/dataset_test.go +++ b/backend/modules/observability/domain/trace/entity/dataset_test.go @@ -8,6 +8,7 @@ import ( "testing" "github.com/bytedance/gg/gptr" + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/common" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/loop_span" "github.com/stretchr/testify/assert" @@ -67,6 +68,7 @@ func TestNewDataset(t *testing.T) { }, }, DatasetCategory: DatasetCategory_General, + Visibility: dataset.DatasetVisibility_Space, }, }, { @@ -90,12 +92,13 @@ func TestNewDataset(t *testing.T) { }, }, DatasetCategory: DatasetCategory_Evaluation, + Visibility: dataset.DatasetVisibility_Space, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { - got := NewDataset(tt.args.id, tt.args.spaceID, tt.args.name, tt.args.category, tt.args.schema, nil, nil) + got := NewDataset(tt.args.id, tt.args.spaceID, tt.args.name, tt.args.category, tt.args.schema, nil, nil, false, 0) assert.Equal(t, tt.want, got) }) } diff --git a/backend/modules/observability/domain/trace/entity/loop_span/annotation.go b/backend/modules/observability/domain/trace/entity/loop_span/annotation.go index ebbad34c4..0cc238744 100644 --- a/backend/modules/observability/domain/trace/entity/loop_span/annotation.go +++ b/backend/modules/observability/domain/trace/entity/loop_span/annotation.go @@ -50,6 +50,7 @@ const ( AnnotationOpenAPIFeedbackFieldPrefix = "feedback_openapi_" AnnotationManualFeedbackFieldPrefix = "manual_feedback_" + AnnotationAutoEvaluateFieldPrefix = "auto_evaluate_" ) type AnnotationValue struct { @@ -72,6 +73,8 @@ type AutoEvaluateMetadata struct { TaskID int64 `json:"task_id"` EvaluatorRecordID int64 `json:"evaluator_record_id"` EvaluatorVersionID int64 `json:"evaluator_version_id"` + ExptID int64 `json:"expt_id"` + ExptTemplateID int64 `json:"expt_template_id"` } type AnnotationManualFeedback struct { diff --git a/backend/modules/observability/domain/trace/entity/loop_span/span.go b/backend/modules/observability/domain/trace/entity/loop_span/span.go index 3642e6f5f..89103b0aa 100644 --- a/backend/modules/observability/domain/trace/entity/loop_span/span.go +++ b/backend/modules/observability/domain/trace/entity/loop_span/span.go @@ -105,6 +105,51 @@ var TimeTagSlice = []string{ SpanFieldLogicDeleteDate, } +var SpanStructFieldKeys = []string{ + SpanFieldStartTime, + SpanFieldSpanId, + SpanFieldParentID, + SpanFieldTraceId, + SpanFieldDuration, + SpanFieldCallType, + SpanFieldPSM, + SpanFieldLogID, + SpanFieldSpaceId, + SpanFieldSpanName, + SpanFieldSpanType, + SpanFieldMethod, + SpanFieldStatusCode, + SpanFieldInput, + SpanFieldOutput, + SpanFieldObjectStorage, +} + +const ( + MetadataValueTypeString = "string" + MetadataValueTypeLong = "long" + MetadataValueTypeDouble = "double" + MetadataValueTypeBool = "bool" +) + +var SpanStructFieldValueTypes = map[string]string{ + SpanFieldStartTime: MetadataValueTypeLong, + SpanFieldSpanId: MetadataValueTypeString, + SpanFieldParentID: MetadataValueTypeString, + SpanFieldTraceId: MetadataValueTypeString, + SpanFieldDuration: MetadataValueTypeLong, + SpanFieldCallType: MetadataValueTypeString, + SpanFieldPSM: MetadataValueTypeString, + SpanFieldLogID: MetadataValueTypeString, + SpanFieldSpaceId: MetadataValueTypeString, + SpanFieldSpanName: MetadataValueTypeString, + SpanFieldSpanType: MetadataValueTypeString, + SpanFieldMethod: MetadataValueTypeString, + SpanFieldStatusCode: MetadataValueTypeLong, + SpanFieldInput: MetadataValueTypeString, + SpanFieldOutput: MetadataValueTypeString, + SpanFieldObjectStorage: MetadataValueTypeString, +} + type SpanList []*Span type Span struct { @@ -409,6 +454,10 @@ func (s *Span) GetFieldValue(fieldName string, isSystem, isCustom bool) any { } else if val, ok := s.TagsByte[fieldName]; ok { return val } + return s.getAnnotationValue(fieldName) +} + +func (s *Span) getAnnotationValue(fieldName string) any { annotationMap := make(map[string]AnnotationValue) for _, annotation := range s.Annotations { var prefix string @@ -417,6 +466,8 @@ func (s *Span) GetFieldValue(fieldName string, isSystem, isCustom bool) any { prefix = AnnotationOpenAPIFeedbackFieldPrefix case AnnotationTypeManualFeedback: prefix = AnnotationManualFeedbackFieldPrefix + case AnnotationTypeAutoEvaluate: + prefix = AnnotationAutoEvaluateFieldPrefix default: continue } @@ -564,7 +615,7 @@ func (s *Span) AddManualDatasetAnnotation(datasetID int64, userID string, annota return a, nil } -func (s *Span) AddAutoEvalAnnotation(taskID, evaluatorRecordID, evaluatorVersionID int64, score float64, reasoning, userID string) (*Annotation, error) { +func (s *Span) AddAutoEvalAnnotation(taskID, evaluatorRecordID, evaluatorVersionID int64, score float64, reasoning, userID string, exptID int64, exptTemplateID int64) (*Annotation, error) { a := &Annotation{} a.SpanID = s.SpanID a.TraceID = s.TraceID @@ -578,6 +629,8 @@ func (s *Span) AddAutoEvalAnnotation(taskID, evaluatorRecordID, evaluatorVersion TaskID: taskID, EvaluatorRecordID: evaluatorRecordID, EvaluatorVersionID: evaluatorVersionID, + ExptID: exptID, + ExptTemplateID: exptTemplateID, } a.Status = AnnotationStatusNormal a.CreatedAt = time.Now() @@ -623,6 +676,35 @@ func (s *Span) extractByJsonpath(ctx context.Context, key string, jsonpath strin } } data = conv.ToString(tag) + } else if strings.HasPrefix(key, "Metadata.") { + key = strings.TrimPrefix(key, "Metadata.") + metadata := s.GetFieldValue(key, true, false) + checkKey := key + if checkKey == SpanFieldStartTime || checkKey == SpanFieldDuration || checkKey == SpanFieldLogicDeleteDate || + slices.Contains(TimeTagSlice, checkKey) { + if integer, ok := metadata.(int64); ok { + metadata = time_util.MicroSec2MillSec(integer) + } + } + data = conv.ToString(metadata) + } else if strings.HasPrefix(key, "Feedback.") { + key = strings.TrimPrefix(key, "Feedback.") + feedback := s.getAnnotationValue(key) + checkKey := key + if strings.HasPrefix(checkKey, AnnotationManualFeedbackFieldPrefix) { + checkKey = strings.TrimPrefix(checkKey, AnnotationManualFeedbackFieldPrefix) + } else if strings.HasPrefix(checkKey, AnnotationOpenAPIFeedbackFieldPrefix) { + checkKey = strings.TrimPrefix(checkKey, AnnotationOpenAPIFeedbackFieldPrefix) + } else if strings.HasPrefix(checkKey, AnnotationAutoEvaluateFieldPrefix) { + checkKey = strings.TrimPrefix(checkKey, AnnotationAutoEvaluateFieldPrefix) + } + if checkKey == SpanFieldStartTime || checkKey == SpanFieldDuration || checkKey == SpanFieldLogicDeleteDate || + slices.Contains(TimeTagSlice, checkKey) { + if integer, ok := feedback.(int64); ok { + feedback = time_util.MicroSec2MillSec(integer) + } + } + data = conv.ToString(feedback) } else { return "", errors.Errorf("unsupported mapping key: %s", key) } diff --git a/backend/modules/observability/domain/trace/entity/loop_span/span_test.go b/backend/modules/observability/domain/trace/entity/loop_span/span_test.go index 28a570254..9ee650ad2 100644 --- a/backend/modules/observability/domain/trace/entity/loop_span/span_test.go +++ b/backend/modules/observability/domain/trace/entity/loop_span/span_test.go @@ -272,6 +272,56 @@ func TestSpan_ExtractByJsonpath(t *testing.T) { assert.NoError(t, err) assert.Equal(t, result, "42") + // 测试从Metadata字段提取数据 + // Metadata 应该取自 TagsString, TagsLong 等字段 + metadataSpan := &Span{ + SystemTagsString: map[string]string{ + "metadata_key": "metadata_value", + }, + Annotations: []*Annotation{ + { + AnnotationType: AnnotationTypeManualFeedback, + Key: "feedback_key", + Value: NewBoolValue(true), + }, + }, + } + // Metadata 直接查 Tags + result, err = metadataSpan.ExtractByJsonpath(ctx, "Metadata.metadata_key", "") + assert.NoError(t, err) + assert.Equal(t, "metadata_value", result) + + // 测试从Feedback字段提取数据 + // Feedback前缀为 "manual_feedback_" + result, err = metadataSpan.ExtractByJsonpath(ctx, "Feedback.manual_feedback_feedback_key", "") + assert.NoError(t, err) + // AnnotationValueTypeBool + assert.Equal(t, "true", result) + + // 测试Metadata时间字段格式化 (从 TagsLong 提取) + timeSpan := &Span{ + SystemTagsLong: map[string]int64{ + SpanFieldLatencyFirstResp: 1000000, // 1s in micros + }, + } + result, err = timeSpan.ExtractByJsonpath(ctx, "Metadata."+SpanFieldLatencyFirstResp, "") + assert.NoError(t, err) + assert.Equal(t, "1000", result) // should be converted to millis + + // 测试Feedback时间字段格式化 (从 Annotations 提取) + timeFeedbackSpan := &Span{ + Annotations: []*Annotation{ + { + AnnotationType: AnnotationTypeManualFeedback, + Key: SpanFieldDuration, + Value: NewLongValue(2000000), // 2s in micros + }, + }, + } + result, err = timeFeedbackSpan.ExtractByJsonpath(ctx, "Feedback.manual_feedback_"+SpanFieldDuration, "") + assert.NoError(t, err) + assert.Equal(t, "2000", result) // should be converted to millis + // 测试空jsonpath的处理 result, err = span.ExtractByJsonpath(ctx, "Input", "") assert.NoError(t, err) @@ -659,7 +709,7 @@ func TestSpan_MergeHistoryContext(t *testing.T) { WorkspaceID: "1", StartTime: time.Now().UnixMicro(), } - anno, err := span.AddAutoEvalAnnotation(1, 2, 3, 0.5, "reason", "user1") + anno, err := span.AddAutoEvalAnnotation(1, 2, 3, 0.5, "reason", "user1", 0, 0) assert.NoError(t, err) assert.NotNil(t, anno) assert.Equal(t, AnnotationTypeAutoEvaluate, anno.AnnotationType) diff --git a/backend/modules/observability/domain/trace/repo/mocks/trace.go b/backend/modules/observability/domain/trace/repo/mocks/trace.go index 0b13d3e70..9c790d721 100644 --- a/backend/modules/observability/domain/trace/repo/mocks/trace.go +++ b/backend/modules/observability/domain/trace/repo/mocks/trace.go @@ -147,6 +147,21 @@ func (mr *MockITraceRepoMockRecorder) ListAnnotations(arg0, arg1 any) *gomock.Ca return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAnnotations", reflect.TypeOf((*MockITraceRepo)(nil).ListAnnotations), arg0, arg1) } +// ListWorkspaceAnnotations mocks base method. +func (m *MockITraceRepo) ListWorkspaceAnnotations(arg0 context.Context, arg1 *repo.ListWorkspaceAnnotationsParam) (loop_span.AnnotationList, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ListWorkspaceAnnotations", arg0, arg1) + ret0, _ := ret[0].(loop_span.AnnotationList) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListWorkspaceAnnotations indicates an expected call of ListWorkspaceAnnotations. +func (mr *MockITraceRepoMockRecorder) ListWorkspaceAnnotations(arg0, arg1 any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListWorkspaceAnnotations", reflect.TypeOf((*MockITraceRepo)(nil).ListWorkspaceAnnotations), arg0, arg1) +} + // ListSpans mocks base method. func (m *MockITraceRepo) ListSpans(arg0 context.Context, arg1 *repo.ListSpansParam) (*repo.ListSpansResult, error) { m.ctrl.T.Helper() diff --git a/backend/modules/observability/domain/trace/repo/trace.go b/backend/modules/observability/domain/trace/repo/trace.go index a7c65b34b..a935b57b5 100644 --- a/backend/modules/observability/domain/trace/repo/trace.go +++ b/backend/modules/observability/domain/trace/repo/trace.go @@ -82,6 +82,16 @@ type ListAnnotationsParam struct { EndAt int64 // ms } +type ListWorkspaceAnnotationsParam struct { + WorkSpaceID string + Tenants []string + AnnotationType string + DescByUpdatedAt bool + StartAt int64 // ms + EndAt int64 // ms + Limit int64 +} + type InsertAnnotationParam struct { WorkSpaceID string Tenant string @@ -115,6 +125,7 @@ type ITraceRepo interface { GetPreSpanIDs(context.Context, *GetPreSpanIDsParam) (preSpanIDs, responseIDs []string, err error) GetTrace(context.Context, *GetTraceParam) (*GetTraceResult, error) ListAnnotations(context.Context, *ListAnnotationsParam) (loop_span.AnnotationList, error) + ListWorkspaceAnnotations(context.Context, *ListWorkspaceAnnotationsParam) (loop_span.AnnotationList, error) GetAnnotation(context.Context, *GetAnnotationParam) (*loop_span.Annotation, error) InsertAnnotations(context.Context, *InsertAnnotationParam) error UpsertTrajectoryConfig(context.Context, *UpsertTrajectoryConfigParam) error diff --git a/backend/modules/observability/domain/trace/service/mocks/trace_service.go b/backend/modules/observability/domain/trace/service/mocks/trace_service.go index b2af2cee9..58635ed97 100644 --- a/backend/modules/observability/domain/trace/service/mocks/trace_service.go +++ b/backend/modules/observability/domain/trace/service/mocks/trace_service.go @@ -233,6 +233,21 @@ func (mr *MockITraceServiceMockRecorder) ListAnnotationEvaluators(ctx, req any) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAnnotationEvaluators", reflect.TypeOf((*MockITraceService)(nil).ListAnnotationEvaluators), ctx, req) } +// ListMetadata mocks base method. +func (m *MockITraceService) ListMetadata(ctx context.Context, req *service.ListMetadataReq) (*service.ListMetadataResp, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ListMetadata", ctx, req) + ret0, _ := ret[0].(*service.ListMetadataResp) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListMetadata indicates an expected call of ListMetadata. +func (mr *MockITraceServiceMockRecorder) ListMetadata(ctx, req any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListMetadata", reflect.TypeOf((*MockITraceService)(nil).ListMetadata), ctx, req) +} + // ListAnnotations mocks base method. func (m *MockITraceService) ListAnnotations(ctx context.Context, req *service.ListAnnotationsReq) (*service.ListAnnotationsResp, error) { m.ctrl.T.Helper() @@ -248,6 +263,21 @@ func (mr *MockITraceServiceMockRecorder) ListAnnotations(ctx, req any) *gomock.C return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAnnotations", reflect.TypeOf((*MockITraceService)(nil).ListAnnotations), ctx, req) } +// ListWorkspaceAnnotations mocks base method. +func (m *MockITraceService) ListWorkspaceAnnotations(ctx context.Context, req *service.ListWorkspaceAnnotationsReq) (*service.ListWorkspaceAnnotationsResp, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ListWorkspaceAnnotations", ctx, req) + ret0, _ := ret[0].(*service.ListWorkspaceAnnotationsResp) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListWorkspaceAnnotations indicates an expected call of ListWorkspaceAnnotations. +func (mr *MockITraceServiceMockRecorder) ListWorkspaceAnnotations(ctx, req any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListWorkspaceAnnotations", reflect.TypeOf((*MockITraceService)(nil).ListWorkspaceAnnotations), ctx, req) +} + // ListPreSpan mocks base method. func (m *MockITraceService) ListPreSpan(ctx context.Context, req *service.ListPreSpanReq) (*service.ListPreSpanResp, error) { m.ctrl.T.Helper() diff --git a/backend/modules/observability/domain/trace/service/trace_export_service.go b/backend/modules/observability/domain/trace/service/trace_export_service.go index 74bb029cb..829be5a48 100644 --- a/backend/modules/observability/domain/trace/service/trace_export_service.go +++ b/backend/modules/observability/domain/trace/service/trace_export_service.go @@ -8,6 +8,13 @@ import ( "strconv" "time" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service/trace/span_filter" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service/trace/span_processor" + + "github.com/coze-dev/coze-loop/backend/modules/observability/application/convertor" + + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/observability/domain/filter" + "github.com/bytedance/gg/gptr" "github.com/coze-dev/coze-loop/backend/infra/middleware/session" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/config" @@ -19,7 +26,6 @@ import ( "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/loop_span" "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/repo" - "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service/trace/span_processor" "github.com/coze-dev/coze-loop/backend/modules/observability/pkg/errno" "github.com/coze-dev/coze-loop/backend/pkg/errorx" "github.com/coze-dev/coze-loop/backend/pkg/lang/goroutine" @@ -46,6 +52,8 @@ type ExportTracesToDatasetRequest struct { // 导入方式，不填默认为追加 ExportType ExportType FieldMappings []entity.FieldMapping + SpanFilters *filter.SpanFilterFields + Limit *int64 } type ExportTracesToDatasetResponse struct { @@ -130,7 +138,7 @@ func (r *TraceExportServiceImpl) ExportTracesToDataset(ctx context.Context, req ) { resp := &ExportTracesToDatasetResponse{} - spans, err := r.getSpans(ctx, req.WorkspaceID, req.SpanIds, req.StartTime, req.EndTime, req.PlatformType) + spans, err := r.getSpans(ctx, req.WorkspaceID, req.SpanIds, nil, req.StartTime, req.EndTime, req.PlatformType, nil) if err != nil { return resp, err } @@ -196,12 +204,31 @@ func (r *TraceExportServiceImpl) PreviewExportTracesToDataset(ctx context.Contex *PreviewExportTracesToDatasetResponse, error, ) { resp := &PreviewExportTracesToDatasetResponse{} - spans, err := r.getSpans(ctx, req.WorkspaceID, req.SpanIds, req.StartTime, req.EndTime, req.PlatformType) + var spans loop_span.SpanList + var err error + + spans, err = r.getSpans(ctx, req.WorkspaceID, req.SpanIds, req.SpanFilters, req.StartTime, req.EndTime, req.PlatformType, req.Limit) if err != nil { return resp, err } logs.CtxInfo(ctx, "Get spans success, total count:%v", len(spans)) + var trajectoryMap map[string]*loop_span.Trajectory + if r.hasTrajectory(req.FieldMappings) && len(req.SpanIds) < 20 { + traceIDs := lo.UniqMap(spans, func(item *loop_span.Span, index int) string { + return item.TraceID + }) + + // 前端传入的是当前span时间，不能直接使用。改为和ListTrajectory逻辑一致。 + finalStartTime := r.traceConfig.GetTraceDataMaxDurationDay(ctx, lo.ToPtr(string(req.PlatformType))) + trajectoryMap, err = r.traceService.GetTrajectories(ctx, req.WorkspaceID, traceIDs, finalStartTime, + time.Now().UnixMilli(), req.PlatformType) + if err != nil { + return resp, err + } + logs.CtxInfo(ctx, "Get trajectories success, total count:%v", len(trajectoryMap)) + } + dataset, err := r.buildPreviewDataset(ctx, req.WorkspaceID, req.Category, req.Config) if err != nil { return resp, err @@ -212,7 +239,7 @@ func (r *TraceExportServiceImpl) PreviewExportTracesToDataset(ctx context.Contex return resp, err } - successItems, failedItems, allItems := r.buildDatasetItems(ctx, spans, req.FieldMappings, req.WorkspaceID, dataset, nil) + successItems, failedItems, allItems := r.buildDatasetItems(ctx, spans, req.FieldMappings, req.WorkspaceID, dataset, trajectoryMap) var ignoreCurrentCount *bool if !req.Config.IsNewDataset && req.ExportType == ExportType_Overwrite { @@ -253,7 +280,7 @@ func (r *TraceExportServiceImpl) createOrUpdateDataset(ctx context.Context, work *config.DatasetName, category, config.DatasetSchema, - nil, nil, + nil, nil, false, 0, )) if err != nil { return nil, err @@ -277,7 +304,7 @@ func (r *TraceExportServiceImpl) createOrUpdateDataset(ctx context.Context, work "", category, config.DatasetSchema, - nil, nil, + nil, nil, false, 0, )); err != nil { return nil, err } @@ -288,17 +315,20 @@ func (r *TraceExportServiceImpl) createOrUpdateDataset(ctx context.Context, work return r.getDatasetProvider(category).GetDataset(ctx, workspaceID, datasetID, category) } -func (r *TraceExportServiceImpl) getSpans(ctx context.Context, workspaceID int64, sids []SpanID, startTime, endTime int64, platformType loop_span.PlatformType) (loop_span.SpanList, error) { - tenant, err := r.tenantProvider.GetTenantsByPlatformType(ctx, platformType) +func (r *TraceExportServiceImpl) getSpans(ctx context.Context, workspaceID int64, sids []SpanID, spanFilters *filter.SpanFilterFields, startTime, endTime int64, platformType loop_span.PlatformType, limit *int64) (loop_span.SpanList, error) { + tenants, err := r.tenantProvider.GetTenantsByPlatformType(ctx, platformType) if err != nil { return nil, err } - spanIDs := lo.Map(sids, func(s SpanID, _ int) string { return s.SpanID }) - traceIDs := lo.UniqMap(sids, func(s SpanID, _ int) string { return s.TraceID }) - result, err := r.traceRepo.ListSpans(ctx, &repo.ListSpansParam{ - WorkSpaceID: strconv.FormatInt(workspaceID, 10), - Tenants: tenant, - Filters: &loop_span.FilterFields{ + var filters *loop_span.FilterFields + var spanLimit int32 = 10 + if limit != nil { + spanLimit = int32(*limit) + } + if len(sids) > 0 { + spanIDs := lo.Map(sids, func(s SpanID, _ int) string { return s.SpanID }) + traceIDs := lo.UniqMap(sids, func(s SpanID, _ int) string { return s.TraceID }) + filters = &loop_span.FilterFields{ FilterFields: []*loop_span.FilterField{ { FieldName: "trace_id", @@ -313,12 +343,47 @@ func (r *TraceExportServiceImpl) getSpans(ctx context.Context, workspaceID int64 QueryType: ptr.Of(loop_span.QueryTypeEnumIn), }, }, - }, - StartAt: startTime, - EndAt: endTime, + } + spanLimit = int32(len(sids)) * 2 + } else { + // align with ListSpans logic + var userFilters *loop_span.FilterFields + var spanListType loop_span.SpanListType + if spanFilters != nil { + userFilters = convertor.FilterFieldsDTO2DO(spanFilters.Filters) + if err := userFilters.Traverse(processSpecificFilter); err != nil { + return nil, errorx.WrapByCode(err, errno.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("invalid filter")) + } + if spanFilters.SpanListType != nil { + spanListType = loop_span.SpanListType(spanFilters.GetSpanListType()) + } + } + + platformFilter, err := r.buildHelper.BuildPlatformRelatedFilter(ctx, platformType) + if err != nil { + return nil, err + } + env := &span_filter.SpanEnv{ + WorkspaceID: workspaceID, + } + builtinFilter, err := BuildBuiltinFilters(ctx, platformFilter, env, spanListType) + if err != nil { + return nil, err + } else if builtinFilter == nil { + return loop_span.SpanList{}, nil + } + filters = CombineFilters(builtinFilter, userFilters) + } + + result, err := r.traceRepo.ListSpans(ctx, &repo.ListSpansParam{ + WorkSpaceID: strconv.FormatInt(workspaceID, 10), + Tenants: tenants, + Filters: filters, + StartAt: startTime, + EndAt: endTime, // May have duplicate Spans // wider limit to avoid emit - Limit: int32(len(sids)) * 2, + Limit: spanLimit, }) if err != nil { return nil, err @@ -342,16 +407,20 @@ func (r *TraceExportServiceImpl) getSpans(ctx context.Context, workspaceID int64 } // sort by sids - spanMap := lo.SliceToMap(spans, func(s *loop_span.Span) (string, *loop_span.Span) { - return s.SpanID, s - }) - sortedSpans := make(loop_span.SpanList, 0, len(sids)) - for _, sid := range sids { - if span, ok := spanMap[sid.SpanID]; ok { - sortedSpans = append(sortedSpans, span) + if len(sids) > 0 { + spanMap := lo.SliceToMap(spans, func(s *loop_span.Span) (string, *loop_span.Span) { + return s.SpanID, s + }) + sortedSpans := make(loop_span.SpanList, 0, len(sids)) + for _, sid := range sids { + if span, ok := spanMap[sid.SpanID]; ok { + sortedSpans = append(sortedSpans, span) + } } + return sortedSpans, nil } - return sortedSpans, nil + + return spans, nil } func (r *TraceExportServiceImpl) clearDataset(ctx context.Context, datasetID int64, req *ExportTracesToDatasetRequest) error { @@ -508,8 +577,9 @@ func (r *TraceExportServiceImpl) buildItem(ctx context.Context, span *loop_span. logs.CtxInfo(ctx, "Extract field failed, err:%v", err) } } - + logs.CtxInfo(ctx, "Extract field value:%v", value) content, errCode := entity.GetContentInfo(ctx, mapping.FieldSchema.ContentType, value) + if errCode == entity.DatasetErrorType_MismatchSchema { item.AddError("invalid multi part", entity.DatasetErrorType_MismatchSchema, nil) continue @@ -543,7 +613,7 @@ func (r *TraceExportServiceImpl) buildPreviewDataset(ctx context.Context, worksp "", category, schema, - nil, nil, + nil, nil, false, 0, ) if config.DatasetID != nil { dataset.ID = *config.DatasetID diff --git a/backend/modules/observability/domain/trace/service/trace_export_service_test.go b/backend/modules/observability/domain/trace/service/trace_export_service_test.go index 918f34eb1..d928bde94 100644 --- a/backend/modules/observability/domain/trace/service/trace_export_service_test.go +++ b/backend/modules/observability/domain/trace/service/trace_export_service_test.go @@ -202,6 +202,91 @@ func TestTraceExportServiceImpl_ExportTracesToDataset(t *testing.T) { want: &ExportTracesToDatasetResponse{}, wantErr: true, }, + { + name: "export traces to dataset with multimodal video", + fieldsGetter: func(ctrl *gomock.Controller) fields { + repoMock := repomocks.NewMockITraceRepo(ctrl) + tenantMock := tenantmocks.NewMockITenantProvider(ctrl) + datasetProviderMock := rpcmocks.NewMockIDatasetProvider(ctrl) + confMock := confmocks.NewMockITraceConfig(ctrl) + traceProducerMock := mqmocks.NewMockITraceProducer(ctrl) + annotationProducerMock := mqmocks.NewMockIAnnotationProducer(ctrl) + metricsMock := metricmocks.NewMockITraceMetrics(ctrl) + filterFactoryMock := filtermocks.NewMockPlatformFilterFactory(ctrl) + buildHelper := NewTraceFilterProcessorBuilder(filterFactoryMock, map[entity.ProcessorScene][]span_processor.Factory{entity.SceneGetTrace: {}, entity.SceneListSpans: {}, entity.SceneAdvanceInfo: {}, entity.SceneIngestTrace: {}, entity.SceneSearchTraceOApi: {}, entity.SceneListSpansOApi: {}}) + adaptor := NewDatasetServiceAdaptor() + adaptor.Register(entity.DatasetCategory_General, datasetProviderMock) + + tenantMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), loop_span.PlatformCozeLoop).Return([]string{"tenant1"}, nil) + repoMock.EXPECT().ListSpans(gomock.Any(), gomock.Any()).Return(&repo.ListSpansResult{ + Spans: []*loop_span.Span{ + { + TraceID: "trace-video", + SpanID: "span-video", + WorkspaceID: "123", + Input: `[{"type": "video_url", "video_url": {"url": "http://example.com/video.mp4", "name": "test_video.mp4"}}]`, + }, + }, + }, nil) + datasetProviderMock.EXPECT().CreateDataset(gomock.Any(), gomock.Any()).Return(int64(101), nil) + datasetProviderMock.EXPECT().GetDataset(gomock.Any(), int64(123), int64(101), entity.DatasetCategory_General).Return(&entity.Dataset{ + ID: 101, + Name: "test-video-dataset", + DatasetCategory: entity.DatasetCategory_General, + DatasetVersion: entity.DatasetVersion{ + DatasetSchema: entity.DatasetSchema{ + FieldSchemas: []entity.FieldSchema{ + {Name: "input", Key: ptr.Of("input"), ContentType: entity.ContentType_MultiPart}, + }, + }, + }, + }, nil) + datasetProviderMock.EXPECT().AddDatasetItems(gomock.Any(), int64(101), entity.DatasetCategory_General, gomock.Any()).DoAndReturn(func(ctx context.Context, datasetID int64, category entity.DatasetCategory, items []*entity.DatasetItem) ([]*entity.DatasetItem, []entity.ItemErrorGroup, error) { + // 可以在这里加断言，验证 items 里的数据对不对 + for _, item := range items { + for k, field := range item.FieldData { + t.Logf("Field key: %v, Field Content: %+v", k, field.Content) + } + } + return items, []entity.ItemErrorGroup{}, nil + }) + + return fields{ + traceRepo: repoMock, + traceConfig: confMock, + traceProducer: traceProducerMock, + annotationProducer: annotationProducerMock, + metrics: metricsMock, + tenantProvider: tenantMock, + DatasetServiceAdaptor: adaptor, + buildHelper: buildHelper, + traceService: &stubTraceService{}, + } + }, + args: args{ + ctx: context.Background(), + req: &ExportTracesToDatasetRequest{ + WorkspaceID: 123, + SpanIds: []SpanID{{TraceID: "trace-video", SpanID: "span-video"}}, + Category: entity.DatasetCategory_General, + Config: DatasetConfig{IsNewDataset: true, DatasetName: ptr.Of("test-video-dataset"), DatasetSchema: entity.DatasetSchema{FieldSchemas: []entity.FieldSchema{{Name: "input", ContentType: entity.ContentType_MultiPart}}}}, + StartTime: time.Now().Unix() - 3600, + EndTime: time.Now().Unix(), + PlatformType: loop_span.PlatformCozeLoop, + ExportType: ExportType_Append, + FieldMappings: []entity.FieldMapping{ + {TraceFieldKey: "input", FieldSchema: entity.FieldSchema{Name: "input", ContentType: entity.ContentType_MultiPart}}, + }, + }, + }, + want: &ExportTracesToDatasetResponse{ + SuccessCount: 0, + DatasetID: 101, + DatasetName: "test-video-dataset", + Errors: []entity.ItemErrorGroup{{Type: 1, Summary: "invalid multi part", ErrorCount: 1}}, + }, + wantErr: false, + }, } for _, tt := range tests { @@ -1005,6 +1090,81 @@ func TestTraceExportServiceImpl_PreviewExportTracesToDataset(t *testing.T) { }, wantErr: false, }, + { + name: "preview with trajectory field mapping", + fieldsGetter: func(ctrl *gomock.Controller) fields { + repoMock := repomocks.NewMockITraceRepo(ctrl) + tenantMock := tenantmocks.NewMockITenantProvider(ctrl) + datasetProviderMock := rpcmocks.NewMockIDatasetProvider(ctrl) + confMock := confmocks.NewMockITraceConfig(ctrl) + traceProducerMock := mqmocks.NewMockITraceProducer(ctrl) + annotationProducerMock := mqmocks.NewMockIAnnotationProducer(ctrl) + metricsMock := metricmocks.NewMockITraceMetrics(ctrl) + filterFactoryMock := filtermocks.NewMockPlatformFilterFactory(ctrl) + buildHelper := NewTraceFilterProcessorBuilder(filterFactoryMock, map[entity.ProcessorScene][]span_processor.Factory{entity.SceneGetTrace: {}, entity.SceneListSpans: {}, entity.SceneAdvanceInfo: {}, entity.SceneIngestTrace: {}, entity.SceneSearchTraceOApi: {}, entity.SceneListSpansOApi: {}}) + + adaptor := NewDatasetServiceAdaptor() + adaptor.Register(entity.DatasetCategory_General, datasetProviderMock) + + testSpan := &loop_span.Span{ + TraceID: "trace-123", + SpanID: "span-456", + WorkspaceID: "123", + Input: `{"question": "test input"}`, + Output: `{"answer": "test output"}`, + } + + tenantMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), loop_span.PlatformCozeLoop).Return([]string{"tenant1"}, nil) + repoMock.EXPECT().ListSpans(gomock.Any(), gomock.Any()).Return(&repo.ListSpansResult{ + Spans: []*loop_span.Span{testSpan}, + }, nil) + confMock.EXPECT().GetTraceDataMaxDurationDay(gomock.Any(), gomock.Any()).Return(int64(7 * 24 * 3600 * 1000)) + datasetProviderMock.EXPECT().ValidateDatasetItems(gomock.Any(), gomock.Any(), gomock.Any(), (*bool)(nil)).Return( + []*entity.DatasetItem{}, []entity.ItemErrorGroup{}, nil) + + traceServiceStub := &stubTraceService{} + traceServiceStub.getTrajectoriesFunc = func(ctx context.Context, workspaceID int64, traceIDs []string, startTime, endTime int64, platformType loop_span.PlatformType) (map[string]*loop_span.Trajectory, error) { + return map[string]*loop_span.Trajectory{}, nil + } + + return fields{ + traceRepo: repoMock, + traceConfig: confMock, + traceProducer: traceProducerMock, + annotationProducer: annotationProducerMock, + metrics: metricsMock, + tenantProvider: tenantMock, + DatasetServiceAdaptor: adaptor, + buildHelper: buildHelper, + traceService: traceServiceStub, + } + }, + args: args{ + ctx: context.Background(), + req: &ExportTracesToDatasetRequest{ + WorkspaceID: 123, + SpanIds: []SpanID{{TraceID: "trace-123", SpanID: "span-456"}}, + Category: entity.DatasetCategory_General, + Config: DatasetConfig{ + IsNewDataset: true, + DatasetName: ptr.Of("test-dataset"), + DatasetSchema: entity.DatasetSchema{ + FieldSchemas: []entity.FieldSchema{ + {Name: "trajectory", ContentType: entity.ContentType_Text, SchemaKey: entity.SchemaKey_Trajectory}, + }, + }, + }, + StartTime: time.Now().Unix() - 3600, + EndTime: time.Now().Unix(), + PlatformType: loop_span.PlatformCozeLoop, + ExportType: ExportType_Append, + FieldMappings: []entity.FieldMapping{ + {TraceFieldKey: "trajectory", FieldSchema: entity.FieldSchema{Name: "trajectory", ContentType: entity.ContentType_Text, SchemaKey: entity.SchemaKey_Trajectory}}, + }, + }, + }, + wantErr: false, + }, } for _, tt := range tests { @@ -1033,7 +1193,11 @@ func TestTraceExportServiceImpl_PreviewExportTracesToDataset(t *testing.T) { assert.Error(t, err) } else { assert.NoError(t, err) - assert.Equal(t, tt.want, got) + if tt.want != nil { + assert.Equal(t, tt.want, got) + } else { + assert.NotNil(t, got) + } } }) } diff --git a/backend/modules/observability/domain/trace/service/trace_filter_helper.go b/backend/modules/observability/domain/trace/service/trace_filter_helper.go new file mode 100644 index 000000000..b76dbd185 --- /dev/null +++ b/backend/modules/observability/domain/trace/service/trace_filter_helper.go @@ -0,0 +1,142 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package service + +import ( + "context" + "fmt" + "strconv" + + "github.com/bytedance/gg/gptr" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/loop_span" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service/trace/span_filter" + obErrorx "github.com/coze-dev/coze-loop/backend/modules/observability/pkg/errno" + "github.com/coze-dev/coze-loop/backend/pkg/errorx" + timeutil "github.com/coze-dev/coze-loop/backend/pkg/time" + "github.com/samber/lo" +) + +func BuildBuiltinFilters(ctx context.Context, f span_filter.Filter, env *span_filter.SpanEnv, spanListType loop_span.SpanListType) (*loop_span.FilterFields, error) { + filters := make([]*loop_span.FilterField, 0) + basicFilter, forceQuery, err := f.BuildBasicSpanFilter(ctx, env) + if err != nil { + return nil, err + } else if len(basicFilter) == 0 && !forceQuery { // if it's null, no need to query from ck + return nil, nil + } + filters = append(filters, basicFilter...) + switch spanListType { + case loop_span.SpanListTypeRootSpan: + subFilter, err := f.BuildRootSpanFilter(ctx, env) + if err != nil { + return nil, err + } + filters = append(filters, subFilter...) + case loop_span.SpanListTypeLLMSpan: + subFilter, err := f.BuildLLMSpanFilter(ctx, env) + if err != nil { + return nil, err + } + filters = append(filters, subFilter...) + case loop_span.SpanListTypeAllSpan: + subFilter, err := f.BuildALLSpanFilter(ctx, env) + if err != nil { + return nil, err + } + filters = append(filters, subFilter...) + case "": + // do nothing + default: + return nil, errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg(fmt.Sprintf("invalid span list type: %s", spanListType))) + } + filterAggr := &loop_span.FilterFields{ + QueryAndOr: lo.ToPtr(loop_span.QueryAndOrEnumAnd), + FilterFields: filters, + } + return filterAggr, nil +} + +func CombineFilters(filters ...*loop_span.FilterFields) *loop_span.FilterFields { + filterAggr := &loop_span.FilterFields{ + QueryAndOr: lo.ToPtr(loop_span.QueryAndOrEnumAnd), + } + for _, f := range filters { + if f == nil { + continue + } + filterAggr.FilterFields = append(filterAggr.FilterFields, &loop_span.FilterField{ + QueryAndOr: lo.ToPtr(loop_span.QueryAndOrEnumAnd), + SubFilter: f, + }) + } + return filterAggr +} + +func processSpecificFilter(f *loop_span.FilterField) error { + switch f.FieldName { + case loop_span.SpanFieldStatus: + if err := processStatusFilter(f); err != nil { + return err + } + case loop_span.SpanFieldDuration, + loop_span.SpanFieldLatencyFirstResp, + loop_span.SpanFieldStartTimeFirstResp, + loop_span.SpanFieldStartTimeFirstTokenResp, + loop_span.SpanFieldLatencyFirstTokenResp, + loop_span.SpanFieldReasoningDuration: + if err := processLatencyFilter(f); err != nil { + return err + } + } + return nil +} + +func processStatusFilter(f *loop_span.FilterField) error { + if f.QueryType == nil || *f.QueryType != loop_span.QueryTypeEnumIn { + return fmt.Errorf("status filter should use in operator") + } + f.FieldName = loop_span.SpanFieldStatusCode + f.FieldType = loop_span.FieldTypeLong + checkSuccess, checkError := false, false + for _, val := range f.Values { + switch val { + case loop_span.SpanStatusSuccess: + checkSuccess = true + case loop_span.SpanStatusError: + checkError = true + default: + return fmt.Errorf("invalid status code field value") + } + } + if checkSuccess && checkError { + f.QueryType = gptr.Of(loop_span.QueryTypeEnumAlwaysTrue) + f.Values = nil + } else if checkSuccess { + f.Values = []string{"0"} + } else if checkError { + f.QueryType = gptr.Of(loop_span.QueryTypeEnumNotIn) + f.Values = []string{"0"} + } else { + return fmt.Errorf("invalid status code query") + } + return nil +} + +// ms -> us +func processLatencyFilter(f *loop_span.FilterField) error { + if f.FieldType != loop_span.FieldTypeLong { + return fmt.Errorf("latency field type should be long ") + } + micros := make([]string, 0) + for _, val := range f.Values { + integer, err := strconv.ParseInt(val, 10, 64) + if err != nil { + return fmt.Errorf("fail to parse long value %s, %v", val, err) + } + integer = timeutil.MillSec2MicroSec(integer) + micros = append(micros, strconv.FormatInt(integer, 10)) + } + f.Values = micros + return nil +} diff --git a/backend/modules/observability/domain/trace/service/trace_service.go b/backend/modules/observability/domain/trace/service/trace_service.go index 965fe2927..e0423b40e 100644 --- a/backend/modules/observability/domain/trace/service/trace_service.go +++ b/backend/modules/observability/domain/trace/service/trace_service.go @@ -6,6 +6,7 @@ package service import ( "context" "fmt" + "sort" "strconv" "sync" "time" @@ -56,6 +57,8 @@ type ListSpansReq struct { PlatformType loop_span.PlatformType SpanListType loop_span.SpanListType Source span_filter.SourceType + SelectColumns []string + OmitColumns []string Scene entity.ProcessorScene } @@ -124,6 +127,18 @@ type GetTraceResp struct { Spans loop_span.SpanList } +type ListMetadataReq struct { + WorkspaceID int64 + StartTime int64 // ms + EndTime int64 // ms + SpanListType loop_span.SpanListType + PlatformType loop_span.PlatformType +} + +type ListMetadataResp struct { + MetadataItemList []*trace.MetadataItemInfo +} + type SearchTraceOApiReq struct { WorkspaceID int64 ThirdPartyWorkspaceID string @@ -277,6 +292,18 @@ type ListAnnotationsResp struct { Annotations loop_span.AnnotationList } +type ListWorkspaceAnnotationsReq struct { + WorkspaceID int64 + StartTime int64 + AnnotationType string + SpanListType loop_span.SpanListType + PlatformType loop_span.PlatformType +} + +type ListWorkspaceAnnotationsResp struct { + Annotations loop_span.AnnotationList +} + type ChangeEvaluatorScoreRequest struct { WorkspaceID int64 AnnotationID string @@ -401,7 +428,9 @@ type ITraceService interface { GetTracesAdvanceInfo(ctx context.Context, req *GetTracesAdvanceInfoReq) (*GetTracesAdvanceInfoResp, error) IngestTraces(ctx context.Context, req *IngestTracesReq) error GetTracesMetaInfo(ctx context.Context, req *GetTracesMetaInfoReq) (*GetTracesMetaInfoResp, error) + ListMetadata(ctx context.Context, req *ListMetadataReq) (*ListMetadataResp, error) ListAnnotations(ctx context.Context, req *ListAnnotationsReq) (*ListAnnotationsResp, error) + ListWorkspaceAnnotations(ctx context.Context, req *ListWorkspaceAnnotationsReq) (*ListWorkspaceAnnotationsResp, error) CreateAnnotation(ctx context.Context, req *CreateAnnotationReq) error DeleteAnnotation(ctx context.Context, req *DeleteAnnotationReq) error CreateManualAnnotation(ctx context.Context, req *CreateManualAnnotationReq) (*CreateManualAnnotationResp, error) @@ -1032,7 +1061,7 @@ func (r *TraceServiceImpl) GetTrace(ctx context.Context, req *GetTraceReq) (*Get QueryType: ptr.Of(loop_span.QueryTypeEnumEq), }) } - queryFilter := r.combineFilters(logTraceFilter, req.Filters) + queryFilter := CombineFilters(logTraceFilter, req.Filters) spans := traceResult.Spans processors, err := r.buildHelper.BuildGetTraceProcessors(ctx, span_processor.Settings{ WorkspaceId: req.WorkspaceID, @@ -1069,13 +1098,18 @@ func (r *TraceServiceImpl) ListSpans(ctx context.Context, req *ListSpansReq) (*L if err != nil { return nil, err } - builtinFilter, err := r.buildBuiltinFilters(ctx, platformFilter, req) + env := &span_filter.SpanEnv{ + WorkspaceID: req.WorkspaceID, + ThirdPartyWorkspaceID: req.ThirdPartyWorkspaceID, + Source: req.Source, + } + builtinFilter, err := BuildBuiltinFilters(ctx, platformFilter, env, req.SpanListType) if err != nil { return nil, err } else if builtinFilter == nil { return &ListSpansResp{Spans: loop_span.SpanList{}}, nil } - filters := r.combineFilters(builtinFilter, req.Filters) + filters := CombineFilters(builtinFilter, req.Filters) tenants, err := r.getTenants(ctx, req.PlatformType) if err != nil { return nil, err @@ -1090,6 +1124,8 @@ func (r *TraceServiceImpl) ListSpans(ctx context.Context, req *ListSpansReq) (*L Limit: req.Limit, DescByStartTime: req.DescByStartTime, PageToken: req.PageToken, + SelectColumns: req.SelectColumns, + OmitColumns: req.OmitColumns, }) r.metrics.EmitListSpans(req.WorkspaceID, string(req.SpanListType), st, err != nil) if err != nil { @@ -1188,17 +1224,17 @@ func (r *TraceServiceImpl) ListSpansOApi(ctx context.Context, req *ListSpansOApi if err != nil { return nil, err } - builtinFilter, err := r.buildBuiltinFilters(ctx, platformFilter, &ListSpansReq{ + env := &span_filter.SpanEnv{ WorkspaceID: req.WorkspaceID, ThirdPartyWorkspaceID: req.ThirdPartyWorkspaceID, - SpanListType: req.SpanListType, - }) + } + builtinFilter, err := BuildBuiltinFilters(ctx, platformFilter, env, req.SpanListType) if err != nil { return nil, err } else if builtinFilter == nil { return &ListSpansOApiResp{Spans: loop_span.SpanList{}}, nil } - filters := r.combineFilters(builtinFilter, req.Filters) + filters := CombineFilters(builtinFilter, req.Filters) tRes, err := r.traceRepo.ListSpans(ctx, &repo.ListSpansParam{ WorkSpaceID: strconv.FormatInt(req.WorkspaceID, 10), Tenants: req.Tenants, @@ -1450,6 +1486,121 @@ func (r *TraceServiceImpl) GetTracesMetaInfo(ctx context.Context, req *GetTraces }, nil } +func (r *TraceServiceImpl) ListMetadata(ctx context.Context, req *ListMetadataReq) (*ListMetadataResp, error) { + const maxListMetadataSpansList = 300 + listSpansResp, err := r.ListSpans(ctx, &ListSpansReq{ + WorkspaceID: req.WorkspaceID, + StartTime: req.StartTime, + EndTime: req.EndTime, + SpanListType: req.SpanListType, + PlatformType: req.PlatformType, + Limit: maxListMetadataSpansList, + OmitColumns: []string{ + "input", + "output", + }, + DescByStartTime: true, + }) + if err != nil { + return nil, err + } + type keyInfo struct { + count int + valueType string + } + keyInfoMap := make(map[string]*keyInfo) + for _, span := range listSpansResp.Spans { + for key := range span.SystemTagsString { + if info, ok := keyInfoMap[key]; ok { + info.count++ + } else { + keyInfoMap[key] = &keyInfo{count: 1, valueType: loop_span.MetadataValueTypeString} + } + } + for key := range span.SystemTagsLong { + if info, ok := keyInfoMap[key]; ok { + info.count++ + } else { + keyInfoMap[key] = &keyInfo{count: 1, valueType: loop_span.MetadataValueTypeLong} + } + } + for key := range span.SystemTagsDouble { + if info, ok := keyInfoMap[key]; ok { + info.count++ + } else { + keyInfoMap[key] = &keyInfo{count: 1, valueType: loop_span.MetadataValueTypeDouble} + } + } + for key := range span.TagsString { + if info, ok := keyInfoMap[key]; ok { + info.count++ + } else { + keyInfoMap[key] = &keyInfo{count: 1, valueType: loop_span.MetadataValueTypeString} + } + } + for key := range span.TagsLong { + if info, ok := keyInfoMap[key]; ok { + info.count++ + } else { + keyInfoMap[key] = &keyInfo{count: 1, valueType: loop_span.MetadataValueTypeLong} + } + } + for key := range span.TagsDouble { + if info, ok := keyInfoMap[key]; ok { + info.count++ + } else { + keyInfoMap[key] = &keyInfo{count: 1, valueType: loop_span.MetadataValueTypeDouble} + } + } + for key := range span.TagsBool { + if info, ok := keyInfoMap[key]; ok { + info.count++ + } else { + keyInfoMap[key] = &keyInfo{count: 1, valueType: loop_span.MetadataValueTypeBool} + } + } + for key := range span.TagsByte { + if info, ok := keyInfoMap[key]; ok { + info.count++ + } else { + keyInfoMap[key] = &keyInfo{count: 1, valueType: loop_span.MetadataValueTypeString} + } + } + } + + keys := lo.Keys(keyInfoMap) + sort.Slice(keys, func(i, j int) bool { + if keyInfoMap[keys[i]].count != keyInfoMap[keys[j]].count { + return keyInfoMap[keys[i]].count > keyInfoMap[keys[j]].count + } + return keys[i] < keys[j] + }) + + structFieldSet := make(map[string]struct{}, len(loop_span.SpanStructFieldKeys)) + for _, k := range loop_span.SpanStructFieldKeys { + structFieldSet[k] = struct{}{} + } + + items := make([]*trace.MetadataItemInfo, 0, len(loop_span.SpanStructFieldKeys)+len(keys)) + for _, key := range loop_span.SpanStructFieldKeys { + items = append(items, &trace.MetadataItemInfo{ + Key: key, + ValueType: loop_span.SpanStructFieldValueTypes[key], + }) + } + for _, key := range keys { + if _, ok := structFieldSet[key]; ok { + continue + } + items = append(items, &trace.MetadataItemInfo{ + Key: key, + ValueType: keyInfoMap[key].valueType, + }) + } + + return &ListMetadataResp{MetadataItemList: items}, nil +} + func (r *TraceServiceImpl) ListAnnotations(ctx context.Context, req *ListAnnotationsReq) (*ListAnnotationsResp, error) { tenants, err := r.getTenants(ctx, req.PlatformType, tenant.WithWorkspaceID(req.WorkspaceID)) if err != nil { @@ -1473,6 +1624,33 @@ func (r *TraceServiceImpl) ListAnnotations(ctx context.Context, req *ListAnnotat }, nil } +func (r *TraceServiceImpl) ListWorkspaceAnnotations(ctx context.Context, req *ListWorkspaceAnnotationsReq) (*ListWorkspaceAnnotationsResp, error) { + tenants, err := r.getTenants(ctx, req.PlatformType, tenant.WithWorkspaceID(req.WorkspaceID)) + if err != nil { + return nil, err + } + const ( + defaultLimit = 300 + defaultDescByUpdatedAt = true + ) + annotations, err := r.traceRepo.ListWorkspaceAnnotations(ctx, &repo.ListWorkspaceAnnotationsParam{ + WorkSpaceID: strconv.FormatInt(req.WorkspaceID, 10), + Tenants: tenants, + AnnotationType: req.AnnotationType, + StartAt: req.StartTime, + EndAt: time.Now().UnixMilli(), + DescByUpdatedAt: defaultDescByUpdatedAt, + Limit: defaultLimit, + }) + if err != nil { + return nil, err + } + + return &ListWorkspaceAnnotationsResp{ + Annotations: annotations, + }, nil +} + func (r *TraceServiceImpl) CreateManualAnnotation(ctx context.Context, req *CreateManualAnnotationReq) (*CreateManualAnnotationResp, error) { tenants, err := r.getTenants(ctx, req.PlatformType) if err != nil { @@ -1857,65 +2035,6 @@ func (r *TraceServiceImpl) getAnnotationCallerCfg(ctx context.Context, caller st return nil, errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode) } -func (r *TraceServiceImpl) buildBuiltinFilters(ctx context.Context, f span_filter.Filter, req *ListSpansReq) (*loop_span.FilterFields, error) { - filters := make([]*loop_span.FilterField, 0) - env := &span_filter.SpanEnv{ - WorkspaceID: req.WorkspaceID, - ThirdPartyWorkspaceID: req.ThirdPartyWorkspaceID, - Source: req.Source, - } - basicFilter, forceQuery, err := f.BuildBasicSpanFilter(ctx, env) - if err != nil { - return nil, err - } else if len(basicFilter) == 0 && !forceQuery { // if it's null, no need to query from ck - return nil, nil - } - filters = append(filters, basicFilter...) - switch req.SpanListType { - case loop_span.SpanListTypeRootSpan: - subFilter, err := f.BuildRootSpanFilter(ctx, env) - if err != nil { - return nil, err - } - filters = append(filters, subFilter...) - case loop_span.SpanListTypeLLMSpan: - subFilter, err := f.BuildLLMSpanFilter(ctx, env) - if err != nil { - return nil, err - } - filters = append(filters, subFilter...) - case loop_span.SpanListTypeAllSpan: - subFilter, err := f.BuildALLSpanFilter(ctx, env) - if err != nil { - return nil, err - } - filters = append(filters, subFilter...) - default: - return nil, errorx.NewByCode(obErrorx.CommercialCommonInvalidParamCodeCode, errorx.WithExtraMsg("invalid span list type: %s")) - } - filterAggr := &loop_span.FilterFields{ - QueryAndOr: ptr.Of(loop_span.QueryAndOrEnumAnd), - FilterFields: filters, - } - return filterAggr, nil -} - -func (r *TraceServiceImpl) combineFilters(filters ...*loop_span.FilterFields) *loop_span.FilterFields { - filterAggr := &loop_span.FilterFields{ - QueryAndOr: ptr.Of(loop_span.QueryAndOrEnumAnd), - } - for _, f := range filters { - if f == nil { - continue - } - filterAggr.FilterFields = append(filterAggr.FilterFields, &loop_span.FilterField{ - QueryAndOr: ptr.Of(loop_span.QueryAndOrEnumAnd), - SubFilter: f, - }) - } - return filterAggr -} - func (r *TraceServiceImpl) getTenants(ctx context.Context, platform loop_span.PlatformType, opts ...tenant.OptFn) ([]string, error) { return r.tenantProvider.GetTenantsByPlatformType(ctx, platform, opts...) } @@ -2375,74 +2494,6 @@ func buildContent(value string) *dataset.Content { return content } -func processSpecificFilter(f *loop_span.FilterField) error { - switch f.FieldName { - case loop_span.SpanFieldStatus: - if err := processStatusFilter(f); err != nil { - return err - } - case loop_span.SpanFieldDuration, - loop_span.SpanFieldLatencyFirstResp, - loop_span.SpanFieldStartTimeFirstResp, - loop_span.SpanFieldStartTimeFirstTokenResp, - loop_span.SpanFieldLatencyFirstTokenResp, - loop_span.SpanFieldReasoningDuration: - if err := processLatencyFilter(f); err != nil { - return err - } - } - return nil -} - -func processStatusFilter(f *loop_span.FilterField) error { - if f.QueryType == nil || *f.QueryType != loop_span.QueryTypeEnumIn { - return fmt.Errorf("status filter should use in operator") - } - f.FieldName = loop_span.SpanFieldStatusCode - f.FieldType = loop_span.FieldTypeLong - checkSuccess, checkError := false, false - for _, val := range f.Values { - switch val { - case loop_span.SpanStatusSuccess: - checkSuccess = true - case loop_span.SpanStatusError: - checkError = true - default: - return fmt.Errorf("invalid status code field value") - } - } - if checkSuccess && checkError { - f.QueryType = ptr.Of(loop_span.QueryTypeEnumAlwaysTrue) - f.Values = nil - } else if checkSuccess { - f.Values = []string{"0"} - } else if checkError { - f.QueryType = ptr.Of(loop_span.QueryTypeEnumNotIn) - f.Values = []string{"0"} - } else { - return fmt.Errorf("invalid status code query") - } - return nil -} - -// ms -> us -func processLatencyFilter(f *loop_span.FilterField) error { - if f.FieldType != loop_span.FieldTypeLong { - return fmt.Errorf("latency field type should be long ") - } - micros := make([]string, 0) - for _, val := range f.Values { - integer, err := strconv.ParseInt(val, 10, 64) - if err != nil { - return fmt.Errorf("fail to parse long value %s, %v", val, err) - } - integer = timeutil.MillSec2MicroSec(integer) - micros = append(micros, strconv.FormatInt(integer, 10)) - } - f.Values = micros - return nil -} - //go:generate mockgen -destination=mocks/span_processor.go -package=mocks . TraceFilterProcessorBuilder type TraceFilterProcessorBuilder interface { BuildPlatformRelatedFilter(context.Context, loop_span.PlatformType) (span_filter.Filter, error) diff --git a/backend/modules/observability/domain/trace/service/trace_service_metadata_annotation_test.go b/backend/modules/observability/domain/trace/service/trace_service_metadata_annotation_test.go new file mode 100644 index 000000000..06f7c6030 --- /dev/null +++ b/backend/modules/observability/domain/trace/service/trace_service_metadata_annotation_test.go @@ -0,0 +1,361 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package service_test + +import ( + "context" + "errors" + "testing" + "time" + + "github.com/stretchr/testify/assert" + "go.uber.org/mock/gomock" + + metricsmock "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/metrics/mocks" + tenantmock "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/tenant/mocks" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/loop_span" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/repo" + repomock "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/repo/mocks" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service" + servicemock "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service/mocks" + filtermock "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service/trace/span_filter/mocks" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/service/trace/span_processor" +) + +func TestTraceServiceImpl_ListMetadata(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + traceRepoMock := repomock.NewMockITraceRepo(ctrl) + tenantProviderMock := tenantmock.NewMockITenantProvider(ctrl) + metricsMock := metricsmock.NewMockITraceMetrics(ctrl) + buildHelperMock := servicemock.NewMockTraceFilterProcessorBuilder(ctrl) + filterMock := filtermock.NewMockFilter(ctrl) + + svc, err := service.NewTraceServiceImpl( + traceRepoMock, + nil, + nil, + nil, + metricsMock, + buildHelperMock, + tenantProviderMock, + nil, + nil, + nil, + ) + assert.NoError(t, err) + + ctx := context.Background() + req := &service.ListMetadataReq{ + WorkspaceID: 123, + StartTime: time.Now().Add(-time.Hour).UnixMilli(), + EndTime: time.Now().UnixMilli(), + PlatformType: loop_span.PlatformCozeLoop, + } + + t.Run("dedup by key and sort by frequency desc", func(t *testing.T) { + buildHelperMock.EXPECT().BuildPlatformRelatedFilter(gomock.Any(), req.PlatformType).Return(filterMock, nil) + filterMock.EXPECT().BuildBasicSpanFilter(gomock.Any(), gomock.Any()).Return(nil, true, nil) + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType).Return([]string{"tenant1"}, nil) + + spans := loop_span.SpanList{ + { + SpanID: "span1", + TagsString: map[string]string{ + "tag_a": "v1", + "tag_b": "v2", + }, + }, + { + SpanID: "span2", + TagsString: map[string]string{ + "tag_a": "v3", + "tag_b": "v4", + "tag_c": "v5", + }, + }, + { + SpanID: "span3", + TagsString: map[string]string{ + "tag_a": "v1", + }, + }, + } + + traceRepoMock.EXPECT().ListSpans(gomock.Any(), gomock.Any()).Return(&repo.ListSpansResult{ + Spans: spans, + }, nil) + + metricsMock.EXPECT().EmitListSpans(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()) + buildHelperMock.EXPECT().BuildListSpansProcessors(gomock.Any(), gomock.Any()).Return([]span_processor.Processor{}, nil) + + resp, err := svc.ListMetadata(ctx, req) + assert.NoError(t, err) + assert.NotNil(t, resp) + structFieldCount := len(loop_span.SpanStructFieldKeys) + assert.Len(t, resp.MetadataItemList, structFieldCount+3) + for i, key := range loop_span.SpanStructFieldKeys { + assert.Equal(t, key, resp.MetadataItemList[i].Key) + assert.Equal(t, loop_span.SpanStructFieldValueTypes[key], resp.MetadataItemList[i].ValueType) + } + assert.Equal(t, "tag_a", resp.MetadataItemList[structFieldCount].Key) + assert.Equal(t, loop_span.MetadataValueTypeString, resp.MetadataItemList[structFieldCount].ValueType) + assert.Equal(t, "tag_b", resp.MetadataItemList[structFieldCount+1].Key) + assert.Equal(t, loop_span.MetadataValueTypeString, resp.MetadataItemList[structFieldCount+1].ValueType) + assert.Equal(t, "tag_c", resp.MetadataItemList[structFieldCount+2].Key) + assert.Equal(t, loop_span.MetadataValueTypeString, resp.MetadataItemList[structFieldCount+2].ValueType) + }) + + t.Run("list spans error", func(t *testing.T) { + buildHelperMock.EXPECT().BuildPlatformRelatedFilter(gomock.Any(), req.PlatformType).Return(nil, errors.New("filter error")) + + resp, err := svc.ListMetadata(ctx, req) + assert.Error(t, err) + assert.Nil(t, resp) + }) + + t.Run("empty spans returns only struct field keys", func(t *testing.T) { + buildHelperMock.EXPECT().BuildPlatformRelatedFilter(gomock.Any(), req.PlatformType).Return(filterMock, nil) + filterMock.EXPECT().BuildBasicSpanFilter(gomock.Any(), gomock.Any()).Return(nil, true, nil) + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType).Return([]string{"tenant1"}, nil) + traceRepoMock.EXPECT().ListSpans(gomock.Any(), gomock.Any()).Return(&repo.ListSpansResult{ + Spans: loop_span.SpanList{}, + }, nil) + metricsMock.EXPECT().EmitListSpans(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()) + buildHelperMock.EXPECT().BuildListSpansProcessors(gomock.Any(), gomock.Any()).Return([]span_processor.Processor{}, nil) + + resp, err := svc.ListMetadata(ctx, req) + assert.NoError(t, err) + assert.NotNil(t, resp) + assert.Len(t, resp.MetadataItemList, len(loop_span.SpanStructFieldKeys)) + for i, key := range loop_span.SpanStructFieldKeys { + assert.Equal(t, key, resp.MetadataItemList[i].Key) + assert.Equal(t, loop_span.SpanStructFieldValueTypes[key], resp.MetadataItemList[i].ValueType) + } + }) + + t.Run("mixed tag types return correct value_type", func(t *testing.T) { + buildHelperMock.EXPECT().BuildPlatformRelatedFilter(gomock.Any(), req.PlatformType).Return(filterMock, nil) + filterMock.EXPECT().BuildBasicSpanFilter(gomock.Any(), gomock.Any()).Return(nil, true, nil) + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType).Return([]string{"tenant1"}, nil) + + spans := loop_span.SpanList{ + { + SpanID: "span1", + TagsString: map[string]string{"str_tag": "val"}, + TagsLong: map[string]int64{"long_tag": 42}, + TagsDouble: map[string]float64{"double_tag": 3.14}, + TagsBool: map[string]bool{"bool_tag": true}, + SystemTagsString: map[string]string{"sys_str": "v"}, + SystemTagsLong: map[string]int64{"sys_long": 1}, + SystemTagsDouble: map[string]float64{"sys_double": 2.0}, + }, + } + + traceRepoMock.EXPECT().ListSpans(gomock.Any(), gomock.Any()).Return(&repo.ListSpansResult{ + Spans: spans, + }, nil) + metricsMock.EXPECT().EmitListSpans(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()) + buildHelperMock.EXPECT().BuildListSpansProcessors(gomock.Any(), gomock.Any()).Return([]span_processor.Processor{}, nil) + + resp, err := svc.ListMetadata(ctx, req) + assert.NoError(t, err) + assert.NotNil(t, resp) + + structFieldCount := len(loop_span.SpanStructFieldKeys) + tagItems := resp.MetadataItemList[structFieldCount:] + + typeMap := make(map[string]string) + for _, item := range tagItems { + typeMap[item.Key] = item.ValueType + } + assert.Equal(t, loop_span.MetadataValueTypeString, typeMap["str_tag"]) + assert.Equal(t, loop_span.MetadataValueTypeLong, typeMap["long_tag"]) + assert.Equal(t, loop_span.MetadataValueTypeDouble, typeMap["double_tag"]) + assert.Equal(t, loop_span.MetadataValueTypeBool, typeMap["bool_tag"]) + assert.Equal(t, loop_span.MetadataValueTypeString, typeMap["sys_str"]) + assert.Equal(t, loop_span.MetadataValueTypeLong, typeMap["sys_long"]) + assert.Equal(t, loop_span.MetadataValueTypeDouble, typeMap["sys_double"]) + }) + + t.Run("TagsByte returns string value type", func(t *testing.T) { + buildHelperMock.EXPECT().BuildPlatformRelatedFilter(gomock.Any(), req.PlatformType).Return(filterMock, nil) + filterMock.EXPECT().BuildBasicSpanFilter(gomock.Any(), gomock.Any()).Return(nil, true, nil) + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType).Return([]string{"tenant1"}, nil) + + spans := loop_span.SpanList{ + { + SpanID: "span1", + TagsByte: map[string]string{"byte_tag": "binary_data"}, + }, + } + + traceRepoMock.EXPECT().ListSpans(gomock.Any(), gomock.Any()).Return(&repo.ListSpansResult{ + Spans: spans, + }, nil) + metricsMock.EXPECT().EmitListSpans(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()) + buildHelperMock.EXPECT().BuildListSpansProcessors(gomock.Any(), gomock.Any()).Return([]span_processor.Processor{}, nil) + + resp, err := svc.ListMetadata(ctx, req) + assert.NoError(t, err) + assert.NotNil(t, resp) + + structFieldCount := len(loop_span.SpanStructFieldKeys) + assert.Len(t, resp.MetadataItemList, structFieldCount+1) + assert.Equal(t, "byte_tag", resp.MetadataItemList[structFieldCount].Key) + assert.Equal(t, loop_span.MetadataValueTypeString, resp.MetadataItemList[structFieldCount].ValueType) + }) + + t.Run("same key across multiple spans increments count for all tag types", func(t *testing.T) { + buildHelperMock.EXPECT().BuildPlatformRelatedFilter(gomock.Any(), req.PlatformType).Return(filterMock, nil) + filterMock.EXPECT().BuildBasicSpanFilter(gomock.Any(), gomock.Any()).Return(nil, true, nil) + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType).Return([]string{"tenant1"}, nil) + + spans := loop_span.SpanList{ + { + SpanID: "span1", + SystemTagsString: map[string]string{"shared_sys": "a"}, + SystemTagsLong: map[string]int64{"shared_sys_long": 1}, + SystemTagsDouble: map[string]float64{"shared_sys_double": 1.0}, + TagsLong: map[string]int64{"shared_long": 1}, + TagsDouble: map[string]float64{"shared_double": 1.1}, + TagsBool: map[string]bool{"shared_bool": true}, + TagsByte: map[string]string{"shared_byte": "x"}, + }, + { + SpanID: "span2", + SystemTagsString: map[string]string{"shared_sys": "b"}, + SystemTagsLong: map[string]int64{"shared_sys_long": 2}, + SystemTagsDouble: map[string]float64{"shared_sys_double": 2.0}, + TagsLong: map[string]int64{"shared_long": 2}, + TagsDouble: map[string]float64{"shared_double": 2.2}, + TagsBool: map[string]bool{"shared_bool": false}, + TagsByte: map[string]string{"shared_byte": "y"}, + }, + } + + traceRepoMock.EXPECT().ListSpans(gomock.Any(), gomock.Any()).Return(&repo.ListSpansResult{ + Spans: spans, + }, nil) + metricsMock.EXPECT().EmitListSpans(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()) + buildHelperMock.EXPECT().BuildListSpansProcessors(gomock.Any(), gomock.Any()).Return([]span_processor.Processor{}, nil) + + resp, err := svc.ListMetadata(ctx, req) + assert.NoError(t, err) + assert.NotNil(t, resp) + + structFieldCount := len(loop_span.SpanStructFieldKeys) + tagItems := resp.MetadataItemList[structFieldCount:] + assert.Len(t, tagItems, 7) + }) + + t.Run("tag key matching struct field key is deduped", func(t *testing.T) { + buildHelperMock.EXPECT().BuildPlatformRelatedFilter(gomock.Any(), req.PlatformType).Return(filterMock, nil) + filterMock.EXPECT().BuildBasicSpanFilter(gomock.Any(), gomock.Any()).Return(nil, true, nil) + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType).Return([]string{"tenant1"}, nil) + + spans := loop_span.SpanList{ + { + SpanID: "span1", + TagsString: map[string]string{ + "trace_id": "should-be-deduped", + "custom_tag": "val", + }, + }, + } + + traceRepoMock.EXPECT().ListSpans(gomock.Any(), gomock.Any()).Return(&repo.ListSpansResult{ + Spans: spans, + }, nil) + metricsMock.EXPECT().EmitListSpans(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()) + buildHelperMock.EXPECT().BuildListSpansProcessors(gomock.Any(), gomock.Any()).Return([]span_processor.Processor{}, nil) + + resp, err := svc.ListMetadata(ctx, req) + assert.NoError(t, err) + assert.NotNil(t, resp) + + structFieldCount := len(loop_span.SpanStructFieldKeys) + assert.Len(t, resp.MetadataItemList, structFieldCount+1) + assert.Equal(t, "custom_tag", resp.MetadataItemList[structFieldCount].Key) + }) +} + +func TestTraceServiceImpl_ListWorkspaceAnnotations(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + traceRepoMock := repomock.NewMockITraceRepo(ctrl) + tenantProviderMock := tenantmock.NewMockITenantProvider(ctrl) + + svc, err := service.NewTraceServiceImpl( + traceRepoMock, + nil, + nil, + nil, + nil, + nil, + tenantProviderMock, + nil, + nil, + nil, + ) + assert.NoError(t, err) + + ctx := context.Background() + req := &service.ListWorkspaceAnnotationsReq{ + WorkspaceID: 123, + StartTime: time.Now().Add(-time.Hour).UnixMilli(), + AnnotationType: "test_type", + PlatformType: loop_span.PlatformCozeLoop, + } + + t.Run("returns raw annotations", func(t *testing.T) { + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType, gomock.Any()).Return([]string{"tenant1"}, nil) + + annotations := loop_span.AnnotationList{ + {ID: "a1", Key: "key_a", AnnotationType: loop_span.AnnotationType("test_type")}, + {ID: "a2", Key: "key_b", AnnotationType: loop_span.AnnotationType("test_type")}, + {ID: "a3", Key: "key_a", AnnotationType: loop_span.AnnotationType("test_type")}, + } + + traceRepoMock.EXPECT().ListWorkspaceAnnotations(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, param *repo.ListWorkspaceAnnotationsParam) (loop_span.AnnotationList, error) { + assert.Equal(t, "123", param.WorkSpaceID) + return annotations, nil + }) + + resp, err := svc.ListWorkspaceAnnotations(ctx, req) + assert.NoError(t, err) + assert.NotNil(t, resp) + assert.Len(t, resp.Annotations, 3) + assert.Equal(t, "key_a", resp.Annotations[0].Key) + assert.Equal(t, "key_b", resp.Annotations[1].Key) + assert.Equal(t, "key_a", resp.Annotations[2].Key) + }) + + t.Run("get tenants error", func(t *testing.T) { + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType, gomock.Any()).Return(nil, errors.New("tenant error")) + resp, err := svc.ListWorkspaceAnnotations(ctx, req) + assert.Error(t, err) + assert.Nil(t, resp) + }) + + t.Run("repo error", func(t *testing.T) { + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType, gomock.Any()).Return([]string{"tenant1"}, nil) + traceRepoMock.EXPECT().ListWorkspaceAnnotations(gomock.Any(), gomock.Any()).Return(nil, errors.New("repo error")) + resp, err := svc.ListWorkspaceAnnotations(ctx, req) + assert.Error(t, err) + assert.Nil(t, resp) + }) + + t.Run("empty annotations", func(t *testing.T) { + tenantProviderMock.EXPECT().GetTenantsByPlatformType(gomock.Any(), req.PlatformType, gomock.Any()).Return([]string{"tenant1"}, nil) + traceRepoMock.EXPECT().ListWorkspaceAnnotations(gomock.Any(), gomock.Any()).Return(loop_span.AnnotationList{}, nil) + + resp, err := svc.ListWorkspaceAnnotations(ctx, req) + assert.NoError(t, err) + assert.NotNil(t, resp) + assert.Len(t, resp.Annotations, 0) + }) +} diff --git a/backend/modules/observability/infra/mq/consumer/backfill_consumer.go b/backend/modules/observability/infra/mq/consumer/backfill_consumer.go index 549c24a8d..c2df36224 100644 --- a/backend/modules/observability/infra/mq/consumer/backfill_consumer.go +++ b/backend/modules/observability/infra/mq/consumer/backfill_consumer.go @@ -59,6 +59,7 @@ func (e *BackFillConsumer) HandleMessage(ctx context.Context, ext *mq.MessageExt logs.CtxError(ctx, "BackFill msg json unmarshal fail, raw: %v, err: %s", conv.UnsafeBytesToString(ext.Body), err) return nil } + newCtx := context.WithoutCancel(ctx) logs.CtxInfo(ctx, "BackFill msg %+v,msgID=%s", event, ext.MsgID) - return e.handler.BackFill(ctx, event) + return e.handler.BackFill(newCtx, event) } diff --git a/backend/modules/observability/infra/repo/ck/annotation.go b/backend/modules/observability/infra/repo/ck/annotation.go index a9cd3f852..20d35abf7 100644 --- a/backend/modules/observability/infra/repo/ck/annotation.go +++ b/backend/modules/observability/infra/repo/ck/annotation.go @@ -76,14 +76,20 @@ func (a *AnnotationCkDaoImpl) Get(ctx context.Context, params *dao.GetAnnotation } func (a *AnnotationCkDaoImpl) List(ctx context.Context, params *dao.ListAnnotationsParam) ([]*dao.Annotation, error) { - if params == nil || len(params.SpanIDs) == 0 { + if params == nil { + return nil, nil + } + if len(params.SpanIDs) == 0 && params.WorkspaceID == "" { + logs.CtxWarn(ctx, "List annotations requires SpanID or WorkspaceID") return nil, nil } db, err := a.buildSql(ctx, &annoSqlParam{ Tables: params.Tables, + WorkspaceID: params.WorkspaceID, StartTime: params.StartTime, EndTime: params.EndTime, SpanIDs: params.SpanIDs, + AnnotationType: params.AnnotationType, DescByUpdatedAt: params.DescByUpdatedAt, Limit: params.Limit, }) @@ -102,10 +108,12 @@ func (a *AnnotationCkDaoImpl) List(ctx context.Context, params *dao.ListAnnotati type annoSqlParam struct { Tables []string + WorkspaceID string StartTime int64 EndTime int64 ID string SpanIDs []string + AnnotationType string DescByUpdatedAt bool Limit int32 } @@ -156,9 +164,15 @@ func (a *AnnotationCkDaoImpl) buildSingleSql(ctx context.Context, db *gorm.DB, t if param.ID != "" { sqlQuery = sqlQuery.Where("id = ?", param.ID) } + if param.WorkspaceID != "" { + sqlQuery = sqlQuery.Where("space_id = ?", param.WorkspaceID) + } if len(param.SpanIDs) > 0 { sqlQuery = sqlQuery.Where("span_id IN (?)", param.SpanIDs) } + if param.AnnotationType != "" { + sqlQuery = sqlQuery.Where("annotation_type = ?", param.AnnotationType) + } sqlQuery = sqlQuery. Where("start_time >= ?", param.StartTime). Where("start_time <= ?", param.EndTime) diff --git a/backend/modules/observability/infra/repo/dao/annotation.go b/backend/modules/observability/infra/repo/dao/annotation.go index a4705b437..23080fce0 100644 --- a/backend/modules/observability/infra/repo/dao/annotation.go +++ b/backend/modules/observability/infra/repo/dao/annotation.go @@ -23,7 +23,9 @@ type GetAnnotationParam struct { type ListAnnotationsParam struct { Tables []string + WorkspaceID string SpanIDs []string + AnnotationType string StartTime int64 // us EndTime int64 // us DescByUpdatedAt bool diff --git a/backend/modules/observability/infra/repo/mysql/convertor/task.go b/backend/modules/observability/infra/repo/mysql/convertor/task.go index 7c735e108..c8a456783 100644 --- a/backend/modules/observability/infra/repo/mysql/convertor/task.go +++ b/backend/modules/observability/infra/repo/mysql/convertor/task.go @@ -30,6 +30,7 @@ func TaskDO2PO(task *entity.ObservabilityTask) *model.ObservabilityTask { CreatedBy: task.CreatedBy, UpdatedBy: task.UpdatedBy, TaskSource: task.TaskSource, + WorkflowID: task.WorkflowID, } } @@ -52,6 +53,7 @@ func TaskPO2DO(task *model.ObservabilityTask) *entity.ObservabilityTask { CreatedBy: task.CreatedBy, UpdatedBy: task.UpdatedBy, TaskSource: task.TaskSource, + WorkflowID: task.WorkflowID, } } diff --git a/backend/modules/observability/infra/repo/mysql/gorm_gen/model/task.gen.go b/backend/modules/observability/infra/repo/mysql/gorm_gen/model/task.gen.go index 5906633a5..832a0adab 100644 --- a/backend/modules/observability/infra/repo/mysql/gorm_gen/model/task.gen.go +++ b/backend/modules/observability/infra/repo/mysql/gorm_gen/model/task.gen.go @@ -29,6 +29,7 @@ type ObservabilityTask struct { CreatedBy string `gorm:"column:created_by;type:varchar(128);not null;comment:创建人" json:"created_by"` // 创建人 UpdatedBy string `gorm:"column:updated_by;type:varchar(128);not null;comment:更新人" json:"updated_by"` // 更新人 TaskSource *string `gorm:"column:task_source;type:varchar(50);default:user;comment:任务来源" json:"task_source"` // 任务来源 + WorkflowID int64 `gorm:"column:workflow_id;type:bigint(20) unsigned;not null;comment:相关工作流ID" json:"workflow_id"` // 相关工作流ID } // TableName ObservabilityTask's table name diff --git a/backend/modules/observability/infra/repo/mysql/gorm_gen/query/task.gen.go b/backend/modules/observability/infra/repo/mysql/gorm_gen/query/task.gen.go index 4ce7251e8..10e49173f 100644 --- a/backend/modules/observability/infra/repo/mysql/gorm_gen/query/task.gen.go +++ b/backend/modules/observability/infra/repo/mysql/gorm_gen/query/task.gen.go @@ -44,6 +44,7 @@ func newObservabilityTask(db *gorm.DB, opts ...gen.DOOption) observabilityTask { _observabilityTask.CreatedBy = field.NewString(tableName, "created_by") _observabilityTask.UpdatedBy = field.NewString(tableName, "updated_by") _observabilityTask.TaskSource = field.NewString(tableName, "task_source") + _observabilityTask.WorkflowID = field.NewInt64(tableName, "workflow_id") _observabilityTask.fillFieldMap() @@ -72,6 +73,7 @@ type observabilityTask struct { CreatedBy field.String // 创建人 UpdatedBy field.String // 更新人 TaskSource field.String // 任务来源 + WorkflowID field.Int64 // 相关工作流ID fieldMap map[string]field.Expr } @@ -105,6 +107,7 @@ func (o *observabilityTask) updateTableName(table string) *observabilityTask { o.CreatedBy = field.NewString(table, "created_by") o.UpdatedBy = field.NewString(table, "updated_by") o.TaskSource = field.NewString(table, "task_source") + o.WorkflowID = field.NewInt64(table, "workflow_id") o.fillFieldMap() @@ -133,7 +136,7 @@ func (o *observabilityTask) GetFieldByName(fieldName string) (field.OrderExpr, b } func (o *observabilityTask) fillFieldMap() { - o.fieldMap = make(map[string]field.Expr, 17) + o.fieldMap = make(map[string]field.Expr, 18) o.fieldMap["id"] = o.ID o.fieldMap["workspace_id"] = o.WorkspaceID o.fieldMap["name"] = o.Name @@ -151,6 +154,7 @@ func (o *observabilityTask) fillFieldMap() { o.fieldMap["created_by"] = o.CreatedBy o.fieldMap["updated_by"] = o.UpdatedBy o.fieldMap["task_source"] = o.TaskSource + o.fieldMap["workflow_id"] = o.WorkflowID } func (o observabilityTask) clone(db *gorm.DB) observabilityTask { diff --git a/backend/modules/observability/infra/repo/mysql/task.go b/backend/modules/observability/infra/repo/mysql/task.go index 837f585a2..c3ebbc9f7 100644 --- a/backend/modules/observability/infra/repo/mysql/task.go +++ b/backend/modules/observability/infra/repo/mysql/task.go @@ -38,6 +38,7 @@ type ListTaskParam struct { ReqLimit int32 ReqOffset int32 OrderBy *common.OrderBy + NeedOnlyOld bool } //go:generate mockgen -destination=mocks/task.go -package=mocks . ITaskDao @@ -96,7 +97,7 @@ func (v *TaskDaoImpl) CreateTask(ctx context.Context, po *model.ObservabilityTas func (v *TaskDaoImpl) UpdateTask(ctx context.Context, po *model.ObservabilityTask) error { q := genquery.Use(v.dbMgr.NewSession(ctx)).ObservabilityTask - if err := q.WithContext(ctx).Save(po); err != nil { + if err := q.WithContext(ctx).Omit(q.CreatedAt).Save(po); err != nil { return errorx.WrapByCode(err, obErrorx.CommonMySqlErrorCode) } else { return nil @@ -121,6 +122,9 @@ func (v *TaskDaoImpl) ListTasks(ctx context.Context, param ListTaskParam) ([]*mo if len(param.WorkspaceIDs) != 0 { qd = qd.Where(q.ObservabilityTask.WorkspaceID.In(param.WorkspaceIDs...)) } + if param.NeedOnlyOld { + qd = qd.Where(q.ObservabilityTask.WorkflowID.Eq(0)) + } // 应用过滤条件 qdf, err := v.applyTaskFilters(q, param.TaskFilters) if err != nil { @@ -201,6 +205,8 @@ func (v *TaskDaoImpl) buildSingleFilterExpr(q *genquery.Query, f *entity.TaskFil return v.buildUpdateAtFilter(q, f) case "task_source": return v.buildTaskSourceFilter(q, f) + case entity.TaskFieldNameUpdatedBy: + return v.buildUpdatedByFilter(q, f) default: return nil, errorx.NewByCode(obErrorx.CommonInvalidParamCode, errorx.WithMsgParam("invalid filter field name: %s", string(*f.FieldName))) } @@ -278,6 +284,31 @@ func (v *TaskDaoImpl) buildCreatedByFilter(q *genquery.Query, f *entity.TaskFilt return q.ObservabilityTask.CreatedBy.In(f.Values...), nil case entity.QueryTypeNotIn: return q.ObservabilityTask.CreatedBy.NotIn(f.Values...), nil + case entity.QueryTypeEq: + return q.ObservabilityTask.CreatedBy.Eq(f.Values[0]), nil + case entity.QueryTypeNotEq: + return q.ObservabilityTask.CreatedBy.Neq(f.Values[0]), nil + default: + return nil, errorx.NewByCode(obErrorx.CommonInvalidParamCode, errorx.WithExtraMsg("invalid query type for created_by")) + } +} + +// 构建更新者过滤条件 +func (v *TaskDaoImpl) buildUpdatedByFilter(q *genquery.Query, f *entity.TaskFilterField) (field.Expr, error) { + if len(f.Values) == 0 { + return nil, errorx.NewByCode(obErrorx.CommonInvalidParamCode, errorx.WithExtraMsg("no values provided for created_by query")) + } + + switch *f.QueryType { + case entity.QueryTypeIn: + return q.ObservabilityTask.UpdatedBy.In(f.Values...), nil + case entity.QueryTypeNotIn: + return q.ObservabilityTask.UpdatedBy.NotIn(f.Values...), nil + case entity.QueryTypeEq: + return q.ObservabilityTask.UpdatedBy.Eq(f.Values[0]), nil + case entity.QueryTypeNotEq: + return q.ObservabilityTask.UpdatedBy.Neq(f.Values[0]), nil + default: return nil, errorx.NewByCode(obErrorx.CommonInvalidParamCode, errorx.WithExtraMsg("invalid query type for created_by")) } diff --git a/backend/modules/observability/infra/repo/mysql/task_run.go b/backend/modules/observability/infra/repo/mysql/task_run.go index 9cce07d99..7159b6166 100755 --- a/backend/modules/observability/infra/repo/mysql/task_run.go +++ b/backend/modules/observability/infra/repo/mysql/task_run.go @@ -123,7 +123,7 @@ func (v *TaskRunDaoImpl) CreateTaskRun(ctx context.Context, po *model.Observabil func (v *TaskRunDaoImpl) UpdateTaskRun(ctx context.Context, po *model.ObservabilityTaskRun) error { q := genquery.Use(v.dbMgr.NewSession(ctx)).ObservabilityTaskRun - if err := q.WithContext(ctx).Save(po); err != nil { + if err := q.WithContext(ctx).Omit(q.CreatedAt).Save(po); err != nil { return errorx.WrapByCode(err, obErrorx.CommonMySqlErrorCode) } else { return nil diff --git a/backend/modules/observability/infra/repo/task.go b/backend/modules/observability/infra/repo/task.go index 67879e887..5ec7b4b0a 100644 --- a/backend/modules/observability/infra/repo/task.go +++ b/backend/modules/observability/infra/repo/task.go @@ -71,6 +71,7 @@ func (v *TaskRepoImpl) ListTasks(ctx context.Context, param repo.ListTaskParam) ReqLimit: param.ReqLimit, ReqOffset: param.ReqOffset, OrderBy: param.OrderBy, + NeedOnlyOld: param.NeedOnlyOld, }) if err != nil { return nil, 0, err @@ -124,6 +125,10 @@ func (v *TaskRepoImpl) CreateTask(ctx context.Context, do *entity.ObservabilityT func (v *TaskRepoImpl) UpdateTask(ctx context.Context, do *entity.ObservabilityTask) error { TaskPo := convertor.TaskDO2PO(do) + if do.TaskType == entity.TaskTypeAutoDataReflow && do.TaskConfig.DataReflowConfig == nil { + logs.CtxInfo(ctx, "Task update config is nil, task:%v", do) + } + // 先执行数据库操作 err := v.TaskDao.UpdateTask(ctx, TaskPo) if err != nil { diff --git a/backend/modules/observability/infra/repo/trace.go b/backend/modules/observability/infra/repo/trace.go index e5844d09a..2814d3a79 100644 --- a/backend/modules/observability/infra/repo/trace.go +++ b/backend/modules/observability/infra/repo/trace.go @@ -467,6 +467,40 @@ func (t *TraceRepoImpl) ListAnnotations(ctx context.Context, param *repo.ListAnn return converter.AnnotationListPO2DO(annotations).Uniq(), nil } +func (t *TraceRepoImpl) ListWorkspaceAnnotations(ctx context.Context, param *repo.ListWorkspaceAnnotationsParam) (loop_span.AnnotationList, error) { + spanStorage := t.storageProvider.GetTraceStorage(ctx, param.WorkSpaceID, param.Tenants) + annoDao := t.annoDaos[spanStorage.StorageName] + if annoDao == nil { + return nil, errorx.WrapByCode(errors.New("invalid storage"), obErrorx.CommercialCommonInvalidParamCodeCode) + } + tableCfg, err := t.getQueryTenantTables(ctx, param.Tenants) + if err != nil { + return nil, err + } else if len(tableCfg.AnnoTables) == 0 { + return loop_span.AnnotationList{}, nil + } + limit := param.Limit + if limit <= 0 || limit > 1000 { + limit = 1000 + } + st := time.Now() + annotations, err := annoDao.List(ctx, &dao.ListAnnotationsParam{ + Tables: tableCfg.AnnoTables, + WorkspaceID: param.WorkSpaceID, + AnnotationType: param.AnnotationType, + StartTime: time_util.MillSec2MicroSec(param.StartAt), + EndTime: time_util.MillSec2MicroSec(param.EndAt), + DescByUpdatedAt: param.DescByUpdatedAt, + Limit: int32(limit), + Extra: spanStorage.StorageConfig, + }) + if err != nil { + return nil, err + } + logs.CtxInfo(ctx, "list workspace annotations successfully, annotations count %d, cost %v", len(annotations), time.Since(st)) + return converter.AnnotationListPO2DO(annotations).Uniq(), nil +} + func (t *TraceRepoImpl) GetAnnotation(ctx context.Context, param *repo.GetAnnotationParam) (*loop_span.Annotation, error) { spanStorage := t.storageProvider.GetTraceStorage(ctx, param.WorkSpaceID, param.Tenants) annoDao := t.annoDaos[spanStorage.StorageName] diff --git a/backend/modules/observability/infra/repo/trace_test.go b/backend/modules/observability/infra/repo/trace_test.go index 67f1ab453..d8b2e239e 100644 --- a/backend/modules/observability/infra/repo/trace_test.go +++ b/backend/modules/observability/infra/repo/trace_test.go @@ -1551,3 +1551,233 @@ func TestTraceRepoImpl_getSpanInsertTable(t *testing.T) { }) } } + +func TestTraceRepoImpl_ListWorkspaceAnnotations(t *testing.T) { + type fields struct { + annoDao dao.IAnnotationDao + traceConfig config.ITraceConfig + spanRedisDao *redis_dao_mock.MockISpansRedisDao + spanProducer mq.ISpanProducer + } + type args struct { + ctx context.Context + param *repo.ListWorkspaceAnnotationsParam + } + tests := []struct { + name string + fieldsGetter func(ctrl *gomock.Controller) fields + args args + want loop_span.AnnotationList + wantErr bool + }{ + { + name: "annoDao is nil returns error", + fieldsGetter: func(ctrl *gomock.Controller) fields { + traceConfigMock := confmocks.NewMockITraceConfig(ctrl) + return fields{ + annoDao: nil, + traceConfig: traceConfigMock, + } + }, + args: args{ + ctx: context.Background(), + param: &repo.ListWorkspaceAnnotationsParam{ + WorkSpaceID: "ws1", + Tenants: []string{"test"}, + }, + }, + wantErr: true, + }, + { + name: "getQueryTenantTables error", + fieldsGetter: func(ctrl *gomock.Controller) fields { + annoDaoMock := daomock.NewMockIAnnotationDao(ctrl) + traceConfigMock := confmocks.NewMockITraceConfig(ctrl) + traceConfigMock.EXPECT().GetTenantConfig(gomock.Any()).Return(nil, assert.AnError) + return fields{ + annoDao: annoDaoMock, + traceConfig: traceConfigMock, + } + }, + args: args{ + ctx: context.Background(), + param: &repo.ListWorkspaceAnnotationsParam{ + WorkSpaceID: "ws1", + Tenants: []string{"test"}, + }, + }, + wantErr: true, + }, + { + name: "empty anno tables returns empty list", + fieldsGetter: func(ctrl *gomock.Controller) fields { + annoDaoMock := daomock.NewMockIAnnotationDao(ctrl) + traceConfigMock := confmocks.NewMockITraceConfig(ctrl) + traceConfigMock.EXPECT().GetTenantConfig(gomock.Any()).Return(&config.TenantCfg{ + TenantTables: map[string]map[loop_span.TTL]config.TableCfg{ + "test": { + loop_span.TTL3d: { + AnnoTable: "", + }, + }, + }, + }, nil) + return fields{ + annoDao: annoDaoMock, + traceConfig: traceConfigMock, + } + }, + args: args{ + ctx: context.Background(), + param: &repo.ListWorkspaceAnnotationsParam{ + WorkSpaceID: "ws1", + Tenants: []string{"test"}, + }, + }, + want: loop_span.AnnotationList{}, + }, + { + name: "annoDao.List error", + fieldsGetter: func(ctrl *gomock.Controller) fields { + annoDaoMock := daomock.NewMockIAnnotationDao(ctrl) + annoDaoMock.EXPECT().List(gomock.Any(), gomock.Any()).Return(nil, assert.AnError) + traceConfigMock := confmocks.NewMockITraceConfig(ctrl) + traceConfigMock.EXPECT().GetTenantConfig(gomock.Any()).Return(&config.TenantCfg{ + TenantTables: map[string]map[loop_span.TTL]config.TableCfg{ + "test": { + loop_span.TTL3d: { + AnnoTable: "annotations", + }, + }, + }, + }, nil) + return fields{ + annoDao: annoDaoMock, + traceConfig: traceConfigMock, + } + }, + args: args{ + ctx: context.Background(), + param: &repo.ListWorkspaceAnnotationsParam{ + WorkSpaceID: "ws1", + Tenants: []string{"test"}, + }, + }, + wantErr: true, + }, + { + name: "success with default limit", + fieldsGetter: func(ctrl *gomock.Controller) fields { + annoDaoMock := daomock.NewMockIAnnotationDao(ctrl) + annoDaoMock.EXPECT().List(gomock.Any(), gomock.Any()).Return([]*dao.Annotation{ + { + ID: "anno1", + TraceID: "trace1", + SpaceID: "ws1", + }, + }, nil) + traceConfigMock := confmocks.NewMockITraceConfig(ctrl) + traceConfigMock.EXPECT().GetTenantConfig(gomock.Any()).Return(&config.TenantCfg{ + TenantTables: map[string]map[loop_span.TTL]config.TableCfg{ + "test": { + loop_span.TTL3d: { + AnnoTable: "annotations", + }, + }, + }, + }, nil) + return fields{ + annoDao: annoDaoMock, + traceConfig: traceConfigMock, + } + }, + args: args{ + ctx: context.Background(), + param: &repo.ListWorkspaceAnnotationsParam{ + WorkSpaceID: "ws1", + Tenants: []string{"test"}, + AnnotationType: "manual", + DescByUpdatedAt: true, + StartAt: 1000, + EndAt: 2000, + Limit: 0, + }, + }, + want: loop_span.AnnotationList{ + { + ID: "anno1", + TraceID: "trace1", + WorkspaceID: "ws1", + StartTime: time.UnixMicro(0), + UpdatedAt: time.UnixMicro(0), + CreatedAt: time.UnixMicro(0), + }, + }, + }, + { + name: "success with explicit limit", + fieldsGetter: func(ctrl *gomock.Controller) fields { + annoDaoMock := daomock.NewMockIAnnotationDao(ctrl) + annoDaoMock.EXPECT().List(gomock.Any(), gomock.Any()).Return([]*dao.Annotation{ + { + ID: "anno2", + TraceID: "trace2", + SpaceID: "ws2", + }, + }, nil) + traceConfigMock := confmocks.NewMockITraceConfig(ctrl) + traceConfigMock.EXPECT().GetTenantConfig(gomock.Any()).Return(&config.TenantCfg{ + TenantTables: map[string]map[loop_span.TTL]config.TableCfg{ + "test": { + loop_span.TTL3d: { + AnnoTable: "annotations", + }, + }, + }, + }, nil) + return fields{ + annoDao: annoDaoMock, + traceConfig: traceConfigMock, + } + }, + args: args{ + ctx: context.Background(), + param: &repo.ListWorkspaceAnnotationsParam{ + WorkSpaceID: "ws2", + Tenants: []string{"test"}, + Limit: 50, + }, + }, + want: loop_span.AnnotationList{ + { + ID: "anno2", + TraceID: "trace2", + WorkspaceID: "ws2", + StartTime: time.UnixMicro(0), + UpdatedAt: time.UnixMicro(0), + CreatedAt: time.UnixMicro(0), + }, + }, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + ctrl := gomock.NewController(t) + defer ctrl.Finish() + fields := tt.fieldsGetter(ctrl) + r, err := NewTraceRepoImpl( + fields.traceConfig, + &mockStorageProvider{}, + fields.spanRedisDao, + fields.spanProducer, + nil, + nil, + WithTraceStorageAnnotationDao("ck", fields.annoDao), + ) + assert.NoError(t, err) + got, err := r.ListWorkspaceAnnotations(tt.args.ctx, tt.args.param) + assert.Equal(t, tt.wantErr, err != nil) + assert.Equal(t, tt.want, got) + }) + } +} diff --git a/backend/modules/observability/infra/rpc/evaluation/evaluation.go b/backend/modules/observability/infra/rpc/evaluation/evaluation.go index 10c59d3c4..5c482b489 100644 --- a/backend/modules/observability/infra/rpc/evaluation/evaluation.go +++ b/backend/modules/observability/infra/rpc/evaluation/evaluation.go @@ -6,11 +6,15 @@ package evaluation import ( "context" + "github.com/bytedance/gg/gptr" + "github.com/bytedance/gg/gslice" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/domain/dataset" + evDomain "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/expt" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/experimentservice" "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/expt" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/rpc" obErrorx "github.com/coze-dev/coze-loop/backend/modules/observability/pkg/errno" "github.com/coze-dev/coze-loop/backend/pkg/errorx" @@ -31,7 +35,7 @@ func (e *EvaluationProvider) SubmitExperiment(ctx context.Context, param *rpc.Su return 0, 0, errorx.NewByCode(obErrorx.CommonInvalidParamCode, errorx.WithExtraMsg("workspace ID is nil")) } logs.CtxInfo(ctx, "SubmitExperiment, param: %+v", param) - resp, err := e.client.SubmitExperiment(ctx, &expt.SubmitExperimentRequest{ + req := &expt.SubmitExperimentRequest{ WorkspaceID: param.WorkspaceID, EvalSetVersionID: param.EvalSetVersionID, TargetVersionID: param.TargetVersionID, @@ -42,7 +46,6 @@ func (e *EvaluationProvider) SubmitExperiment(ctx context.Context, param *rpc.Su TargetID: param.TargetID, TargetFieldMapping: param.TargetFieldMapping, EvaluatorFieldMapping: param.EvaluatorFieldMapping, - ItemConcurNum: param.ItemConcurNum, EvaluatorsConcurNum: param.EvaluatorsConcurNum, CreateEvalTargetParam: param.CreateEvalTargetParam, ExptType: param.ExptType, @@ -50,7 +53,16 @@ func (e *EvaluationProvider) SubmitExperiment(ctx context.Context, param *rpc.Su SourceType: param.SourceType, SourceID: param.SourceID, Session: param.Session, - }) + EnableWeightedScore: gptr.Of(true), + ExptTemplateID: param.ExptTemplateID, + ItemConcurNum: param.ItemConcurNum, + ItemRetryNum: param.ItemRetryNum, + TimeRange: param.TimeRange, + } + if param.IsWorkflowScheduled != nil && *param.IsWorkflowScheduled { + req.TriggerType = gptr.Of(evDomain.Schedule) + } + resp, err := e.client.SubmitExperiment(ctx, req) if err != nil { logs.CtxError(ctx, "SubmitExperiment failed, err: %v", err) return 0, 0, errorx.NewByCode(obErrorx.CommonRPCErrorCode, errorx.WithExtraMsg("SubmitExperiment failed")) diff --git a/backend/modules/observability/infra/rpc/evaluation/evaluation_test.go b/backend/modules/observability/infra/rpc/evaluation/evaluation_test.go new file mode 100644 index 000000000..0d654e962 --- /dev/null +++ b/backend/modules/observability/infra/rpc/evaluation/evaluation_test.go @@ -0,0 +1,103 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package evaluation + +import ( + "context" + "testing" + + "github.com/bytedance/gg/gptr" + "github.com/coze-dev/coze-loop/backend/modules/observability/infra/rpc/evaluation/mocks" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" + "go.uber.org/mock/gomock" + + domainExpt "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/domain/expt" + expt "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/expt" + "github.com/coze-dev/coze-loop/backend/modules/observability/domain/component/rpc" +) + +//go:generate mockgen -destination=mock_expt_client_test.go -package=evaluation github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/expt/experimentservice Client + +func TestSubmitExperiment_WorkspaceIDZero(t *testing.T) { + t.Parallel() + p := &EvaluationProvider{} + _, _, err := p.SubmitExperiment(context.Background(), &rpc.SubmitExperimentReq{WorkspaceID: 0}) + assert.Error(t, err) +} + +func TestSubmitExperiment_Success(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + mockClient := mocks.NewMockClient(ctrl) + tplID := int64(999) + itemConcur := int32(5) + itemRetry := int32(3) + + mockClient.EXPECT().SubmitExperiment(gomock.Any(), gomock.Any()).DoAndReturn( + func(ctx context.Context, req *expt.SubmitExperimentRequest, opts ...interface{}) (*expt.SubmitExperimentResponse, error) { + assert.Equal(t, int64(123), req.WorkspaceID) + assert.Equal(t, &tplID, req.ExptTemplateID) + assert.Equal(t, &itemConcur, req.ItemConcurNum) + assert.Equal(t, &itemRetry, req.ItemRetryNum) + assert.Equal(t, gptr.Of(true), req.EnableWeightedScore) + return &expt.SubmitExperimentResponse{ + Experiment: &domainExpt.Experiment{ID: gptr.Of(int64(100))}, + RunID: gptr.Of(int64(200)), + }, nil + }, + ) + + p := &EvaluationProvider{client: mockClient} + exptID, runID, err := p.SubmitExperiment(context.Background(), &rpc.SubmitExperimentReq{ + WorkspaceID: 123, + ExptTemplateID: &tplID, + ItemConcurNum: &itemConcur, + ItemRetryNum: &itemRetry, + }) + require.NoError(t, err) + assert.Equal(t, int64(100), exptID) + assert.Equal(t, int64(200), runID) +} + +func TestSubmitExperiment_WithIsWorkflowScheduled(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + mockClient := mocks.NewMockClient(ctrl) + mockClient.EXPECT().SubmitExperiment(gomock.Any(), gomock.Any()).DoAndReturn( + func(ctx context.Context, req *expt.SubmitExperimentRequest, opts ...interface{}) (*expt.SubmitExperimentResponse, error) { + require.NotNil(t, req.TriggerType) + return &expt.SubmitExperimentResponse{ + Experiment: &domainExpt.Experiment{ID: gptr.Of(int64(1))}, + RunID: gptr.Of(int64(2)), + }, nil + }, + ) + + p := &EvaluationProvider{client: mockClient} + _, _, err := p.SubmitExperiment(context.Background(), &rpc.SubmitExperimentReq{ + WorkspaceID: 123, + IsWorkflowScheduled: gptr.Of(true), + }) + assert.NoError(t, err) +} + +func TestSubmitExperiment_ClientError(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + defer ctrl.Finish() + + mockClient := mocks.NewMockClient(ctrl) + mockClient.EXPECT().SubmitExperiment(gomock.Any(), gomock.Any()).Return(nil, assert.AnError) + + p := &EvaluationProvider{client: mockClient} + _, _, err := p.SubmitExperiment(context.Background(), &rpc.SubmitExperimentReq{ + WorkspaceID: 123, + }) + assert.Error(t, err) +} diff --git a/backend/modules/observability/infra/rpc/evaluation/mocks/mock_expt_client.go b/backend/modules/observability/infra/rpc/evaluation/mocks/mock_expt_client.go new file mode 100644 index 000000000..ab4db7c11 --- /dev/null +++ b/backend/modules/observability/infra/rpc/evaluation/mocks/mock_expt_client.go @@ -0,0 +1,843 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/expt/experimentservice (interfaces: Client) +// +// Generated by this command: +// +// mockgen -destination=modules/observability/infra/rpc/evaluation/mocks/mock_expt_client.go -package=mocks github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/expt/experimentservice Client +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + callopt "github.com/cloudwego/kitex/client/callopt" + expt "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/expt" + gomock "go.uber.org/mock/gomock" +) + +// MockClient is a mock of Client interface. +type MockClient struct { + ctrl *gomock.Controller + recorder *MockClientMockRecorder + isgomock struct{} +} + +// MockClientMockRecorder is the mock recorder for MockClient. +type MockClientMockRecorder struct { + mock *MockClient +} + +// NewMockClient creates a new mock instance. +func NewMockClient(ctrl *gomock.Controller) *MockClient { + mock := &MockClient{ctrl: ctrl} + mock.recorder = &MockClientMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockClient) EXPECT() *MockClientMockRecorder { + return m.recorder +} + +// AssociateAnnotationTag mocks base method. +func (m *MockClient) AssociateAnnotationTag(ctx context.Context, req *expt.AssociateAnnotationTagReq, callOptions ...callopt.Option) (*expt.AssociateAnnotationTagResp, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "AssociateAnnotationTag", varargs...) + ret0, _ := ret[0].(*expt.AssociateAnnotationTagResp) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// AssociateAnnotationTag indicates an expected call of AssociateAnnotationTag. +func (mr *MockClientMockRecorder) AssociateAnnotationTag(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AssociateAnnotationTag", reflect.TypeOf((*MockClient)(nil).AssociateAnnotationTag), varargs...) +} + +// BatchDeleteExperiments mocks base method. +func (m *MockClient) BatchDeleteExperiments(ctx context.Context, req *expt.BatchDeleteExperimentsRequest, callOptions ...callopt.Option) (*expt.BatchDeleteExperimentsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchDeleteExperiments", varargs...) + ret0, _ := ret[0].(*expt.BatchDeleteExperimentsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchDeleteExperiments indicates an expected call of BatchDeleteExperiments. +func (mr *MockClientMockRecorder) BatchDeleteExperiments(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchDeleteExperiments", reflect.TypeOf((*MockClient)(nil).BatchDeleteExperiments), varargs...) +} + +// BatchGetExperimentAggrResult_ mocks base method. +func (m *MockClient) BatchGetExperimentAggrResult_(ctx context.Context, req *expt.BatchGetExperimentAggrResultRequest, callOptions ...callopt.Option) (*expt.BatchGetExperimentAggrResultResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchGetExperimentAggrResult_", varargs...) + ret0, _ := ret[0].(*expt.BatchGetExperimentAggrResultResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchGetExperimentAggrResult_ indicates an expected call of BatchGetExperimentAggrResult_. +func (mr *MockClientMockRecorder) BatchGetExperimentAggrResult_(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetExperimentAggrResult_", reflect.TypeOf((*MockClient)(nil).BatchGetExperimentAggrResult_), varargs...) +} + +// BatchGetExperimentResult_ mocks base method. +func (m *MockClient) BatchGetExperimentResult_(ctx context.Context, req *expt.BatchGetExperimentResultRequest, callOptions ...callopt.Option) (*expt.BatchGetExperimentResultResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchGetExperimentResult_", varargs...) + ret0, _ := ret[0].(*expt.BatchGetExperimentResultResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchGetExperimentResult_ indicates an expected call of BatchGetExperimentResult_. +func (mr *MockClientMockRecorder) BatchGetExperimentResult_(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetExperimentResult_", reflect.TypeOf((*MockClient)(nil).BatchGetExperimentResult_), varargs...) +} + +// BatchGetExperimentTemplate mocks base method. +func (m *MockClient) BatchGetExperimentTemplate(ctx context.Context, req *expt.BatchGetExperimentTemplateRequest, callOptions ...callopt.Option) (*expt.BatchGetExperimentTemplateResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchGetExperimentTemplate", varargs...) + ret0, _ := ret[0].(*expt.BatchGetExperimentTemplateResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchGetExperimentTemplate indicates an expected call of BatchGetExperimentTemplate. +func (mr *MockClientMockRecorder) BatchGetExperimentTemplate(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetExperimentTemplate", reflect.TypeOf((*MockClient)(nil).BatchGetExperimentTemplate), varargs...) +} + +// BatchGetExperiments mocks base method. +func (m *MockClient) BatchGetExperiments(ctx context.Context, req *expt.BatchGetExperimentsRequest, callOptions ...callopt.Option) (*expt.BatchGetExperimentsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchGetExperiments", varargs...) + ret0, _ := ret[0].(*expt.BatchGetExperimentsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchGetExperiments indicates an expected call of BatchGetExperiments. +func (mr *MockClientMockRecorder) BatchGetExperiments(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetExperiments", reflect.TypeOf((*MockClient)(nil).BatchGetExperiments), varargs...) +} + +// CalculateExperimentAggrResult_ mocks base method. +func (m *MockClient) CalculateExperimentAggrResult_(ctx context.Context, req *expt.CalculateExperimentAggrResultRequest, callOptions ...callopt.Option) (*expt.CalculateExperimentAggrResultResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CalculateExperimentAggrResult_", varargs...) + ret0, _ := ret[0].(*expt.CalculateExperimentAggrResultResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CalculateExperimentAggrResult_ indicates an expected call of CalculateExperimentAggrResult_. +func (mr *MockClientMockRecorder) CalculateExperimentAggrResult_(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CalculateExperimentAggrResult_", reflect.TypeOf((*MockClient)(nil).CalculateExperimentAggrResult_), varargs...) +} + +// CheckExperimentName mocks base method. +func (m *MockClient) CheckExperimentName(ctx context.Context, req *expt.CheckExperimentNameRequest, callOptions ...callopt.Option) (*expt.CheckExperimentNameResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CheckExperimentName", varargs...) + ret0, _ := ret[0].(*expt.CheckExperimentNameResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CheckExperimentName indicates an expected call of CheckExperimentName. +func (mr *MockClientMockRecorder) CheckExperimentName(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckExperimentName", reflect.TypeOf((*MockClient)(nil).CheckExperimentName), varargs...) +} + +// CheckExperimentTemplateName mocks base method. +func (m *MockClient) CheckExperimentTemplateName(ctx context.Context, req *expt.CheckExperimentTemplateNameRequest, callOptions ...callopt.Option) (*expt.CheckExperimentTemplateNameResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CheckExperimentTemplateName", varargs...) + ret0, _ := ret[0].(*expt.CheckExperimentTemplateNameResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CheckExperimentTemplateName indicates an expected call of CheckExperimentTemplateName. +func (mr *MockClientMockRecorder) CheckExperimentTemplateName(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckExperimentTemplateName", reflect.TypeOf((*MockClient)(nil).CheckExperimentTemplateName), varargs...) +} + +// CloneExperiment mocks base method. +func (m *MockClient) CloneExperiment(ctx context.Context, req *expt.CloneExperimentRequest, callOptions ...callopt.Option) (*expt.CloneExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CloneExperiment", varargs...) + ret0, _ := ret[0].(*expt.CloneExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CloneExperiment indicates an expected call of CloneExperiment. +func (mr *MockClientMockRecorder) CloneExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloneExperiment", reflect.TypeOf((*MockClient)(nil).CloneExperiment), varargs...) +} + +// CreateAnnotateRecord mocks base method. +func (m *MockClient) CreateAnnotateRecord(ctx context.Context, req *expt.CreateAnnotateRecordReq, callOptions ...callopt.Option) (*expt.CreateAnnotateRecordResp, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CreateAnnotateRecord", varargs...) + ret0, _ := ret[0].(*expt.CreateAnnotateRecordResp) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateAnnotateRecord indicates an expected call of CreateAnnotateRecord. +func (mr *MockClientMockRecorder) CreateAnnotateRecord(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateAnnotateRecord", reflect.TypeOf((*MockClient)(nil).CreateAnnotateRecord), varargs...) +} + +// CreateExperiment mocks base method. +func (m *MockClient) CreateExperiment(ctx context.Context, req *expt.CreateExperimentRequest, callOptions ...callopt.Option) (*expt.CreateExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CreateExperiment", varargs...) + ret0, _ := ret[0].(*expt.CreateExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateExperiment indicates an expected call of CreateExperiment. +func (mr *MockClientMockRecorder) CreateExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateExperiment", reflect.TypeOf((*MockClient)(nil).CreateExperiment), varargs...) +} + +// CreateExperimentTemplate mocks base method. +func (m *MockClient) CreateExperimentTemplate(ctx context.Context, req *expt.CreateExperimentTemplateRequest, callOptions ...callopt.Option) (*expt.CreateExperimentTemplateResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CreateExperimentTemplate", varargs...) + ret0, _ := ret[0].(*expt.CreateExperimentTemplateResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateExperimentTemplate indicates an expected call of CreateExperimentTemplate. +func (mr *MockClientMockRecorder) CreateExperimentTemplate(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateExperimentTemplate", reflect.TypeOf((*MockClient)(nil).CreateExperimentTemplate), varargs...) +} + +// DeleteAnnotationTag mocks base method. +func (m *MockClient) DeleteAnnotationTag(ctx context.Context, req *expt.DeleteAnnotationTagReq, callOptions ...callopt.Option) (*expt.DeleteAnnotationTagResp, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "DeleteAnnotationTag", varargs...) + ret0, _ := ret[0].(*expt.DeleteAnnotationTagResp) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DeleteAnnotationTag indicates an expected call of DeleteAnnotationTag. +func (mr *MockClientMockRecorder) DeleteAnnotationTag(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAnnotationTag", reflect.TypeOf((*MockClient)(nil).DeleteAnnotationTag), varargs...) +} + +// DeleteExperiment mocks base method. +func (m *MockClient) DeleteExperiment(ctx context.Context, req *expt.DeleteExperimentRequest, callOptions ...callopt.Option) (*expt.DeleteExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "DeleteExperiment", varargs...) + ret0, _ := ret[0].(*expt.DeleteExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DeleteExperiment indicates an expected call of DeleteExperiment. +func (mr *MockClientMockRecorder) DeleteExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteExperiment", reflect.TypeOf((*MockClient)(nil).DeleteExperiment), varargs...) +} + +// DeleteExperimentTemplate mocks base method. +func (m *MockClient) DeleteExperimentTemplate(ctx context.Context, req *expt.DeleteExperimentTemplateRequest, callOptions ...callopt.Option) (*expt.DeleteExperimentTemplateResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "DeleteExperimentTemplate", varargs...) + ret0, _ := ret[0].(*expt.DeleteExperimentTemplateResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DeleteExperimentTemplate indicates an expected call of DeleteExperimentTemplate. +func (mr *MockClientMockRecorder) DeleteExperimentTemplate(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteExperimentTemplate", reflect.TypeOf((*MockClient)(nil).DeleteExperimentTemplate), varargs...) +} + +// DeleteExptInsightAnalysisRecord mocks base method. +func (m *MockClient) DeleteExptInsightAnalysisRecord(ctx context.Context, req *expt.DeleteExptInsightAnalysisRecordRequest, callOptions ...callopt.Option) (*expt.DeleteExptInsightAnalysisRecordResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "DeleteExptInsightAnalysisRecord", varargs...) + ret0, _ := ret[0].(*expt.DeleteExptInsightAnalysisRecordResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DeleteExptInsightAnalysisRecord indicates an expected call of DeleteExptInsightAnalysisRecord. +func (mr *MockClientMockRecorder) DeleteExptInsightAnalysisRecord(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteExptInsightAnalysisRecord", reflect.TypeOf((*MockClient)(nil).DeleteExptInsightAnalysisRecord), varargs...) +} + +// ExportExptResult_ mocks base method. +func (m *MockClient) ExportExptResult_(ctx context.Context, req *expt.ExportExptResultRequest, callOptions ...callopt.Option) (*expt.ExportExptResultResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ExportExptResult_", varargs...) + ret0, _ := ret[0].(*expt.ExportExptResultResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ExportExptResult_ indicates an expected call of ExportExptResult_. +func (mr *MockClientMockRecorder) ExportExptResult_(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExportExptResult_", reflect.TypeOf((*MockClient)(nil).ExportExptResult_), varargs...) +} + +// FeedbackExptInsightAnalysisReport mocks base method. +func (m *MockClient) FeedbackExptInsightAnalysisReport(ctx context.Context, req *expt.FeedbackExptInsightAnalysisReportRequest, callOptions ...callopt.Option) (*expt.FeedbackExptInsightAnalysisReportResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "FeedbackExptInsightAnalysisReport", varargs...) + ret0, _ := ret[0].(*expt.FeedbackExptInsightAnalysisReportResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// FeedbackExptInsightAnalysisReport indicates an expected call of FeedbackExptInsightAnalysisReport. +func (mr *MockClientMockRecorder) FeedbackExptInsightAnalysisReport(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FeedbackExptInsightAnalysisReport", reflect.TypeOf((*MockClient)(nil).FeedbackExptInsightAnalysisReport), varargs...) +} + +// FinishExperiment mocks base method. +func (m *MockClient) FinishExperiment(ctx context.Context, req *expt.FinishExperimentRequest, callOptions ...callopt.Option) (*expt.FinishExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "FinishExperiment", varargs...) + ret0, _ := ret[0].(*expt.FinishExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// FinishExperiment indicates an expected call of FinishExperiment. +func (mr *MockClientMockRecorder) FinishExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FinishExperiment", reflect.TypeOf((*MockClient)(nil).FinishExperiment), varargs...) +} + +// GetAnalysisRecordFeedbackVote mocks base method. +func (m *MockClient) GetAnalysisRecordFeedbackVote(ctx context.Context, req *expt.GetAnalysisRecordFeedbackVoteRequest, callOptions ...callopt.Option) (*expt.GetAnalysisRecordFeedbackVoteResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetAnalysisRecordFeedbackVote", varargs...) + ret0, _ := ret[0].(*expt.GetAnalysisRecordFeedbackVoteResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetAnalysisRecordFeedbackVote indicates an expected call of GetAnalysisRecordFeedbackVote. +func (mr *MockClientMockRecorder) GetAnalysisRecordFeedbackVote(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAnalysisRecordFeedbackVote", reflect.TypeOf((*MockClient)(nil).GetAnalysisRecordFeedbackVote), varargs...) +} + +// GetExptInsightAnalysisRecord mocks base method. +func (m *MockClient) GetExptInsightAnalysisRecord(ctx context.Context, req *expt.GetExptInsightAnalysisRecordRequest, callOptions ...callopt.Option) (*expt.GetExptInsightAnalysisRecordResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetExptInsightAnalysisRecord", varargs...) + ret0, _ := ret[0].(*expt.GetExptInsightAnalysisRecordResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetExptInsightAnalysisRecord indicates an expected call of GetExptInsightAnalysisRecord. +func (mr *MockClientMockRecorder) GetExptInsightAnalysisRecord(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExptInsightAnalysisRecord", reflect.TypeOf((*MockClient)(nil).GetExptInsightAnalysisRecord), varargs...) +} + +// GetExptResultExportRecord mocks base method. +func (m *MockClient) GetExptResultExportRecord(ctx context.Context, req *expt.GetExptResultExportRecordRequest, callOptions ...callopt.Option) (*expt.GetExptResultExportRecordResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetExptResultExportRecord", varargs...) + ret0, _ := ret[0].(*expt.GetExptResultExportRecordResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetExptResultExportRecord indicates an expected call of GetExptResultExportRecord. +func (mr *MockClientMockRecorder) GetExptResultExportRecord(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExptResultExportRecord", reflect.TypeOf((*MockClient)(nil).GetExptResultExportRecord), varargs...) +} + +// InsightAnalysisExperiment mocks base method. +func (m *MockClient) InsightAnalysisExperiment(ctx context.Context, req *expt.InsightAnalysisExperimentRequest, callOptions ...callopt.Option) (*expt.InsightAnalysisExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "InsightAnalysisExperiment", varargs...) + ret0, _ := ret[0].(*expt.InsightAnalysisExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// InsightAnalysisExperiment indicates an expected call of InsightAnalysisExperiment. +func (mr *MockClientMockRecorder) InsightAnalysisExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsightAnalysisExperiment", reflect.TypeOf((*MockClient)(nil).InsightAnalysisExperiment), varargs...) +} + +// InvokeExperiment mocks base method. +func (m *MockClient) InvokeExperiment(ctx context.Context, req *expt.InvokeExperimentRequest, callOptions ...callopt.Option) (*expt.InvokeExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "InvokeExperiment", varargs...) + ret0, _ := ret[0].(*expt.InvokeExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// InvokeExperiment indicates an expected call of InvokeExperiment. +func (mr *MockClientMockRecorder) InvokeExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InvokeExperiment", reflect.TypeOf((*MockClient)(nil).InvokeExperiment), varargs...) +} + +// KillExperiment mocks base method. +func (m *MockClient) KillExperiment(ctx context.Context, req *expt.KillExperimentRequest, callOptions ...callopt.Option) (*expt.KillExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "KillExperiment", varargs...) + ret0, _ := ret[0].(*expt.KillExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// KillExperiment indicates an expected call of KillExperiment. +func (mr *MockClientMockRecorder) KillExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "KillExperiment", reflect.TypeOf((*MockClient)(nil).KillExperiment), varargs...) +} + +// ListExperimentStats mocks base method. +func (m *MockClient) ListExperimentStats(ctx context.Context, req *expt.ListExperimentStatsRequest, callOptions ...callopt.Option) (*expt.ListExperimentStatsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListExperimentStats", varargs...) + ret0, _ := ret[0].(*expt.ListExperimentStatsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListExperimentStats indicates an expected call of ListExperimentStats. +func (mr *MockClientMockRecorder) ListExperimentStats(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListExperimentStats", reflect.TypeOf((*MockClient)(nil).ListExperimentStats), varargs...) +} + +// ListExperimentTemplates mocks base method. +func (m *MockClient) ListExperimentTemplates(ctx context.Context, req *expt.ListExperimentTemplatesRequest, callOptions ...callopt.Option) (*expt.ListExperimentTemplatesResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListExperimentTemplates", varargs...) + ret0, _ := ret[0].(*expt.ListExperimentTemplatesResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListExperimentTemplates indicates an expected call of ListExperimentTemplates. +func (mr *MockClientMockRecorder) ListExperimentTemplates(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListExperimentTemplates", reflect.TypeOf((*MockClient)(nil).ListExperimentTemplates), varargs...) +} + +// ListExperiments mocks base method. +func (m *MockClient) ListExperiments(ctx context.Context, req *expt.ListExperimentsRequest, callOptions ...callopt.Option) (*expt.ListExperimentsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListExperiments", varargs...) + ret0, _ := ret[0].(*expt.ListExperimentsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListExperiments indicates an expected call of ListExperiments. +func (mr *MockClientMockRecorder) ListExperiments(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListExperiments", reflect.TypeOf((*MockClient)(nil).ListExperiments), varargs...) +} + +// ListExptInsightAnalysisComment mocks base method. +func (m *MockClient) ListExptInsightAnalysisComment(ctx context.Context, req *expt.ListExptInsightAnalysisCommentRequest, callOptions ...callopt.Option) (*expt.ListExptInsightAnalysisCommentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListExptInsightAnalysisComment", varargs...) + ret0, _ := ret[0].(*expt.ListExptInsightAnalysisCommentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListExptInsightAnalysisComment indicates an expected call of ListExptInsightAnalysisComment. +func (mr *MockClientMockRecorder) ListExptInsightAnalysisComment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListExptInsightAnalysisComment", reflect.TypeOf((*MockClient)(nil).ListExptInsightAnalysisComment), varargs...) +} + +// ListExptInsightAnalysisRecord mocks base method. +func (m *MockClient) ListExptInsightAnalysisRecord(ctx context.Context, req *expt.ListExptInsightAnalysisRecordRequest, callOptions ...callopt.Option) (*expt.ListExptInsightAnalysisRecordResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListExptInsightAnalysisRecord", varargs...) + ret0, _ := ret[0].(*expt.ListExptInsightAnalysisRecordResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListExptInsightAnalysisRecord indicates an expected call of ListExptInsightAnalysisRecord. +func (mr *MockClientMockRecorder) ListExptInsightAnalysisRecord(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListExptInsightAnalysisRecord", reflect.TypeOf((*MockClient)(nil).ListExptInsightAnalysisRecord), varargs...) +} + +// ListExptResultExportRecord mocks base method. +func (m *MockClient) ListExptResultExportRecord(ctx context.Context, req *expt.ListExptResultExportRecordRequest, callOptions ...callopt.Option) (*expt.ListExptResultExportRecordResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListExptResultExportRecord", varargs...) + ret0, _ := ret[0].(*expt.ListExptResultExportRecordResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListExptResultExportRecord indicates an expected call of ListExptResultExportRecord. +func (mr *MockClientMockRecorder) ListExptResultExportRecord(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListExptResultExportRecord", reflect.TypeOf((*MockClient)(nil).ListExptResultExportRecord), varargs...) +} + +// RetryExperiment mocks base method. +func (m *MockClient) RetryExperiment(ctx context.Context, req *expt.RetryExperimentRequest, callOptions ...callopt.Option) (*expt.RetryExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "RetryExperiment", varargs...) + ret0, _ := ret[0].(*expt.RetryExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// RetryExperiment indicates an expected call of RetryExperiment. +func (mr *MockClientMockRecorder) RetryExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RetryExperiment", reflect.TypeOf((*MockClient)(nil).RetryExperiment), varargs...) +} + +// RunExperiment mocks base method. +func (m *MockClient) RunExperiment(ctx context.Context, req *expt.RunExperimentRequest, callOptions ...callopt.Option) (*expt.RunExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "RunExperiment", varargs...) + ret0, _ := ret[0].(*expt.RunExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// RunExperiment indicates an expected call of RunExperiment. +func (mr *MockClientMockRecorder) RunExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunExperiment", reflect.TypeOf((*MockClient)(nil).RunExperiment), varargs...) +} + +// SubmitExperiment mocks base method. +func (m *MockClient) SubmitExperiment(ctx context.Context, req *expt.SubmitExperimentRequest, callOptions ...callopt.Option) (*expt.SubmitExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "SubmitExperiment", varargs...) + ret0, _ := ret[0].(*expt.SubmitExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// SubmitExperiment indicates an expected call of SubmitExperiment. +func (mr *MockClientMockRecorder) SubmitExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SubmitExperiment", reflect.TypeOf((*MockClient)(nil).SubmitExperiment), varargs...) +} + +// UpdateAnnotateRecord mocks base method. +func (m *MockClient) UpdateAnnotateRecord(ctx context.Context, req *expt.UpdateAnnotateRecordReq, callOptions ...callopt.Option) (*expt.UpdateAnnotateRecordResp, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpdateAnnotateRecord", varargs...) + ret0, _ := ret[0].(*expt.UpdateAnnotateRecordResp) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateAnnotateRecord indicates an expected call of UpdateAnnotateRecord. +func (mr *MockClientMockRecorder) UpdateAnnotateRecord(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAnnotateRecord", reflect.TypeOf((*MockClient)(nil).UpdateAnnotateRecord), varargs...) +} + +// UpdateExperiment mocks base method. +func (m *MockClient) UpdateExperiment(ctx context.Context, req *expt.UpdateExperimentRequest, callOptions ...callopt.Option) (*expt.UpdateExperimentResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpdateExperiment", varargs...) + ret0, _ := ret[0].(*expt.UpdateExperimentResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateExperiment indicates an expected call of UpdateExperiment. +func (mr *MockClientMockRecorder) UpdateExperiment(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateExperiment", reflect.TypeOf((*MockClient)(nil).UpdateExperiment), varargs...) +} + +// UpdateExperimentTemplate mocks base method. +func (m *MockClient) UpdateExperimentTemplate(ctx context.Context, req *expt.UpdateExperimentTemplateRequest, callOptions ...callopt.Option) (*expt.UpdateExperimentTemplateResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpdateExperimentTemplate", varargs...) + ret0, _ := ret[0].(*expt.UpdateExperimentTemplateResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateExperimentTemplate indicates an expected call of UpdateExperimentTemplate. +func (mr *MockClientMockRecorder) UpdateExperimentTemplate(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateExperimentTemplate", reflect.TypeOf((*MockClient)(nil).UpdateExperimentTemplate), varargs...) +} + +// UpdateExperimentTemplateMeta mocks base method. +func (m *MockClient) UpdateExperimentTemplateMeta(ctx context.Context, req *expt.UpdateExperimentTemplateMetaRequest, callOptions ...callopt.Option) (*expt.UpdateExperimentTemplateMetaResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpdateExperimentTemplateMeta", varargs...) + ret0, _ := ret[0].(*expt.UpdateExperimentTemplateMetaResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateExperimentTemplateMeta indicates an expected call of UpdateExperimentTemplateMeta. +func (mr *MockClientMockRecorder) UpdateExperimentTemplateMeta(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateExperimentTemplateMeta", reflect.TypeOf((*MockClient)(nil).UpdateExperimentTemplateMeta), varargs...) +} + +// UpsertExptTurnResultFilter mocks base method. +func (m *MockClient) UpsertExptTurnResultFilter(ctx context.Context, req *expt.UpsertExptTurnResultFilterRequest, callOptions ...callopt.Option) (*expt.UpsertExptTurnResultFilterResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpsertExptTurnResultFilter", varargs...) + ret0, _ := ret[0].(*expt.UpsertExptTurnResultFilterResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpsertExptTurnResultFilter indicates an expected call of UpsertExptTurnResultFilter. +func (mr *MockClientMockRecorder) UpsertExptTurnResultFilter(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertExptTurnResultFilter", reflect.TypeOf((*MockClient)(nil).UpsertExptTurnResultFilter), varargs...) +} diff --git a/idl/thrift/coze/loop/observability/coze.loop.observability.trace.thrift b/idl/thrift/coze/loop/observability/coze.loop.observability.trace.thrift index 2cf8b8efd..b4edbd8f6 100644 --- a/idl/thrift/coze/loop/observability/coze.loop.observability.trace.thrift +++ b/idl/thrift/coze/loop/observability/coze.loop.observability.trace.thrift @@ -10,6 +10,7 @@ include "./domain/annotation.thrift" include "./domain/export_dataset.thrift" include "./domain/task.thrift" include "../trajectory.thrift" +include "./domain/annotation.thrift" struct ListSpansRequest { 1: required i64 workspace_id (api.js_conv='true', go.tag='json:"workspace_id"', api.body="workspace_id") @@ -268,6 +269,22 @@ struct ListAnnotationsResponse { 255: optional base.BaseResp BaseResp } +struct ListWorkspaceAnnotationsRequest { + 1: required i64 workspace_id (api.js_conv='true', go.tag='json:"workspace_id"', api.body="workspace_id", vt.gt="0") + 2: optional common.PlatformType platform_type (api.body="platform_type") + 3: optional annotation.AnnotationType annotation_type(api.body="annotation_type") + 4: optional common.SpanListType span_list_type (api.body="span_list_type") + + 255: optional base.Base Base +} + +struct ListWorkspaceAnnotationsResponse { + 1: required list simple_annotation_list + + 255: optional base.BaseResp BaseResp +} + + struct ExportTracesToDatasetRequest { 1: required i64 workspace_id (api.js_conv="true", go.tag='json:"workspace_id"', api.body="workspace_id", vt.gt="0") 2: required list span_ids (api.body="span_ids", vt.min_size="1", vt.max_size="500") @@ -307,7 +324,7 @@ struct ExportTracesToDatasetResponse { struct PreviewExportTracesToDatasetRequest { 1: required i64 workspace_id (api.js_conv="true", go.tag='json:"workspace_id"', api.body="workspace_id", vt.gt="0") - 2: required list span_ids (api.body="span_ids", vt.min_size="1", vt.max_size="500") + 2: required list span_ids (api.body="span_ids", vt.max_size="500") 3: required dataset.DatasetCategory category (api.body="category") 4: required DatasetConfig config (api.body="config") 5: required i64 start_time (api.js_conv="true", go.tag='json:"start_time"', api.body="start_time") @@ -315,7 +332,8 @@ struct PreviewExportTracesToDatasetRequest { 7: optional common.PlatformType platform_type (api.body="platform_type") 8: required export_dataset.ExportType export_type (api.body="export_type") // 导入方式，不填默认为追加 9: optional list field_mappings (api.body="field_mappings", vt.min_size="1", vt.max_size="100") - + 10: optional filter.SpanFilterFields span_filters (api.body="span_filters") + 11: optional i64 limit (api.js_conv="true", api.body="limit", vt.gt="0") 255: optional base.Base Base (api.none="true") } @@ -421,6 +439,24 @@ struct ListTrajectoryResponse { 255: optional base.BaseResp BaseResp } +struct ListMetadataRequest { + 1: required i64 workspace_id (api.js_conv='true', go.tag='json:"workspace_id"', api.body="workspace_id") + 2: optional common.PlatformType platform_type (api.body="platform_type") + 3: optional common.SpanListType span_list_type (api.body="span_list_type") + + 255: optional base.Base Base +} + +struct MetadataItemInfo { + 1: required string key + 2: required span.MetadataValueType value_type +} +struct ListMetadataResponse { + 1: required list metadataItemList , + + 255: optional base.BaseResp BaseResp +} + service TraceService { ListSpansResponse ListSpans(1: ListSpansRequest req) (api.post = '/api/observability/v1/spans/list') ListPreSpanResponse ListPreSpan(1: ListPreSpanRequest req) (api.post = '/api/observability/v1/spans/pre_list') @@ -437,6 +473,7 @@ service TraceService { UpdateManualAnnotationResponse UpdateManualAnnotation(1: UpdateManualAnnotationRequest req) (api.put = '/api/observability/v1/annotations/:annotation_id') DeleteManualAnnotationResponse DeleteManualAnnotation(1: DeleteManualAnnotationRequest req) (api.delete = '/api/observability/v1/annotations/:annotation_id') ListAnnotationsResponse ListAnnotations(1: ListAnnotationsRequest req) (api.post = '/api/observability/v1/annotations/list') + ListWorkspaceAnnotationsResponse ListWorkspaceAnnotations(1: ListWorkspaceAnnotationsRequest req) (api.post = '/api/observability/v1/annotations/list_by_workspace') ExportTracesToDatasetResponse ExportTracesToDataset(1: ExportTracesToDatasetRequest Req)(api.post = '/api/observability/v1/traces/export_to_dataset') PreviewExportTracesToDatasetResponse PreviewExportTracesToDataset(1: PreviewExportTracesToDatasetRequest Req)(api.post = '/api/observability/v1/traces/preview_export_to_dataset') ChangeEvaluatorScoreResponse ChangeEvaluatorScore(1: ChangeEvaluatorScoreRequest req) (api.post = '/api/observability/v1/traces/change_eval_score') @@ -445,4 +482,5 @@ service TraceService { UpsertTrajectoryConfigResponse UpsertTrajectoryConfig(1: UpsertTrajectoryConfigRequest req) (api.post = '/api/observability/v1/traces/trajectory_config') GetTrajectoryConfigResponse GetTrajectoryConfig(1: GetTrajectoryConfigRequest req) (api.get = '/api/observability/v1/traces/trajectory_config') ListTrajectoryResponse ListTrajectory(1: ListTrajectoryRequest req) (api.post = '/api/observability/v1/traces/trajectory') + ListMetadataResponse ListMetadata(1: ListMetadataRequest req) (api.post = '/api/observability/v1/traces/metadata/list') } diff --git a/idl/thrift/coze/loop/observability/domain/annotation.thrift b/idl/thrift/coze/loop/observability/domain/annotation.thrift index ca54ce19d..75f2ecf79 100644 --- a/idl/thrift/coze/loop/observability/domain/annotation.thrift +++ b/idl/thrift/coze/loop/observability/domain/annotation.thrift @@ -36,6 +36,8 @@ struct AutoEvaluate { 4: optional EvaluatorResult evaluator_result 5: required i64 record_id (api.js_conv='true', go.tag='json:"record_id"') 6: required string task_id + 7: optional i64 expt_id (api.js_conv='true', go.tag='json:"expt_id"') + 8: optional i64 expt_template_id (api.js_conv='true', go.tag='json:"expt_template_id"') } struct ManualFeedback { @@ -67,4 +69,10 @@ struct AnnotationEvaluator { 1: required i64 evaluator_version_id, 2: required string evaluator_name, 3: required string evaluator_version, +} + +struct SimpleAnnotationInfo { + 1: required string key + 2: optional AnnotationType annotation_type + 3: optional string original_key } \ No newline at end of file diff --git a/idl/thrift/coze/loop/observability/domain/filter.thrift b/idl/thrift/coze/loop/observability/domain/filter.thrift index 12837cec8..2cb564a5d 100644 --- a/idl/thrift/coze/loop/observability/domain/filter.thrift +++ b/idl/thrift/coze/loop/observability/domain/filter.thrift @@ -32,6 +32,9 @@ const TaskFieldName TaskFieldName_TaskName = "task_name" const TaskFieldName TaskFieldName_TaskType = "task_type" const TaskFieldName TaskFieldName_SampleRate = "sample_rate" const TaskFieldName TaskFieldName_CreatedBy = "created_by" +const TaskFieldName TaskFieldName_TaskID = "task_id" +const TaskFieldName TaskFieldName_WorkflowID = "workflow_id" +const TaskFieldName TaskFieldName_UpdatedBy = "updated_by" struct FilterFields { diff --git a/idl/thrift/coze/loop/observability/domain/span.thrift b/idl/thrift/coze/loop/observability/domain/span.thrift index f6e61783c..ea5d6ae52 100644 --- a/idl/thrift/coze/loop/observability/domain/span.thrift +++ b/idl/thrift/coze/loop/observability/domain/span.thrift @@ -25,6 +25,13 @@ const SpanType SpanType_VectorRetriever= "vector_retriever" const SpanType SpanType_Agent = "agent" const SpanType SpanType_LLMCall = "LLMCall" +typedef string MetadataValueType (ts.enum="true") +const MetadataValueType ValueType_String = "string" +const MetadataValueType ValueType_Byte = "byte" +const MetadataValueType ValueType_Long = "long" +const MetadataValueType ValueType_Double = "double" +const MetadataValueType ValueType_Bool = "bool" + struct AttrTos { 1: optional string input_data_url 2: optional string output_data_url diff --git a/idl/thrift/coze/loop/observability/domain/task.thrift b/idl/thrift/coze/loop/observability/domain/task.thrift index 67f2cb2e9..4b0e69a67 100644 --- a/idl/thrift/coze/loop/observability/domain/task.thrift +++ b/idl/thrift/coze/loop/observability/domain/task.thrift @@ -3,6 +3,7 @@ namespace go coze.loop.observability.domain.task include "common.thrift" include "filter.thrift" include "export_dataset.thrift" +include "../../data/domain/dataset.thrift" typedef string TimeUnit (ts.enum="true") const TimeUnit TimeUnit_Day = "day" @@ -46,7 +47,8 @@ struct Task { 9: optional RunDetail task_detail // 任务状态详情 10: optional RunDetail backfill_task_detail // 任务历史数据执行详情 11: optional TaskSource task_source // 创建来源 - + 12: optional i64 workflow_id (api.js_conv="true", go.tag='json:"workflow_id"') // 对应工作流 ID + 13: optional list task_runs // task runs 100: optional common.BaseInfo base_info // 基础信息 } @@ -72,11 +74,26 @@ struct EffectiveTime { 2: optional i64 end_at (api.js_conv="true", go.tag='json:"end_at"') // ms timestamp } +struct EvaluationExperimentConfig { + 1: optional i32 item_concurrency_count (api.js_conv="true", go.tag='json:"item_concurrency_count"') + 2: optional i32 item_max_retry_count (api.js_conv="true", go.tag='json:"item_max_retry_count"') + 3: optional string source_target_id (go.tag='json:"source_target_id"') + 4: optional i64 expt_template_id (go.tag='json:"expt_template_id"') + 5: optional string source_target_version (go.tag='json:"source_target_version"') + 6: optional list full_eval_set_field_mappings (go.tag='json:"full_eval_set_field_mappings"') +} +struct SourceInfo { + 1: optional string name + 2: optional string version +} // TaskConfig struct TaskConfig { - 1: optional list auto_evaluate_configs // 配置的评测规则信息 + 1: optional list auto_evaluate_configs // 配置的评测规则信息 evaluator 维度 2: optional list data_reflow_config // 配置的数据回流的数据集信息 + 3: optional EvaluationExperimentConfig evaluation_experiment_config // 评测实验配置 task 维度 + 4: optional list source_info // 数据源信息 + 5: optional bool is_workflow_scheduled // 工作流周期性任务 } struct DataReflowConfig { @@ -84,12 +101,15 @@ struct DataReflowConfig { 2: optional string dataset_name // 数据集名称 3: optional export_dataset.DatasetSchema dataset_schema (vt.not_nil="true") // 数据集列数据schema 4: optional list field_mappings (vt.min_size="1", vt.max_size="100") + 5: optional dataset.DatasetCategory dataset_category // 数据集类型 } struct AutoEvaluateConfig { 1: required i64 evaluator_version_id (api.js_conv="true", go.tag='json:"evaluator_version_id"') 2: required i64 evaluator_id (api.js_conv="true", go.tag='json:"evaluator_id"') 3: required list field_mappings + 4: optional double score_weight (api.js_conv="true", go.tag='json:"score_weight"') + 5: optional string evaluator_version } // RunDetail diff --git a/release/deployment/docker-compose/bootstrap/mysql-init/init-sql/task.sql b/release/deployment/docker-compose/bootstrap/mysql-init/init-sql/task.sql index e68741fa0..51cf677c3 100644 --- a/release/deployment/docker-compose/bootstrap/mysql-init/init-sql/task.sql +++ b/release/deployment/docker-compose/bootstrap/mysql-init/init-sql/task.sql @@ -16,6 +16,7 @@ CREATE TABLE IF NOT EXISTS `task` ( `created_by` varchar(128) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT '创建人', `updated_by` varchar(128) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT '更新人', `task_source` VARCHAR(50) DEFAULT 'user' COMMENT '任务来源', + `workflow_id` bigint unsigned NOT NULL DEFAULT 0 COMMENT '相关工作流ID', PRIMARY KEY (`id`), KEY `idx_space_id_status` (`workspace_id`,`task_status`), KEY `idx_space_id_type` (`workspace_id`,`task_type`) diff --git a/release/deployment/docker-compose/bootstrap/mysql-init/patch-sql/task_alter.sql.sql b/release/deployment/docker-compose/bootstrap/mysql-init/patch-sql/task_alter.sql.sql new file mode 100644 index 000000000..87fd87d54 --- /dev/null +++ b/release/deployment/docker-compose/bootstrap/mysql-init/patch-sql/task_alter.sql.sql @@ -0,0 +1 @@ +ALTER TABLE task ADD COLUMN `workflow_id` bigint unsigned NOT NULL DEFAULT '0' COMMENT '相关工作流 ID'; \ No newline at end of file diff --git a/release/deployment/helm-chart/charts/app/bootstrap/init/mysql/init-sql/task.sql b/release/deployment/helm-chart/charts/app/bootstrap/init/mysql/init-sql/task.sql index e68741fa0..51cf677c3 100644 --- a/release/deployment/helm-chart/charts/app/bootstrap/init/mysql/init-sql/task.sql +++ b/release/deployment/helm-chart/charts/app/bootstrap/init/mysql/init-sql/task.sql @@ -16,6 +16,7 @@ CREATE TABLE IF NOT EXISTS `task` ( `created_by` varchar(128) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT '创建人', `updated_by` varchar(128) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT '更新人', `task_source` VARCHAR(50) DEFAULT 'user' COMMENT '任务来源', + `workflow_id` bigint unsigned NOT NULL DEFAULT 0 COMMENT '相关工作流ID', PRIMARY KEY (`id`), KEY `idx_space_id_status` (`workspace_id`,`task_status`), KEY `idx_space_id_type` (`workspace_id`,`task_type`) diff --git a/release/deployment/helm-chart/charts/app/bootstrap/init/mysql/init-sql/task_alter.sql.sql b/release/deployment/helm-chart/charts/app/bootstrap/init/mysql/init-sql/task_alter.sql.sql new file mode 100644 index 000000000..87fd87d54 --- /dev/null +++ b/release/deployment/helm-chart/charts/app/bootstrap/init/mysql/init-sql/task_alter.sql.sql @@ -0,0 +1 @@ +ALTER TABLE task ADD COLUMN `workflow_id` bigint unsigned NOT NULL DEFAULT '0' COMMENT '相关工作流 ID'; \ No newline at end of file