github.com/condensat/bank-core@v0.1.0/database/query/migrate.go (about)

     1  // Copyright 2020 Condensat Tech. All rights reserved.
     2  // Use of this source code is governed by a MIT
     3  // license that can be found in the LICENSE file.
     4  
     5  package query
     6  
     7  import (
     8  	"github.com/condensat/bank-core/database"
     9  	"github.com/condensat/bank-core/database/model"
    10  )
    11  
    12  func UserModel() []database.Model {
    13  	return []database.Model{
    14  		database.Model(new(model.User)),
    15  		database.Model(new(model.UserRole)),
    16  	}
    17  }
    18  
    19  func AccountModel() []database.Model {
    20  	return append(UserModel(), []database.Model{
    21  		database.Model(new(model.Currency)),
    22  		database.Model(new(model.Account)),
    23  	}...)
    24  }
    25  
    26  func AccountStateModel() []database.Model {
    27  	return append(AccountModel(), new(model.AccountState))
    28  }
    29  
    30  func AccountOperationModel() []database.Model {
    31  	return append(AccountStateModel(), new(model.AccountOperation))
    32  }
    33  
    34  func CurrencyModel() []database.Model {
    35  	return []database.Model{
    36  		database.Model(new(model.Currency)),
    37  		database.Model(new(model.CurrencyRate)),
    38  	}
    39  }
    40  
    41  func CryptoAddressModel() []database.Model {
    42  	return []database.Model{
    43  		database.Model(new(model.CryptoAddress)),
    44  	}
    45  }
    46  
    47  func SsmAddressModel() []database.Model {
    48  	return []database.Model{
    49  		database.Model(new(model.SsmAddress)),
    50  		database.Model(new(model.SsmAddressInfo)),
    51  		database.Model(new(model.SsmAddressState)),
    52  	}
    53  }
    54  
    55  func OperationInfoModel() []database.Model {
    56  	return append(CryptoAddressModel(), []database.Model{
    57  		database.Model(new(model.OperationInfo)),
    58  		database.Model(new(model.OperationStatus)),
    59  	}...)
    60  }
    61  
    62  func AssetModel() []database.Model {
    63  	return []database.Model{
    64  		database.Model(new(model.Asset)),
    65  		database.Model(new(model.AssetInfo)),
    66  		database.Model(new(model.AssetIcon)),
    67  	}
    68  }
    69  
    70  func SwapModel() []database.Model {
    71  	return []database.Model{
    72  		database.Model(new(model.Swap)),
    73  		database.Model(new(model.SwapInfo)),
    74  	}
    75  }
    76  
    77  func WithdrawModel() []database.Model {
    78  	return append(AccountOperationModel(), []database.Model{
    79  		database.Model(new(model.Withdraw)),
    80  		database.Model(new(model.WithdrawInfo)),
    81  		database.Model(new(model.WithdrawTarget)),
    82  		database.Model(new(model.Fee)),
    83  		database.Model(new(model.FeeInfo)),
    84  		database.Model(new(model.Batch)),
    85  		database.Model(new(model.BatchInfo)),
    86  		database.Model(new(model.BatchWithdraw)),
    87  	}...)
    88  }
    89  
    90  func FeeModel() []database.Model {
    91  	return []database.Model{
    92  		database.Model(new(model.Fee)),
    93  		database.Model(new(model.FeeInfo)),
    94  	}
    95  }