code.vegaprotocol.io/vega@v0.79.0/datanode/gateway/graphql/marshallers/timestamp.go (about)

     1  // Copyright (C) 2023 Gobalsky Labs Limited
     2  //
     3  // This program is free software: you can redistribute it and/or modify
     4  // it under the terms of the GNU Affero General Public License as
     5  // published by the Free Software Foundation, either version 3 of the
     6  // License, or (at your option) any later version.
     7  //
     8  // This program is distributed in the hope that it will be useful,
     9  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    10  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    11  // GNU Affero General Public License for more details.
    12  //
    13  // You should have received a copy of the GNU Affero General Public License
    14  // along with this program.  If not, see <http://www.gnu.org/licenses/>.
    15  
    16  package marshallers
    17  
    18  import (
    19  	"errors"
    20  	"io"
    21  	"strconv"
    22  
    23  	"code.vegaprotocol.io/vega/datanode/vegatime"
    24  
    25  	"github.com/99designs/gqlgen/graphql"
    26  )
    27  
    28  func MarshalTimestamp(t int64) graphql.Marshaler {
    29  	return graphql.WriterFunc(func(w io.Writer) {
    30  		// special case for 0 timestamps, returns null
    31  		if t == 0 {
    32  			io.WriteString(w, "null")
    33  			return
    34  		}
    35  		io.WriteString(w, strconv.Quote(vegatime.Format(vegatime.UnixNano(t))))
    36  	})
    37  }
    38  
    39  func UnmarshalTimestamp(v interface{}) (int64, error) {
    40  	s, ok := v.(string)
    41  	if !ok {
    42  		return 0, errors.New("expected timestamp to be a string")
    43  	}
    44  
    45  	t, err := vegatime.Parse(s)
    46  	if err != nil {
    47  		return 0, err
    48  	}
    49  
    50  	return t.UnixNano(), nil
    51  }