From ad64a291455282819feddd33c9de4aea79b260e2 Mon Sep 17 00:00:00 2001 From: John Guo Date: Tue, 7 Dec 2021 21:03:06 +0800 Subject: [PATCH] add MustSet/MusetRemove for package genv --- os/genv/genv_must.go | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 os/genv/genv_must.go diff --git a/os/genv/genv_must.go b/os/genv/genv_must.go new file mode 100644 index 000000000..3ee5539f2 --- /dev/null +++ b/os/genv/genv_must.go @@ -0,0 +1,21 @@ +// Copyright GoFrame Author(https://goframe.org). All Rights Reserved. +// +// This Source Code Form is subject to the terms of the MIT License. +// If a copy of the MIT was not distributed with this file, +// You can obtain one at https://github.com/gogf/gf. + +package genv + +// MustSet performs as Set, but it panics if any error occurs. +func MustSet(key, value string) { + if err := Set(key, value); err != nil { + panic(err) + } +} + +// MustRemove performs as Remove, but it panics if any error occurs. +func MustRemove(key ...string) { + if err := Remove(key...); err != nil { + panic(err) + } +}