use defer

1 files changed, 9 insertions, 2 deletions

diff --git a/go/main.go b/go/main.go
index 242aa62..f4ee030 100644
--- a/go/main.go
+++ b/go/main.go
@@ -13,18 +13,27 @@ func main() {
 	fmt.Println("init")
 	C.blsInit()
 
+	id := C.blsIdCreate()
+	defer C.blsIdDestroy(id)
+
+	C.blsIdSet(id, (*C.uint64_t)(unsafe.Pointer(&[]uint64{1, 2, 3, 4}[0])))
+	C.blsIdPut(id)
+
 	fmt.Println("create secret key")
 	sec := C.blsSecretKeyCreate()
+	defer C.blsSecretKeyDestroy(sec)
 	C.blsSecretKeyInit(sec)
 	C.blsSecretKeyPut(sec)
 
 	fmt.Println("create public key")
 	pub := C.blsPublicKeyCreate()
+	defer C.blsPublicKeyDestroy(pub)
 	C.blsSecretKeyGetPublicKey(sec, pub)
 
 	C.blsPublicKeyPut(pub)
 
 	sign := C.blsSignCreate()
+	defer C.blsSignDestroy(sign)
 
 	msg := []byte("Hello bls")
 	fmt.Println("sign message")
@@ -34,6 +43,4 @@ func main() {
 
 	fmt.Println("verify:", C.blsSignVerify(sign, pub, (*C.char)(unsafe.Pointer(&msg[0])), C.size_t(len(msg))))
 
-	C.blsPublicKeyDestroy(pub)
-	C.blsSecretKeyDestroy(sec)
 }