1// Copyright 2021 Google Inc. All rights reserved. 2// 3// Licensed under the Apache License, Version 2.0 (the "License"); 4// you may not use this file except in compliance with the License. 5// You may obtain a copy of the License at 6// 7// http://www.apache.org/licenses/LICENSE-2.0 8// 9// Unless required by applicable law or agreed to in writing, software 10// distributed under the License is distributed on an "AS IS" BASIS, 11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12// See the License for the specific language governing permissions and 13// limitations under the License. 14 15package shared 16 17import ( 18 "io/ioutil" 19 "os" 20 21 "google.golang.org/protobuf/proto" 22) 23 24// Save takes a protobuf message, marshals to an array of bytes 25// and is then saved to a file. 26func Save(pb proto.Message, filepath string) (err error) { 27 data, err := proto.Marshal(pb) 28 if err != nil { 29 return err 30 } 31 tempFilepath := filepath + ".tmp" 32 if err := ioutil.WriteFile(tempFilepath, []byte(data), 0644 /* rw-r--r-- */); err != nil { 33 return err 34 } 35 36 if err := os.Rename(tempFilepath, filepath); err != nil { 37 return err 38 } 39 40 return nil 41} 42