// SaveVendor will create a vendor if the ID field is not initialized
// if ID field is initialized, it will update the conresponding vendor.
func (r *Registry) SaveVendor(vendor *models.Vendor, reply *models.Vendor) error {
db, err := getDB()
if err != nil {
return err
}
err = db.Save(vendor).Error
if err != nil {
return err
}
key, err := r.keygen.GenRandomKey(int64(vendor.ID))
if err != nil {
return err
}
vendor.VendorKey = key
err = db.Save(vendor).Error
if err != nil {
return err
}
reply.ID = vendor.ID
reply.VendorName = vendor.VendorName
reply.VendorDescription = vendor.VendorDescription
reply.VendorKey = vendor.VendorKey
reply.CreatedAt = vendor.CreatedAt
reply.UpdatedAt = vendor.UpdatedAt
return nil
}
func addVendor() error {
args := models.Vendor{}
reader := bufio.NewReader(os.Stdin)
fmt.Printf("vendor name: ")
name, err := reader.ReadString('\n')
if err != nil {
return err
}
args.VendorName = strings.Replace(name, "\n", "", -1)
fmt.Printf("vendor description: ")
desc, err := reader.ReadString('\n')
if err != nil {
return err
}
args.VendorDescription = strings.Replace(desc, "\n", "", -1)
reply := &models.Vendor{}
err = server.RPCCallByName("registry", "Registry.SaveVendor", &args, reply)
if err != nil {
return err
}
fmt.Println("=======> vendor created successfully:")
printStruct(reply)
fmt.Println("=======")
return nil
}
func setVendor(target *models.Vendor, src *models.Vendor) {
target.ID = src.ID
target.VendorName = src.VendorName
target.VendorDescription = src.VendorDescription
target.VendorKey = src.VendorKey
target.CreatedAt = src.CreatedAt
target.UpdatedAt = src.UpdatedAt
}