mirror of
https://github.com/ultravioletrs/cocos.git
synced 2026-08-07 07:14:50 +00:00
Sammy Kerata Oina
ee7159a406
* Refactor RunRequest to use structured Computation The protobuf and associated service implementations for the RunRequest message were refactored to replace the raw Computation byte slice with a structured ComputationReq object. This allows clearer and more type-safe manipulation of computation requests. The grpc, http, and agent service layers were updated to build and parse ComputationReq accordingly. The ComputationReq structure includes details like IDs, names, time stamps, and metadata, forming a well-defined contract for computation tasks. This change aligns with efforts to standardize request formats and improve clarity in inter-service communication. It impacts all systems interfacing with the RunRequest service and thus requires coordinated updates to the entire stack. Signed-off-by: SammyOina <sammyoina@gmail.com> * Initialize metadata maps and handle nil values Improved the robustness of metadata handling in gRPC endpoints and SDK by initializing metadata maps and explicitly checking for nil values before converting them. This ensures that both the agent's gRPC endpoint and the SDK properly handle cases where metadata fields may be uninitialized or contain nil values, preventing potential null pointer exceptions. Signed-off-by: SammyOina <sammyoina@gmail.com> * Refactor computation request handling Refactored the endpoint to construct Computation object from gRPC request, incorporating structpb for metadata handling and timestamppb for StartTime and EndTime fields. The management service and API requests are also updated to align with these changes, improving type safety and ensuring data is correctly marshalled when making service calls. Resolves data marshalling issues for computation requests. Signed-off-by: SammyOina <sammyoina@gmail.com> * use singular Signed-off-by: SammyOina <sammyoina@gmail.com> * remove unuse fields Signed-off-by: SammyOina <sammyoina@gmail.com> * remove unused fields Signed-off-by: SammyOina <sammyoina@gmail.com> --------- Signed-off-by: SammyOina <sammyoina@gmail.com>
74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
// Copyright (c) Ultraviolet
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
package grpc
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/ultravioletrs/cocos/agent"
|
|
)
|
|
|
|
type runReq struct {
|
|
Computation *agent.ComputationReq
|
|
}
|
|
|
|
func (req *runReq) validate() error {
|
|
return nil
|
|
}
|
|
|
|
type algoReq struct {
|
|
Algorithm []byte `protobuf:"bytes,1,opt,name=algorithm,proto3" json:"algorithm,omitempty"`
|
|
Provider string `protobuf:"bytes,2,opt,name=provider,proto3" json:"provider,omitempty"`
|
|
Id string `protobuf:"bytes,3,opt,name=id,proto3" json:"id,omitempty"`
|
|
}
|
|
|
|
func (req algoReq) validate() error {
|
|
if len(req.Algorithm) == 0 {
|
|
return errors.New("algorithm binary is required")
|
|
}
|
|
if req.Id == "" {
|
|
return errors.New("malformed entity")
|
|
}
|
|
if req.Provider == "" {
|
|
return errors.New("malformed entity")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type dataReq struct {
|
|
Dataset []byte `protobuf:"bytes,1,opt,name=dataset,proto3" json:"dataset,omitempty"`
|
|
Provider string `protobuf:"bytes,2,opt,name=provider,proto3" json:"provider,omitempty"`
|
|
Id string `protobuf:"bytes,3,opt,name=id,proto3" json:"id,omitempty"`
|
|
}
|
|
|
|
func (req dataReq) validate() error {
|
|
if len(req.Dataset) == 0 {
|
|
return errors.New("dataset CSV file is required")
|
|
}
|
|
if req.Id == "" {
|
|
return errors.New("malformed entity")
|
|
}
|
|
if req.Provider == "" {
|
|
return errors.New("malformed entity")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type resultReq struct {
|
|
Consumer string `protobuf:"bytes,1,opt,name=consumer,proto3" json:"consumer,omitempty"`
|
|
}
|
|
|
|
func (req resultReq) validate() error {
|
|
// No request parameters to validate, so no validation logic needed
|
|
return nil
|
|
}
|
|
|
|
type attestationReq struct {
|
|
// No request parameters needed for retrieving attestation output
|
|
}
|
|
|
|
func (req attestationReq) validate() error {
|
|
// No request parameters to validate, so no validation logic needed
|
|
return nil
|
|
}
|