private <T extends ServiceResponse> void invokeOperation( final OperationRequest request, OperationParameterValueProvider parameters, T response) throws ODataApplicationException, ODataLibraryException { checkExpand(request.getUriInfo().asUriInfoResource()); final ODataSQLBuilder visitor = new ODataSQLBuilder( odata, getClient().getMetadataStore(), this.prepared, true, request.getODataRequest().getRawBaseUri(), this.serviceMetadata, this.nameGenerator); visitor.setOperationParameterValueProvider(parameters); visitor.visit(request.getUriInfo()); final OperationResponseImpl queryResponse; try { if (visitor.getContext() instanceof NoDocumentNode) { NoDocumentNode cdn = (NoDocumentNode) visitor.getContext(); ProcedureReturn procReturn = cdn.getProcedureReturn(); queryResponse = new OperationResponseImpl(procReturn); getClient().executeCall(cdn.getQuery(), visitor.getParameters(), procReturn, queryResponse); } else { Query query = visitor.selectQuery(); queryResponse = (OperationResponseImpl) executeQuery(request, request.isCountRequest(), visitor, query); } } catch (Throwable e) { throw new ODataApplicationException( e.getMessage(), HttpStatusCode.INTERNAL_SERVER_ERROR.getStatusCode(), Locale.getDefault(), e); } /* try { MetadataStore store = getClient().getMetadataStore(); ProcedureSQLBuilder builder = new ProcedureSQLBuilder(store.getSchema(schemaName), request); ProcedureReturn procedureReturn = builder.getReturn(); result = new OperationResponseImpl(procedureReturn); getClient().executeCall(builder.buildProcedureSQL(), builder.getSqlParameters(), procedureReturn, result); } catch (SQLException e) { throw new ODataApplicationException(e.getMessage(), HttpStatusCode.INTERNAL_SERVER_ERROR.getStatusCode(), Locale.getDefault(), e); } catch (TeiidException e) { throw new ODataApplicationException(e.getMessage(), HttpStatusCode.INTERNAL_SERVER_ERROR.getStatusCode(), Locale.getDefault(), e); } */ final OperationResponseImpl operationResult = queryResponse; response.accepts( new ServiceResponseVisior() { @Override public void visit(PropertyResponse response) throws ODataLibraryException, ODataApplicationException { Property property = (Property) operationResult.getResult(); Object value = property.getValue(); if (value instanceof SQLXML || value instanceof Blob || value instanceof Clob) { try { handleLobResult(getClient().getProperty(Client.CHARSET), value, response); } catch (SQLException e) { LogManager.logDetail(LogConstants.CTX_ODATA, e); response.writeServerError(true); } } else { response.writeProperty(request.getReturnType().getType(), property); } } }); }
@Override public <T extends ServiceResponse> void read(final DataRequest request, T response) throws ODataLibraryException, ODataApplicationException { final ODataSQLBuilder visitor = new ODataSQLBuilder( odata, getClient().getMetadataStore(), this.prepared, true, request.getODataRequest().getRawBaseUri(), this.serviceMetadata, this.nameGenerator); visitor.visit(request.getUriInfo()); final BaseResponse queryResponse; try { Query query = visitor.selectQuery(); queryResponse = executeQuery(request, request.isCountRequest(), visitor, query); } catch (Throwable e) { throw new ODataApplicationException( e.getMessage(), HttpStatusCode.INTERNAL_SERVER_ERROR.getStatusCode(), Locale.getDefault(), e); } response.accepts( new ServiceResponseVisior() { public void visit(CountResponse response) throws ODataLibraryException, ODataApplicationException { org.teiid.odata.api.CountResponse cr = (org.teiid.odata.api.CountResponse) queryResponse; response.writeCount(cr.getCount()); } public void visit(PrimitiveValueResponse response) throws ODataLibraryException, ODataApplicationException { EntityCollection entitySet = (EntityCollection) queryResponse; Entity entity = entitySet.getEntities().get(0); EdmProperty edmProperty = request.getUriResourceProperty().getProperty(); Property property = entity.getProperty(edmProperty.getName()); if (property == null) { response.writeNotFound(true); } else if (property.getValue() == null) { response.writeNoContent(true); } response.write(property.getValue()); } public void visit(PropertyResponse response) throws ODataLibraryException, ODataApplicationException { EntityCollection entitySet = (EntityCollection) queryResponse; if (!entitySet.getEntities().isEmpty()) { Entity entity = entitySet.getEntities().get(0); EdmProperty edmProperty = request.getUriResourceProperty().getProperty(); Property property = entity.getProperty(edmProperty.getName()); response.writeProperty(edmProperty.getType(), property); } else { response.writeNotFound(true); } } public void visit(StreamResponse response) throws ODataLibraryException, ODataApplicationException { EntityCollectionResponse entitySet = (EntityCollectionResponse) queryResponse; EdmProperty edmProperty = request.getUriResourceProperty().getProperty(); Object value = entitySet.getStream(edmProperty.getName()); if (value == null) { response.writeNoContent(true); } else { try { handleLobResult(getClient().getProperty(Client.CHARSET), value, response); } catch (SQLException e) { LogManager.logDetail(LogConstants.CTX_ODATA, e); response.writeServerError(true); } } } public void visit(EntityResponse response) throws ODataLibraryException, ODataApplicationException { EntityCollection entitySet = (EntityCollection) queryResponse; if (entitySet.getEntities().isEmpty()) { if (visitor.hasNavigation()) { response.writeNoContent(true); } else { response.writeNotFound(true); } } else { response.writeReadEntity( visitor.getContext().getEdmEntityType(), entitySet.getEntities().get(0)); } } public void visit(EntitySetResponse response) throws ODataLibraryException, ODataApplicationException { sendResults(request, visitor, queryResponse, response); } }); }