ASqlModel.h

/*
 * AUI Framework - Declarative UI toolkit for modern C++20
 * Copyright (C) 2020-2025 Alex2772 and Contributors
 *
 * SPDX-License-Identifier: MPL-2.0
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
 
#pragma once
 
#include "AModelMeta.h"
#include <AUI/Data/ASqlBuilder.h>
 
#include <utility>

template<typename Model>
struct ASqlModel {
   class NoSuchRowException: public AException {};

   class TooManyRowsException: public AException {};
 
 
   typedef AModelMeta<Model> Meta;
   id_t id = 0;

   void save() {
       if (id == 0) {
           id = table(Meta::getSqlTable()).insertORM((Model&)*this);
       } else {
           table(Meta::getSqlTable()).updateORM((Model&)*this);
       }
   }

   void remove() {
       AUI_ASSERT(id != 0);
       table(Meta::getSqlTable()).removeORM((Model&)*this);
   }
 
   class IncompleteSelectRequest {
       friend struct ASqlModel<Model>;
   private:
       AString mSql;
       AString mWhereExpr;
       AVector<AVariant> mWhereParams;
 
       IncompleteSelectRequest(AString sql, const ASqlBuilder::WhereStatement::WhereExpr& expression):
           mSql(std::move(sql)) {
           where(expression);
       }
 
   public:
       IncompleteSelectRequest(const IncompleteSelectRequest&) = delete;
       ~IncompleteSelectRequest() = default;
 
       IncompleteSelectRequest& where(const ASqlBuilder::WhereStatement::WhereExpr& w) {
           auto[exprString, whereParams] = ASqlBuilder::WhereStatement::WhereExpr::unpack(w);
           if (!exprString.empty()) {
               if (mWhereExpr.empty()) {
                   mWhereExpr = "WHERE " + exprString;
                   mWhereParams = std::move(whereParams);
               } else {
                   mWhereExpr += "AND " + exprString;
                   mWhereParams.insertAll(whereParams);
               }
           }
           return *this;
       }

       AVector<Model> get() {
           auto idField = AField<ASqlModel<Model>>::make(&ASqlModel<Model>::id);
           AVector<Model> result;
           result.reserve(0x100);
 
           mSql += " ";
           mSql += mWhereExpr;
           auto dbResult = Autumn::get<ASqlDatabase>()->query(mSql, mWhereParams);
 
           AVector<size_t> sqlColumnToModelFieldIndexMapping;
           AVector<_<AField<Model>>> fields;
           fields << AModelMeta<Model>::getFields().valueVector();
 
           for (auto& row : dbResult) {
               Model m;
               idField->set(m, row->getValue(0));
               for (size_t columnIndex = 1; columnIndex < dbResult->getColumns().size(); ++columnIndex) {
                   fields[columnIndex - 1]->set(m, row->getValue(columnIndex));
               }
               result << std::move(m);
           }
 
           return result;
       }

       inline Model first() {
           auto result = get();
           if (result.size() == 0)
               throw NoSuchRowException();
           if (result.size() != 1)
               throw TooManyRowsException();
           return result.first();
       }
   };
   static _<IncompleteSelectRequest> where(const ASqlBuilder::WhereStatement::WhereExpr& expression) {
       AStringVector columnNames;
       columnNames << "id";
       columnNames << Meta::getFields().keyVector();
       return aui::ptr::manage(new IncompleteSelectRequest("SELECT " + (columnNames.empty() ? '*'
           : columnNames.join(',')) + " FROM " + AModelMeta<Model>::getSqlTable(), expression));
   }

   static Model byId(id_t id) {
       AStringVector columns;
       columns << "id";
       columns << Meta::getFields().keyVector();
       auto result = table(Meta::getSqlTable()).select(columns).where(col("id") == id)
               .template as<Model>();
 
       if (result.empty())
           throw NoSuchRowException();
       AUI_ASSERT(result.size() == 1);
       return result.first();
   }
 
   static _<IncompleteSelectRequest> all() {
       AStringVector columnNames;
       columnNames << "id";
       columnNames << Meta::getFields().keyVector();
       return aui::ptr::manage(new IncompleteSelectRequest("SELECT " + (columnNames.empty() ? '*'
           : columnNames.join(',')) + " FROM " + AModelMeta<Model>::getSqlTable(), {}));
   }
 

   template<typename ... Args>
   static Model make(Args&&... args) {
       Model m = {0, std::forward<Args>(args)...};
       m.toJson();
       return m;
   }
 

   static AString getIdColumnNameInOtherTables() {
       AString tableName = AModelMeta<Model>::getSqlTable();
       if (tableName.endsWith("s")) {
           tableName = tableName.substr(0, tableName.length() - 1);
       }
       tableName += "_id";
       return tableName;
   }
 
protected:    /* ORM RELATIONSHIP */

   template<typename Other>
   _<typename Other::IncompleteSelectRequest> hasMany() {
       const AString columnName = getIdColumnNameInOtherTables();
       return Other::where(col(columnName) == id);
   }
 
   template<typename Other>
   Other belongsTo(id_t desiredId) {
       return Other::where(col("id") == desiredId)->first();
   }
};
 

© 2025 AUI Framework Contributors. Documentation contributions included herein are the copyrights of their respective owners. The documentation provided herein is licensed under the terms of the GNU Free Documentation License version 1.3 as published by the Free Software Foundation. All trademarks are property of their respective owners.