Add scanner/valuer interface to babyjub.SignatureComp

babyjub/eddsa.go

@@ -177,6 +177,24 @@ func (s *SignatureComp) Decompress() (*Signature, error) {
 	return new(Signature).Decompress(*s)
 }
 
+// Scan implements Scanner for database/sql.
+func (s *SignatureComp) Scan(src interface{}) error {
+	srcB, ok := src.([]byte)
+	if !ok {
+		return fmt.Errorf("can't scan %T into Signature", src)
+	}
+	if len(srcB) != 64 {
+		return fmt.Errorf("can't scan []byte of len %d into Signature, want %d", len(srcB), 64)
+	}
+	copy(s[:], srcB[:])
+	return nil
+}
+
+// Value implements valuer for database/sql.
+func (s SignatureComp) Value() (driver.Value, error) {
+	return s[:], nil
+}
+
 // Scan implements Scanner for database/sql.
 func (s *Signature) Scan(src interface{}) error {
 	srcB, ok := src.([]byte)

babyjub/eddsa_test.go

@@ -135,6 +135,17 @@ func TestCompressDecompress(t *testing.T) {
 	}
 }
 
+func TestSignatureCompScannerValuer(t *testing.T) {
+	privK := NewRandPrivKey()
+	var value driver.Valuer //nolint:gosimple this is done to ensure interface compability
+	value = privK.SignPoseidon(big.NewInt(674238462)).Compress()
+	scan := privK.SignPoseidon(big.NewInt(1)).Compress()
+	fromDB, err := value.Value()
+	assert.Nil(t, err)
+	assert.Nil(t, scan.Scan(fromDB))
+	assert.Equal(t, value, scan)
+}
+
 func TestSignatureScannerValuer(t *testing.T) {
 	privK := NewRandPrivKey()
 	var value driver.Valuer
@@ -142,8 +153,8 @@ func TestSignatureScannerValuer(t *testing.T) {
 	value = privK.SignPoseidon(big.NewInt(674238462))
 	scan = privK.SignPoseidon(big.NewInt(1))
 	fromDB, err := value.Value()
-	assert.NoError(t, err)
-	assert.NoError(t, scan.Scan(fromDB))
+	assert.Nil(t, err)
+	assert.Nil(t, scan.Scan(fromDB))
 	assert.Equal(t, value, scan)
 }
 
@@ -157,8 +168,8 @@ func TestPubKeyScannerValuer(t *testing.T) {
 	value = pubKValue
 	scan = pubKScan
 	fromDB, err := value.Value()
-	assert.NoError(t, err)
-	assert.NoError(t, scan.Scan(fromDB))
+	assert.Nil(t, err)
+	assert.Nil(t, scan.Scan(fromDB))
 	assert.Equal(t, value, scan)
 }