proto.go 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  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. package shared
  15. import (
  16. "io/ioutil"
  17. "os"
  18. "google.golang.org/protobuf/proto"
  19. )
  20. // Save takes a protobuf message, marshals to an array of bytes
  21. // and is then saved to a file.
  22. func Save(pb proto.Message, filepath string) (err error) {
  23. data, err := proto.Marshal(pb)
  24. if err != nil {
  25. return err
  26. }
  27. tempFilepath := filepath + ".tmp"
  28. if err := ioutil.WriteFile(tempFilepath, []byte(data), 0644 /* rw-r--r-- */); err != nil {
  29. return err
  30. }
  31. if err := os.Rename(tempFilepath, filepath); err != nil {
  32. return err
  33. }
  34. return nil
  35. }